def test_read_usage_records_given_no_records_found(self):
acs_mock = self.mock_cloudstack({'project': [{'name': 'name'}]}, {'usagerecord': []})
measure_mock = self.mock_measure()
reader = UsageRecordReader('region')
reader.index_usage()
self.assertEquals(1, measure_mock.delete.call_count)
self.assertEquals(0, measure_mock.create.call_count)
self.assertEquals(1, acs_mock.listUsageRecords.called)
def test_read_usage_records_given_no_records_found(self):
acs_mock = self.mock_cloudstack({'project': [{'name': 'name'}]}, {'usagerecord': []}, {'serviceoffering': [self.service_offerings]})
measure_mock = self.mock_elk()
reader = UsageRecordReader('region')
reader.index_usage()
self.assertEquals(1, measure_mock.delete_usage_records.call_count)
self.assertEquals(0, measure_mock.create_usage_record.call_count)
self.assertEquals(1, acs_mock.listUsageRecords.called)
def test_read_usage_records(self):
acs_mock = self.mock_cloudstack({'project': [{'name': 'name'}]}, {'usagerecord': [{
"enddate": "2016-03-14'T'23:59:59-03:00",
"offeringid": "8d67bc56-d1dd-4f19-83bd-b3b546a2a4f7",
"project": "Admin",
"projectid": "8ea2cf55-2b12-4821-bac7-3c96a4cf1ac2",
"rawusage": "24",
"startdate": "2016-03-14'T'00:00:00-03:00",
"usagetype": 2
}]})
measure_mock = self.mock_measure()
reader = UsageRecordReader('region')
reader.index_usage()
self.assertEquals(1, measure_mock.delete.call_count)
self.assertEquals(4, measure_mock.create.call_count)
self.assertEquals(8, acs_mock.listUsageRecords.call_count)
def test_read_usage_records(self):
acs_mock = self.mock_cloudstack({'project': [{'name': 'name'}]}, {'usagerecord': [{
"enddate": "2016-03-14'T'23:59:59-03:00",
"offeringid": "8d67bc56-d1dd-4f19-83bd-b3b546a2a4f7",
"project": "Admin",
"projectid": "8ea2cf55-2b12-4821-bac7-3c96a4cf1ac2",
"rawusage": "24",
"startdate": "2016-03-14'T'00:00:00-03:00",
"usagetype": 2
}]}, {'serviceoffering': [self.service_offerings]})
measure_mock = self.mock_elk()
reader = UsageRecordReader('region')
reader.index_usage()
self.assertEquals(1, measure_mock.delete_usage_records.call_count)
self.assertEquals(2, measure_mock.create_usage_record.call_count)
self.assertEquals(4, acs_mock.listUsageRecords.call_count)
