good/buy/tests/test_buy_order.py

# Copyright 2016 上海开阖软件有限公司 (http://www.osbzr.com)
# License AGPL-3.0 or later (http://www.gnu.org/licenses/agpl).

from odoo.tests.common import TransactionCase
from odoo.exceptions import UserError


class TestBuyOrder(TransactionCase):

    def setUp(self):
        super(TestBuyOrder, self).setUp()
        # 给buy_order_1中的商品“键盘”的分类设置科目
        self.env.ref('core.goods_category_1').account_id = self.env.ref(
            'finance.account_goods').id
        self.order = self.env.ref('buy.buy_order_1')
        self.order.bank_account_id = False

        # 因同一个业务伙伴不能存在两张未审核的收付款单，把系统里已有的相关业务伙伴未审核的收付款单审核
        self.env.ref('money.get_40000').money_order_done()
        self.env.ref('money.pay_2000').money_order_done()

    def test_get_paid_amount(self):
        ''' 测试：计算采购订单付款/退款 '''
        self.order.buy_order_done()
        receipt = self.env['buy.receipt'].search([('order_id', '=', self.order.id)])
        receipt.payment = 1
        receipt.bank_account_id = self.env.ref('core.comm').id
        receipt.buy_receipt_done()
        money = self.env['money.order'].search([('buy_id', '=', self.order.id)])
        money.money_order_done()
        self.assertTrue(self.order.paid_amount == 1)

    def test_get_paid_amount_pay_divided(self):
        ''' 测试：分批付款时，采购订单付款/退款 '''
        self.order.invoice_by_receipt = False
        plan = self.env['payment.plan'].create({
            'buy_id': self.order.id,
            'name': 'first',
            'amount_money': 10,
        })
        plan.request_payment()
        # 生成的发票未核销，已付金额为0
        self.assertTrue(self.order.paid_amount == 0)

    def test_onchange_discount_rate(self):
        ''' 优惠率改变时，改变优惠金额，成交金额也改变'''
        amount_before = self.order.amount
        discount_amount_before = self.order.discount_amount
        self.order.discount_rate = 10
        self.order.onchange_discount_rate()
        if amount_before:
            self.assertTrue(self.order.amount != amount_before)
            self.assertTrue(self.order.discount_amount !=
                            discount_amount_before)

    def test_get_buy_goods_state(self):
        '''返回收货状态'''
        # order商品总数量为10
        self.order.buy_order_done()
        # 采购订单行的已入库数量为0时，将商品状态写为未入库
        self.assertTrue(self.order.goods_state == '未入库')
        receipt = self.env['buy.receipt'].search(
            [('order_id', '=', self.order.id)])
        # 采购订单行的已入库数量等于商品数量时，将商品状态写为全部入库
        receipt.buy_receipt_done()
        self.assertTrue(self.order.goods_state == '全部入库')

    def test_get_buy_goods_state_part_in(self):
        '''返回收货状态：部分入库'''
        # 采购订单行的已入库数量小于商品数量时，将商品状态写为部分入库
        self.order.buy_order_done()
        receipt = self.env['buy.receipt'].search(
            [('order_id', '=', self.order.id)])
        for line in receipt.line_in_ids:
            line.goods_qty = 5
        receipt.buy_receipt_done()
        self.assertTrue(self.order.goods_state == '部分入库')
        back_rec = self.env['buy.receipt'].search(
            [('order_id', '=', self.order.id), ('state', '=', 'draft')])
        back_rec.action_cancel()
        self.assertTrue(self.order.goods_state == '部分入库剩余作废')

    def test_default_warehouse_dest(self):
        '''新建采购订单时默认调入仓库'''
        order = self.env['buy.order'].with_context({
            'warehouse_dest_type': 'stock'
        }).create({})
        self.assertTrue(order.warehouse_dest_id.type == 'stock')

    def test_unlink(self):
        '''测试删除已审核的采购订单'''
        self.order.buy_order_done()
        with self.assertRaises(UserError):
            self.order.unlink()
        # 删除草稿状态的采购订单
        self.order.buy_order_draft()
        self.order.unlink()

    def test_buy_order_done(self):
        '''采购订单审核'''
        # 正常审核
        self.order.buy_order_done()
        self.assertTrue(self.order.state == 'done')
        # 重复审核报错
        with self.assertRaises(UserError):
            self.order.buy_order_done()
        # 数量单价小于0应报错
        self.order.buy_order_draft()
        for line in self.order.line_ids:
            line.quantity = 0
        with self.assertRaises(UserError):
            self.order.buy_order_done()

        # 输入预付款和结算账户
        bank_account = self.env.ref('core.alipay')
        bank_account.balance = 1000000
        self.order.prepayment = 50.0
        self.order.bank_account_id = bank_account
        for line in self.order.line_ids:
            line.quantity = 1
            line.tax_rate = 0
        self.order.buy_order_done()

        # 预付款不为空时，请选择结算账户
        self.order.buy_order_draft()
        self.order.bank_account_id = False
        self.order.prepayment = 50.0
        with self.assertRaises(UserError):
            self.order.buy_order_done()

        # 没有订单行时审核报错
        for line in self.order.line_ids:
            line.unlink()
        with self.assertRaises(UserError):
            self.order.buy_order_done()

    def test_buy_order_done_foreign_currency(self):
        """测试审核采购订单，外币免税"""
        self.order.currency_id = self.env.ref('base.USD')
        for line in self.order.line_ids:
            line.price_taxed = 1170
            line.tax_rate = 17
        with self.assertRaises(UserError):
            self.order.buy_order_done()

    def test_buy_order_draft(self):
        '''采购订单反审核'''
        # 正常反审核
        self.order.buy_order_done()
        self.order.buy_order_draft()
        receipt = self.env['buy.receipt'].search(
            [('order_id', '=', self.order.id)])
        self.assertTrue(not receipt)
        self.assertTrue(self.order.state == 'draft')
        self.assertTrue(not self.order.approve_uid.id)
        # 重复反审核报错
        self.order.buy_order_done()
        self.order.buy_order_draft()
        with self.assertRaises(UserError):
            self.order.buy_order_draft()
        # 订单已收货不能反审核
        self.order.buy_order_done()
        receipt = self.env['buy.receipt'].search(
            [('order_id', '=', self.order.id)])
        receipt.buy_receipt_done()
        with self.assertRaises(UserError):
            self.order.buy_order_draft()

    def test_buy_order_draft_with_prepayment(self):
        '''有预付款的采购订单反审核'''
        # 输入预付款和结算账户
        bank_account = self.env.ref('core.alipay')
        bank_account.balance = 1000000
        self.order.prepayment = 100
        self.order.bank_account_id = bank_account
        self.order.buy_order_done()
        money_order = self.env['money.order'].search([
            ('origin_name', '=', self.order.name)])
        money_order.money_order_done()
        with self.assertRaises(UserError):
            self.order.buy_order_draft()

    def test_buy_generate_receipt(self):
        '''测试采购订单生成入库单,批次管理拆分折扣金额'''
        # 采购订单
        # 全部入库
        self.order.buy_order_done()
        receipt = self.env['buy.receipt'].search(
            [('order_id', '=', self.order.id)])
        receipt.buy_receipt_done()
        self.assertTrue(self.order.goods_state == '全部入库')
        # 批次管理拆分订单行
        new_order = self.order.copy()
        for line in new_order.line_ids:
            line.goods_id = self.env.ref('goods.mouse').id
            new_order.buy_generate_receipt()
        receipt = self.env['buy.receipt'].search(
            [('order_id', '=', new_order.id)])
        self.assertTrue(len(receipt.line_in_ids) == 10)
        # 退货订单
        # 全部入库
        return_receipt = self.env.ref('buy.buy_return_order_1')
        return_receipt.buy_order_done()
        receipt = self.env['buy.receipt'].search(
            [('order_id', '=', return_receipt.id)])
        receipt.buy_receipt_done()
        return_receipt.buy_generate_receipt()

    def test_onchange_partner_id(self):
        # partner 无 税率，采购单行商品无税率
        self.env.ref('core.lenovo').tax_rate = 0
        self.env.ref('goods.keyboard').tax_rate = 0
        self.order.onchange_partner_id()
        # partner 有 税率，采购单行商品无税率
        self.env.ref('core.lenovo').tax_rate = 10
        self.env.ref('goods.keyboard').tax_rate = 0
        self.order.onchange_partner_id()
        # partner 无税率，采购单行商品无税率
        self.env.ref('core.lenovo').tax_rate = 0
        self.env.ref('goods.keyboard').tax_rate = 10
        self.order.onchange_partner_id()
        # partner 税率 > 采购单行商品税率
        self.env.ref('core.lenovo').tax_rate = 11
        self.env.ref('goods.keyboard').tax_rate = 10
        self.order.onchange_partner_id()
        # partner 税率 =< 采购单行商品税率
        self.env.ref('core.lenovo').tax_rate = 11
        self.env.ref('goods.keyboard').tax_rate = 12
        self.order.onchange_partner_id()

    def test_compute_receipt_and_invoice(self):
        '''计算生成入库单和结算单的个数'''
        self.order.buy_order_done()
        self.assertTrue(self.order.receipt_count == 1)

        self.order.receipt_ids[0].buy_receipt_done()
        self.assertTrue(self.order.invoice_count == 1)
        self.assertTrue(self.order.invoice_ids ==
                        self.order.receipt_ids[0].invoice_id)

    def test_action_view_receipt(self):
        '''查看生成的入库单'''
        # 生成一张入库单
        self.order.buy_order_done()
        self.order.action_view_receipt()
        # 生成两张入库单
        self.order.buy_order_draft()
        self.order.buy_order_done()
        for line in self.order.receipt_ids[0].line_in_ids:
            line.goods_qty = 3
        self.order.receipt_ids[0].buy_receipt_done()
        self.order.action_view_receipt()

    def test_action_view_return(self):
        '''查看生成的采购退货单'''
        self.order.type = 'return'
        self.env.ref('buy.buy_order_line_1').goods_id = self.env.ref('goods.cable').id
        self.env.ref('buy.buy_order_line_1').attribute_id = False
        self.env.ref('warehouse.wh_in_whin0').warehouse_dest_id = self.env.ref('warehouse.sh_stock').id
        self.env.ref('warehouse.wh_in_whin0').approve_order()
        # 生成一张采购退货单
        self.order.buy_order_done()
        self.order.action_view_return()
        # 生成两张采购退货单
        self.order.buy_order_draft()
        self.order.buy_order_done()
        for line in self.order.receipt_ids[0].line_out_ids:
            line.goods_qty = 3
        self.order.receipt_ids[0].buy_receipt_done()
        self.order.action_view_return()

    def test_action_view_invoice(self):
        '''查看生成的结算单'''
        # 生成一张入库单，审核后查看结算单
        self.order.buy_order_done()
        self.order.action_view_invoice()
        self.order.receipt_ids[0].buy_receipt_done()
        self.order.action_view_invoice()
        # 生成两张入库单，都审核后在采购订单上查看结算单
        self.order.receipt_ids[0].buy_receipt_draft()
        self.order.buy_order_draft()
        self.order.buy_order_done()
        for line in self.order.receipt_ids[0].line_in_ids:
            line.goods_qty = 3
        self.order.receipt_ids[0].buy_receipt_done()
        self.order.action_view_invoice()

    def test_action_view_receipt(self):
        """ 测试 查看发货/退货单 """
        self.order.buy_order_done()
        self.order.action_view_receipt()

        receipt = self.env['buy.receipt'].search(
            [('order_id', '=', self.order.id)])
        for line in receipt.line_in_ids:
            line.goods_qty = 1
        receipt.buy_receipt_done()
        self.order.action_view_receipt()

    def test_buy_order_done_no_attribute(self):
        '''检查属性是否填充'''
        self.order.line_ids[0].attribute_id = False
        with self.assertRaises(UserError):
            self.order.buy_order_done()


