from onionrblocks.generators import anonvdf
import unittest
import kasten
from onionrblocks.blockcreator import create_anonvdf_block

class TestBlockCreator(unittest.TestCase):
    def test_create_anonvdf(self):
        bl = create_anonvdf_block(b"Test", "txt", 3600)
        # (rounds - (size_bytes * cls.byte_cost)) // cls.second_cost
        expected_rounds = (3600 * 4) + (len(bl.get_packed()) * 100) + 100
        self.assertEqual(expected_rounds, bl.get_metadata()['rds'])

    def test_create_anonvdf_half_hour(self):
        bl = create_anonvdf_block(b"Test", "txt", 1800)
        expected_rounds = (len(bl.get_packed()) * anonvdf.AnonVDFGenerator.byte_cost) + (1800 * anonvdf.AnonVDFGenerator.second_cost) + 100
        self.assertEqual(expected_rounds, bl.get_metadata()['rds'])

    def test_create_anonvdf_odd(self):
        #(rounds - (size_bytes * cls.byte_cost)) // cls.second_cost
        bl = create_anonvdf_block(b"Test", "txt", 1303)
        expected_rounds = (len(bl.get_packed()) * anonvdf.AnonVDFGenerator.byte_cost) + (1303 * anonvdf.AnonVDFGenerator.second_cost) + 100
        self.assertEqual(expected_rounds, bl.get_metadata()['rds'])

    def test_create_anonvdf_verify(self):
        bl = create_anonvdf_block(b"Test", "txt", 3600)
        expected_rounds = (len(bl.get_packed()) * anonvdf.AnonVDFGenerator.byte_cost) + (3600 * anonvdf.AnonVDFGenerator.second_cost) + 100
        self.assertEqual(expected_rounds, bl.get_metadata()['rds'])

        packed = bl.get_packed()
        id = bl.id
        kasten.Kasten(id, packed, anonvdf.AnonVDFGenerator, auto_check_generator=True)

unittest.main()