71 lines
3.0 KiB
Python
71 lines
3.0 KiB
Python
|
# -*- coding: utf-8 -*-
|
||
|
|
# Part of Flectra. See LICENSE file for full copyright and licensing details.
|
||
|
|
import time
|
||
|
|
from flectra.addons.purchase.tests import test_purchase_order
|
||
|
|
|
||
|
|
|
||
|
|
class TestPurchaseBranchAuthentication(test_purchase_order.TestPurchaseOrder):
|
||
|
|
def setUp(self):
|
||
|
|
super(TestPurchaseBranchAuthentication, self).setUp()
|
||
|
|
self.user_1 = self.create_test_users(self.main_company, 'user_1', self.branch_1, [self.branch_1],
|
||
|
|
[self.purchase_user_group, self.stock_user_group])
|
||
|
|
self.user_2 = self.create_test_users(self.child_company, 'user_2', self.branch_2, [self.branch_2],
|
||
|
|
[self.purchase_user_group, self.stock_user_group])
|
||
|
|
product_list = [(self.product_id_1, 500),
|
||
|
|
(self.product_id_2, 1000),
|
||
|
|
(self.product_id_3, 800)]
|
||
|
|
order_lines = []
|
||
|
|
for product_id, quantity in product_list:
|
||
|
|
line_values = {
|
||
|
|
'name': product_id.name,
|
||
|
|
'product_id': product_id.id,
|
||
|
|
'price_unit': 80,
|
||
|
|
'product_qty': quantity,
|
||
|
|
'product_uom': product_id.uom_id.id,
|
||
|
|
'date_planned': self.current_time,
|
||
|
|
}
|
||
|
|
order_lines.append((0, 0, line_values))
|
||
|
|
self.purchase = self.PurchaseOrder.sudo(self.user_1).create({
|
||
|
|
'partner_id': self.partner_id.id,
|
||
|
|
'company_id': self.main_company.id,
|
||
|
|
'branch_id': self.branch_1.id,
|
||
|
|
'order_line': order_lines,
|
||
|
|
|
||
|
|
})
|
||
|
|
|
||
|
|
self.purchase.sudo(self.user_1).button_confirm()
|
||
|
|
self.create_purchase_invoice(self.partner_id, self.purchase, self.account)
|
||
|
|
|
||
|
|
def create_test_users(self, company_id, login, branch_id, branch_ids, group_ids):
|
||
|
|
group_ids = [group_id.id for group_id in group_ids]
|
||
|
|
user_obj = \
|
||
|
|
self.env['res.users'].with_context({'no_reset_password': True}). \
|
||
|
|
create({
|
||
|
|
'company_id': company_id.id,
|
||
|
|
'company_ids': [(4, company_id.id)],
|
||
|
|
'default_branch_id': branch_id.id,
|
||
|
|
'branch_ids': [(4, branch.id) for branch in branch_ids],
|
||
|
|
'name': 'Alex Purchase User',
|
||
|
|
'login': login,
|
||
|
|
'password': '123',
|
||
|
|
'email': 'alex@yourcompany.com',
|
||
|
|
'groups_id': [(6, 0, group_ids)]
|
||
|
|
})
|
||
|
|
return user_obj.id
|
||
|
|
|
||
|
|
def create_purchase_invoice(self, partner_id, purchase, account):
|
||
|
|
context = {
|
||
|
|
'active_model': 'purchase.order',
|
||
|
|
'active_ids': purchase.ids,
|
||
|
|
'active_id': purchase.id,
|
||
|
|
|
||
|
|
}
|
||
|
|
invoice_vals = {
|
||
|
|
'type': 'in_invoice',
|
||
|
|
'partner_id': partner_id.id,
|
||
|
|
'purchase_id': purchase.id,
|
||
|
|
'account_id': self.partner_id.property_account_payable_id.id,
|
||
|
|
}
|
||
|
|
self.env['account.invoice'].with_context(context).create(invoice_vals)
|
||
|
|
return True
|