import os, uuid
import base64
import secrets

import time
from nacl import signing

import kasten

TEST_DIR = 'testdata/%s-%s' % (str(uuid.uuid4())[:6], os.path.basename(__file__)) + '/'
print("Test directory:", TEST_DIR)
os.environ["ONIONR_HOME"] = TEST_DIR

import unittest
import sys


import kasten
from kasten.generator import pack as kasten_pack

sys.path.append('static-data/official-plugins/rpc/rpc')
sys.path.append("src/")

import queue

import onionrblocks

from gossip import blockqueues


import blocks

byte_cost = 10
second_cost = 4

def _get_rounds(seconds: int, size_bytes: int):
    return (seconds * second_cost) + (size_bytes * byte_cost)

class RPCPrepareBlockForVDFTest(unittest.TestCase):

    def test_prepare_block_for_vdf(self):
        data_bytes = b'block data'
        data = base64.b64encode(data_bytes)

        resp_dict = blocks.prepare_block_for_vdf(data, 'txt', 3600, {})
        expected_kasten_packed = kasten_pack.pack(data_bytes, 'txt', {'ttl': 3600}, int(time.time()))
        expected_kasten_obj = kasten.Kasten('', expected_kasten_packed, kasten.generator.KastenBaseGenerator, auto_check_generator=False)
        self.assertTrue(resp_dict['raw'])
        self.assertEqual(base64.b64decode(resp_dict['raw']),expected_kasten_packed)
        self.assertEqual(resp_dict['rounds_needed'], _get_rounds(3600, len(expected_kasten_obj.get_packed())))



unittest.main()