41 lines
1.5 KiB
Python
41 lines
1.5 KiB
Python
import unittest
|
|
import os
|
|
|
|
from kasten import pack
|
|
from kasten import exceptions
|
|
|
|
|
|
class TestPack(unittest.TestCase):
|
|
|
|
def test_unsigned_pack(self):
|
|
data = os.urandom(10)
|
|
packed = pack.pack(data, 'bin', 0)
|
|
parts = packed.split(b'\n', 1)
|
|
self.assertEqual(parts[0], b'\x92\xa3bin\x00\xc4\x01')
|
|
self.assertEqual(parts[1], data)
|
|
|
|
def test_linebreak_data(self):
|
|
data = os.urandom(9) + b'\n' + b"okay"
|
|
packed = pack.pack(data, 'bin', 0)
|
|
parts = packed.split(b'\n', 1)
|
|
self.assertEqual(parts[0], b'\x92\xa3bin\x00\xc4\x01')
|
|
self.assertEqual(parts[1], data)
|
|
|
|
def test_invalid_data_type(self):
|
|
data = os.urandom(10)
|
|
self.assertRaises(exceptions.InvalidKastenTypeLength, pack.pack, data, '', 1)
|
|
data = os.urandom(10)
|
|
self.assertRaises(exceptions.InvalidKastenTypeLength, pack.pack, data, 'aaaaa', 1)
|
|
|
|
def test_invalid_enc_mode(self):
|
|
data = os.urandom(10)
|
|
self.assertRaises(exceptions.InvalidEncryptionMode, pack.pack, data, 'txt', None)
|
|
self.assertRaises(exceptions.InvalidEncryptionMode, pack.pack, data, 'txt', "100")
|
|
self.assertRaises(exceptions.InvalidEncryptionMode, pack.pack, data, 'txt', 100)
|
|
self.assertRaises(exceptions.InvalidEncryptionMode, pack.pack, data, 'txt', -1)
|
|
self.assertRaises(exceptions.InvalidEncryptionMode, pack.pack, data, 'txt', -5)
|
|
self.assertRaises(exceptions.InvalidEncryptionMode, pack.pack, data, 'txt', "test")
|
|
|
|
|
|
|
|
unittest.main() |