def test_initialize_packet(self): with self.assertRaises(packet.PacketError): pack = packet.AccessRequest(SECRET_KEY, identifier=1024) with self.assertRaises(packet.PacketError): pack = packet.AccessRequest(SECRET_KEY, identifier='blafoo') with self.assertRaises(packet.PacketError) as ctx: pack = packet.AccessRequest(SECRET_KEY) pack.attributes = None
def test_access_request(self): pack = packet.AccessRequest(SECRET_KEY) pack.attributes.add('User-Name', 'TestUser') pack.attributes.add('User-Name', 'TestUser2') pack.attributes.extend( ('User-Password', 'blafoo'), ('Cisco-AVPair', 'bla=foo'), ('NAS-IP-Address', ipaddress.IPv4Address('127.0.0.1')), ('Service-Type', 'Login-User'), ('DHCP-Your-IP-Address', ipaddress.IPv4Address('192.168.0.1')), 'not_pair' ) with self.assertRaises(KeyError): pack.attributes.add('Bla-Foo', 'blafoo') self.assertEqual(pack.attributes.get('User-Name'), 'TestUser2') self.assertEqual(pack.attributes.get('User-Password'), 'blafoo') self.assertEqual(pack.attributes.get('Cisco-AVPair'), 'bla=foo') self.assertEqual(pack.attributes.get('NAS-IP-Address'), ipaddress.IPv4Address('127.0.0.1')) self.assertEqual(pack.attributes.get('Service-Type'), 'Login-User') self.assertEqual(pack.attributes.get('DHCP-Your-IP-Address'), ipaddress.IPv4Address('192.168.0.1')) encoded_ = bytes(pack) decoded_pack = packet.decode_request(SECRET_KEY, encoded_) self.assertEqual(pack, decoded_pack)
def test_access_challenge(self): request = packet.AccessRequest(SECRET_KEY) request.attributes.extend( ('User-Name', 'username'), ('User-Password', 'password'), ('NAS-Identifier', 'nas-id') ) encoded_ = bytes(request) decoded_request = packet.decode_request(SECRET_KEY, encoded_) self.assertEqual(request, decoded_request) response = decoded_request.create_reply(packet.ACCESS_CHALLENGE) response.attributes.add('Reply-Message', 'ACCESS_CHALANGE') encoded_ = bytes(response) decoded_response = packet.decode_response(SECRET_KEY, encoded_, request=request) self.assertEqual(response, decoded_response) response = decoded_request.create_reply(packet.ACCESS_CHALLENGE) response.attributes.add('User-Name', 'username') with self.assertRaises(packet.PacketError): bytes(response)
def test_access_accept(self): attributes = packet.AttributesSet( ('User-Name', 'username'), ('User-Password', 'password'), ('NAS-Identifier', 'nas-id') ) request = packet.AccessRequest(SECRET_KEY) request.attributes = attributes encoded_ = bytes(request) decoded_request = packet.decode_request(SECRET_KEY, encoded_) self.assertEqual(request, decoded_request) response = decoded_request.create_reply(packet.ACCESS_ACCEPT) encoded_ = bytes(response) decoded_response = packet.decode_response(SECRET_KEY, encoded_, request=request) self.assertEqual(response, decoded_response)