From 76ea68489c4a5691cc43329609cb41d53d6f38ef Mon Sep 17 00:00:00 2001
From: Frederik Jaeckel <frederik.jaeckel@m-ds.de>
Date: Mon, 19 Dec 2022 21:53:00 +0100
Subject: [PATCH] =?UTF-8?q?test=20f=C3=BCr=20wizard?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 import_wiz.py        |  1 -
 tests/__init__.py    |  2 ++
 tests/test_wizard.py | 78 ++++++++++++++++++++++++++++++++++++++++++++
 3 files changed, 80 insertions(+), 1 deletion(-)
 create mode 100644 tests/test_wizard.py

diff --git a/import_wiz.py b/import_wiz.py
index d33d954..389456f 100644
--- a/import_wiz.py
+++ b/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
diff --git a/tests/__init__.py b/tests/__init__.py
index 1d46b3f..bb6349b 100644
--- a/tests/__init__.py
+++ b/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,\
diff --git a/tests/test_wizard.py b/tests/test_wizard.py
new file mode 100644
index 0000000..8ddf9a0
--- /dev/null
+++ b/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