OCA
/
bank-statement-import
5
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity
Browse Source

[FIX] account_bank_statement_import_online: correct balance recalculation

12.0
Alexey Pelykh 5 years ago
parent
cff2defc8b
commit
2c52971966
3 changed files with 83 additions and 18 deletions
Whitespace
Show all changes Ignore whitespace when comparing lines Ignore changes in amount of whitespace Ignore changes in whitespace at EOL
Split View
Diff Options
Show Stats Download Patch File Download Diff File
  1. 6
      account_bank_statement_import_online/models/online_bank_statement_provider.py
  2. 15
      account_bank_statement_import_online/tests/online_bank_statement_provider_dummy.py
  3. 80
      account_bank_statement_import_online/tests/test_account_bank_statement_import_online.py

6
account_bank_statement_import_online/models/online_bank_statement_provider.py
View File

@ -226,7 +226,7 @@ class OnlineBankStatementProvider(models.Model):
filtered_lines = []
for line_values in lines_data:
date = fields.Datetime.from_string(line_values['date'])
if date < statement_date_since:
if date < statement_date_since or date < date_since:
if 'balance_start' in statement_values:
statement_values['balance_start'] = (
Decimal(
@ -236,7 +236,7 @@ class OnlineBankStatementProvider(models.Model):
)
)
continue
elif date >= statement_date_until:
elif date >= statement_date_until or date >= date_until:
if 'balance_end_real' in statement_values:
statement_values['balance_end_real'] = (
Decimal(
@ -246,8 +246,6 @@ class OnlineBankStatementProvider(models.Model):
)
)
continue
elif date <= date_since or date > date_until:
continue
unique_import_id = line_values.get('unique_import_id')
if unique_import_id:
unique_import_id = provider._generate_unique_import_id(

15
account_bank_statement_import_online/tests/online_bank_statement_provider_dummy.py
View File

@ -33,8 +33,10 @@ class OnlineBankStatementProviderDummy(models.Model):
})
line_step = relativedelta(**line_step_options)
expand_by = self.env.context.get('expand_by', 0)
date_since -= expand_by * line_step
date_until += expand_by * line_step
data_since = self.env.context.get('data_since', date_since)
data_until = self.env.context.get('data_until', date_until)
data_since -= expand_by * line_step
data_until += expand_by * line_step
balance_start = self.env.context.get(
'balance_start',
@ -42,9 +44,12 @@ class OnlineBankStatementProviderDummy(models.Model):
)
balance = balance_start
lines = []
date = date_since
while date < date_until:
amount = randrange(-100, 100, 1) * 0.1
date = data_since
while date < data_until:
amount = self.env.context.get(
'amount',
randrange(-100, 100, 1) * 0.1
)
lines.append({
'name': 'payment',
'amount': amount,

80
account_bank_statement_import_online/tests/test_account_bank_statement_import_online.py
View File

@ -2,6 +2,7 @@
# Copyright 2019-2020 Dataplug (https://dataplug.io)
# License AGPL-3.0 or later (https://www.gnu.org/licenses/agpl).
from datetime import datetime
from dateutil.relativedelta import relativedelta
from psycopg2 import IntegrityError
from urllib.error import HTTPError
@ -97,8 +98,8 @@ class TestAccountBankAccountStatementImportOnline(common.TransactionCase):
statement = self.AccountBankStatement.search(
[('journal_id', '=', journal.id)],
)
self.assertEquals(len(statement), 1)
self.assertEquals(len(statement.line_ids), 12)
self.assertEqual(len(statement), 1)
self.assertEqual(len(statement.line_ids), 12)
def test_pull_mode_daily(self):
journal = self.AccountJournal.create({
@ -117,7 +118,7 @@ class TestAccountBankAccountStatementImportOnline(common.TransactionCase):
self.now - relativedelta(days=1),
self.now,
)
self.assertEquals(
self.assertEqual(
len(self.AccountBankStatement.search(
[('journal_id', '=', journal.id)]
)),
@ -141,7 +142,7 @@ class TestAccountBankAccountStatementImportOnline(common.TransactionCase):
self.now - relativedelta(weeks=1),
self.now,
)
self.assertEquals(
self.assertEqual(
len(self.AccountBankStatement.search(
[('journal_id', '=', journal.id)]
)),
@ -165,7 +166,7 @@ class TestAccountBankAccountStatementImportOnline(common.TransactionCase):
self.now - relativedelta(months=1),
self.now,
)
self.assertEquals(
self.assertEqual(
len(self.AccountBankStatement.search(
[('journal_id', '=', journal.id)]
)),
@ -196,7 +197,7 @@ class TestAccountBankAccountStatementImportOnline(common.TransactionCase):
statement = self.AccountBankStatement.search(
[('journal_id', '=', journal.id)],
)
self.assertEquals(len(statement), 1)
self.assertEqual(len(statement), 1)
def test_pull_skip_duplicates_by_unique_import_id(self):
journal = self.AccountJournal.create({
@ -215,7 +216,7 @@ class TestAccountBankAccountStatementImportOnline(common.TransactionCase):
self.now - relativedelta(weeks=2),
self.now,
)
self.assertEquals(
self.assertEqual(
len(self.AccountBankStatementLine.search(
[('journal_id', '=', journal.id)]
)),
@ -226,7 +227,7 @@ class TestAccountBankAccountStatementImportOnline(common.TransactionCase):
self.now - relativedelta(weeks=3),
self.now - relativedelta(weeks=1),
)
self.assertEquals(
self.assertEqual(
len(self.AccountBankStatementLine.search(
[('journal_id', '=', journal.id)]
)),
@ -237,7 +238,7 @@ class TestAccountBankAccountStatementImportOnline(common.TransactionCase):
self.now - relativedelta(weeks=1),
self.now,
)
self.assertEquals(
self.assertEqual(
len(self.AccountBankStatementLine.search(
[('journal_id', '=', journal.id)]
)),
@ -421,3 +422,64 @@ class TestAccountBankAccountStatementImportOnline(common.TransactionCase):
self.assertTrue(self.AccountBankStatement.search(
[('journal_id', '=', journal.id)],
))
def test_pull_statement_partially(self):
journal = self.AccountJournal.create({
'name': 'Bank',
'type': 'bank',
'code': 'BANK',
'bank_statements_source': 'online',
'online_bank_statement_provider': 'dummy',
})
provider = journal.online_bank_statement_provider_id
provider.active = True
provider.statement_creation_mode = 'monthly'
provider_context = {
'step': {'hours': 24},
'data_since': datetime(2020, 1, 1),
'data_until': datetime(2020, 2, 29),
'amount': 1.0,
'balance_start': 0,
}
provider.with_context(**provider_context)._pull(
datetime(2020, 1, 1),
datetime(2020, 1, 31),
)
statements = self.AccountBankStatement.search(
[('journal_id', '=', journal.id)],
order='date asc',
)
self.assertEqual(len(statements), 1)
self.assertEqual(statements[0].balance_start, 0.0)
self.assertEqual(statements[0].balance_end_real, 30.0)
provider.with_context(**provider_context)._pull(
datetime(2020, 1, 1),
datetime(2020, 2, 15),
)
statements = self.AccountBankStatement.search(
[('journal_id', '=', journal.id)],
order='date asc',
)
self.assertEqual(len(statements), 2)
self.assertEqual(statements[0].balance_start, 0.0)
self.assertEqual(statements[0].balance_end_real, 31.0)
self.assertEqual(statements[1].balance_start, 31.0)
self.assertEqual(statements[1].balance_end_real, 45.0)
provider.with_context(**provider_context)._pull(
datetime(2020, 1, 1),
datetime(2020, 2, 29),
)
statements = self.AccountBankStatement.search(
[('journal_id', '=', journal.id)],
order='date asc',
)
self.assertEqual(len(statements), 2)
self.assertEqual(statements[0].balance_start, 0.0)
self.assertEqual(statements[0].balance_end_real, 31.0)
self.assertEqual(statements[1].balance_start, 31.0)
self.assertEqual(statements[1].balance_end_real, 59.0)
Write Preview
Loading…
Cancel
Save