audrius/bank-statement-import
1
0
Fork 1
mirror of https://github.com/OCA/bank-statement-import.git synced 2025-01-20 12:37:43 +02:00
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #188 from hbrunn/8.0-account_bank_statement_import_save_file-fix_test

Browse Source 
[FIX][account_bank_statement_import_save_file] don't mess with the registry for tests
This commit is contained in:
Pedro M. Baeza
2018-12-23 13:52:47 +01:00
committed by GitHub
parent 95fb15d938 44828e8dd3
commit 2be276e2f0
1 changed files with 41 additions and 49 deletions
Download Patch File Download Diff File Expand all files Collapse all files

90
account_bank_statement_import_save_file/tests/test_save_file.py
View File

@@ -19,60 +19,52 @@
#
##############################################################################
import base64
from openerp import api, models
from mock import patch
from openerp.tests.common import TransactionCase
acc_number = 'BE1234567890'
class HelloWorldParser(models.TransientModel):
_inherit = 'account.bank.statement.import'
@api.model
def _parse_file(self, data_file):
return (
'EUR',
acc_number,
[{
'name': '000000123',
'date': '2013-06-26',
'transactions': [{
'name': 'KBC-INVESTERINGSKREDIET 787-5562831-01',
'date': '2013-06-26',
'amount': 42,
'unique_import_id': 'hello',
}],
}],
)
class TestSaveFile(TransactionCase):
def test_SaveFile(self):
HelloWorldParser._build_model(self.registry, self.cr)
import_wizard = self.env['account.bank.statement.import']
import_wizard._prepare_setup()
import_wizard._setup_base(False)
import_wizard._setup_fields()
import_wizard._setup_complete()
import_wizard._auto_init()
journal_id = self.env['res.partner.bank'].search([
('acc_number', '=', acc_number),
]).journal_id.id
if not journal_id:
account = import_wizard._create_bank_account(acc_number)
journal_id = self.env['account.journal']\
.search([
'|',
('currency.name', '=', 'EUR'),
('currency', '=', False)
]).ids[0]
account.journal_id = journal_id
action = import_wizard.with_context(journal_id=journal_id)\
.create({'data_file': base64.b64encode('hello world')})\
.import_file()
for statement in self.env['account.bank.statement'].browse(
action['context']['statement_ids']):
self.assertEqual(
base64.b64decode(statement.import_file.datas),
'hello world')
with patch.object(
self.env.registry
.models['account.bank.statement.import'].__class__,
'_parse_file'
) as _parse_file:
_parse_file.side_effect = lambda data_file: (
'EUR',
acc_number,
[{
'name': '000000123',
'date': '2013-06-26',
'transactions': [{
'name': 'KBC-INVESTERINGSKREDIET 787-5562831-01',
'date': '2013-06-26',
'amount': 42,
'unique_import_id': 'hello',
}],
}],
)
import_wizard = self.env['account.bank.statement.import']
journal_id = self.env['res.partner.bank'].search([
('acc_number', '=', acc_number),
]).journal_id.id
if not journal_id:
account = import_wizard._create_bank_account(acc_number)
journal_id = self.env['account.journal']\
.search([
'|',
('currency.name', '=', 'EUR'),
('currency', '=', False)
]).ids[0]
account.journal_id = journal_id
action = import_wizard.with_context(journal_id=journal_id)\
.create({'data_file': base64.b64encode('hello world')})\
.import_file()
for statement in self.env['account.bank.statement'].browse(
action['context']['statement_ids']):
self.assertEqual(
base64.b64decode(statement.import_file.datas),
'hello world')