Skip to content
Snippets Groups Projects

[IMP] energy_selfconsumption: add unit testing invoicing energy delivered

1 file
+ 47
3
Compare changes
  • Side-by-side
  • Inline
@@ -66,7 +66,7 @@ class TestContractGenerationWizard(TransactionCase):
"coefficient": 1,
}
)
self.define_invoicing_mode_wizard = self.env[
self.define_invoicing_mode_power_acquired_wizard = self.env[
"energy_selfconsumption.define_invoicing_mode.wizard"
].create(
{
@@ -77,6 +77,17 @@ class TestContractGenerationWizard(TransactionCase):
"invoicing_mode": "power_acquired",
}
)
self.define_invoicing_mode_energy_delivered_wizard = self.env[
"energy_selfconsumption.define_invoicing_mode.wizard"
].create(
{
"selfconsumption_id": self.selfconsumption.id,
"price": 0.1,
"recurring_interval": 1,
"recurring_rule_type": "monthly",
"invoicing_mode": "energy_delivered",
}
)
self.contract_generation_wizard = self.env[
"energy_selfconsumption.contract_generation.wizard"
].create(
@@ -85,8 +96,10 @@ class TestContractGenerationWizard(TransactionCase):
}
)
def test_generation_contracts(self):
res = self.define_invoicing_mode_wizard.save_data_to_selfconsumption()
def test_power_acquired_generation_contracts(self):
res = (
self.define_invoicing_mode_power_acquired_wizard.save_data_to_selfconsumption()
)
self.assertEqual(
res,
{
@@ -109,3 +122,34 @@ class TestContractGenerationWizard(TransactionCase):
invoice = related_contract._get_related_invoices()
self.assertEqual(invoice.invoice_line_ids[0].quantity, expected_quantity)
self.assertEqual(invoice.invoice_line_ids[0].price_unit, 0.1)
def test_energy_delivered_generation_contracts(self):
res = (
self.define_invoicing_mode_energy_delivered_wizard.save_data_to_selfconsumption()
)
self.assertEqual(
res,
{
"type": "ir.actions.act_window_close",
},
)
res = self.contract_generation_wizard.generate_contracts_button()
self.assertEqual(res, True)
related_contract = self.env["contract.contract"].search(
[("project_id", "=", self.selfconsumption.project_id.id)]
)
wizard_id = self.env["energy_selfconsumption.invoicing.wizard"].create(
{
"contract_ids": [(6, 0, related_contract.ids)],
"power": 200,
}
)
expected_quantity = 200 * 1
wizard_id.generate_invoices()
invoice = related_contract._get_related_invoices()
self.assertEqual(invoice.invoice_line_ids[0].quantity, expected_quantity)
self.assertEqual(invoice.invoice_line_ids[0].price_unit, 0.1)
Loading