Add X509 to __all__. Add pem() method to X509

[oss/ctypescrypto.git] / tests / testcipher.py

diff --git a/tests/testcipher.py b/tests/testcipher.py
index 052193c2143062046ed8ae615b03874486622523..7fa5a97b6473e866b4e50f3f8b9a021ed15e98ac 100644 (file)
--- a/tests/testcipher.py
+++ b/tests/testcipher.py
@@ -1,6 +1,7 @@
 from ctypescrypto.oid import Oid
 from ctypescrypto import cipher
 import unittest
+import sys
 
 
 class TestCipherType(unittest.TestCase):
@@ -9,6 +10,7 @@ class TestCipherType(unittest.TestCase):
                self.assertEqual(ct.block_size(),8)
                self.assertEqual(ct.key_length(),8)
                self.assertEqual(ct.iv_length(),8)
+               self.assertEqual(ct.algo(),'DES-CBC')
                self.assertEqual(ct.oid().shortname(),"DES-CBC")
                self.assertEqual(ct.mode(),"CBC")
        def test_ciphaesofb(self):
@@ -16,10 +18,18 @@ class TestCipherType(unittest.TestCase):
                self.assertEqual(ct.block_size(),1)
                self.assertEqual(ct.key_length(),32)
                self.assertEqual(ct.iv_length(),16)
+               self.assertEqual(ct.algo(),'AES-256-OFB')
                self.assertEqual(ct.oid().shortname(),"AES-256-OFB")
                self.assertEqual(ct.mode(),"OFB")
+       def test_unknowncipher(self):
+               with self.assertRaises(cipher.CipherError):
+                       ct=cipher.CipherType("no-such-cipher")
 
 class TestEncryptDecrypt(unittest.TestCase):
+       def test_cons_nokey(self):
+               ct=cipher.CipherType("DES-CBC")
+               with self.assertRaises(ValueError):
+                       c=cipher.Cipher(ct,None,None)
        def test_blockcipher(self):
                data="sdfdsddf"
                key='abcdabcd'
@@ -56,13 +66,37 @@ class TestEncryptDecrypt(unittest.TestCase):
 
        def test_ofb_noiv(self):
                data="sdfdsddfxx"
-               key='abcdabcd'
-               c=cipher.new("AES-256-OFB",key)
+               encryptkey='abcdabcd'*4
+               decryptkey=encryptkey[0:5]+encryptkey[5:]
+
+
+               c=cipher.new("AES-256-OFB",encryptkey)
                enc=c.update(data)+c.finish()
                # See if padding is added by default
                self.assertEqual(len(enc),len(data))
-               d=cipher.new("AES-256-OFB",key,encrypt=False)
+               d=cipher.new("AES-256-OFB",decryptkey,encrypt=False)
                dec=d.update(enc)+d.finish()
                self.assertEqual(data,dec)
+       def test_wrong_keylength(self):
+               data="sdfsdfxxx"
+               key="abcdabcd"
+               with self.assertRaises(ValueError):
+                       c=cipher.new("AES-128-OFB",key)
+       def test_wrong_ivlength(self):
+               key="abcdabcdabcdabcd"
+               iv="xxxxx"
+               with self.assertRaises(ValueError):
+                       c=cipher.new("AES-128-OFB",key,iv=iv)
+       def test_variable_keylength(self):
+               encryptkey="abcdefabcdefghtlgvasdgdgsdgdsg"
+               data="asdfsdfsdfsdfsdfsdfsdfsdf"
+               iv="abcdefgh"
+               c=cipher.new("bf-ofb",encryptkey,iv=iv)
+               ciphertext=c.update(data)+c.finish()
+               decryptkey=encryptkey[0:5]+encryptkey[5:]
+               d=cipher.new("bf-ofb",decryptkey,encrypt=False,iv=iv)
+               deciph=d.update(ciphertext)+d.finish()
+               self.assertEqual(deciph,data)
+
 if __name__ == '__main__':
        unittest.main()