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")