def test_fixed_payload(self):
start_time = datetime.datetime.utcnow().replace(microsecond=0) -\
datetime.timedelta(days=1)
end_time = datetime.datetime.utcnow().replace(microsecond=0)
ipaddress = get_fake_fixed_address()
ipaddress.allocated_at = start_time
ipaddress.deallocated_at = end_time
ipaddress.address_type = 'fixed'
payload = billing.build_payload(ipaddress,
'ip.exists',
start_time=start_time,
end_time=end_time)
self.assertEqual(payload['event_type'], u'USAGE',
'event_type is wrong')
self.assertEqual(payload['tenant_id'], TENANT_ID, 'tenant_id is wrong')
self.assertEqual(payload['ip_address'], IP_READABLE,
'ip_address is wrong')
self.assertEqual(payload['subnet_id'], SUBNET_ID, 'subnet_id is wrong')
self.assertEqual(payload['network_id'], PUB_NETWORK_ID,
'network_id is wrong')
self.assertEqual(payload['public'], True, 'public should be true')
self.assertEqual(payload['ip_version'], 4, 'ip_version should be 4')
self.assertEqual(payload['ip_type'], 'fixed',
'ip_type should be fixed')
self.assertEqual(payload['id'], IP_ID, 'ip_id is wrong')
self.assertEqual(payload['startTime'],
billing.convert_timestamp(start_time),
'startTime is wrong')
self.assertEqual(payload['endTime'],
billing.convert_timestamp(end_time),
'endTime is wrong')
def test_fixed_payload(self):
start_time = datetime.datetime.utcnow().replace(microsecond=0) -\
datetime.timedelta(days=1)
end_time = datetime.datetime.utcnow().replace(microsecond=0)
ipaddress = get_fake_fixed_address()
ipaddress.allocated_at = start_time
ipaddress.deallocated_at = end_time
ipaddress.address_type = 'fixed'
payload = billing.build_payload(ipaddress, billing.IP_EXISTS,
start_time=start_time,
end_time=end_time)
self.assertEqual(payload['event_type'], billing.IP_EXISTS,
'event_type is wrong')
self.assertEqual(payload['tenant_id'], TENANT_ID,
'tenant_id is wrong')
self.assertEqual(payload['ip_address'], IP_READABLE,
'ip_address is wrong')
self.assertEqual(payload['ip_version'], 4,
'ip_version should be 4')
self.assertEqual(payload['ip_type'], 'fixed',
'ip_type should be fixed')
self.assertEqual(payload['id'], IP_ID, 'ip_id is wrong')
self.assertEqual(payload['startTime'],
billing.convert_timestamp(start_time),
'startTime is wrong')
self.assertEqual(payload['endTime'],
billing.convert_timestamp(end_time),
'endTime is wrong')
def test_disassociate_flip_payload(self):
event_time = datetime.datetime.utcnow().replace(microsecond=0)
ipaddress = get_fake_fixed_address()
# allocated_at and deallocated_at could be anything for testing this
ipaddress.allocated_at = event_time
ipaddress.deallocated_at = event_time
ipaddress.address_type = 'floating'
payload = billing.build_payload(ipaddress,
'ip.disassociate',
event_time=event_time)
self.assertEqual(payload['event_type'], u'DOWN', 'event_type is wrong')
self.assertEqual(payload['tenant_id'], TENANT_ID, 'tenant_id is wrong')
self.assertEqual(payload['ip_address'], IP_READABLE,
'ip_address is wrong')
self.assertEqual(payload['subnet_id'], SUBNET_ID, 'subnet_id is wrong')
self.assertEqual(payload['network_id'], PUB_NETWORK_ID,
'network_id is wrong')
self.assertEqual(payload['public'], True, 'public should be true')
self.assertEqual(payload['ip_version'], 4, 'ip_version should be 4')
self.assertEqual(payload['ip_type'], 'floating',
'ip_type should be fixed')
self.assertEqual(payload['id'], IP_ID, 'ip_id is wrong')
self.assertEqual(payload['eventTime'],
billing.convert_timestamp(event_time),
'eventTime is wrong')
def test_disassociate_flip_payload(self):
event_time = datetime.datetime.utcnow().replace(microsecond=0)
ipaddress = get_fake_fixed_address()
# allocated_at and deallocated_at could be anything for testing this
ipaddress.allocated_at = event_time
ipaddress.deallocated_at = event_time
ipaddress.address_type = 'floating'
payload = billing.build_payload(ipaddress, billing.IP_DISASSOC,
event_time=event_time)
self.assertEqual(payload['event_type'], billing.IP_DISASSOC,
'event_type is wrong')
self.assertEqual(payload['tenant_id'], TENANT_ID, 'tenant_id is wrong')
self.assertEqual(payload['ip_address'], IP_READABLE,
'ip_address is wrong')
self.assertEqual(payload['subnet_id'], SUBNET_ID, 'subnet_id is wrong')
self.assertEqual(payload['network_id'], PUB_NETWORK_ID,
'network_id is wrong')
self.assertEqual(payload['public'], True, 'public should be true')
self.assertEqual(payload['ip_version'], 4, 'ip_version should be 4')
self.assertEqual(payload['ip_type'], 'floating',
'ip_type should be fixed')
self.assertEqual(payload['id'], IP_ID, 'ip_id is wrong')
self.assertEqual(payload['eventTime'],
billing.convert_timestamp(event_time),
'eventTime is wrong')
def test_fixed_payload(self):
start_time = datetime.datetime.utcnow().replace(microsecond=0) - datetime.timedelta(days=1)
end_time = datetime.datetime.utcnow().replace(microsecond=0)
ipaddress = get_fake_fixed_address()
ipaddress.allocated_at = start_time
ipaddress.deallocated_at = end_time
ipaddress.address_type = "fixed"
payload = billing.build_payload(ipaddress, "ip.exists", start_time=start_time, end_time=end_time)
self.assertEqual(payload["event_type"], u"USAGE", "event_type is wrong")
self.assertEqual(payload["tenant_id"], TENANT_ID, "tenant_id is wrong")
self.assertEqual(payload["ip_address"], IP_READABLE, "ip_address is wrong")
self.assertEqual(payload["subnet_id"], SUBNET_ID, "subnet_id is wrong")
self.assertEqual(payload["network_id"], PUB_NETWORK_ID, "network_id is wrong")
self.assertEqual(payload["public"], True, "public should be true")
self.assertEqual(payload["ip_version"], 4, "ip_version should be 4")
self.assertEqual(payload["ip_type"], "fixed", "ip_type should be fixed")
self.assertEqual(payload["id"], IP_ID, "ip_id is wrong")
self.assertEqual(payload["startTime"], billing.convert_timestamp(start_time), "startTime is wrong")
self.assertEqual(payload["endTime"], billing.convert_timestamp(end_time), "endTime is wrong")
def test_disassociate_flip_payload(self):
event_time = datetime.datetime.utcnow().replace(microsecond=0)
ipaddress = get_fake_fixed_address()
# allocated_at and deallocated_at could be anything for testing this
ipaddress.allocated_at = event_time
ipaddress.deallocated_at = event_time
ipaddress.address_type = "floating"
payload = billing.build_payload(ipaddress, "ip.disassociate", event_time=event_time)
self.assertEqual(payload["event_type"], u"DOWN", "event_type is wrong")
self.assertEqual(payload["tenant_id"], TENANT_ID, "tenant_id is wrong")
self.assertEqual(payload["ip_address"], IP_READABLE, "ip_address is wrong")
self.assertEqual(payload["subnet_id"], SUBNET_ID, "subnet_id is wrong")
self.assertEqual(payload["network_id"], PUB_NETWORK_ID, "network_id is wrong")
self.assertEqual(payload["public"], True, "public should be true")
self.assertEqual(payload["ip_version"], 4, "ip_version should be 4")
self.assertEqual(payload["ip_type"], "floating", "ip_type should be fixed")
self.assertEqual(payload["id"], IP_ID, "ip_id is wrong")
self.assertEqual(payload["eventTime"], billing.convert_timestamp(event_time), "eventTime is wrong")
