From 675d25d3381fb19c1007ac863e917d01f29b9f6d Mon Sep 17 00:00:00 2001
From: Albert Cervera i Areny <albert@nan-tic.com>
Date: Thu, 1 Apr 2021 16:45:16 +0200
Subject: [PATCH] Fix tests.

---
 invoice.py                             | 3 ---
 tests/test_account_invoice_facturae.py | 4 ++++
 2 files changed, 4 insertions(+), 3 deletions(-)

diff --git a/invoice.py b/invoice.py
index 2eb3f92..be09611 100644
--- a/invoice.py
+++ b/invoice.py
@@ -21,9 +21,6 @@ from trytond import backend
 from trytond.i18n import gettext
 from trytond.exceptions import UserError
 
-__all__ = ['Invoice', 'InvoiceLine', 'CreditInvoiceStart', 'CreditInvoice',
-    'GenerateFacturaeStart', 'GenerateFacturae']
-
 # Get from XSD scheme of Facturae 3.2.1
 # http://www.facturae.gob.es/formato/Versiones/Facturaev3_2_1.xml
 RECTIFICATIVE_REASON_CODES = [
diff --git a/tests/test_account_invoice_facturae.py b/tests/test_account_invoice_facturae.py
index 17d6f4f..225dad1 100644
--- a/tests/test_account_invoice_facturae.py
+++ b/tests/test_account_invoice_facturae.py
@@ -199,6 +199,8 @@ class TestAccountInvoiceFacturaeCase(ModuleTestCase):
                 invoice.account = invoice.on_change_with_account()
 
                 line1 = InvoiceLine()
+                line1.company = company
+                line1.currency = currency
                 line1.product = product
                 line1.on_change_product()
                 line1.on_change_account()
@@ -206,6 +208,8 @@ class TestAccountInvoiceFacturaeCase(ModuleTestCase):
                 line1.unit_price = Decimal('40')
 
                 line2 = InvoiceLine()
+                line2.company = company
+                line2.currency = currency
                 line2.account = revenue
                 line2.on_change_account()
                 line2.product = product