Update tests. Add tools to work with POS.

tests/tools.py

@@ -0,0 +1,79 @@
+# This file is part of the account_invoice_ar module for Tryton.
+# The COPYRIGHT file at the top level of this repository contains
+# the full copyright notices and license terms.
+
+from proteus import Model
+
+from trytond.modules.company.tests.tools import get_company
+
+__all__ = ['create_pos', 'get_pos', 'get_invoice_types']
+
+
+def create_pos(company=None, config=None):
+    "Create a Point of Sale"
+    Pos = Model.get('account.pos', config=config)
+    Sequence = Model.get('ir.sequence', config=config)
+
+    if not company:
+        company = get_company()
+
+    pos = Pos(
+        company=company.id,
+        number=1,
+        pos_type='manual',
+        )
+
+    for attr, name in (
+            ('1', '01-Factura A'),
+            ('2', '02-Nota de Debito A'),
+            ('3', '03-Nota de Credito A'),
+            ('6', '06-Factura B'),
+            ('7', '07-Nota de Debito B'),
+            ('8', '08-Nota de Credito B'),
+            ('11', '11-Factura C'),
+            ('12', '12-Nota de Debito C'),
+            ('13', '13-Nota de Credito C')):
+        sequence = Sequence(
+            name='%s %s' % (name, 'manual'),
+            code='account.invoice',
+            company=company)
+        sequence.save()
+        pos.pos_sequences.new(
+            invoice_type=attr,
+            invoice_sequence=sequence,
+            )
+    pos.save()
+    return pos
+
+
+def get_pos(company=None, config=None):
+    "Return the only pos"
+    Pos = Model.get('account.pos', config=config)
+
+    if not company:
+        company = get_company()
+
+    pos, = Pos.find([('company', '=', company.id)])
+    return pos
+
+
+def get_invoice_types(company=None, pos=None, config=None):
+    "Return invoices types per pos and company"
+    Account = Model.get('account.account', config=config)
+    PosSequence = Model.get('account.pos.sequence', config=config)
+
+    if not company:
+        company = get_company()
+
+    if not pos:
+        pos = get_pos(company)
+
+    accounts = Account.find([
+            ('kind', 'in', ['receivable', 'payable', 'revenue', 'expense']),
+            ('company', '=', company.id),
+            ])
+    invoice_types = PosSequence.find([
+            ('pos', '=', pos.id),
+            ])
+    invoice_types = {i.invoice_type: i for i in invoice_types}
+    return invoice_types