Quellcode durchsuchen

added full process test

marton levente vor 7 Monaten
Ursprung
Commit
73004e6ad8
1 geänderte Dateien mit 17 neuen und 5 gelöschten Zeilen
  1. 17 5
      tests/test_csv2mentor.py

+ 17 - 5
tests/test_csv2mentor.py

@@ -3,10 +3,11 @@
 import unittest
 import os
 
-from pymt940 import (PaymentsProcessorBtrl, private, utils)
+from pymt940 import (PaymentsProcessorBtrl, private, utils, languages)
 from jinja2 import (Environment, PackageLoader)
 import pandas as pd
 
+
 home = os.path.expanduser('~')
 downloads_dir = os.path.join(home, 'Downloads', 'mentor', 'import', 'MZK')
 currency = 'ron'
@@ -21,6 +22,9 @@ acc_currency = 'accountron'
 class CsvTest(unittest.TestCase):
 
     templates = Environment(loader=PackageLoader('pymt940', 'templates'))
+    body_template = 'body_incomes.txt'
+    # body_template = 'body_payments.txt'
+    template = templates.get_template(body_template)
 
     processor = PaymentsProcessorBtrl(os.path.join(downloads_dir, private.MZK[currency]), private.MZK['name'],
                                       account_name=private.MZK[acc_currency]['name'],
@@ -43,12 +47,20 @@ class CsvTest(unittest.TestCase):
         print(self.df['Name'])
 
     def test_process_income(self):
-        df = self.processor.process_income_data()
-        print(df['Name'])
+        df = self.processor.process_income_data(lang=languages.LANG.EN)
+        print(df['symbols'], df['Name'], df['Amount'])
+        data = self.processor.to_dict(kind='TotalIncasari', month_back=1,
+                                      std_client='MOZAIK CONSULTING SRL', acc_symbol=411.01)
+        output = self.template.render(data)
+        print(output)
 
     def test_process_payment(self):
-        df = self.processor.process_payment_data()
-        print(df['Name'])
+        df = self.processor.process_payment_data(lang=languages.LANG.EN)
+        # print(df['symbols'], df['Name'], df['Amount'], df['types'])
+        data = self.processor.to_dict(kind='TotalPlati', month_back=1,
+                                      std_client='MOZAIK CONSULTING SRL', acc_symbol=401)
+        output = self.template.render(data)
+        print(output)
 
     def test_mentor_names(self):
         df_dict = utils.get_partners(self.processor.partner_data)