diff options
Diffstat (limited to 'tests/test_compat.py')
-rw-r--r-- | tests/test_compat.py | 78 |
1 files changed, 0 insertions, 78 deletions
diff --git a/tests/test_compat.py b/tests/test_compat.py deleted file mode 100644 index 9c65315..0000000 --- a/tests/test_compat.py +++ /dev/null @@ -1,78 +0,0 @@ -# Copyright 2011 Sybren A. Stüvel <sybren@stuvel.eu> -# -# Licensed under the Apache License, Version 2.0 (the "License"); -# you may not use this file except in compliance with the License. -# You may obtain a copy of the License at -# -# https://www.apache.org/licenses/LICENSE-2.0 -# -# Unless required by applicable law or agreed to in writing, software -# distributed under the License is distributed on an "AS IS" BASIS, -# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. -# See the License for the specific language governing permissions and -# limitations under the License. - -import unittest -import struct - -from rsa._compat import byte, xor_bytes - - -class TestByte(unittest.TestCase): - """Tests for single bytes.""" - - def test_byte(self): - for i in range(256): - byt = byte(i) - self.assertIsInstance(byt, bytes) - self.assertEqual(ord(byt), i) - - def test_raises_StructError_on_overflow(self): - self.assertRaises(struct.error, byte, 256) - self.assertRaises(struct.error, byte, -1) - - def test_byte_literal(self): - self.assertIsInstance(b"abc", bytes) - - -class TestBytes(unittest.TestCase): - """Tests for bytes objects.""" - - def setUp(self): - self.b1 = b"\xff\xff\xff\xff" - self.b2 = b"\x00\x00\x00\x00" - self.b3 = b"\xf0\xf0\xf0\xf0" - self.b4 = b"\x4d\x23\xca\xe2" - self.b5 = b"\x9b\x61\x3b\xdc" - self.b6 = b"\xff\xff" - - self.byte_strings = (self.b1, self.b2, self.b3, self.b4, self.b5, self.b6) - - def test_xor_bytes(self): - self.assertEqual(xor_bytes(self.b1, self.b2), b"\xff\xff\xff\xff") - self.assertEqual(xor_bytes(self.b1, self.b3), b"\x0f\x0f\x0f\x0f") - self.assertEqual(xor_bytes(self.b1, self.b4), b"\xb2\xdc\x35\x1d") - self.assertEqual(xor_bytes(self.b1, self.b5), b"\x64\x9e\xc4\x23") - self.assertEqual(xor_bytes(self.b2, self.b3), b"\xf0\xf0\xf0\xf0") - self.assertEqual(xor_bytes(self.b2, self.b4), b"\x4d\x23\xca\xe2") - self.assertEqual(xor_bytes(self.b2, self.b5), b"\x9b\x61\x3b\xdc") - self.assertEqual(xor_bytes(self.b3, self.b4), b"\xbd\xd3\x3a\x12") - self.assertEqual(xor_bytes(self.b3, self.b5), b"\x6b\x91\xcb\x2c") - self.assertEqual(xor_bytes(self.b4, self.b5), b"\xd6\x42\xf1\x3e") - - def test_xor_bytes_length(self): - self.assertEqual(xor_bytes(self.b1, self.b6), b"\x00\x00") - self.assertEqual(xor_bytes(self.b2, self.b6), b"\xff\xff") - self.assertEqual(xor_bytes(self.b3, self.b6), b"\x0f\x0f") - self.assertEqual(xor_bytes(self.b4, self.b6), b"\xb2\xdc") - self.assertEqual(xor_bytes(self.b5, self.b6), b"\x64\x9e") - self.assertEqual(xor_bytes(self.b6, b""), b"") - - def test_xor_bytes_commutative(self): - for first in self.byte_strings: - for second in self.byte_strings: - min_length = min(len(first), len(second)) - result = xor_bytes(first, second) - - self.assertEqual(result, xor_bytes(second, first)) - self.assertEqual(len(result), min_length) |