commit afc760d4e243a13b581434b38daa5417bcbe2273
author Paul Kehrer <paul.l.kehrer@gmail.com> Sat Mar 15 12:39:34 2014 -0400
committer Paul Kehrer <paul.l.kehrer@gmail.com> Sat Mar 15 12:39:34 2014 -0400
tree 33dd9a90b435b65a5ff87c47bf452b86e38515c6
parent b8a9c9e6243716b353b0786dae1e6e7d94f474a8

test IDEA key_size properly

tests/hazmat/primitives/test_ciphers.py

diff --git a/tests/hazmat/primitives/test_ciphers.py b/tests/hazmat/primitives/test_ciphers.py
index 50cadf6..d9f8353 100644
--- a/tests/hazmat/primitives/test_ciphers.py
+++ b/tests/hazmat/primitives/test_ciphers.py
@@ -18,7 +18,7 @@
 import pytest
 
 from cryptography.hazmat.primitives.ciphers.algorithms import (
-    AES, Camellia, TripleDES, Blowfish, ARC4, CAST5
+    AES, Camellia, TripleDES, Blowfish, ARC4, CAST5, IDEA
 )
 
 
@@ -110,3 +110,13 @@
     def test_invalid_key_size(self):
         with pytest.raises(ValueError):
             ARC4(binascii.unhexlify(b"0" * 34))
+
+
+class TestIDEA(object):
+    def test_key_size(self):
+        cipher = IDEA(b"\x00" * 16)
+        assert cipher.key_size == 128
+
+    def test_invalid_key_size(self):
+        with pytest.raises(ValueError):
+            IDEA(b"\x00" * 17)