class TestBuyOrderLine(TransactionCase):

    def setUp(self):
        super(TestBuyOrderLine, self).setUp()
        self.order = self.env.ref('buy.buy_order_1')
        self.cable = self.env.ref('goods.cable')

    def test_compute_using_attribute(self):
        '''返回订单行中商品是否使用属性'''
        for line in self.order.line_ids:
            self.assertTrue(line.using_attribute)
            line.goods_id = self.env.ref('goods.mouse')
            self.assertTrue(not line.using_attribute)

    def test_compute_all_amount(self):
        '''当订单行的数量、含税单价、折扣额、税率改变时，改变采购金额、税额、价税合计'''
        for line in self.order.line_ids:
            line.price_taxed = 11.7
            self.assertTrue(line.amount == 100)
            self.assertTrue(line.tax_amount == 17)
            self.assertAlmostEqual(line.price_taxed, 11.7)
            self.assertTrue(line.subtotal == 117)

    def test_compute_all_amount_foreign_currency(self):
        '''外币测试：当订单行的数量、含税单价、折扣额、税率改变时，改变采购金额、税额、价税合计'''
        # 单据上外币是公司本位币
        self.order.currency_id = self.env.ref('base.CNY')
        for line in self.order.line_ids:
            line.price_taxed = 11.7
        # 外币
        self.order.currency_id = self.env.ref('base.EUR')
        for line in self.order.line_ids:
            line.price_taxed = 11.7

    def test_compute_all_amount_wrong_tax_rate(self):
        '''明细行上输入错误税率，应报错'''
        for line in self.order.line_ids:
            with self.assertRaises(UserError):
                line.tax_rate = -1
            with self.assertRaises(UserError):
                line.tax_rate = 102

    def test_onchange_price(self):
        '''当订单行的不含税单价改变时，改变含税单价'''
        for line in self.order.line_ids:
            line.price_taxed = 0
            line.price = 10
            line.onchange_price()
            self.assertAlmostEqual(line.price_taxed, 11.7)

    def test_onchange_goods_id(self):
        '''当订单行的商品变化时，带出商品上的单位、成本'''
        for line in self.order.line_ids:
            line.goods_id = self.cable
            line.onchange_goods_id()
            self.assertTrue(line.uom_id.name == '件')

            # 测试价格是否是商品的成本
            self.assertTrue(line.price == self.cable.cost)

    def test_onchange_goods_id_tax_rate(self):
        ''' 测试 修改商品时，采购单行税率变化 '''
        self.order_line = self.env.ref('buy.buy_order_line_1')
        # partner 无 税率，采购单行商品无税率
        self.env.ref('core.lenovo').tax_rate = 0
        self.env.ref('goods.keyboard').tax_rate = 0
        self.order_line.onchange_goods_id()
        # partner 有 税率，采购单行商品无税率
        self.env.ref('core.lenovo').tax_rate = 10
        self.env.ref('goods.keyboard').tax_rate = 0
        self.order_line.onchange_goods_id()
        # partner 无税率，采购单行商品有税率
        self.env.ref('core.lenovo').tax_rate = 0
        self.env.ref('goods.keyboard').tax_rate = 10
        self.order_line.onchange_goods_id()
        # partner 税率 > 采购单行商品税率
        self.env.ref('core.lenovo').tax_rate = 11
        self.env.ref('goods.keyboard').tax_rate = 10
        self.order_line.onchange_goods_id()
        # partner 税率 =< 采购单行商品税率
        self.env.ref('core.lenovo').tax_rate = 9
        self.env.ref('goods.keyboard').tax_rate = 10
        self.order_line.onchange_goods_id()

    def test_onchange_goods_id_vendor(self):
        '''当订单行的商品变化时，带出商品上供应商价'''
        # 添加网线的供应商价
        self.cable.vendor_ids.create({
            'goods_id': self.cable.id,
            'vendor_id': self.env.ref('core.lenovo').id,
            'price': 2, })
        # 不选择供应商时，应弹出警告
        self.order.partner_id = False
        for line in self.order.line_ids:
            with self.assertRaises(UserError):
                line.onchange_goods_id()
        # 选择供应商联想，得到供应商价
        self.order.partner_id = self.env.ref('core.lenovo')
        for line in self.order.line_ids:
            line.goods_id = self.cable
            line.onchange_goods_id()
            self.assertTrue(line.price == 2)

    def test_onchange_discount_rate(self):
        ''' 订单行优惠率改变时，改变优惠金额'''
        for line in self.order.line_ids:
            line.price_taxed = 11.7
            line.discount_rate = 10
            line.onchange_discount_rate()
            self.assertTrue(line.discount_amount == 10)