good/task/tests/test_task.py

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

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


class TestTask(TransactionCase):

    def setUp(self):
        super(TestTask, self).setUp()
        self.task = self.env.ref('task.task_sell')
        self.project_id = self.env.ref('task.project_gooderp')
        self.status = self.env.ref('task.task_status_doing')

    def test_compute_hours(self):
        '''计算任务的实际时间'''
        self.assertTrue(self.task.hours == 1)

    def test_assign_to_me(self):
        '''将任务指派给自己，并修改状态'''
        self.task.assign_to_me()
        self.assertTrue(self.task.user_id == self.env.ref('base.user_root'))
        self.assertTrue(self.task.status == self.status)

    def test_default_status(self):
        '''创建任务时，任务阶段默认为doing状态的阶段'''
        new_task = self.env['task'].create({
            'name': '测试创建任务时任务阶段的默认值',
            'project_id': self.project_id.id,
        })
        self.assertTrue(new_task.status == self.env.ref('task.task_status_todo'))


class TestTimesheet(TransactionCase):

    def setUp(self):
        super(TestTimesheet, self).setUp()
        self.timesheet = self.env.ref('task.timesheet_20161110')

    def test_name_get(self):
        '''测试今日工作日志的name_get'''
        name = self.timesheet.name_get()
        real_name = '%s %s' % (self.env.ref(
            'base.user_admin').name, '2016-11-10')
        self.assertEqual(name[0][1], real_name)


class TestTimeline(TransactionCase):

    def setUp(self):
        super(TestTimeline, self).setUp()
        self.task = self.env.ref('task.task_sell')
        self.status_doing = self.env.ref('task.task_status_doing')

    def test_create(self):
        '''创建工作记录时应更新对应task的status等字段'''
        timeline = self.env['timeline'].create({
            'task_id': self.task.id,
            'just_done': '创建一个销售订单',
            'next_action': '测试其他内容',
            'next_datetime': (datetime.now()).strftime('%Y-%m-%d %H:%M:%S'),
            'set_status': self.status_doing.id,
            'user_id': self.env.ref('base.user_root').id,
        })
        self.assertEqual(self.task.status, self.status_doing)
        self.assertEqual(self.task.next_action, timeline.next_action)
        self.assertEqual(self.task.next_datetime, timeline.next_datetime)
        self.assertEqual(self.task.user_id, timeline.user_id)


class TestProjectInvoice(TransactionCase):

    def setUp(self):
        super(TestProjectInvoice, self).setUp()
        self.project = self.env.ref('task.project_gooderp')
        self.invoice1 = self.env.ref('task.project_invoice_1')

    def test_compute_tax_amount(self):
        '''计算税额'''
        self.assertTrue(self.invoice1.tax_amount == 17.0)

    def test_compute_tax_amount_wrong_tax_rate(self):
        '''输入错误税率，应报错'''
        with self.assertRaises(UserError):
            self.invoice1.tax_rate = -1
            self.invoice1._compute_tax_amount()  # 不调用此方法，测试中UserError报不出来
        with self.assertRaises(UserError):
            self.invoice1.tax_rate = 102
            self.invoice1._compute_tax_amount()

    def test_make_invoice(self):
        '''生成结算单'''
        # 不输入客户点击生成发票按钮时应报错
        with self.assertRaises(UserError):
            self.invoice1.make_invoice()
        # 输入客户产生结算单，验证如下
        self.invoice1.project_id.customer_id = self.env.ref('core.jd')
        invoice = self.invoice1.make_invoice()
        self.assertTrue(self.invoice1.invoice_id == invoice)
        self.assertTrue(self.invoice1.project_id.auxiliary_id ==
                        invoice.auxiliary_id)
        self.assertTrue(self.invoice1.tax_amount == invoice.tax_amount)
        self.assertTrue(self.invoice1.amount == invoice.amount)


class TestProject(TransactionCase):

    def setUp(self):
        super(TestProject, self).setUp()
        self.project_id = self.env.ref('task.project_gooderp')

    def test_compute_hours(self):
        '''计算项目的实际工时'''
        self.assertTrue(self.project_id.hours == 1)