diff options
author | Yesudeep Mangalapilly <yesudeep@gmail.com> | 2011-08-11 04:48:11 +0530 |
---|---|---|
committer | Yesudeep Mangalapilly <yesudeep@gmail.com> | 2011-08-11 04:48:11 +0530 |
commit | c19a21bac0fef5691143959d799e5490537fe312 (patch) | |
tree | 141c0cf97d5cc7b05cf1909d786fa681c8e008cb | |
parent | c5e027ff9e69faaecf101000194f1a53b2db388c (diff) | |
download | rsa-git-c19a21bac0fef5691143959d799e5490537fe312.tar.gz |
Adds tests for int2bytes.
-rw-r--r-- | tests/test_transform.py | 24 |
1 files changed, 24 insertions, 0 deletions
diff --git a/tests/test_transform.py b/tests/test_transform.py new file mode 100644 index 0000000..ecc1a30 --- /dev/null +++ b/tests/test_transform.py @@ -0,0 +1,24 @@ +# -*- coding: utf-8 -*- + + +import unittest2 +from rsa._compat import b +from rsa.transform import int2bytes + + +class Test_integer_to_bytes(unittest2.TestCase): + def test_chunk_size(self): + self.assertEqual(int2bytes(123456789, 6), + b('\x00\x00\x07[\xcd\x15')) + self.assertEqual(int2bytes(123456789, 7), + b('\x00\x00\x00\x07[\xcd\x15')) + + def test_raises_OverflowError_when_chunk_size_is_insufficient(self): + self.assertRaises(OverflowError, int2bytes, 123456789, 3) + self.assertRaises(OverflowError, int2bytes, 299999999999, 4) + + def test_raises_ValueError_when_negative_integer(self): + self.assertRaises(ValueError, int2bytes, -1) + + def test_raises_TypeError_when_not_integer(self): + self.assertRaises(TypeError, int2bytes, None) |