Coverage

OCA · May 3, 2019 · ff902e7 · ff902e7
1 parent 4f8c0df
commit ff902e7
Showing 1 changed file with 9 additions and 3 deletions.
diff --git a/l10n_nl_tax_statement/tests/test_l10n_nl_vat_statement.py b/l10n_nl_tax_statement/tests/test_l10n_nl_vat_statement.py
@@ -299,14 +299,20 @@ def test_11_wizard_execute(self):
     def test_12_undeclared_invoice(self):
         self.invoice_1._onchange_invoice_line_ids()
         self.invoice_1.action_invoice_open()
-        self.invoice_1.move_id.l10n_nl_add_move_in_statement()
+        move = self.invoice_1.move_id
+        move.with_context(params={
+            'model': 'l10n.nl.vat.statement',
+            'id': self.statement_1.id
+        }).l10n_nl_add_move_in_statement()
         for line in self.invoice_1.move_id.line_ids:
             self.assertTrue(line.l10n_nl_vat_statement_include)
-        self.invoice_1.move_id.l10n_nl_unlink_move_from_statement()
+        move.with_context(params={
+            'model': 'l10n.nl.vat.statement',
+            'id': self.statement_1.id
+        }).l10n_nl_unlink_move_from_statement()
         for line in self.invoice_1.move_id.line_ids:
             self.assertFalse(line.l10n_nl_vat_statement_include)
 
-        self.statement_1.statement_update()
         self.assertEqual(len(self.statement_1.line_ids.ids), 22)
 
         for line in self.statement_1.line_ids: