kasten/tests/test_generator_base.py

import unittest
from hashlib import sha3_384

from kasten import exceptions
from kasten.generator import KastenBaseGenerator


class TestBaseGenerator(unittest.TestCase):
    def test_base_generator(self):

        k = b'\x93\xa3tst\xce_\xe7\xfb\xfb\xc0\nTest'
        invalid = b'\x93\xa3tst\xce_\xe7\xfb\xfb\xc0\nTes2'

        correct_K = KastenBaseGenerator.generate(k)
        hash_test = sha3_384(k).digest()

        self.assertTrue(len(correct_K.get_packed()) > 0)

        # k = b'\x93\xa3txt\x01\xce^\x97\xe3\xdc\ntest'
        # invalid = b'\x93\xa3txt\x01\xce^\x97\xe3\xdc\ntes2'
        # K = KastenBaseGenerator.generate(k)
        # h = sha3_384(k).digest()
        # self.assertTrue(len(K.get_packed()) > 0)
        # KastenBaseGenerator.validate_id(h, k)
        # self.assertRaises(exceptions.InvalidID, KastenBaseGenerator.validate_id, h, invalid)
        # self.assertEqual(K.get_packed(), k)


unittest.main()