test für wizard

import_wiz.py

@@ -124,7 +124,6 @@ class ImportWizard(Wizard):
             done_dates.append(rate['date'])
 
         if len(to_create) > 0:
-            print('-- to_create:', to_create)
             Rate.create(to_create)
 
     @classmethod

tests/__init__.py

@@ -7,12 +7,14 @@ import unittest
 from trytond.modules.investment.tests.test_asset import AssetTestCase
 from trytond.modules.investment.tests.test_rate import RateTestCase
 from trytond.modules.investment.tests.test_source import SourceTestCase
+from trytond.modules.investment.tests.test_wizard import WizardTestCase
 
 
 __all__ = ['suite']
 
 
 class InvestmentTestCase(\
+    WizardTestCase, \
     SourceTestCase, \
     RateTestCase,\
     AssetTestCase,\

tests/test_wizard.py

@@ -0,0 +1,78 @@
+# -*- coding: utf-8 -*-
+# This file is part of the investment-module from m-ds for Tryton.
+# The COPYRIGHT file at the top level of this repository contains the
+# full copyright notices and license terms.
+
+from trytond.tests.test_tryton import ModuleTestCase, with_transaction
+from trytond.pool import Pool
+from trytond.modules.company.tests import create_company
+from trytond.transaction import Transaction
+from decimal import Decimal
+from datetime import date
+
+
+class WizardTestCase(ModuleTestCase):
+    'Test wizard module'
+    module = 'investment'
+
+    @with_transaction()
+    def test_wiz_run_import(self):
+        """ run import wizard
+        """
+        pool = Pool()
+        Asset = pool.get('investment.asset')
+        ImportWiz = pool.get('investment.imp_wiz', type='wizard')
+
+        company = self.prep_asset_company()
+        product = self.prep_asset_product(
+            name='Product 1',
+            description='some asset')
+
+        asset = self.prep_asset_item(
+            company=company,
+            product = product)
+        self.assertEqual(len(asset.rates), 0)
+
+        with Transaction().set_context({
+            'active_id': asset.id,
+            'active_model': 'investment.asset',
+            }):
+            (sess_id, start_state, end_state) = ImportWiz.create()
+            w_obj = ImportWiz(sess_id)
+            self.assertEqual(start_state, 'start')
+            self.assertEqual(end_state, 'end')
+
+            # run start
+            result = ImportWiz.execute(sess_id, {}, start_state)
+            self.assertEqual(list(result.keys()), ['view'])
+
+            self.assertEqual(result['view']['defaults']['asset'], asset.id)
+            self.assertEqual(result['view']['defaults']['dec_divider'], ',')
+            self.assertEqual(result['view']['defaults']['date_fmt'], '%d.%m.%Y')
+            self.assertEqual(result['view']['defaults']['field_delimiter'], ';')
+
+            w_obj.start.asset = asset
+            w_obj.start.dec_divider = ','
+            w_obj.start.date_fmt = '%d.%m.%Y'
+            w_obj.start.field_delimiter = ';'
+
+            result = ImportWiz.execute(sess_id, {'start': {
+                'asset': asset.id,
+                'dec_divider': ',',
+                'date_fmt': '%d.%m.%Y',
+                'field_delimiter': ';',
+                'file_': b'"date";"rate"\n"03.05.2022";"23,56"\n"05.05.2022";"24,22"\n"06.05.2022";"25,43"',
+                }}, 'importf')
+            self.assertEqual(list(result.keys()), [])
+            # finish wizard
+            ImportWiz.delete(sess_id)
+
+        self.assertEqual(len(asset.rates), 3)
+        self.assertEqual(asset.rates[0].date, date(2022, 5, 6))
+        self.assertEqual(asset.rates[0].rate, Decimal('25.43'))
+        self.assertEqual(asset.rates[1].date, date(2022, 5, 5))
+        self.assertEqual(asset.rates[1].rate, Decimal('24.22'))
+        self.assertEqual(asset.rates[2].date, date(2022, 5, 3))
+        self.assertEqual(asset.rates[2].rate, Decimal('23.56'))
+
+# end WizardTestCase