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)