20 lines
572 B
Python
20 lines
572 B
Python
|
import unittest
|
||
|
from hashlib import sha3_384
|
||
|
|
||
|
from kasten import Kasten
|
||
|
from kasten import exceptions
|
||
|
from kasten.generator import KastenBaseGenerator
|
||
|
|
||
|
|
||
|
class TestKastenBaseGenerator(unittest.TestCase):
|
||
|
def test_kasten(self):
|
||
|
k = b'\x93\xa3txt\x01\xce^\x97\xe3\xdc\ntest'
|
||
|
K = Kasten(sha3_384(k).digest(), k, KastenBaseGenerator)
|
||
|
|
||
|
def test_kasten_invalid(self):
|
||
|
k = b'\x93\xa3txt\x01\xce^\x97\xe3\xdc\ntest'
|
||
|
self.assertRaises(exceptions.InvalidID, Kasten, sha3_384(k + b'invalid').digest(), k, KastenBaseGenerator)
|
||
|
|
||
|
|
||
|
unittest.main()
|