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_create_with_too_long_name(self): srp_extension = SRPExtension() with self.assertRaises(ValueError): srp_extension = srp_extension.create(bytearray(b'a' * 256))