114 lines
2.8 KiB
ReStructuredText
114 lines
2.8 KiB
ReStructuredText
========================
|
|
Agent Selection Scenario
|
|
========================
|
|
|
|
Imports::
|
|
|
|
>>> import datetime
|
|
>>> from dateutil.relativedelta import relativedelta
|
|
>>> from decimal import Decimal
|
|
>>> from proteus import Model, Wizard
|
|
>>> from trytond.tests.tools import activate_modules, set_user
|
|
>>> from trytond.modules.company.tests.tools import create_company, \
|
|
... get_company
|
|
>>> from trytond.modules.account.tests.tools import create_fiscalyear, \
|
|
... create_chart, get_accounts
|
|
>>> today = datetime.date.today()
|
|
>>> yesterday = today - relativedelta(days=1)
|
|
|
|
Activate modules::
|
|
|
|
>>> config = activate_modules(['commission', 'sale'])
|
|
|
|
Create company::
|
|
|
|
>>> _ = create_company()
|
|
>>> company = get_company()
|
|
|
|
Create chart of accounts::
|
|
|
|
>>> _ = create_chart(company)
|
|
>>> accounts = get_accounts(company)
|
|
|
|
Create employee::
|
|
|
|
>>> Party = Model.get('party.party')
|
|
>>> Employee = Model.get('company.employee')
|
|
>>> employee = Employee()
|
|
>>> party = Party(name='Employee')
|
|
>>> party.save()
|
|
>>> employee.party = party
|
|
>>> employee.company = company
|
|
>>> employee.save()
|
|
|
|
Create customers::
|
|
|
|
>>> customer = Party(name='Customer')
|
|
>>> customer.save()
|
|
>>> other_customer = Party(name='Other Customer')
|
|
>>> other_customer.save()
|
|
|
|
Create agent::
|
|
|
|
>>> Agent = Model.get('commission.agent')
|
|
>>> agent_party = Party(name='Agent')
|
|
>>> agent_party.save()
|
|
>>> agent = Agent(party=agent_party)
|
|
>>> agent.type_ = 'agent'
|
|
>>> agent.currency = company.currency
|
|
>>> selection = agent.selections.new()
|
|
>>> selection.start_date = today
|
|
>>> selection.party = customer
|
|
>>> selection = agent.selections.new()
|
|
>>> selection.employee = employee
|
|
>>> agent.save()
|
|
|
|
Create sale::
|
|
|
|
>>> Sale = Model.get('sale.sale')
|
|
>>> sale = Sale()
|
|
>>> sale.party = customer
|
|
>>> sale.save()
|
|
|
|
The agent is assigned on quotation::
|
|
|
|
>>> sale.agent
|
|
>>> sale.click('quote')
|
|
>>> sale.state
|
|
'quotation'
|
|
>>> sale.agent == agent
|
|
True
|
|
|
|
Agent is not set for yesterday sales::
|
|
|
|
>>> Sale = Model.get('sale.sale')
|
|
>>> sale = Sale()
|
|
>>> sale.sale_date = yesterday
|
|
>>> sale.party = customer
|
|
>>> sale.click('quote')
|
|
>>> sale.state
|
|
'quotation'
|
|
>>> sale.agent
|
|
|
|
Link user to employee::
|
|
|
|
>>> User = Model.get('res.user')
|
|
>>> user = User(config.user)
|
|
>>> user.employees.append(employee)
|
|
>>> user.employee = employee
|
|
>>> user.save()
|
|
>>> set_user(user.id)
|
|
|
|
Creating a sale and test agent is assigned::
|
|
|
|
>>> sale = Sale()
|
|
>>> sale.party = other_customer
|
|
>>> sale.agent
|
|
>>> sale.click('quote')
|
|
>>> sale.state
|
|
'quotation'
|
|
>>> sale.quoted_by == employee
|
|
True
|
|
>>> sale.agent == agent
|
|
True
|