def test_init_data_row(self):
"""Test the init data row method."""
generator = OCIGenerator(self.six_hours_ago, self.now, self.currency)
test_row = {}
for report_type in OCI_REPORT_TYPE_TO_COLS:
kwargs = {"report_type": report_type}
test_row.update(
generator._init_data_row(self.six_hours_ago, self.now,
**kwargs))
self.assertIsInstance(test_row, dict)
def test_add_common_usage_info(self):
"""Test that add_common_usage_info updates usage timestamps."""
generator = OCIGenerator(self.six_hours_ago, self.now, self.currency)
test_row_in = {}
for report_type in OCI_REPORT_TYPE_TO_COLS:
kwargs = {"report_type": report_type}
test_row_in.update(
generator._init_data_row(self.six_hours_ago, self.now,
**kwargs))
test_row_out = generator._add_common_usage_info(
test_row_in, self.six_hours_ago, self.now)
self.assertIn("lineItem/intervalUsageStart", test_row_out)
self.assertIn("lineItem/intervalUsageEnd", test_row_out)
def test_init_data_row_start_none(self):
"""Test the init data row method none start date."""
generator = OCIGenerator(self.six_hours_ago, self.now, self.currency)
with self.assertRaises(ValueError):
generator._init_data_row(None, self.now)
def test_init_data_row_end_invalid(self):
"""Test the init data row method invalid end date."""
generator = OCIGenerator(self.six_hours_ago, self.now, self.currency)
with self.assertRaises(ValueError):
generator._init_data_row(self.six_hours_ago, "invalid")