kasten/tests/test_kasten_with_base_generator.py

import unittest
from hashlib import sha3_384

from kasten import Kasten
from kasten import exceptions
from kasten.generator import KastenBaseGenerator, pack


class TestKastenBaseGenerator(unittest.TestCase):
    def test_kasten(self):
        k = b'\x93\xa3tst\xce_\xe7\xfb\xfb\xc0\nTest'
        K = Kasten(sha3_384(k).digest(), k, KastenBaseGenerator)

    def test_kasten_invalid(self):
        k = b'\x93\xa3tst\xce_\xe7\xfb\xfb\xc0\nTest'
        self.assertRaises(
            exceptions.InvalidID,
            Kasten, sha3_384(k + b'invalid').digest(), k, KastenBaseGenerator)

    def test_kasten_get_metadata(self):
        metadata = {"name": "john", "raw": b"are we having fun yet?"}
        packed = pack.pack(b"test msg", "tst", app_metadata=metadata)
        K = Kasten(sha3_384(packed).digest(), packed, KastenBaseGenerator)
        self.assertEqual(K.get_metadata(), metadata)

    def test_kasten_unsafe_deserialize(self):
        class Person:
            def __init__(self, n):
                self.name = n
        kevin = Person("kevin")
        metadata = {
            "name": "john", "raw": b"are we having fun yet?", "person": kevin}
        try:
            packed = pack.pack(b"test msg", "tst", app_metadata=metadata)
            K = Kasten(sha3_384(packed).digest(), packed, KastenBaseGenerator)
        except TypeError:
            pass
        else:
            raise Exception("Serialized custom type in packer")



unittest.main()