From 117e80f381aaca4f7c973e665d5b450e07f5beb6 Mon Sep 17 00:00:00 2001 From: Andrea Date: Fri, 14 Jun 2019 09:00:30 +0200 Subject: [PATCH] Add partner in move lines --- .../models/account_spread_line.py | 2 ++ .../tests/test_account_invoice_spread.py | 12 ++++-------- 2 files changed, 6 insertions(+), 8 deletions(-) diff --git a/account_spread_cost_revenue/models/account_spread_line.py b/account_spread_cost_revenue/models/account_spread_line.py index caba7743..b5b986db 100644 --- a/account_spread_cost_revenue/models/account_spread_line.py +++ b/account_spread_cost_revenue/models/account_spread_line.py @@ -94,6 +94,7 @@ class AccountInvoiceSpreadLine(models.Model): 'account_id': spread.debit_account_id.id, 'debit': amount if amount > 0.0 else 0.0, 'credit': -amount if amount < 0.0 else 0.0, + 'partner_id': self.spread_id.invoice_id.partner_id.id, 'analytic_account_id': analytic.id, 'analytic_tag_ids': analytic_tags, 'currency_id': not_same_curr and current_currency.id or False, @@ -103,6 +104,7 @@ class AccountInvoiceSpreadLine(models.Model): 'account_id': spread.credit_account_id.id, 'credit': amount if amount > 0.0 else 0.0, 'debit': -amount if amount < 0.0 else 0.0, + 'partner_id': self.spread_id.invoice_id.partner_id.id, 'analytic_account_id': analytic.id, 'analytic_tag_ids': analytic_tags, 'currency_id': not_same_curr and current_currency.id or False, diff --git a/account_spread_cost_revenue/tests/test_account_invoice_spread.py b/account_spread_cost_revenue/tests/test_account_invoice_spread.py index e89f91f4..57a3d9c1 100644 --- a/account_spread_cost_revenue/tests/test_account_invoice_spread.py +++ b/account_spread_cost_revenue/tests/test_account_invoice_spread.py @@ -512,10 +512,8 @@ class TestAccountInvoiceSpread(common.TransactionCase): invoice_mls = self.invoice.move_id.mapped('line_ids') self.assertTrue(invoice_mls) - count_balance_sheet = 0 - for invoice_ml in invoice_mls: - if invoice_ml.account_id == balance_sheet: - count_balance_sheet += 1 + count_balance_sheet = len(invoice_mls.filtered( + lambda x: x.account_id == balance_sheet)) self.assertEqual(count_balance_sheet, 1) self.spread.line_ids.create_and_reconcile_moves() @@ -572,10 +570,8 @@ class TestAccountInvoiceSpread(common.TransactionCase): invoice_mls = self.invoice.move_id.mapped('line_ids') self.assertTrue(invoice_mls) - count_balance_sheet = 0 - for invoice_ml in invoice_mls: - if invoice_ml.account_id == balance_sheet: - count_balance_sheet += 1 + count_balance_sheet = len(invoice_mls.filtered( + lambda x: x.account_id == balance_sheet)) self.assertEqual(count_balance_sheet, 1) self.spread.company_id.force_move_auto_post = True