tradon/modules/project_invoice/tests/scenario_project_invoice_timesheet.rst

==================================
Project Invoice Timesheet Scenario
==================================

Imports::

    >>> import datetime as dt
    >>> from decimal import Decimal

    >>> from proteus import Model
    >>> from trytond.modules.account.tests.tools import create_chart, get_accounts
    >>> from trytond.modules.account_invoice.tests.tools import create_payment_term
    >>> from trytond.modules.company.tests.tools import create_company, get_company
    >>> from trytond.tests.tools import activate_modules, set_user

    >>> today = dt.date.today()
    >>> yesterday = today - dt.timedelta(days=1)

Activate modules::

    >>> config = activate_modules('project_invoice')

Create company::

    >>> _ = create_company()
    >>> company = get_company()

Create project user::

    >>> User = Model.get('res.user')
    >>> Group = Model.get('res.group')
    >>> project_user = User()
    >>> project_user.name = 'Project'
    >>> project_user.login = 'project'
    >>> project_group, = Group.find([('name', '=', 'Project Administration')])
    >>> timesheet_group, = Group.find([('name', '=', 'Timesheet Administration')])
    >>> project_user.groups.extend([project_group, timesheet_group])
    >>> project_user.save()

Create project invoice user::

    >>> project_invoice_user = User()
    >>> project_invoice_user.name = 'Project Invoice'
    >>> project_invoice_user.login = 'project_invoice'
    >>> project_invoice_group, = Group.find([('name', '=', 'Project Invoice')])
    >>> project_group, = Group.find([('name', '=', 'Project Administration')])
    >>> invoice_group, = Group.find([('name', '=', 'Account')])
    >>> project_invoice_user.groups.extend(
    ...     [project_invoice_group, project_group, invoice_group])
    >>> project_invoice_user.save()

Create chart of accounts::

    >>> _ = create_chart(company)
    >>> accounts = get_accounts(company)
    >>> revenue = accounts['revenue']

Create payment term::

    >>> payment_term = create_payment_term()
    >>> payment_term.save()

Create customer::

    >>> Party = Model.get('party.party')
    >>> customer = Party(name='Customer')
    >>> customer.customer_payment_term = payment_term
    >>> customer.save()

Create employee::

    >>> Employee = Model.get('company.employee')
    >>> employee = Employee()
    >>> party = Party(name='Employee')
    >>> party.save()
    >>> employee.party = party
    >>> employee.company = company
    >>> employee.save()

Create account category::

    >>> ProductCategory = Model.get('product.category')
    >>> account_category = ProductCategory(name="Account Category")
    >>> account_category.accounting = True
    >>> account_category.account_revenue = revenue
    >>> account_category.save()

Create product::

    >>> ProductUom = Model.get('product.uom')
    >>> hour, = ProductUom.find([('name', '=', 'Hour')])
    >>> ProductTemplate = Model.get('product.template')

    >>> template = ProductTemplate()
    >>> template.name = 'Service'
    >>> template.default_uom = hour
    >>> template.type = 'service'
    >>> template.list_price = Decimal('20')
    >>> template.account_category = account_category
    >>> template.save()
    >>> product, = template.products

Create a Project::

    >>> set_user(project_user)
    >>> ProjectWork = Model.get('project.work')
    >>> project = ProjectWork()
    >>> project.name = 'Test timesheet'
    >>> project.type = 'project'
    >>> project.party = customer
    >>> project.project_invoice_method = 'timesheet'
    >>> project.product = product
    >>> project.timesheet_available = True
    >>> task = ProjectWork()
    >>> task.name = 'Task 1'
    >>> task.timesheet_available = True
    >>> task.type = 'task'
    >>> task.product = product
    >>> project.children.append(task)
    >>> project.save()
    >>> task, = project.children

Add a task without timesheet work::

    >>> task2 = project.children.new()
    >>> task2.name = 'Task 2'
    >>> task2.type = 'task'
    >>> project.save()

Create timesheets::

    >>> TimesheetLine = Model.get('timesheet.line')
    >>> line = TimesheetLine()
    >>> line.date = yesterday
    >>> line.employee = employee
    >>> line.duration = dt.timedelta(hours=3)
    >>> line.work, = task.timesheet_works
    >>> line.save()
    >>> line = TimesheetLine()
    >>> line.date = today
    >>> line.employee = employee
    >>> line.duration = dt.timedelta(hours=2)
    >>> line.work, = project.timesheet_works
    >>> line.save()

Check project amounts::

    >>> project.reload()
    >>> project.amount_to_invoice
    Decimal('100.00')
    >>> project.invoiced_amount
    Decimal('0.00')

Invoice project up to yesterday::

    >>> set_user(project_user)
    >>> project.project_invoice_timesheet_up_to = yesterday
    >>> project.save()
    >>> set_user(project_invoice_user)
    >>> project.click('invoice')
    >>> project.amount_to_invoice
    Decimal('0.00')
    >>> project.invoiced_amount
    Decimal('60.00')

    >>> project.project_invoice_timesheet_up_to = today
    >>> project.save()
    >>> project.amount_to_invoice
    Decimal('40.00')

    >>> set_user(project_invoice_user)
    >>> Invoice = Model.get('account.invoice')
    >>> invoice, = Invoice.find([])
    >>> invoice.total_amount
    Decimal('60.00')

Invoice all project::

    >>> set_user(project_user)
    >>> project.project_invoice_timesheet_up_to = None
    >>> project.save()
    >>> set_user(project_invoice_user)
    >>> project.click('invoice')
    >>> project.amount_to_invoice
    Decimal('0.00')
    >>> project.invoiced_amount
    Decimal('100.00')

    >>> set_user(project_invoice_user)
    >>> _, invoice = Invoice.find([], order=[('id', 'ASC')])
    >>> invoice.total_amount
    Decimal('40.00')

Create more timesheets::

    >>> set_user(project_user)
    >>> TimesheetLine = Model.get('timesheet.line')
    >>> line = TimesheetLine()
    >>> line.employee = employee
    >>> line.duration = dt.timedelta(hours=4)
    >>> line.work, = task.timesheet_works
    >>> line.save()

Check project amounts::

    >>> project.reload()
    >>> project.amount_to_invoice
    Decimal('80.00')
    >>> project.invoiced_amount
    Decimal('100.00')

Invoice again project::

    >>> set_user(project_invoice_user)
    >>> project.click('invoice')
    >>> project.amount_to_invoice
    Decimal('0.00')
    >>> project.invoiced_amount
    Decimal('180.00')

    >>> set_user(project_invoice_user)
    >>> _, _, invoice = Invoice.find([], order=[('id', 'ASC')])
    >>> invoice.total_amount
    Decimal('80.00')

Try to change invoice line quantity::

    >>> set_user(1)
    >>> TimesheetLine = Model.get('timesheet.line')
    >>> line = TimesheetLine(line.id)
    >>> line.invoice_line.quantity = 5
    >>> line.invoice_line.save()
    Traceback (most recent call last):
        ...
    InvoiceLineValidationError: ...
    >>> line.invoice_line.quantity = 4
    >>> line.invoice_line.save()