def test_create(self):
srp_extension = SRPExtension()
srp_extension = srp_extension.create()
self.assertEqual(None, srp_extension.identity)
self.assertEqual(12, srp_extension.ext_type)
self.assertEqual(bytearray(0), srp_extension.ext_data)
def test_create(self):
srp_extension = SRPExtension()
srp_extension = srp_extension.create()
self.assertIsNone(srp_extension.identity)
self.assertEqual(12, srp_extension.extType)
self.assertEqual(bytearray(0), srp_extension.extData)
def test_create_with_name(self):
srp_extension = SRPExtension()
srp_extension = srp_extension.create(bytearray(b'username'))
self.assertEqual(bytearray(b'username'), srp_extension.identity)
self.assertEqual(bytearray(
b'\x08' + # length of string - 8 bytes
b'username'), srp_extension.ext_data)
def test_write(self):
srp_extension = SRPExtension()
srp_extension = srp_extension.create(bytearray(b'username'))
self.assertEqual(bytearray(
b'\x00\x0c' + # type of extension - SRP (12)
b'\x00\x09' + # length of extension - 9 bytes
b'\x08' + # length of encoded name
b'username'), srp_extension.write())
def test_create_with_name(self):
srp_extension = SRPExtension()
srp_extension = srp_extension.create(bytearray(b'username'))
self.assertEqual(bytearray(b'username'), srp_extension.identity)
self.assertEqual(
bytearray(b'\x08' + # length of string - 8 bytes
b'username'),
srp_extension.extData)
def test_write(self):
srp_extension = SRPExtension()
srp_extension = srp_extension.create(bytearray(b'username'))
self.assertEqual(
bytearray(b'\x00\x0c' + # type of extension - SRP (12)
b'\x00\x09' + # length of extension - 9 bytes
b'\x08' + # length of encoded name
b'username'),
srp_extension.write())
def test___repr__(self):
srp_extension = SRPExtension()
srp_extension = srp_extension.create(bytearray(b'user'))
self.assertEqual("SRPExtension(identity=bytearray(b'user'))",
repr(srp_extension))
def test_create_with_too_long_name(self):
srp_extension = SRPExtension()
with self.assertRaises(ValueError):
srp_extension = srp_extension.create(bytearray(b'a'*256))
def test___repr__(self):
srp_extension = SRPExtension()
srp_extension = srp_extension.create(bytearray(b'user'))
self.assertEqual("SRPExtension(identity=bytearray(b'user'))",
repr(srp_extension))
def test_create_with_too_long_name(self):
srp_extension = SRPExtension()
with self.assertRaises(ValueError):
srp_extension = srp_extension.create(bytearray(b'a' * 256))
