def test_generate_data(self):
"""Test that the RDS generate_data method works."""
generator = RDSGenerator(self.two_hours_ago, self.now,
self.payer_account, self.usage_accounts,
self.attributes)
data = generator.generate_data()
self.assertNotEqual(data, [])
def test_update_data(self):
"""Test RDS specific update data method."""
generator = RDSGenerator(self.two_hours_ago, self.now,
self.payer_account, self.usage_accounts,
self.attributes)
start_row = {}
row = generator._update_data(start_row, self.two_hours_ago, self.now)
self.assertEqual(row["lineItem/ProductCode"], "AmazonRDS")
self.assertEqual(row["lineItem/Operation"], "CreateDBInstance")
self.assertEqual(row["product/ProductName"],
"Amazon Relational Database Service")
def test_init_with_attributes(self):
"""Test the unique init options for RDS."""
generator = RDSGenerator(self.two_hours_ago, self.now,
self.payer_account, self.usage_accounts,
self.attributes)
self.assertEqual(generator._product_sku, self.product_sku)
self.assertEqual(generator._resource_id, "i-" + self.resource_id)
self.assertEqual(generator._tags, self.tags)
self.assertEqual(generator._instance_type[:-1],
tuple(self.instance_type.values()))
def test_init_no_attributes(self):
"""Test the init wihout attributes."""
generator = RDSGenerator(self.two_hours_ago,
self.now,
self.payer_account,
self.usage_accounts,
attributes={"empty": "dictionary"})
self.assertIsNotNone(generator._product_sku)
self.assertIsNotNone(generator._resource_id)
self.assertIsNotNone(generator._instance_type)
def test_update_data(self):
"""Test RDS specific update data method."""
test_args = self.test_config_kwargs
test_args["generator"] = "RDSGenerator"
test_config = create_test_config(**test_args)
generator = RDSGenerator(self.two_hours_ago,
self.now,
user_config=test_config)
self.assertEqual(generator.config[0].get("product_sku"),
self.product_sku)
start_row = {}
row = generator._update_data(start_row, self.two_hours_ago, self.now,
**{"config": generator.config[0]})
self.assertEqual(row["lineItem/ProductCode"], "AmazonRDS")
self.assertEqual(row["lineItem/Operation"], "CreateDBInstance")
self.assertEqual(row["product/ProductName"],
"Amazon Relational Database Service")
os.remove(test_config)
def test_init_with_attributes(self):
"""Test the unique init options for RDS."""
test_args = self.test_config_kwargs
test_args["generator"] = "RDSGenerator"
test_config = create_test_config(**test_args)
generator = RDSGenerator(self.two_hours_ago,
self.now,
user_config=test_config)
self.assertEqual(generator.config[0].get("product_sku"),
self.product_sku)
self.assertIn(self.tags["key"], generator.config[0].get("tags").keys())
self.assertEqual(generator.config[0].get("resource_id"),
self.resource_id)
self.assertEqual(
tuple(generator.config[0].get("instance_type").values()),
tuple(self.instance_type.values()))
os.remove(test_config)
def test_generate_data(self):
"""Test that the RDS generate_data method works."""
generator = RDSGenerator(self.two_hours_ago, self.now)
data = generator.generate_data()
self.assertNotEqual(data, [])
def test_init_no_attributes(self):
"""Test the init wihout attributes."""
generator = RDSGenerator(self.two_hours_ago, self.now)
self.assertIsNotNone(generator.config[0].get("product_sku"))
self.assertIsNotNone(generator.config[0].get("resource_id"))
self.assertIsNotNone(generator.config[0].get("instance_type"))
