testPayroll.py 157 KB
Newer Older
1
# -*- coding: utf-8 -*-
2 3
##############################################################################
#
4
# Copyright (c) 2009-2010 Nexedi SA and Contributors. All Rights Reserved.
5 6 7
#          Fabien Morin <fabien.morin@gmail.com>
#
# WARNING: This program as such is intended to be used by professional
8
# programmers who take the whole responsibility of assessing all potential
9 10
# consequences resulting from its eventual inadequacies and bugs
# End users who are looking for a ready-to-use solution with commercial
11
# guarantees and support are strongly adviced to contract a Free Software
12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28
# Service Company
#
# This program is Free Software; you can redistribute it and/or
# modify it under the terms of the GNU General Public License
# as published by the Free Software Foundation; either version 2
# of the License, or (at your option) any later version.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with this program; if not, write to the Free Software
# Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA.
#
##############################################################################
29
from Products.ERP5Type.tests.ERP5TypeTestCase import ERP5ReportTestCase
30
from Products.ERP5.tests.testTradeModelLine import TestTradeModelLineMixin
31
from AccessControl.SecurityManagement import newSecurityManager
32 33
from Products.ERP5Type.tests.Sequence import SequenceList
from Products.ERP5Type.tests.utils import reindex
34
from DateTime import DateTime
35
import transaction
36
from Products.ERP5Type.tests.utils import createZODBPythonScript
37

38
class TestPayrollMixin(TestTradeModelLineMixin, ERP5ReportTestCase):
39 40
  BUSINESS_PATH_CREATION_SEQUENCE_STRING = """
               CreateBusinessProcess
41
               CreateBusinessLink
42
               CreateUrssafRoubaixOrganisation
43
               ModifyBusinessLinkTradePhase
44 45 46 47 48 49 50 51 52 53
               ModelSpecialiseBusinessProcess
               Tic
  """
  COMMON_BASIC_DOCUMENT_CREATION_SEQUENCE_STRING = """
               CreateUrssafService
               CreateLabourService
               CreateEmployer
               CreateEmployee
               CreatePriceCurrency
               CreateBasicModel
54
               Tic
55 56 57 58 59 60 61
               ModelCreateUrssafModelLine
               UrssafModelLineCreateMovements
               CreateBasicPaysheet
               PaysheetCreateLabourPaySheetLine
               Tic
  """ + BUSINESS_PATH_CREATION_SEQUENCE_STRING

62
  amount_generator_line_portal_type = 'Pay Sheet Model Line'
63

64 65 66
  def getTitle(self):
    return "Payroll"

67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88
  def setPayrollBaseAmountQuantityMethod(self, base_amount_id, text):
    """Populate TradeModelLine_getBaseAmountQuantityMethod shared script

    This helper method edits the script so that:
    - there's no need to do any cleanup
    - data produced by previous still behaves as expected
    """
    skin = self.portal.portal_skins.custom
    script_id = self.amount_generator_line_portal_type.replace(' ', '') \
                + '_getBaseAmountQuantityMethod'
    test = "\nif base_application.startswith(%r):\n  " % base_amount_id
    try:
      old_text = '\n' + skin[script_id].body()
    except KeyError:
      old_text = ''
    else:
      skin._delObject(script_id)
    text = "context.log('ba', base_application)\n\n" + test + '\n  '.join(text.splitlines()) + old_text
    createZODBPythonScript(skin, script_id, "base_application", text)
    return base_amount_id


89 90
  def afterSetUp(self):
    """Prepare the test."""
91
    TestTradeModelLineMixin.afterSetUp(self)
92
    self.createCategories()
93
    self.fixed_quantity = self.setBaseAmountQuantityMethod('fixed_quantity',
94
      "return lambda *args, **kw: 1")
95
    self.setPayrollBaseAmountQuantityMethod("base_amount/payroll/base/contribution",
96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119
                                            """\
  def getBaseAmountQuantity(delivery_amount, base_application,
                            variation_category_list=(), **kw):
    if variation_category_list:
      for variation in variation_category_list:
        if variation.startswith('salary_range'):
          cell = delivery_amount.getSpecialiseValue().getCell(variation)
          if cell is not None:
            model_slice_min = cell.getQuantityRangeMin()
            model_slice_max = cell.getQuantityRangeMax()
            base_application = delivery_amount.getTotalPrice(0.0)
            if base_application <= model_slice_min:
              # if base_application is not in the slice range, quantity is 0
              return 0
            elif base_application-model_slice_min > 0:
              if base_application <= model_slice_max:
                quantity = base_application - model_slice_min
              elif model_slice_max:
                quantity = model_slice_max - model_slice_min
            return quantity
      quantity = delivery_amount.getGeneratedAmountQuantity(base_application)
      return quantity
    return context.getBaseAmountQuantity(delivery_amount, base_application, **kw)
  return getBaseAmountQuantity
120 121
""")

122 123 124 125 126 127 128
    self.setPayrollBaseAmountQuantityMethod("base_amount/payroll/base/income_tax",
                                            """\
  def getBaseAmountQuantity(delivery_amount, base_application,
                            variation_category_list=(), **kw):
    if variation_category_list:
      return delivery_amount.getGeneratedAmountQuantity(base_application)
    return context.getBaseAmountQuantity(delivery_amount, base_application, **kw)
129
  return getBaseAmountQuantity
130 131
""")

132

133
  @reindex
134
  def beforeTearDown(self):
135
    TestTradeModelLineMixin.beforeTearDown(self)
136 137 138 139
    transaction.abort()
    for module in (
      self.portal.organisation_module,
      self.portal.person_module,
140
      self.portal.account_module,
141 142 143 144 145 146
      self.portal.paysheet_model_module,
      self.portal.accounting_module,
      self.portal.service_module,
      self.portal.portal_simulation,):
      module.manage_delObjects(list(module.objectIds()))

147
    self.portal.business_process_module.manage_delObjects(list([x for x in self.portal.business_process_module.objectIds() if x != "erp5_default_business_process"]))
148

149
  def login(self):
150
    uf = self.getPortal().acl_users
151
    uf._doAddUser('admin', '', ['Manager', 'Assignee', 'Assignor',
152 153 154 155
                               'Associate', 'Auditor', 'Author'], [])
    user = uf.getUserById('admin').__of__(uf)
    newSecurityManager(None, user)

156
  @reindex
157 158 159 160 161 162 163 164 165 166 167 168 169 170 171 172 173 174 175 176 177 178 179 180 181 182 183
  def createCategories(self):
    """Create the categories for our test. """
    # create categories
    for cat_string in self.getNeededCategoryList() :
      base_cat = cat_string.split("/")[0]
      # if base_cat not exist, create it
      if getattr(self.getPortal().portal_categories, base_cat, None) == None:
        self.getPortal().portal_categories.newContent(\
                                          portal_type='Base Category',
                                          id=base_cat)
      path = self.getPortal().portal_categories[base_cat]
      for cat in cat_string.split("/")[1:] :
        if not cat in path.objectIds() :
          path = path.newContent(
                    portal_type='Category',
                    id=cat,
                    title=cat.replace('_', ' ').title(),)
        else:
          path = path[cat]
    # check categories have been created
    for cat_string in self.getNeededCategoryList() :
      self.assertNotEquals(None,
                self.getCategoryTool().restrictedTraverse(cat_string),
                cat_string)

  def getNeededCategoryList(self):
    """return a list of categories that should be created."""
184 185
    return ('contribution_share/employer',
            'contribution_share/employee',
186 187 188 189
            'base_amount/payroll/base/income_tax',
            'base_amount/payroll/base/contribution',
            'base_amount/payroll/report/salary/net',
            'base_amount/payroll/report/salary/gross',
190 191 192 193 194 195 196 197
            'base_amount/payroll/l10n/fr/salary_range/a',
            'base_amount/payroll/l10n/fr/salary_range/b',
            'base_amount/payroll/l10n/fr/salary_range/c',
            'base_amount/payroll/l10n/fr/salary_range/forfait',
            'base_amount/payroll/l10n/fr/salary_range/slice_0_to_200',
            'base_amount/payroll/l10n/fr/salary_range/slice_200_to_400',
            'base_amount/payroll/l10n/fr/salary_range/slice_400_to_5000',
            'base_amount/payroll/l10n/fr/salary_range/slice_600_to_800',
198 199
            'grade/worker',
            'grade/engineer',
200
            'quantity_unit/time/month',
201
            'product_line/base_salary',
202 203
            'product_line/labour',
            'product_line/state_insurance',
204
            'product_line/payroll_tax_1',
205 206 207 208 209 210 211 212 213 214 215 216 217 218 219 220 221
            'use/payroll/tax',
            'use/payroll/base_salary',
            'use/payroll/output',
            'trade_phase/payroll/france/urssaf',
            'trade_phase/payroll/france/labour',
            'time/hours',
            'salary_range/france',
            'salary_range/france/slice_a',
            'salary_range/france/slice_b',
            'salary_range/france/slice_c',
            'salary_range/france/forfait',
            'salary_range/france/slice_0_to_200',
            'salary_range/france/slice_200_to_400',
            'salary_range/france/slice_400_to_5000',
            'salary_range/france/slice_600_to_800',
            'marital_status/married',
            'marital_status/single',
222
            'group/demo_group',
223 224 225
           )

  def getBusinessTemplateList(self):
226
    return TestTradeModelLineMixin.getBusinessTemplateList(self) + ('erp5_payroll', 'erp5_core_proxy_field_legacy')
227

228 229 230 231 232 233 234 235 236 237 238
  def stepCreatePriceCurrency(self, sequence):
    sequence.edit(price_currency = self.createResource('Currency',
        title='Currency', base_unit_quantity=self.base_unit_quantity))

  def stepCreateBusinessProcess(self, sequence):
    sequence.edit(business_process=self.createBusinessProcess(title=self.id()))

  def stepCreateBusinessLink(self, sequence):
    business_process = sequence.get('business_process')
    sequence.edit(business_link=self.createBusinessLink(business_process))

239 240 241 242 243 244 245 246
  def createService(self):
    module = self.portal.getDefaultModule(portal_type='Service')
    return module.newContent(portal_type='Service')

  def stepCreateUrssafService(self, sequence=None, **kw):
    node = self.createService()
    node.edit(title='Urssaf',
        product_line='state_insurance', quantity_unit='time/month',
247
        variation_base_category_list=['contribution_share', 'salary_range'],
248
        use='payroll/tax')
249 250 251 252 253 254
    node.setVariationCategoryList(['contribution_share/employee',
                                   'contribution_share/employer',
                                    'salary_range/france/slice_0_to_200',
                                    'salary_range/france/slice_200_to_400',
                                    'salary_range/france/slice_400_to_5000',
                                    'salary_range/france/slice_600_to_800',
Fabien Morin's avatar
Fabien Morin committed
255
                                   ])
256 257 258 259 260 261 262 263 264 265 266 267 268 269 270 271 272 273 274 275
    sequence.edit(urssaf_service = node)

  def stepCreateLabourService(self, sequence=None, **kw):
    node = self.createService()
    node.edit(title='Labour', quantity_unit='time/month',
        product_line='labour', use='payroll/base_salary')
    sequence.edit(labour_service = node)

  def stepCreateLabourOutputService(self, sequence=None, **kw):
    '''In case we want to have the labour line in the model, the use category
    should not be base_salary (=input) because in this case the line will be
    added on each calculation'''
    node = self.createService()
    node.edit(title='Labour', quantity_unit='time/month',
        product_line='labour', use='payroll/output')
    sequence.edit(labour_service = node)

  def stepCreateBonusService(self, sequence=None, **kw):
    node = self.createService()
    node.edit(title='Bonus', quantity_unit='time/month',
276 277 278 279
        variation_base_category_list=['contribution_share'],
        product_line='labour', use='payroll/base_salary')
    node.setVariationCategoryList(['contribution_share/employee',
                                   'contribution_share/employer'])
280 281 282 283 284
    sequence.edit(bonus_service = node)

  def stepCreateOldAgeInsuranaceService(self, sequence=None, **kw):
    node = self.createService()
    node.edit(title='Old Age Insurance', quantity_unit='time/month',
285 286 287 288
        variation_base_category_list=['contribution_share', 'salary_range'],
        product_line='state_insurance', use='payroll/tax')
    node.setVariationCategoryList(['contribution_share/employee',
                                   'contribution_share/employer'])
289 290 291 292 293
    sequence.edit(old_age_insurance_service = node)

  def stepCreateSicknessInsuranceService(self, sequence=None, **kw):
    node = self.createService()
    node.edit(title='Sickness Insurance', quantity_unit='time/month',
294 295 296 297
        variation_base_category_list=['contribution_share', 'salary_range'],
        product_line='state_insurance', use='payroll/tax')
    node.setVariationCategoryList(['contribution_share/employee',
                                   'contribution_share/employer'])
298 299 300 301 302 303 304 305 306 307 308 309 310 311 312 313 314 315 316 317 318 319 320 321 322 323 324 325 326 327 328
    sequence.edit(sickness_insurance_service = node)

  def createModel(self):
    module = self.portal.getDefaultModule(portal_type='Pay Sheet Model')
    return module.newContent(portal_type='Pay Sheet Model')

  def createPerson(self):
    module = self.portal.getDefaultModule(portal_type='Person')
    return module.newContent(portal_type='Person')

  def createOrganisation(self):
    module = self.portal.getDefaultModule(portal_type='Organisation')
    return module.newContent(portal_type='Organisation')

  def stepCreateEmployee(self, sequence=None, **kw):
    employer = sequence.get('employer')
    node = self.createPerson()
    node.edit(title='Employee',
              career_subordination_value=employer)
    sequence.edit(employee = node)

  def stepCreateEmployer(self, sequence=None, **kw):
    node = self.createOrganisation()
    node.edit(title='Employer')
    sequence.edit(employer = node)

  def stepCreateBasicModel(self, sequence=None, **kw):
    model = self.createModel()
    employer = sequence.get('employer')
    employee = sequence.get('employee')
    model.edit(destination_section_value=employer,
329 330 331 332 333 334
        source_section_value=employee,
        effective_date=DateTime(2009,01,01),
        expiration_date=DateTime(2009,12,31),
        version='001',
        reference='basic_model',
    )
335
    sequence.edit(model = model)
336 337

  def addSlice(self, model, slice, min_value, max_value, base_id='cell'):
338
    '''add a new slice in the model'''
339
    slice_value = model.newCell(slice, portal_type='Pay Sheet Model Slice',
340
        base_id=base_id)
341 342 343
    slice_value.setQuantityRangeMax(max_value)
    slice_value.setQuantityRangeMin(min_value)
    return slice_value
Fabien Morin's avatar
Fabien Morin committed
344

345 346 347 348
  def stepSetCurrencyOnModel(self, sequence=None, **kw):
    model = sequence.get('model')
    currency = sequence.get('price_currency')
    model.setPriceCurrencyValue(currency)
349

350 351 352 353 354 355 356 357 358 359 360 361 362 363 364 365 366 367 368 369 370 371 372 373
  def stepCreateModelWithSlices(self, sequence=None, **kw):
    model = self.createModel()
    employer = sequence.get('employer')
    employee = sequence.get('employee')
    model.edit(destination_section_value=employer,
        source_section_value=employee)
    model.setVariationSettingsCategoryList(\
        ['salary_range/france'])
    self.addSlice(model, 'salary_range/france/slice_0_to_200', 0, 200)
    self.addSlice(model, 'salary_range/france/slice_200_to_400', 200, 400)
    self.addSlice(model, 'salary_range/france/slice_400_to_5000', 400, 5000)
    self.addSlice(model, 'salary_range/france/slice_600_to_800', 600, 800)
    sequence.edit(model = model)

  def createModelLine(self, document, **kw):
    return document.newContent(portal_type='Pay Sheet Model Line', **kw)

  def stepModelCreateUrssafModelLine(self, sequence=None, **kw):
    model = sequence.get('model')
    model_line = self.createModelLine(model)
    model_line.edit(title='Urssaf',
                    reference='urssaf_model_line',
                    trade_phase='payroll/france/urssaf',
                    resource_value=sequence.get('urssaf_service'),
374 375 376 377 378
                    variation_category_list=['contribution_share/employee',
                                             'contribution_share/employer'],
                    base_application_list=[ 'base_amount/payroll/base/contribution'],
                    base_contribution_list=['base_amount/payroll/base/income_tax'])
    #model_line.setQuantity(0.0)
379 380 381 382 383 384 385 386 387
    sequence.edit(urssaf_model_line = model_line)

  def stepModelCreateUrssafModelLineWithSlices(self, sequence=None, **kw):
    model = sequence.get('model')
    model_line = self.createModelLine(model)
    model_line.edit(title='Urssaf',
                    reference='urssaf_model_line_2',
                    trade_phase='payroll/france/urssaf',
                    resource_value=sequence.get('urssaf_service'),
388 389 390 391 392 393
                    variation_category_list=['contribution_share/employee',
                                       'contribution_share/employer',
                                       'salary_range/france/slice_0_to_200',
                                       'salary_range/france/slice_200_to_400',
                                       'salary_range/france/slice_400_to_5000'],
                    base_application_list=[ 'base_amount/payroll/base/contribution',],
394
                    base_contribution_list=['base_amount/payroll/base/income_tax'])
395 396 397 398 399 400 401 402 403
    sequence.edit(urssaf_model_line_with_slices = model_line)

  def stepModelCreateUrssafModelLineWithComplexSlices(self, sequence=None, **kw):
    model = sequence.get('model')
    model_line = self.createModelLine(model)
    model_line.edit(title='Urssaf',
                    reference='urssaf_model_line_3',
                    trade_phase='payroll/france/urssaf',
                    resource_value=sequence.get('urssaf_service'),
404 405 406 407 408
                    variation_category_list=['contribution_share/employee',
                                       'contribution_share/employer',
                                       'salary_range/france/slice_200_to_400',
                                       'salary_range/france/slice_600_to_800'],
                    base_application_list=[ 'base_amount/payroll/base/contribution'],
409
                    base_contribution_list=['base_amount/payroll/base/income_tax'])
410
    #model_line.setQuantity(0.0)
411 412 413 414
    sequence.edit(urssaf_model_line_with_slices = model_line)

  def stepUrssafModelLineCreateMovements(self, sequence=None, **kw):
    model_line = sequence.get('urssaf_model_line')
415
    cell1 = model_line.newCell('contribution_share/employee',
416 417
                               portal_type='Pay Sheet Model Cell',
                               base_id='movement',
418
                               base_application = "contribution_share/employee",
419
                               mapped_value_property_list=('quantity', 'price'))
420 421
    cell1.edit(price=0.1, contribution_share='employee', quantity=None)
    cell2 = model_line.newCell('contribution_share/employer',
422 423
                               portal_type='Pay Sheet Model Cell',
                               base_id='movement',
424
                               base_application = "contribution_share/employer",
425
                               mapped_value_property_list=('quantity', 'price'))
426
    cell2.edit(price=0.5, contribution_share='employer', quantity=None)
427 428 429

  def stepUrssafModelLineCreateMovementsWithQuantityOnly(self, sequence=None, **kw):
    model_line = sequence.get('urssaf_model_line')
430 431 432 433 434 435 436 437 438 439 440 441 442 443 444 445 446
    cell1 = model_line.newCell('contribution_share/employee',
        portal_type='Pay Sheet Model Cell',
        base_id='movement',
        base_application = "contribution_share/employee",
        mapped_value_property_list=('quantity', 'price'))
    cell1.edit(quantity=-100.0, contribution_share='employee')
    cell2 = model_line.newCell('contribution_share/employer',
        portal_type='Pay Sheet Model Cell',
        base_id='movement',
        base_application = "contribution_share/employer",
        mapped_value_property_list=('quantity', 'price'))
    cell2.edit(quantity=-200.0, contribution_share='employer')

  def stepUrssafModelLineWithSlicesCreateMovements(self, sequence=None, **kw):
    model_line = sequence.get('urssaf_model_line_with_slices')
    cell1 = model_line.newCell('contribution_share/employee',
                               'salary_range/france/slice_0_to_200',
447 448
                               portal_type='Pay Sheet Model Cell',
                               base_id='movement',
449 450
                               base_application_list=['contribution_share/employee',
                                                      'salary_range/france/slice_0_to_200',],
451
                               mapped_value_property_list=('quantity', 'price'))
452 453 454 455
    cell1.edit(price=0.1, contribution_share='employee', quantity=None,
               salary_range='france/slice_0_to_200')
    cell2 = model_line.newCell('contribution_share/employer',
                               'salary_range/france/slice_0_to_200',
456 457
                               portal_type='Pay Sheet Model Cell',
                               base_id='movement',
458 459
                               base_application_list=['contribution_share/employer',
                                                      'salary_range/france/slice_0_to_200',],
460
                               mapped_value_property_list=('quantity', 'price'))
461 462 463 464 465 466 467 468 469 470 471 472 473 474 475 476 477 478 479 480 481 482 483 484 485 486 487 488 489 490 491 492 493 494 495 496 497 498
    cell2.edit(price=0.2, contribution_share='employer', quantity=None,
        salary_range='france/slice_0_to_200')
    cell3 = model_line.newCell('contribution_share/employee',
                               'salary_range/france/slice_200_to_400',
                               portal_type='Pay Sheet Model Cell',
                               base_id='movement',
                               base_application_list=['contribution_share/employee',
                                                      'salary_range/france/slice_200_to_400',],
                               mapped_value_property_list=('quantity', 'price'))
    cell3.edit(price=0.3, contribution_share='employee', quantity=None,
        salary_range='france/slice_200_to_400')
    cell4 = model_line.newCell('contribution_share/employer',
                               'salary_range/france/slice_200_to_400',
                               portal_type='Pay Sheet Model Cell',
                               base_id='movement',
                               base_application_list=['contribution_share/employer',
                                                      'salary_range/france/slice_200_to_400',],
                               mapped_value_property_list=('quantity', 'price'))
    cell4.edit(price=0.4, contribution_share='employer', quantity=None,
        salary_range='france/slice_200_to_400')
    cell5 = model_line.newCell('contribution_share/employee',
                               'salary_range/france/slice_400_to_5000',
                               portal_type='Pay Sheet Model Cell',
                               base_id='movement',
                               base_application_list=['contribution_share/employee',
                                                      'salary_range/france/slice_400_to_5000',],
                               mapped_value_property_list=('quantity', 'price'))
    cell5.edit(price=0.5, contribution_share='employee', quantity=None,
        salary_range='france/slice_400_to_5000')
    cell6 = model_line.newCell('contribution_share/employer',
                               'salary_range/france/slice_400_to_5000',
                               portal_type='Pay Sheet Model Cell',
                               base_id='movement',
                               base_application_list=['contribution_share/employer',
                                                      'salary_range/france/slice_400_to_5000',],
                               mapped_value_property_list=('quantity', 'price'))
    cell6.edit(price=0.6, contribution_share='employer', quantity=None,
               salary_range='france/slice_400_to_5000')
499 500 501 502

  def stepUrssafModelLineWithComplexSlicesCreateMovements(self,
      sequence=None, **kw):
    model_line = sequence.get('urssaf_model_line_with_slices')
503 504 505 506 507 508 509 510 511 512 513 514 515 516 517 518 519 520 521 522 523 524 525 526 527 528 529 530 531 532 533 534 535 536 537 538 539 540 541 542 543 544 545 546 547 548 549 550 551 552 553
    cell1 = model_line.newCell('contribution_share/employee',
                               'salary_range/france/slice_200_to_400',
                               portal_type='Pay Sheet Model Cell',
                               base_id='movement',
                               base_application_list=['contribution_share/employee',
                                                      'salary_range/france/slice_200_to_400',],
                               mapped_value_property_list=('quantity', 'price'))
    cell1.edit(price=0.1, contribution_share='employee', quantity=None,
               salary_range='france/slice_200_to_400')
    cell2 = model_line.newCell('contribution_share/employer',
                               'salary_range/france/slice_200_to_400',
                               portal_type='Pay Sheet Model Cell',
                               base_id='movement',
                               base_application_list=['contribution_share/employer',
                                                      'salary_range/france/slice_200_to_400',],
                               mapped_value_property_list=('quantity', 'price'))
    cell2.edit(price=0.2, contribution_share='employer', quantity=None,
        salary_range='france/slice_200_to_400')
    cell3 = model_line.newCell('contribution_share/employee',
                               'salary_range/france/slice_600_to_800',
                               portal_type='Pay Sheet Model Cell',
                               base_id='movement',
                               base_application_list=['contribution_share/employee',
                                                      'salary_range/france/slice_600_to_800',],
                               mapped_value_property_list=('quantity', 'price'))
    cell3.edit(price=0.3, contribution_share='employee', quantity=None,
               salary_range='france/slice_600_to_800')
    cell4 = model_line.newCell('contribution_share/employer',
                               'salary_range/france/slice_600_to_800',
                               portal_type='Pay Sheet Model Cell',
                               base_id='movement',
                               base_application_list=['contribution_share/employer',
                                                      'salary_range/france/slice_600_to_800',],
                               mapped_value_property_list=('quantity', 'price'))
    cell4.edit(price=0.4, contribution_share='employer', quantity=None,
               salary_range='france/slice_600_to_800')

  def stepPaysheetUrssafModelLineCreateMovements(self, sequence=None, **kw):
    model_line = sequence.get('urssaf_model_line')
    cell1 = model_line.newCell('contribution_share/employee',
        portal_type='Pay Sheet Model Cell',
        base_id='movement',
        base_application = "contribution_share/employee",
        mapped_value_property_list=('quantity', 'price'))
    cell1.edit(price=0.3, contribution_share='employee', quantity=None)
    cell2 = model_line.newCell('contribution_share/employer',
        portal_type='Pay Sheet Model Cell',
        base_id='movement',
        base_application = "contribution_share/employer",
        mapped_value_property_list=('quantity', 'price'))
    cell2.edit(price=0.7, contribution_share='employer', quantity=None)
554 555 556 557 558 559 560 561 562 563 564 565

  def createPaysheet(self, sequence=None, **kw):
    module = self.portal.getDefaultModule(portal_type='Pay Sheet Transaction')
    return module.newContent(portal_type='Pay Sheet Transaction')

  def stepCreateBasicPaysheet(self, sequence=None, **kw):
    paysheet = self.createPaysheet()
    paysheet.edit(title='test 1',
                  specialise_value=sequence.get('model'),
                  source_section_value=sequence.get('employee'),
                  destination_section_value=sequence.get('employer'),
                  resource_value=sequence.get('price_currency'),
566 567
                  start_date=DateTime(2009,06,01),
                  stop_date=DateTime(2009,06,30))
568 569 570
    sequence.edit(paysheet = paysheet)

  def createPaysheetLine(self, document, **kw):
571
    return document.newContent(portal_type='Pay Sheet Line', quantity=0.0, **kw)
572 573 574 575 576 577 578 579

  def stepPaysheetCreateLabourPaySheetLine(self, sequence=None, **kw):
    paysheet = sequence.get('paysheet')
    paysheet_line = self.createPaysheetLine(paysheet)
    paysheet_line.edit(title='Labour',
                    price=20,
                    quantity=150,
                    resource_value=sequence.get('labour_service'),
580
                    base_contribution_list=['base_amount/payroll/base/contribution',
581 582 583 584
                                            'base_amount/payroll/l10n/fr/salary_range/slice_0_to_200',
                                            'base_amount/payroll/l10n/fr/salary_range/slice_200_to_400',
                                            'base_amount/payroll/l10n/fr/salary_range/slice_400_to_5000',
                                            'base_amount/payroll/l10n/fr/salary_range/slice_600_to_800',
585
                                            'base_amount/payroll/report/salary/gross'])
586 587 588 589 590 591
    sequence.edit(labour_paysheet_line = paysheet_line)

  def stepPaysheetCreateBonusPaySheetLine(self, sequence=None, **kw):
    paysheet = sequence.get('paysheet')
    paysheet_line = self.createPaysheetLine(paysheet)
    paysheet_line.edit(title='Bonus',
592
                       resource_value=sequence.get('bonus_service'),
593 594
                       quantity=1000, price=1,
                       base_contribution_list=[ 'base_amount/payroll/base/contribution'])
595 596
    sequence.edit(bonus_paysheet_line = paysheet_line)

597
  def checkUpdateAggregatedAmountListReturn(self, paysheet,
598
      expected_movement_to_delete_count, expected_movement_to_add_count):
599
    movement_dict = paysheet.updateAggregatedAmountList()
600 601 602 603 604 605 606 607
    movement_to_delete = movement_dict['movement_to_delete_list']
    movement_to_add = movement_dict['movement_to_add_list']
    self.assertEquals(len(movement_to_delete),
        expected_movement_to_delete_count)
    self.assertEquals(len(movement_to_add), expected_movement_to_add_count)

  def stepCheckUpdateAggregatedAmountListReturn(self, sequence=None, **kw):
    paysheet = sequence.get('paysheet')
608
    self.checkUpdateAggregatedAmountListReturn(paysheet, 0, 2)
609 610 611 612

  def stepCheckUpdateAggregatedAmountListReturnUsingSlices(self,
      sequence=None, **kw):
    paysheet = sequence.get('paysheet')
613
    self.checkUpdateAggregatedAmountListReturn(paysheet, 0, 6)
614 615 616 617

  def stepCheckUpdateAggregatedAmountListReturnUsingComplexSlices(self,
      sequence=None, **kw):
    paysheet = sequence.get('paysheet')
618
    self.checkUpdateAggregatedAmountListReturn(paysheet, 0, 4)
619 620 621 622

  def stepCheckUpdateAggregatedAmountListReturnUsingPredicate(self,
      sequence=None, **kw):
    paysheet = sequence.get('paysheet')
623
    self.checkUpdateAggregatedAmountListReturn(paysheet, 0, 4)
624 625 626 627 628 629 630 631

  def stepCheckUpdateAggregatedAmountListReturnAfterChangePredicate(self,
      sequence=None, **kw):
    # the marital status of the employe is change in the way that sickness
    # insurance model_line will not be applied but old age insurance yes.
    # So two movements will be deleted (from sickness insurance) and two should
    # be added (from old age insurance)
    paysheet = sequence.get('paysheet')
632
    self.checkUpdateAggregatedAmountListReturn(paysheet, 2, 2)
633 634 635 636

  def stepCheckUpdateAggregatedAmountListReturnAfterRemoveLine(self,
      sequence=None, **kw):
    paysheet = sequence.get('paysheet')
637
    self.checkUpdateAggregatedAmountListReturn(paysheet, 2, 0)
638 639 640 641 642 643 644 645

  def stepPaysheetApplyTransformation(self, sequence=None, **kw):
    paysheet = sequence.get('paysheet')
    paysheet.applyTransformation()

  def stepCheckPaysheetLineAreCreated(self, sequence=None, **kw):
    paysheet = sequence.get('paysheet')
    paysheet_line_list = paysheet.contentValues(portal_type='Pay Sheet Line')
646
    self.assertEqual(len(paysheet_line_list), 2)
647 648 649 650 651 652 653 654
    self.assertEqual(len(paysheet.getMovementList(portal_type=\
        'Pay Sheet Cell')), 2) # 2 because labour line contain no movement

  def stepCheckNoPaysheetLineAreCreated(self, sequence=None, **kw):
    paysheet = sequence.get('paysheet')
    paysheet_line_list = paysheet.contentValues(portal_type='Pay Sheet Line')
    self.assertEqual(len(paysheet_line_list), 0)
    self.assertEqual(len(paysheet.getMovementList(portal_type=\
655
        'Pay Sheet Cell')), 0) # 2 because labour line contain no movement
656 657 658 659

  def stepCheckPaysheetLineAreCreatedUsingBonus(self, sequence=None, **kw):
    paysheet = sequence.get('paysheet')
    paysheet_line_list = paysheet.contentValues(portal_type='Pay Sheet Line')
660
    self.assertEqual(len(paysheet_line_list), 3)
661
    self.assertEqual(len(paysheet.getMovementList(portal_type=\
662 663
        'Pay Sheet Cell')), 2) # 2 because labour line contain no movement
                               # 2 for urssaf
664 665 666 667 668 669 670 671 672 673 674

  def stepCheckThereIsOnlyOnePaysheetLine(self, sequence=None, **kw):
    paysheet = sequence.get('paysheet')
    paysheet_line_list = paysheet.contentValues(portal_type='Pay Sheet Line')
    self.assertEqual(len(paysheet_line_list), 1)
    self.assertEqual(len(paysheet.getMovementList(portal_type=\
        'Pay Sheet Cell')), 0) # 0 because labour line contain no movement

  def stepCheckPaysheetLineAreCreatedUsingSlices(self, sequence=None, **kw):
    paysheet = sequence.get('paysheet')
    paysheet_line_list = paysheet.contentValues(portal_type='Pay Sheet Line')
675
    self.assertEqual(len(paysheet_line_list), 2)
676 677
    self.assertEqual(len(paysheet.getMovementList(portal_type=\
        'Pay Sheet Cell')), 6) # 6 because labour line contain no movement and
678
                               # because of the 3 slice and 2 contribution_shares
679 680 681 682

  def stepCheckPaysheetLineAreCreatedUsingComplexSlices(self, sequence=None, **kw):
    paysheet = sequence.get('paysheet')
    paysheet_line_list = paysheet.contentValues(portal_type='Pay Sheet Line')
683
    self.assertEqual(len(paysheet_line_list), 2)
684 685
    self.assertEqual(len(paysheet.getMovementList(portal_type=\
        'Pay Sheet Cell')), 4) # 4 because labour line contain no movement and
686
                               # because of the 2 slice and 2 contribution_shares
687 688 689 690

  def stepCheckPaysheetLineAreCreatedUsingWith3Lines(self, sequence=None, **kw):
    paysheet = sequence.get('paysheet')
    paysheet_line_list = paysheet.contentValues(portal_type='Pay Sheet Line')
691
    self.assertEqual(len(paysheet_line_list), 3)
692
    self.assertEqual(len(paysheet.getMovementList(portal_type=\
693 694 695 696
        'Pay Sheet Cell')), 4) # 4 because labour line contain no movement and
                               # because of the two lines and 2 contribution_shares
                               # (urssaf and sickness insurance. old age
                               # insurance does not match predicate)
697

698 699 700
  def stepCheckPaysheetLineAreCreatedAfterUpdateWithLinesWithSameResource(self, sequence=None, **kw):
    paysheet = sequence.get('paysheet')
    paysheet_line_list = paysheet.contentValues(portal_type='Pay Sheet Line')
701
    self.assertEqual(len(paysheet_line_list), 3)
702 703
    self.assertEqual(len(paysheet.getMovementList(portal_type=\
        'Pay Sheet Cell')), 8) # 8 because labour line contain no movement and
704 705
                               # because of the 3 slice and 2 contribution_shares
                               # + the first model line with 2 contribution_shares
706

707 708 709 710 711 712
  def stepCheckPaysheetLineAmounts(self, sequence=None, **kw):
    paysheet = sequence.get('paysheet')
    paysheet_line_list = paysheet.contentValues(portal_type='Pay Sheet Line')
    for paysheet_line in paysheet_line_list:
      service = paysheet_line.getResourceTitle()
      if service == 'Urssaf':
713 714 715 716 717 718
        cell1 = paysheet_line.getCell('contribution_share/employee')
        self.assertEquals(cell1.getQuantity(), 3000)
        self.assertEquals(cell1.getPrice(), 0.1)
        cell2 = paysheet_line.getCell('contribution_share/employer')
        self.assertEquals(cell2.getQuantity(), 3000)
        self.assertEquals(cell2.getPrice(), 0.5)
719 720 721 722 723 724 725 726 727 728 729
      elif service == 'Labour':
        self.assertEqual(paysheet_line.getTotalPrice(), 3000.0)
      else:
        self.fail("Unknown service for line %s" % paysheet_line.getTitle())

  def stepCheckPaysheetLineWithBonusAmounts(self, sequence=None, **kw):
    paysheet = sequence.get('paysheet')
    paysheet_line_list = paysheet.contentValues(portal_type='Pay Sheet Line')
    for paysheet_line in paysheet_line_list:
      service = paysheet_line.getResourceTitle()
      if service == 'Urssaf':
730 731 732 733 734 735
        cell1 = paysheet_line.getCell('contribution_share/employee')
        self.assertEquals(cell1.getQuantity(), 4000)
        self.assertEquals(cell1.getPrice(), 0.1)
        cell2 = paysheet_line.getCell('contribution_share/employer')
        self.assertEquals(cell2.getQuantity(), 4000)
        self.assertEquals(cell2.getPrice(), 0.5)
736 737 738
      elif service == 'Labour':
        self.assertEqual(paysheet_line.getTotalPrice(), 3000.0)
      elif service == 'Bonus':
739
        self.assertEqual(paysheet_line.getTotalPrice(), 1000.0)
740 741 742 743 744 745 746 747 748
      else:
        self.fail("Unknown service for line %s" % paysheet_line.getTitle())

  def stepCheckPaysheetLineAmountsWithQuantityOnly(self, sequence=None, **kw):
    paysheet = sequence.get('paysheet')
    paysheet_line_list = paysheet.contentValues(portal_type='Pay Sheet Line')
    for paysheet_line in paysheet_line_list:
      service = paysheet_line.getResourceTitle()
      if service == 'Urssaf':
749 750 751 752 753 754 755 756
        cell1 = paysheet_line.getCell('contribution_share/employee')
        # XXX-Aurel quantity from model line is multiply by total price of labour line
        # price remains None
        self.assertEquals(cell1.getQuantity(), -300000)
        self.assertEquals(cell1.getPrice(), None)
        cell2 = paysheet_line.getCell('contribution_share/employer')
        self.assertEquals(cell2.getQuantity(), -600000)
        self.assertEquals(cell2.getPrice(), None)
757 758 759 760 761 762 763 764 765 766 767
      elif service == 'Labour':
        self.assertEqual(paysheet_line.getTotalPrice(), 3000.0)
      else:
        self.fail("Unknown service for line %s" % paysheet_line.getTitle())

  def stepCheckPaysheetLineAmountsUsingSlices(self, sequence=None, **kw):
    paysheet = sequence.get('paysheet')
    paysheet_line_list = paysheet.contentValues(portal_type='Pay Sheet Line')
    for paysheet_line in paysheet_line_list:
      service = paysheet_line.getResourceTitle()
      if service == 'Urssaf':
768 769 770 771 772 773 774 775 776 777 778 779 780 781 782 783 784 785 786 787 788 789 790 791 792 793 794
        if paysheet_line.getSalaryRange() == 'france/slice_0_to_200':
          cell1 = paysheet_line.getCell('contribution_share/employee',
              'salary_range/france/slice_0_to_200')
          self.assertEquals(cell1.getQuantity(), 200)
          self.assertEquals(cell1.getPrice(), 0.1)
          cell2 = paysheet_line.getCell('contribution_share/employer',
              'salary_range/france/slice_0_to_200')
          self.assertEquals(cell2.getQuantity(), 200)
          self.assertEquals(cell2.getPrice(), 0.2)
        elif paysheet_line.getSalaryRange() == 'france/slice_200_to_400':
          cell3 = paysheet_line.getCell('contribution_share/employee',
              'salary_range/france/slice_200_to_400')
          self.assertEquals(cell3.getQuantity(), 200)
          self.assertEquals(cell3.getPrice(), 0.3)
          cell4 = paysheet_line.getCell('contribution_share/employer',
              'salary_range/france/slice_200_to_400')
          self.assertEquals(cell4.getQuantity(), 200)
          self.assertEquals(cell4.getPrice(), 0.4)
        elif paysheet_line.getSalaryRange() == 'france/slice_400_to_5000':
          cell5 = paysheet_line.getCell('contribution_share/employee',
              'salary_range/france/slice_400_to_5000')
          self.assertEquals(cell5.getQuantity(), 2600)
          self.assertEquals(cell5.getPrice(), 0.5)
          cell6 = paysheet_line.getCell('contribution_share/employer',
              'salary_range/france/slice_400_to_5000')
          self.assertEquals(cell6.getQuantity(), 2600)
          self.assertEquals(cell6.getPrice(), 0.6)
795
        else:
796
          self.fail("Unknown salary range for line %s" % paysheet_line.getTitle())
797 798 799 800 801 802 803 804 805 806 807
      elif service == 'Labour':
        self.assertEqual(paysheet_line.getTotalPrice(), 3000.0)
      else:
        self.fail("Unknown service for line %s" % paysheet_line.getTitle())

  def stepCheckPaysheetLineAmountsUsingComplexSlices(self, sequence=None, **kw):
    paysheet = sequence.get('paysheet')
    paysheet_line_list = paysheet.contentValues(portal_type='Pay Sheet Line')
    for paysheet_line in paysheet_line_list:
      service = paysheet_line.getResourceTitle()
      if service == 'Urssaf':
808 809 810 811 812 813 814 815 816 817 818 819 820 821 822 823 824 825
        if paysheet_line.getSalaryRange() == 'france/slice_200_to_400':
          cell1 = paysheet_line.getCell('contribution_share/employee',
              'salary_range/france/slice_200_to_400')
          self.assertEquals(cell1.getQuantity(), 200)
          self.assertEquals(cell1.getPrice(), 0.1)
          cell2 = paysheet_line.getCell('contribution_share/employer',
              'salary_range/france/slice_200_to_400')
          self.assertEquals(cell2.getQuantity(), 200)
          self.assertEquals(cell2.getPrice(), 0.2)
        elif paysheet_line.getSalaryRange() == 'france/slice_600_to_800':
          cell3 = paysheet_line.getCell('contribution_share/employee',
              'salary_range/france/slice_600_to_800')
          self.assertEquals(cell3.getQuantity(), 200)
          self.assertEquals(cell3.getPrice(), 0.3)
          cell4 = paysheet_line.getCell('contribution_share/employer',
              'salary_range/france/slice_600_to_800')
          self.assertEquals(cell4.getQuantity(), 200)
          self.assertEquals(cell4.getPrice(), 0.4)
826
        else:
827
          self.fail("Unknown salary range for line %s" % paysheet_line.getTitle())
828 829 830 831 832
      elif service == 'Labour':
        self.assertEqual(paysheet_line.getTotalPrice(), 3000.0)
      else:
        self.fail("Unknown service for line %s" % paysheet_line.getTitle())

833
  def stepCheckPaysheetLineAmountsAfterUpdateWithLinesWithSameResource(self,
834
      sequence=None, **kw):
Fabien Morin's avatar
Fabien Morin committed
835 836 837 838 839
    paysheet = sequence.get('paysheet')
    paysheet_line_list = paysheet.contentValues(portal_type='Pay Sheet Line')
    for paysheet_line in paysheet_line_list:
      service = paysheet_line.getResourceTitle()
      if service == 'Urssaf':
840 841 842 843 844 845 846 847 848 849 850 851 852 853 854 855 856 857 858 859 860 861 862 863 864 865 866 867 868 869 870 871 872 873
        if paysheet_line.getSalaryRange() == 'france/slice_0_to_200':
          cell1 = paysheet_line.getCell('contribution_share/employee',
                                        'salary_range/france/slice_0_to_200')
          self.assertEquals(cell1.getQuantity(), 200)
          self.assertEquals(cell1.getPrice(), 0.1)
          cell2 = paysheet_line.getCell('contribution_share/employer',
              'salary_range/france/slice_0_to_200')
          self.assertEquals(cell2.getQuantity(), 200)
          self.assertEquals(cell2.getPrice(), 0.2)
        elif paysheet_line.getSalaryRange() == 'france/slice_200_to_400':
          cell3 = paysheet_line.getCell('contribution_share/employee',
              'salary_range/france/slice_200_to_400')
          self.assertEquals(cell3.getQuantity(), 200)
          self.assertEquals(cell3.getPrice(), 0.3)
          cell4 = paysheet_line.getCell('contribution_share/employer',
              'salary_range/france/slice_200_to_400')
          self.assertEquals(cell4.getQuantity(), 200)
          self.assertEquals(cell4.getPrice(), 0.4)
        elif paysheet_line.getSalaryRange() == 'france/slice_400_to_5000':
          cell5 = paysheet_line.getCell('contribution_share/employee',
              'salary_range/france/slice_400_to_5000')
          self.assertEquals(cell5.getQuantity(), 2600)
          self.assertEquals(cell5.getPrice(), 0.5)
          cell6 = paysheet_line.getCell('contribution_share/employer',
              'salary_range/france/slice_400_to_5000')
          self.assertEquals(cell6.getQuantity(), 2600)
          self.assertEquals(cell6.getPrice(), 0.6)
        else:
          cell1 = paysheet_line.getCell('contribution_share/employee')
          self.assertEquals(cell1.getQuantity(), 3000)
          self.assertEquals(cell1.getPrice(), 0.1)
          cell2 = paysheet_line.getCell('contribution_share/employer')
          self.assertEquals(cell2.getQuantity(), 3000)
          self.assertEquals(cell2.getPrice(), 0.5)
Fabien Morin's avatar
Fabien Morin committed
874 875 876 877 878
      elif service == 'Labour':
        self.assertEqual(paysheet_line.getTotalPrice(), 3000.0)
      else:
        self.fail("Unknown service for line %s" % paysheet_line.getTitle())

879 880 881 882 883 884
  def stepCheckPaysheetLineAmountsWithSicknessInsuranceAndUrssaf(self, sequence=None, **kw):
    paysheet = sequence.get('paysheet')
    paysheet_line_list = paysheet.contentValues(portal_type='Pay Sheet Line')
    for paysheet_line in paysheet_line_list:
      service = paysheet_line.getResourceTitle()
      if service == 'Urssaf':
885 886 887 888 889 890
        cell1 = paysheet_line.getCell('contribution_share/employee')
        self.assertEquals(cell1.getQuantity(), 3000)
        self.assertEquals(cell1.getPrice(), 0.1)
        cell2 = paysheet_line.getCell('contribution_share/employer')
        self.assertEquals(cell2.getQuantity(), 3000)
        self.assertEquals(cell2.getPrice(), 0.5)
891 892 893
      elif service == 'Labour':
        self.assertEqual(paysheet_line.getTotalPrice(), 3000.0)
      elif service == 'Sickness Insurance':
894 895 896 897 898 899
        cell1 = paysheet_line.getCell('contribution_share/employee')
        self.assertEquals(cell1.getQuantity(), 3000)
        self.assertEquals(cell1.getPrice(), 0.4)
        cell2 = paysheet_line.getCell('contribution_share/employer')
        self.assertEquals(cell2.getQuantity(), 3000)
        self.assertEquals(cell2.getPrice(), 0.3)
900 901 902 903 904 905 906 907 908
      else:
        self.fail("Unknown service for line %s" % paysheet_line.getTitle())

  def stepCheckPaysheetLineAmountsWithOldAgeInsuranceAndUrssaf(self, sequence=None, **kw):
    paysheet = sequence.get('paysheet')
    paysheet_line_list = paysheet.contentValues(portal_type='Pay Sheet Line')
    for paysheet_line in paysheet_line_list:
      service = paysheet_line.getResourceTitle()
      if service == 'Urssaf':
909 910 911 912 913 914
        cell1 = paysheet_line.getCell('contribution_share/employee')
        self.assertEquals(cell1.getQuantity(), 3000)
        self.assertEquals(cell1.getPrice(), 0.1)
        cell2 = paysheet_line.getCell('contribution_share/employer')
        self.assertEquals(cell2.getQuantity(), 3000)
        self.assertEquals(cell2.getPrice(), 0.5)
915 916 917
      elif service == 'Labour':
        self.assertEqual(paysheet_line.getTotalPrice(), 3000.0)
      elif service == 'Old Age Insurance':
918 919 920 921 922 923
        cell1 = paysheet_line.getCell('contribution_share/employee')
        self.assertEquals(cell1.getQuantity(), 3000)
        self.assertEquals(cell1.getPrice(), 0.5)
        cell2 = paysheet_line.getCell('contribution_share/employer')
        self.assertEquals(cell2.getQuantity(), 3000)
        self.assertEquals(cell2.getPrice(), 0.8)
924 925 926 927 928
      else:
        self.fail("Unknown service for line %s" % paysheet_line.getTitle())

  def stepCheckUpdateAggregatedAmountListReturnNothing(self, sequence=None, **kw):
    paysheet = sequence.get('paysheet')
929
    movement_dict = paysheet.updateAggregatedAmountList()
930 931 932 933 934 935 936 937 938 939
    movement_to_delete = movement_dict['movement_to_delete_list']
    movement_to_add = movement_dict['movement_to_add_list']
    self.assertEquals(len(movement_to_delete), 0)
    self.assertEquals(len(movement_to_add), 0)

  def stepCreateUrssafRoubaixOrganisation(self, sequence=None, **kw):
    node = self.createOrganisation()
    node.edit(title='Urssaf de Roubaix Tourcoing')
    sequence.edit(urssaf_roubaix = node)

940 941 942
  def stepModifyBusinessLinkTradePhase(self, sequence=None, **kw):
    business_link = sequence.get('business_link')
    business_link.setTradePhaseList(['payroll/france/urssaf',
943
                                     'payroll/france/labour'])
944 945 946 947
    business_link.setSourceValue(sequence.get('urssaf_roubaix'))
    business_link.setSourceSectionValue(sequence.get('urssaf_roubaix'))
    business_link.setDeliveryBuilderList(('portal_deliveries/pay_sheet_builder',))
    sequence.edit(business_link=business_link)
948 949 950 951 952 953 954 955 956 957 958
    # Add a trade model path
    business_process = sequence.get('business_process')
    urssaf_roubaix = sequence.get('urssaf_roubaix')
    business_process.newContent(portal_type='Trade Model Path',
                                trade_phase_list=['payroll/france/urssaf',
                                                  'payroll/france/labour'],
                                reference="paysheet_path",
                                efficiency=1.0,
                                source_section_value=urssaf_roubaix,
                                )

959 960 961 962 963 964 965 966 967 968 969 970 971 972 973
  def stepModelSpecialiseBusinessProcess(self, sequence=None, **kw):
    model = sequence.get('model')
    business_process = sequence.get('business_process')
    model.setSpecialiseValueList(business_process)

  def stepCheckSourceSectionOnMovements(self, sequence=None, **kw):
    '''Check that the cell contain urssaf_roubaix as source section. This should
    be the case if getAggregatedAmountList from PaySheetModelLine is able to
    use Business Process to find the good source_section'''
    paysheet = sequence.get('paysheet')
    paysheet_line_list = paysheet.contentValues(portal_type='Pay Sheet Line')
    urssaf_roubaix = sequence.get('urssaf_roubaix')
    for paysheet_line in paysheet_line_list:
      service = paysheet_line.getResourceTitle()
      if service == 'Urssaf':
974 975 976 977
        cell1 = paysheet_line.getCell('contribution_share/employee')
        self.assertEquals(cell1.getSourceSectionValue(), urssaf_roubaix)
        cell2 = paysheet_line.getCell('contribution_share/employer')
        self.assertEquals(cell2.getSourceSectionValue(), urssaf_roubaix)
978 979 980 981 982 983 984 985 986 987 988 989 990
      elif service == 'Labour':
        pass
      else:
        self.fail("Unknown service for line %s" % paysheet_line.getTitle())

  def stepModelSetCategories(self, sequence=None, **kw):
    model = sequence.get('model')
    currency = sequence.get('price_currency')
    employer = sequence.get('employer')
    employee = sequence.get('employee')
    model.edit(\
        price_currency_value=currency,
        default_payment_condition_trade_date='custom',
991
        default_payment_condition_payment_date=DateTime(2009,05,25),
992 993 994 995 996 997 998 999 1000
        work_time_annotation_line_quantity=151.67,
        work_time_annotation_line_quantity_unit='time/hours',
        )

  def stepPaysheetSetModelAndApplyIt(self, sequence=None, **kw):
    paysheet = sequence.get('paysheet')
    model = sequence.get('model')
    paysheet.setSpecialiseValue(model)
    paysheet.PaySheetTransaction_applyModel(force=1)
1001

1002 1003 1004 1005 1006 1007 1008 1009 1010 1011
  def stepCheckCategoriesOnPaySheet(self, sequence=None, **kw):
    paysheet = sequence.get('paysheet')
    employer = sequence.get('employer')
    employee = sequence.get('employee')
    currency = sequence.get('price_currency')
    self.assertEquals(paysheet.getSourceSectionValue(), employee)
    self.assertEquals(paysheet.getDestinationSectionValue(), employer)
    self.assertEquals(paysheet.getPriceCurrencyValue(), currency)
    self.assertEquals(paysheet.getDefaultPaymentConditionTradeDate(), 'custom')
    self.assertEquals(paysheet.getDefaultPaymentConditionPaymentDate(),
1012
        DateTime(2009,05,25))
1013 1014 1015 1016 1017 1018 1019 1020 1021 1022 1023 1024 1025 1026 1027 1028 1029 1030 1031 1032 1033 1034 1035 1036 1037 1038 1039 1040 1041 1042 1043 1044 1045 1046 1047 1048 1049 1050 1051 1052 1053 1054 1055 1056 1057 1058 1059 1060 1061 1062 1063 1064 1065 1066 1067 1068 1069 1070 1071 1072 1073 1074 1075 1076 1077 1078 1079 1080 1081 1082 1083 1084 1085 1086 1087 1088 1089 1090 1091 1092 1093 1094 1095 1096 1097 1098 1099 1100 1101 1102 1103 1104 1105 1106 1107 1108 1109 1110 1111 1112 1113 1114 1115 1116 1117 1118 1119 1120 1121 1122 1123 1124 1125 1126 1127 1128 1129 1130 1131 1132 1133 1134 1135 1136 1137 1138 1139 1140 1141 1142 1143 1144 1145 1146 1147 1148 1149
    self.assertEquals(paysheet.getWorkTimeAnnotationLineQuantity(), 151.67)
    self.assertEquals(paysheet.getWorkTimeAnnotationLineQuantityUnit(),
      'time/hours')

  def stepCheckPaysheetContainNoAnnotationLine(self, sequence=None, **kw):
    paysheet = sequence.get('paysheet')
    self.assertEquals(len(paysheet.contentValues(portal_type=\
        'Annotation Line')), 0)

  def stepCheckPaysheetContainOneAnnotationLine(self, sequence=None, **kw):
    paysheet = sequence.get('paysheet')
    self.assertEquals(len(paysheet.contentValues(portal_type=\
        'Annotation Line')), 1)

  def stepCheckPaysheetContainNoPaymentCondition(self, sequence=None, **kw):
    paysheet = sequence.get('paysheet')
    self.assertEquals(len(paysheet.contentValues(portal_type=\
        'Payment Condition')), 0)

  def stepCheckPaysheetContainOnePaymentCondition(self, sequence=None, **kw):
    paysheet = sequence.get('paysheet')
    self.assertEquals(len(paysheet.contentValues(portal_type=\
        'Payment Condition')), 1)

  def stepCreateModelTree(self, sequence=None, **kw):
    '''
      Create three models inheriting from each other. Set slices values on a
      model.

      the inheritance tree look like this :

                                Employee Model
                                 /        \
                                /          \
                               /            \
                    Company Model         Second Company Alt
                         /
                        /
                       /
                Country Model
    '''
    model_employee = self.createModel()
    model_employee.edit(title='Employee Model', reference='model_employee',
        variation_settings_category_list='salary_range/france')
    model_company = self.createModel()
    model_company.edit(title='Company Model', reference='model_company',
        variation_settings_category_list='salary_range/france')
    model_company_alt = self.createModel()
    model_company_alt.edit(title='Second Company Model',
        reference='model_company_alt',
        variation_settings_category_list='salary_range/france')
    model_country = self.createModel()
    model_country.edit(title='Country Model', reference='model_country',
        variation_settings_category_list='salary_range/france')
    # add some cells in the models
    slice1 = model_employee.newCell('salary_range/france/slice_a',
                            portal_type='Pay Sheet Model Slice',
                            base_id='cell')
    slice1.setQuantityRangeMin(0)
    slice1.setQuantityRangeMax(1)

    slice2 = model_company.newCell('salary_range/france/slice_b',
                            portal_type='Pay Sheet Model Slice',
                            base_id='cell')
    slice2.setQuantityRangeMin(2)
    slice2.setQuantityRangeMax(3)

    slice3 = model_company_alt.newCell('salary_range/france/forfait',
                            portal_type='Pay Sheet Model Slice',
                            base_id='cell')
    slice3.setQuantityRangeMin(20)
    slice3.setQuantityRangeMax(30)

    slice4 = model_country.newCell('salary_range/france/slice_c',
                            portal_type='Pay Sheet Model Slice',
                            base_id='cell')
    slice4.setQuantityRangeMin(4)
    slice4.setQuantityRangeMax(5)

    # inherite from each other
    model_employee.setSpecialiseValueList((model_company, model_company_alt))
    model_company.setSpecialiseValue(model_country)

    sequence.edit(model_employee = model_employee,
                  model_company = model_company,
                  model_company_alt = model_company_alt,
                  model_country = model_country)

  def assertCell(self, model, salary_range, range_min, range_max):
    cell = model.getCell(salary_range)
    self.assertNotEqual(cell, None)
    self.assertEqual(cell.getQuantityRangeMin(), range_min)
    self.assertEqual(cell.getQuantityRangeMax(), range_max)

  def assertCellIsNone(self, model, salary_range):
    cell = model.getCell(salary_range)
    self.assertEqual(cell, None)

  def stepCheckgetCellResults(self, sequence=None, **kw):
    model_employee = sequence.get('model_employee')
    model_company = sequence.get('model_company')
    model_company_alt = sequence.get('model_company_alt')
    model_country = sequence.get('model_country')

    # check model_employee could access all cells
    self.assertCell(model_employee, 'salary_range/france/slice_a', 0, 1)
    self.assertCell(model_employee, 'salary_range/france/slice_b', 2, 3)
    self.assertCell(model_employee, 'salary_range/france/forfait', 20, 30)
    self.assertCell(model_employee, 'salary_range/france/slice_c', 4, 5)

    # check model_company could access just it's own cell
    # and this of the country model
    self.assertCellIsNone(model_company, 'salary_range/france/slice_a')
    self.assertCell(model_company, 'salary_range/france/slice_b', 2, 3)
    self.assertCellIsNone(model_company, 'salary_range/france/forfait')
    self.assertCell(model_company, 'salary_range/france/slice_c', 4, 5)

    # model_company_alt could access just it's own cell
    self.assertCellIsNone(model_company_alt, 'salary_range/france/slice_a')
    self.assertCellIsNone(model_company_alt, 'salary_range/france/slice_b')
    self.assertCell(model_company_alt, 'salary_range/france/forfait', 20, 30)
    self.assertCellIsNone(model_company_alt, 'salary_range/france/slice_c')

    # check model_country could access just it's own cell
    self.assertCellIsNone(model_country, 'salary_range/france/slice_a')
    self.assertCellIsNone(model_country, 'salary_range/france/slice_b')
    self.assertCellIsNone(model_country, 'salary_range/france/forfait')
    self.assertCell(model_country, 'salary_range/france/slice_c', 4, 5)

  def stepModelCreateIntermediateModelLine(self, sequence=None, **kw):
    '''
      create an intermediate_line wich contribute to tax and applied to
      base_salary
    '''
    model = sequence.get('model')
    model_line = self.createModelLine(model)
    model_line.edit(title='intermediate line',
1150
                    int_index = 10,
1151
                    reference='intermediate_line',
1152
                    price=0.2,
1153 1154
                    base_contribution_list=['base_amount/payroll/base/income_tax'],
                    base_application_list=['base_amount/payroll/base/contribution'])
1155 1156 1157 1158 1159 1160 1161 1162 1163
    sequence.edit(intermediate_model_line = model_line)

  def stepModelCreateAppliedOnTaxModelLine(self, sequence=None, **kw):
    '''
      create a model line applied on tax
    '''
    model = sequence.get('model')
    model_line = self.createModelLine(model)
    model_line.edit(title='line applied on intermediate line',
1164
                    int_index = 50,
1165 1166 1167
                    trade_phase='payroll/france/urssaf',
                    resource_value=sequence.get('urssaf_service'),
                    reference='line_applied_on_intermediate_line',
1168 1169 1170 1171
                    variation_category_list=['contribution_share/employee',
                                             'contribution_share/employer'],
                    base_contribution_list=['base_amount/payroll/report/salary/net'],
                    base_application_list=['base_amount/payroll/base/income_tax'])
1172 1173 1174 1175
    sequence.edit(model_line_applied_on_tax = model_line)

  def stepAppliedOnTaxModelLineCreateMovements(self, sequence=None, **kw):
    model_line = sequence.get('model_line_applied_on_tax')
1176 1177 1178 1179 1180 1181 1182 1183 1184 1185 1186 1187
    cell1 = model_line.newCell('contribution_share/employee',
        portal_type='Pay Sheet Model Cell',
        base_id='movement',
        base_application = "contribution_share/employee",
        mapped_value_property_list=('quantity', 'price'))
    cell1.edit(price=0.1, quantity=None, contribution_share='employee')
    cell2 = model_line.newCell('contribution_share/employer',
        portal_type='Pay Sheet Model Cell',
        base_id='movement',
        base_application = "contribution_share/employer",
        mapped_value_property_list=('quantity', 'price'))
    cell2.edit(price=0.5, quantity=None, contribution_share='employer')
1188 1189 1190 1191 1192 1193 1194 1195

  def stepModelCreateOldAgeInsuranceModelLine(self, sequence=None, **kw):
    model = sequence.get('model')
    model_line = self.createModelLine(model)
    model_line.edit(title='Old Age Insurance',
                    trade_phase='payroll/france/urssaf',
                    resource_value=sequence.get('old_age_insurance_service'),
                    reference='old_age_insurance',
1196 1197 1198
                    variation_category_list=['contribution_share/employee',
                                             'contribution_share/employer'],
                    base_application_list=[ 'base_amount/payroll/base/contribution'],
1199
                    base_contribution_list=['base_amount/payroll/base/income_tax'])
1200 1201 1202 1203
    sequence.edit(old_age_insurance = model_line)

  def stepOldAgeInsuranceModelLineCreateMovements(self, sequence=None, **kw):
    model_line = sequence.get('old_age_insurance')
1204
    cell1 = model_line.newCell('contribution_share/employee',
1205 1206
                               portal_type='Pay Sheet Model Cell',
                               base_id='movement',
1207 1208
                               base_application = "contribution_share/employee",
                               quantity=None,
1209
                               mapped_value_property_list=('quantity', 'price'))
1210 1211
    cell1.edit(price=0.5, contribution_share='employee')
    cell2 = model_line.newCell('contribution_share/employer',
1212 1213
                               portal_type='Pay Sheet Model Cell',
                               base_id='movement',
1214 1215
                               base_application = "contribution_share/employer",
                               quantity=None,
1216
                               mapped_value_property_list=('quantity', 'price'))
1217
    cell2.edit(price=0.8, contribution_share='employer')
1218 1219 1220 1221 1222 1223 1224 1225

  def stepModelCreateSicknessInsuranceModelLine(self, sequence=None, **kw):
    model = sequence.get('model')
    model_line = self.createModelLine(model)
    model_line.edit(title='Sickness Insurance',
                    trade_phase='payroll/france/urssaf',
                    resource_value=sequence.get('sickness_insurance_service'),
                    reference='sickness_insurance',
1226 1227 1228 1229
                    variation_category_list=['contribution_share/employee',
                                             'contribution_share/employer'],
                    base_application_list=[ 'base_amount/payroll/base/contribution'],
                    base_contribution_list=['base_amount/payroll/base/income_tax'])
1230 1231 1232 1233
    sequence.edit(sickness_insurance = model_line)

  def stepSicknessInsuranceModelLineCreateMovements(self, sequence=None, **kw):
    model_line = sequence.get('sickness_insurance')
1234
    cell1 = model_line.newCell('contribution_share/employee',
1235 1236
                               portal_type='Pay Sheet Model Cell',
                               base_id='movement',
1237 1238
                               base_application = "contribution_share/employee",
                               quantity=None,
1239
                               mapped_value_property_list=('quantity', 'price'))
1240 1241
    cell1.edit(price=0.4, contribution_share='employee')
    cell2 = model_line.newCell('contribution_share/employer',
1242 1243
                               portal_type='Pay Sheet Model Cell',
                               base_id='movement',
1244 1245
                               base_application = "contribution_share/employer",
                               quantity=None,
1246
                               mapped_value_property_list=('quantity', 'price'))
1247
    cell2.edit(price=0.3, contribution_share='employer')
1248 1249 1250 1251

  def stepCheckPaysheetIntermediateLines(self, sequence=None, **kw):
    paysheet = sequence.get('paysheet')

1252
    # paysheet should contain only two lines (labour and urssaf, but not
1253 1254
    # intermediate urssaf
    self.assertEquals(len(paysheet.contentValues(portal_type=\
1255
        'Pay Sheet Line')), 2)
1256 1257 1258 1259 1260 1261

    # check amounts
    paysheet_line_list = paysheet.contentValues(portal_type='Pay Sheet Line')
    for paysheet_line in paysheet_line_list:
      service = paysheet_line.getResourceTitle()
      if service == 'Urssaf':
1262 1263 1264 1265 1266 1267 1268 1269
        cell1 = paysheet_line.getCell('contribution_share/employee')
        self.assertEquals(cell1.getQuantity(), 600) # here it's 600 of tax
                                  # because of the intermediate line (3000*0.2)
        self.assertEquals(cell1.getPrice(), 0.1)
        cell2 = paysheet_line.getCell('contribution_share/employer')
        self.assertEquals(cell2.getQuantity(), 600) # here it's 600 of tax
                                  # because of the intermediate line (3000*0.2)
        self.assertEquals(cell2.getPrice(), 0.5)
1270 1271 1272 1273 1274 1275 1276
      elif service == 'Labour':
        self.assertEqual(paysheet_line.getTotalPrice(), 3000.0)
      else:
        self.fail("Unknown service for line %s" % paysheet_line.getTitle())

  def stepModelModifyUrssafModelLine(self, sequence=None, **kw):
    model_line = sequence.get('urssaf_model_line')
1277 1278 1279
    # modify price on movements :
    cell_1 = model_line.getCell('contribution_share/employee',
        base_id='movement')
1280
    self.assertNotEquals(cell_1, None)
1281 1282 1283
    cell_1.edit(price=0.2)
    cell_2 = model_line.getCell('contribution_share/employer',
        base_id='movement')
1284
    self.assertNotEquals(cell_2, None)
1285
    cell_2.edit(price=0.6)
1286 1287 1288 1289

  def stepModelDelUrssafModelLine(self, sequence=None, **kw):
    model_line = sequence.get('urssaf_model_line')
    model = sequence.get('model')
1290
    model.manage_delObjects([model_line.getId(),])
1291 1292 1293 1294 1295 1296 1297

  def stepCheckPaysheetLineNewAmountsAfterUpdate(self, sequence=None, **kw):
    paysheet = sequence.get('paysheet')
    paysheet_line_list = paysheet.contentValues(portal_type='Pay Sheet Line')
    for paysheet_line in paysheet_line_list:
      service = paysheet_line.getResourceTitle()
      if service == 'Urssaf':
1298 1299 1300 1301 1302 1303
        cell1 = paysheet_line.getCell('contribution_share/employee')
        self.assertEquals(cell1.getQuantity(), 3000)
        self.assertEquals(cell1.getPrice(), 0.2)
        cell2 = paysheet_line.getCell('contribution_share/employer')
        self.assertEquals(cell2.getQuantity(), 3000)
        self.assertEquals(cell2.getPrice(), 0.6)
1304 1305 1306 1307 1308 1309 1310 1311 1312 1313 1314 1315 1316 1317 1318 1319 1320 1321 1322 1323 1324 1325 1326 1327 1328 1329 1330 1331 1332 1333 1334 1335 1336 1337 1338 1339 1340 1341 1342 1343 1344 1345 1346 1347 1348 1349 1350 1351 1352 1353 1354 1355 1356 1357 1358 1359 1360 1361 1362 1363 1364 1365 1366 1367 1368 1369 1370 1371 1372 1373 1374 1375 1376 1377 1378 1379 1380 1381
      elif service == 'Labour':
        self.assertEqual(paysheet_line.getTotalPrice(), 3000.0)
      else:
        self.fail("Unknown service for line %s" % paysheet_line.getTitle())

  def stepCheckPaysheetLineLabourAmountOnly(self, sequence=None, **kw):
    paysheet = sequence.get('paysheet')
    paysheet_line_list = paysheet.contentValues(portal_type='Pay Sheet Line')
    for paysheet_line in paysheet_line_list:
      service = paysheet_line.getResourceTitle()
      if service == 'Labour':
        self.assertEqual(paysheet_line.getTotalPrice(), 3000.0)
      else:
        self.fail("Unknown service for line %s" % paysheet_line.getTitle())

  def stepCheckPaysheetConsistency(self, sequence=None, **kw):
    paysheet = sequence.get('paysheet')
    self.assertEquals([], paysheet.checkConsistency())

  def stepCheckModelConsistency(self, sequence=None, **kw):
    model = sequence.get('model')
    self.assertEquals([], model.checkConsistency())

  def stepCheckServiceConsistency(self, sequence=None, **kw):
    service = sequence.get('urssaf_service')
    self.assertEquals([], service.checkConsistency())

  def stepAddPredicateOnOldAgeInsuranceModelLineForSinglePerson(self,
      sequence=None, **kw):
    model_line = sequence.get('old_age_insurance')
    model_line.setMembershipCriterionBaseCategoryList(\
        ['source_marital_status'])
    model_line.setMembershipCriterionCategoryList(\
        ['source_marital_status/marital_status/single'])

  def stepAddPredicateOnSicknessInsuranceModelLineForMarriedPerson(self,
      sequence=None, **kw):
    model_line = sequence.get('sickness_insurance')
    model_line.setMembershipCriterionBaseCategoryList(\
        ['source_marital_status'])
    model_line.setMembershipCriterionCategoryList(\
        ['source_marital_status/marital_status/married'])

  def stepSetMaritalStatusMarriedOnEmployee(self, sequence=None, **kw):
    employee = sequence.get('employee')
    employee.setMaritalStatus('married')

  def stepSetMaritalStatusSingleOnEmployee(self, sequence=None, **kw):
    employee = sequence.get('employee')
    employee.setMaritalStatus('single')

  def stepModelTreeAddAnnotationLines(self, sequence=None, **kw):
    model_employee = sequence.get('model_employee')
    model_employee.newContent(id='over_time_duration',
                              title='Over time duration',
                              portal_type='Annotation Line',
                              reference='over_time_duration',
                              quantity=1)
    model_company = sequence.get('model_company')
    model_company.newContent(id='worked_time_duration',
                             title='Worked time duration',
                             portal_type='Annotation Line',
                             reference='worked_time_duration',
                             quantity=2)
    model_company_alt = sequence.get('model_company_alt')
    model_company_alt.newContent(id='social_insurance',
                                 title='Social insurance',
                                 portal_type='Annotation Line',
                                 reference='social_insurance',
                                 quantity=3)
    model_country = sequence.get('model_country')
    model_country.newContent(id='social_insurance',
                             title='Social insurance',
                             portal_type='Annotation Line',
                             reference='social_insurance',
                             quantity=4)

  def stepCheckInheritanceModelReferenceDict(self, sequence=None, **kw):
1382
    paysheet = self.createPaysheet()
1383
    model_employee = sequence.get('model_employee')
1384 1385
    paysheet.setSpecialiseValue(model_employee)

1386 1387 1388 1389
    model_employee_url = model_employee.getRelativeUrl()
    model_company_url = sequence.get('model_company').getRelativeUrl()
    model_company_alt_url = sequence.get('model_company_alt').getRelativeUrl()
    model_country_url = sequence.get('model_country').getRelativeUrl()
1390 1391 1392

    model_reference_dict = model_employee.getInheritanceReferenceDict(
        paysheet, portal_type_list=('Annotation Line',))
1393 1394 1395 1396 1397 1398 1399 1400 1401 1402 1403 1404 1405
    self.assertEquals(len(model_reference_dict), 3) # there is 4 model but two
                                                    # models have the same
                                                    # reference.
    self.assertEquals(model_reference_dict.has_key(model_employee_url), True)
    self.assertEquals(model_reference_dict[model_employee_url],
        ['over_time_duration'])
    self.assertEquals(model_reference_dict.has_key(model_company_url), True)
    self.assertEquals(model_reference_dict[model_company_url],
        ['worked_time_duration'])
    self.assertEquals(model_reference_dict.has_key(model_company_alt_url), True)
    self.assertEquals(model_reference_dict[model_company_alt_url],
        ['social_insurance'])
    self.assertNotEquals(model_reference_dict.has_key(model_country_url), True)
1406
    
1407 1408 1409 1410 1411 1412 1413 1414 1415 1416 1417 1418 1419 1420 1421 1422 1423 1424 1425 1426 1427 1428 1429 1430 1431 1432 1433 1434 1435 1436 1437 1438 1439 1440
    # check the object list :
    object_list = paysheet.getInheritedObjectValueList(portal_type_list=\
        ('Annotation Line',))
    self.assertEquals(len(object_list), 3) # one line have the same reference
                                           # than another, so each reference
                                           # should be prensent only one time
                                           # in the list
    over_time_duration = paysheet.getAnnotationLineFromReference(\
        'over_time_duration')
    self.assertNotEquals(over_time_duration, None)
    over_time_duration.getQuantity(1)
    worked_time_duration = paysheet.getAnnotationLineFromReference(\
        'worked_time_duration')
    self.assertNotEquals(worked_time_duration, None)
    worked_time_duration.getQuantity(2)
    social_insurance = paysheet.getAnnotationLineFromReference(\
        'social_insurance')
    self.assertNotEquals(social_insurance, None)
    social_insurance.getQuantity(3)

  def stepCheckPaySheetTransaction_getMovementListReturn(self,
      sequence=None, **kw):
    paysheet = sequence.get('paysheet')
    # when pay sheet has no line, the script returns an empty list
    self.assertEquals(len(paysheet.PaySheetTransaction_getMovementList()), 0)

    # we add a line, then it is returned in the list
    line = self.createPaysheetLine(paysheet)
    self.assertEquals(len(paysheet.PaySheetTransaction_getMovementList()), 1)

    # if the line has cells with different tax categories, new properties are
    # added to this line.
    urssaf_service = sequence.get('urssaf_service')
    line.setResourceValue(urssaf_service)
1441 1442 1443
    line.setVariationCategoryList(['contribution_share/employee',
                                   'contribution_share/employer'])
    cell0 = line.newCell('contribution_share/employee',
1444 1445
                         portal_type='Pay Sheet Cell', base_id='movement')
    cell0.setMappedValuePropertyList(['quantity', 'price'])
1446
    cell0.setVariationCategoryList(('contribution_share/employee',))
1447 1448
    cell0.setPrice(2.0)
    cell0.setQuantity(3.0)
1449
    cell1 = line.newCell('contribution_share/employer',
1450 1451
                         portal_type='Pay Sheet Cell', base_id='movement')
    cell1.setMappedValuePropertyList(['quantity', 'price'])
1452
    cell1.setVariationCategoryList(('contribution_share/employer',))
1453 1454
    cell1.setPrice(4.0)
    cell1.setQuantity(5.0)
1455 1456 1457 1458

    movement_list = paysheet.PaySheetTransaction_getMovementList()
    self.assertEquals(1, len(movement_list))
    movement = movement_list[0]
1459 1460 1461 1462 1463 1464
    self.assertEquals(2, movement.employee_price)
    self.assertEquals(3, movement.employee_quantity)
    self.assertEquals(2*3, movement.employee_total_price)
    self.assertEquals(4, movement.employer_price)
    self.assertEquals(5, movement.employer_quantity)
    self.assertEquals(4*5, movement.employer_total_price)
1465 1466 1467 1468 1469 1470 1471 1472

  def stepCheckModelWithoutRefValidity(self, sequence=None, **kw):
    '''
    If no reference are defined on a model, the behavior is that this model is
    always valid. So check a Pay Sheet Transaction Line is created after
    calling the calculation script
    '''
    eur = sequence.get('currency')
1473
    labour = sequence.get('labour_service')
1474 1475 1476 1477 1478 1479 1480 1481 1482 1483 1484 1485 1486 1487 1488

    model_without_ref = self.getPortalObject().paysheet_model_module.newContent( \
        specialise_value=sequence.get('business_process'),
        portal_type='Pay Sheet Model',
        variation_settings_category_list=['salary_range/france',],
        effective_date=DateTime(2009, 1, 1),
        expiration_date=DateTime(2009, 12, 31))
    model_without_ref.setPriceCurrencyValue(eur)


    model_line_1 = self.createModelLine(model_without_ref)
    model_line_1.edit(
        trade_phase='payroll/france/labour',
        reference='model_without_ref',
        resource_value=labour,
1489 1490
        target_delivery=True,
        base_application=self.fixed_quantity,
1491 1492
        base_contribution_list=['base_amount/payroll/base/contribution',
          'base_amount/payroll/report/salary/gross'],
1493
        quantity=10000.0)
1494
    
1495 1496 1497
    # create the paysheet
    paysheet = self.createPaysheet()
    paysheet.edit(specialise_value=model_without_ref,
1498 1499
                  start_date=DateTime(2009, 1, 1),
                  stop_date=DateTime(2009, 1, 31),
1500 1501 1502 1503 1504 1505 1506 1507
                  price_currency_value=eur)
    paysheet.PaySheetTransaction_applyModel()
    self.stepTic()

    portal_type_list = ['Pay Sheet Model Line',]

    # if no reference, we don't care about dates
    sub_object_list = paysheet.getInheritedObjectValueList(portal_type_list)
1508
    
1509 1510 1511 1512 1513 1514 1515 1516
    self.assertEquals(len(paysheet.contentValues(\
        portal_type='Pay Sheet Line')), 0)
    # calculate the pay sheet
    paysheet.applyTransformation()
    self.stepTic()
    self.assertEquals(len(paysheet.contentValues(
        portal_type='Pay Sheet Line')), 1)
    # check values on the paysheet
1517 1518
    # XXX-Aurel : no price here
    self.assertEquals(paysheet.contentValues()[0].getQuantity(), 10000)
1519 1520 1521 1522

  def stepCheckModelWithoutDateValidity(self, sequence=None, **kw):
    '''
    If no date are defined on a model, the behavior is that this model
1523
    is always valid.
1524 1525
    '''
    eur = sequence.get('currency')
1526
    labour = sequence.get('labour_service')
1527 1528 1529 1530 1531 1532 1533 1534 1535 1536 1537

    model_without_date = self.getPortalObject().paysheet_model_module.newContent( \
        specialise_value=sequence.get('business_process'),
        portal_type='Pay Sheet Model',
        variation_settings_category_list=['salary_range/france',],
        reference='fabien_model_without_date')
    model_line_2 = self.createModelLine(model_without_date)
    model_line_2.edit(
        trade_phase='payroll/france/labour',
        reference='model_without_date',
        resource_value=labour,
1538 1539
        target_delivery=True,
        base_application=self.fixed_quantity,
1540 1541
        base_contribution_list=['base_amount/payroll/base/contribution',
          'base_amount/payroll/report/salary/gross'],
1542
        quantity=10000.0)
1543
    self.stepTic()
1544

1545 1546 1547
    # create a paysheet without date
    paysheet_without_date = self.createPaysheet()
    paysheet_without_date.edit(specialise_value=model_without_date,
1548
                  price_currency_value=eur)
1549
    paysheet_without_date.PaySheetTransaction_applyModel()
1550 1551 1552 1553
    self.stepTic()

    portal_type_list = ['Pay Sheet Model Line',]

1554 1555
    # check the paysheet contail no lines before calculation
    self.assertEquals(len(paysheet_without_date.contentValues(\
1556 1557
        portal_type='Pay Sheet Line')), 0)
    # calculate the pay sheet
1558
    paysheet_without_date.applyTransformation()
1559
    self.stepTic()
1560
    self.assertEquals(len(paysheet_without_date.contentValues(\
1561
        portal_type='Pay Sheet Line')), 1)
1562
    # check values on the paysheet_without_date
1563 1564
    # XXX-Aurel getTotalPrice is None as no price defined on the model
    self.assertEquals(paysheet_without_date.contentValues()[0].getQuantity(),
1565 1566 1567 1568 1569 1570 1571 1572 1573 1574 1575 1576 1577
        10000)

    # create a paysheet with dates
    paysheet_with_date = self.createPaysheet()
    paysheet_with_date.edit(specialise_value=model_without_date,
                  start_date=DateTime(2009, 1, 1),
                  stop_date=DateTime(2009, 1, 31),
                  price_currency_value=eur)
    paysheet_with_date.PaySheetTransaction_applyModel()
    self.stepTic()

    portal_type_list = ['Pay Sheet Model Line',]

1578
    # check the paysheet contains no lines before calculation
1579 1580 1581 1582 1583
    self.assertEquals(len(paysheet_with_date.contentValues(\
        portal_type='Pay Sheet Line')), 0)
    # calculate the pay sheet
    paysheet_with_date.applyTransformation()
    self.stepTic()
1584
    # after calculation, paysheet contains one line, because the model applies.
1585
    self.assertEquals(len(paysheet_with_date.contentValues(\
1586
        portal_type='Pay Sheet Line')), 1)
1587 1588
    # XXX-Aurel same as previous one
    self.assertEquals(paysheet_without_date.contentValues()[0].getQuantity(),
1589
        10000)
1590 1591 1592 1593 1594 1595

  def stepCheckModelDateValidity(self, sequence=None, **kw):
    '''
    check that model effective_date and expiration_date are take into account.
    '''
    eur = sequence.get('currency')
1596
    labour = sequence.get('labour_service')
1597 1598 1599 1600 1601 1602 1603 1604

    model_1 = self.getPortalObject().paysheet_model_module.newContent( \
        specialise_value=sequence.get('business_process'),
        portal_type='Pay Sheet Model',
        variation_settings_category_list=['salary_range/france',],
        reference='fabien_model_2009',
        effective_date=DateTime(2009, 1, 1),
        expiration_date=DateTime(2009, 06, 30))
1605
    
1606 1607 1608 1609 1610 1611 1612 1613 1614 1615 1616 1617 1618
    model_2 = self.getPortalObject().paysheet_model_module.newContent( \
        specialise_value=sequence.get('business_process'),
        portal_type='Pay Sheet Model',
        variation_settings_category_list=['salary_range/france',],
        reference='fabien_model_2009',
        effective_date=DateTime(2009, 07, 1),
        expiration_date=DateTime(2009, 12, 31))

    model_line_3 = self.createModelLine(model_1)
    model_line_3.edit(
        trade_phase='payroll/france/labour',
        reference='check_model_date_validity_1',
        resource_value=labour,
1619 1620
        target_delivery=True,
        base_application=self.fixed_quantity,
1621 1622
        base_contribution_list=['base_amount/payroll/base/contribution',
          'base_amount/payroll/report/salary/gross'],
1623 1624 1625 1626 1627 1628 1629 1630
        quantity=20000,
        price=1)

    model_line_4 = self.createModelLine(model_2)
    model_line_4.edit(
        trade_phase='payroll/france/labour',
        reference='check_model_date_validity_2',
        resource_value=labour,
1631 1632
        target_delivery=True,
        base_application=self.fixed_quantity,
1633 1634
        base_contribution_list=['base_amount/payroll/base/contribution',
          'base_amount/payroll/report/salary/gross'],
1635 1636 1637 1638 1639 1640 1641 1642 1643 1644 1645 1646 1647 1648 1649 1650 1651 1652 1653
        quantity=30000,
        price=1)
    self.stepTic()

    # create the paysheet
    paysheet = self.portal.accounting_module.newContent(
                              portal_type='Pay Sheet Transaction',
                              specialise_value=model_1,
                              start_date=DateTime(2009, 07, 1),
                              stop_date=DateTime(2009, 07, 31),
                              price_currency_value=eur)
    paysheet.PaySheetTransaction_applyModel()
    self.stepTic()

    self.assertEquals(len(paysheet.contentValues(\
        portal_type='Pay Sheet Line')), 0)
    # calculate the pay sheet
    paysheet.applyTransformation()
    self.stepTic()
1654
    # XXX-Aurel Why it is one as the model should not apply since date are not in the range ??
1655 1656
    self.assertEquals(len(paysheet.contentValues(\
        portal_type='Pay Sheet Line')), 1)
1657
    # check values on the paysheet, if it's model_2, the total_price 
1658
    # should be 30000.
1659
    # self.assertEquals(paysheet.contentValues()[0].getTotalPrice(), 30000)
1660 1661 1662 1663 1664 1665 1666 1667 1668 1669 1670 1671 1672 1673 1674 1675 1676 1677 1678 1679 1680 1681 1682 1683 1684 1685 1686 1687 1688 1689 1690 1691 1692 1693 1694 1695 1696

  def stepCheckModelVersioning(self, sequence=None, **kw):
    '''
    check that latest version is used in case of more thant one model is matching
    using dates
    '''
    eur = sequence.get('currency')

    # define a non effective model
    model_1 = self.getPortalObject().paysheet_model_module.newContent( \
        portal_type='Pay Sheet Model',
        variation_settings_category_list=['salary_range/france',],
        reference='fabien_model_2009',
        effective_date=DateTime(2009, 01, 1),
        expiration_date=DateTime(2009, 02, 28),
        specialise_value=sequence.get('business_process'))

    # define two models with same references and same dates
    # but different version number
    model_2 = self.getPortalObject().paysheet_model_module.newContent( \
        portal_type='Pay Sheet Model',
        variation_settings_category_list=['salary_range/france',],
        reference='fabien_model_2009',
        effective_date=DateTime(2009, 07, 1),
        expiration_date=DateTime(2009, 12, 31),
        version='002',
        specialise_value=sequence.get('business_process'))

    model_3 = self.getPortalObject().paysheet_model_module.newContent( \
        portal_type='Pay Sheet Model',
        variation_settings_category_list=['salary_range/france',],
        reference='fabien_model_2009',
        effective_date=DateTime(2009, 07, 1),
        expiration_date=DateTime(2009, 12, 31),
        version='001',
        specialise_value=sequence.get('business_process'))
    self.stepTic()
1697

1698 1699 1700 1701 1702 1703 1704 1705 1706 1707 1708 1709 1710
    # create the paysheet
    paysheet = self.portal.accounting_module.newContent(
                              portal_type='Pay Sheet Transaction',
                              specialise_value=model_1,
                              start_date=DateTime(2009, 07, 1),
                              stop_date=DateTime(2009, 07, 31),
                              price_currency_value=eur)
    paysheet.PaySheetTransaction_applyModel()
    self.stepTic()

    # the effective model should be model_2 because of the effective date and
    # version number
    specialise_value = paysheet.getSpecialiseValue()
1711 1712 1713 1714
    effective_model = specialise_value.getEffectiveModel(\
        start_date=paysheet.getStartDate(),
        stop_date=paysheet.getStopDate())
    self.assertEquals(effective_model, model_2)
1715 1716 1717

    # check the effective model tree list
    effective_value_list = specialise_value.findEffectiveSpecialiseValueList(\
1718
        context=paysheet)
1719 1720 1721 1722 1723 1724
    self.assertEquals(effective_value_list, [model_2])

  def stepCreateModelLineZeroPrice(self, sequence=None, **kw):
    '''Test the creation of lines when the price is set to zero: the line should
    not be created.'''
    model = sequence.get('model')
1725
    labour = sequence.get('labour_service')
1726 1727
    line = model.newContent(
          id='line',
1728
          reference='zero_price_line',
1729 1730
          portal_type='Pay Sheet Model Line',
          resource_value=labour,
1731 1732
          base_contribution_list=['base_amount/payroll/base/contribution',
            'base_amount/payroll/report/salary/gross'],
1733 1734 1735 1736 1737
          quantity=5,
          price=0)

  def stepComplexModelInheritanceScheme(self, sequence=None, **kw):
    '''
Fabien Morin's avatar
Fabien Morin committed
1738
    check inheritance and effective model with a more complex inheritance tree
1739 1740 1741 1742 1743 1744 1745 1746 1747 1748 1749 1750 1751 1752 1753 1754 1755 1756

    # the inheritance tree look like this :

                                    model_employee
(model_1, 01/01/09, 28/02/09) ; (model_2, 01/07/09, 31/12/09) ; (model_3, 01/07/09, 31/12/09)
                                           |
                                           |
                                           |
                                     model_company
             (model_4, 01/07/09, 31/12/09), (model_5, 01/07/09, 31/12/09)
                                           |
                                           |
                                           |
                                     model_company
             (model_6, 01/07/09, 31/12/09), (model_7, 01/07/09, 31/12/09)
    '''

    eur = sequence.get('currency')
1757
    labour = sequence.get('labour_service')
1758 1759 1760 1761 1762 1763 1764 1765 1766 1767 1768 1769
    paysheet_model_module = self.getPortalObject().paysheet_model_module

    # define a non effective model
    model_1 = paysheet_model_module.newContent( \
        portal_type='Pay Sheet Model',
        specialise_value=sequence.get('business_process'),
        reference='fabien_model_2009',
        effective_date=DateTime(2009, 01, 1),
        expiration_date=DateTime(2009, 02, 28))
    model_line_1 = self.createModelLine(model_1)
    model_line_1.edit(
        resource_value=labour,
1770 1771
        base_contribution_list=['base_amount/payroll/base/contribution',
          'base_amount/payroll/report/salary/gross'],
1772 1773 1774 1775 1776 1777 1778 1779 1780 1781 1782 1783 1784 1785
        quantity=10000)

    # define two models with same references and same dates
    # but different version number
    model_2 = paysheet_model_module.newContent( \
        portal_type='Pay Sheet Model',
        specialise_value=sequence.get('business_process'),
        reference='fabien_model_2009',
        effective_date=DateTime(2009, 07, 1),
        expiration_date=DateTime(2009, 12, 31),
        version='002')
    model_line_2 = self.createModelLine(model_2)
    model_line_2.edit(
        resource_value=labour,
1786 1787
        base_contribution_list=['base_amount/payroll/base/contribution',
          'base_amount/payroll/report/salary/gross'],
1788 1789 1790 1791 1792 1793 1794 1795 1796 1797 1798 1799
        quantity=20000)

    model_3 = paysheet_model_module.newContent( \
        portal_type='Pay Sheet Model',
        specialise_value=sequence.get('business_process'),
        reference='fabien_model_2009',
        effective_date=DateTime(2009, 07, 1),
        expiration_date=DateTime(2009, 12, 31),
        version='001')
    model_line_3 = self.createModelLine(model_3)
    model_line_3.edit(
        resource_value=labour,
1800 1801
        base_contribution_list=['base_amount/payroll/base/contribution',
          'base_amount/payroll/report/salary/gross'],
1802 1803 1804 1805 1806 1807 1808 1809 1810 1811 1812 1813 1814 1815
        quantity=30000)

    # define two models with same references and same dates
    # but different version number
    model_4 = paysheet_model_module.newContent( \
        portal_type='Pay Sheet Model',
        specialise_value=sequence.get('business_process'),
        reference='fabien_model_level_2_2009',
        effective_date=DateTime(2009, 01, 1),
        expiration_date=DateTime(2009, 06, 30),
        version='002')
    model_line_4 = self.createModelLine(model_4)
    model_line_4.edit(
        resource_value=labour,
1816 1817
        base_contribution_list=['base_amount/payroll/base/contribution',
          'base_amount/payroll/report/salary/gross'],
1818 1819 1820 1821 1822 1823 1824 1825 1826 1827 1828 1829
        quantity=40000)

    model_5 = paysheet_model_module.newContent( \
        portal_type='Pay Sheet Model',
        specialise_value=sequence.get('business_process'),
        reference='fabien_model_level_2_2009',
        effective_date=DateTime(2009, 07, 1),
        expiration_date=DateTime(2009, 12, 31),
        version='001')
    model_line_5 = self.createModelLine(model_5)
    model_line_5.edit(
        resource_value=labour,
1830 1831
        base_contribution_list=['base_amount/payroll/base/contribution',
          'base_amount/payroll/report/salary/gross'],
1832 1833 1834 1835 1836 1837 1838 1839 1840 1841 1842 1843 1844 1845
        quantity=50000)

    # third level : define two models with same references and same dates
    # but different version number
    model_6 = paysheet_model_module.newContent( \
        portal_type='Pay Sheet Model',
        specialise_value=sequence.get('business_process'),
        reference='fabien_model_level_3_2009',
        effective_date=DateTime(2009, 01, 1),
        expiration_date=DateTime(2009, 06, 30),
        version='002')
    model_line_6 = self.createModelLine(model_6)
    model_line_6.edit(
        resource_value=labour,
1846 1847
        base_contribution_list=['base_amount/payroll/base/contribution',
          'base_amount/payroll/report/salary/gross'],
1848 1849 1850 1851 1852 1853 1854 1855 1856 1857 1858 1859
        quantity=60000)

    model_7 = paysheet_model_module.newContent( \
        portal_type='Pay Sheet Model',
        specialise_value=sequence.get('business_process'),
        reference='fabien_model_level_3_2009',
        effective_date=DateTime(2009, 07, 1),
        expiration_date=DateTime(2009, 12, 31),
        version='001')
    model_line_7 = self.createModelLine(model_7)
    model_line_7.edit(
        resource_value=labour,
1860 1861
        base_contribution_list=['base_amount/payroll/base/contribution',
          'base_amount/payroll/report/salary/gross'],
1862 1863 1864 1865 1866 1867 1868 1869 1870 1871 1872 1873 1874 1875 1876 1877 1878
        quantity=70000)

    self.stepTic()

    # create the paysheet
    paysheet = self.portal.accounting_module.newContent(
                              portal_type='Pay Sheet Transaction',
                              specialise_value=model_1,
                              start_date=DateTime(2009, 07, 1),
                              stop_date=DateTime(2009, 07, 31),
                              price_currency_value=eur)
    specialise_value = paysheet.getSpecialiseValue()

    # design some heritance trees, and check them:
    model_1.setSpecialiseValue(model_4)
    model_4.setSpecialiseValue(model_6)
    paysheet.PaySheetTransaction_applyModel()
1879 1880
    self.assertEquals([model_2],
      specialise_value.findEffectiveSpecialiseValueList(context=paysheet))
1881 1882 1883 1884 1885

    model_1.setSpecialiseValue(None)
    model_2.setSpecialiseValue(model_5)
    model_5.setSpecialiseValue(model_6)
    paysheet.PaySheetTransaction_applyModel()
1886 1887
    self.assertEquals([model_2, model_5, model_7],
      specialise_value.findEffectiveSpecialiseValueList(context=paysheet))
1888

Fabien Morin's avatar
Fabien Morin committed
1889 1890 1891 1892 1893 1894 1895 1896 1897 1898 1899 1900 1901 1902 1903 1904 1905 1906 1907 1908 1909 1910 1911 1912 1913 1914 1915 1916 1917 1918 1919 1920
  def stepCheckPropertiesAreCopiedFromModelLineToPaySheetLine(self,
      sequence=None, **kw):
    model = sequence.get('model')
    paysheet = sequence.get('paysheet')
    property_list = ('title', 'description', 'int_index')
    for model_line in model.contentValues(portal_type='Pay Sheet Model Line'):
      model_line_resource = model_line.getResource()
      line_found = False
      for paysheet_line in paysheet.contentValues(portal_type='Pay Sheet Line'):
        if paysheet_line.getResource() == model_line_resource:
          line_found = True
          for prop in property_list:
            prop_from_model_line = getattr(model_line, prop, None)
            # check a value is set on the model line
            self.assertNotEquals(prop_from_model_line, None)
            prop_from_paysheet_line = getattr(paysheet_line, prop, None)
            # check the property is the same on model_line and paysheet_line
            self.assertEquals(prop_from_model_line, prop_from_paysheet_line)
          break

      # check that for each model line, we foud a corresponding paysheet_line
      self.assertEquals(line_found, True)

  def stepSetProperiesOnModelLines(self, sequence=None, **kw):
    model = sequence.get('model')
    index = 0
    for model_line in model.contentValues(portal_type='Pay Sheet Model Line'):
      model_line.setTitle('Model line title %s' % index)
      model_line.setDescription('Model line description %s' % index)
      model_line.setIntIndex(index)
      index += 1

1921 1922 1923 1924 1925 1926 1927 1928 1929 1930 1931 1932
  def checkPrecisionOfListBox(self, report_section, precision):
    here = report_section.getObject(self.portal)
    report_section.pushReport(self.portal)
    form = getattr(here, report_section.getFormId())
    self.portal.REQUEST['here'] = here
    if form.has_field('listbox'):
      result = form.listbox.get_value('default',
                                      render_format='list',
                                      REQUEST=self.portal.REQUEST)
      self.assertEquals(precision, self.portal.REQUEST.get('precision'))
    report_section.popReport(self.portal)

1933 1934
class TestPayroll(TestPayrollMixin):

1935
  def test_modelGetCell(self):
1936
    """
1937 1938 1939 1940 1941 1942 1943 1944 1945 1946
      Model objects have a overload method called getCell. This method first
      call the XMLMatrix.getCell and if the cell is not found, call
      getCell method in all it's inherited model until the cell is found or
      the cell have been searched on all inherited models.

      TODO : Currently, the method use a Depth-First Search algorithm, it will
      be better to use Breadth-First Search one.
      more about this on :
        - http://en.wikipedia.org/wiki/Breadth-first_search
        - http://en.wikipedia.org/wiki/Depth-first_search
1947
    """
1948 1949 1950
    sequence_list = SequenceList()
    sequence_string = """
               CreateModelTree
1951
               Tic
1952 1953 1954 1955 1956 1957 1958 1959 1960 1961 1962 1963 1964 1965 1966 1967 1968 1969
               CheckgetCellResults
    """
    sequence_list.addSequenceString(sequence_string)
    sequence_list.play(self)

  def test_applyModelSetCategories(self):
    '''
      check that when the model is set on the Pay Sheet Transaction, properties
      like source, destination, currency, ... and sub-object like
      annotation_line are copied from the model to the Pay Sheet Transaction
    '''
    sequence_list = SequenceList()
    sequence_string = """
               CreateUrssafService
               CreateLabourService
               CreateEmployer
               CreateEmployee
               CreateBasicModel
1970
               Tic
1971 1972 1973 1974 1975 1976 1977 1978 1979 1980 1981 1982 1983 1984 1985 1986 1987 1988 1989 1990
               CreatePriceCurrency
               ModelSetCategories
               CreateBasicPaysheet
               PaysheetSetModelAndApplyIt
               CheckCategoriesOnPaySheet
    """
    sequence_list.addSequenceString(sequence_string)
    sequence_list.play(self)

  def test_applyModelTwice(self):
    '''
      apply model twice does not copy subdocument twice
    '''
    sequence_list = SequenceList()
    sequence_string = """
               CreateUrssafService
               CreateLabourService
               CreateEmployer
               CreateEmployee
               CreateBasicModel
1991
               Tic
1992 1993 1994 1995 1996 1997 1998 1999 2000 2001 2002 2003 2004 2005 2006 2007 2008 2009 2010 2011 2012 2013 2014 2015 2016 2017 2018 2019 2020 2021 2022 2023 2024 2025 2026 2027 2028 2029 2030 2031 2032 2033 2034 2035 2036 2037 2038
               CreatePriceCurrency
               ModelSetCategories
               CreateBasicPaysheet
               CheckPaysheetContainNoAnnotationLine
               CheckPaysheetContainNoPaymentCondition
               PaysheetSetModelAndApplyIt
               CheckPaysheetContainOneAnnotationLine
               CheckPaysheetContainOnePaymentCondition
               PaysheetSetModelAndApplyIt
               CheckPaysheetContainOneAnnotationLine
               CheckPaysheetContainOnePaymentCondition
    """
    sequence_list.addSequenceString(sequence_string)
    sequence_list.play(self)

  def test_checkConsistency(self):
    '''
      minimal test for checkConsistency on a Pay Sheet Transaction and it's
      subdocuments (may have to be updated when we'll add more constraints).
    '''
    sequence_list = SequenceList()
    sequence_string = self.COMMON_BASIC_DOCUMENT_CREATION_SEQUENCE_STRING + """
               CheckPaysheetConsistency
               CheckModelConsistency
               CheckServiceConsistency
    """
    sequence_list.addSequenceString(sequence_string)
    sequence_list.play(self)

  def test_basicPaySheetCalculation(self):
    '''
      test applyTransformation method. It should create new movements
    '''
    sequence_list = SequenceList()
    sequence_string = self.COMMON_BASIC_DOCUMENT_CREATION_SEQUENCE_STRING + """
               CheckUpdateAggregatedAmountListReturn
               PaysheetApplyTransformation
               Tic
               CheckPaysheetLineAreCreated
               CheckPaysheetLineAmounts
               CheckUpdateAggregatedAmountListReturnNothing
               CheckPaysheetLineAmounts
    """
    sequence_list.addSequenceString(sequence_string)
    sequence_list.play(self)

  def test_intermediateLines(self):
2039
    """
2040 2041 2042
      Intermediate lines are paysheet model lines usefull to calcul, but we
      don't want to have on paysheet. So a checkbox on paysheet model lines
      permit to create it or not (created by default)
2043
    """
2044 2045 2046 2047 2048 2049 2050
    sequence_list = SequenceList()
    sequence_string = """
               CreateUrssafService
               CreateLabourService
               CreateEmployer
               CreateEmployee
               CreateBasicModel
2051
               Tic
2052 2053 2054 2055 2056 2057 2058 2059 2060 2061 2062 2063 2064 2065 2066 2067 2068 2069 2070 2071 2072 2073 2074 2075 2076 2077 2078 2079 2080 2081 2082 2083 2084 2085 2086 2087 2088 2089 2090 2091 2092
               ModelCreateIntermediateModelLine
               ModelCreateAppliedOnTaxModelLine
               AppliedOnTaxModelLineCreateMovements
               CreateBasicPaysheet
               PaysheetCreateLabourPaySheetLine
  """ + self.BUSINESS_PATH_CREATION_SEQUENCE_STRING + """
               PaysheetApplyTransformation
               Tic
               CheckPaysheetIntermediateLines
    """
    sequence_list.addSequenceString(sequence_string)
    sequence_list.play(self)

  def test_updateModifyMovements(self):
    '''
      Calculate the paySheet using a model, modify one value in the model and
      check that updateAggregatedAmount return nothing but modifed movements
      that need to be modified
    '''
    sequence_list = SequenceList()
    sequence_string = self.COMMON_BASIC_DOCUMENT_CREATION_SEQUENCE_STRING + """
               CheckUpdateAggregatedAmountListReturn
               PaysheetApplyTransformation
               Tic
               CheckPaysheetLineAreCreated
               CheckPaysheetLineAmounts
               CheckUpdateAggregatedAmountListReturnNothing
               ModelModifyUrssafModelLine
               CheckUpdateAggregatedAmountListReturnNothing
               PaysheetApplyTransformation
               Tic
               CheckPaysheetLineAreCreated
               CheckPaysheetLineNewAmountsAfterUpdate
               CheckUpdateAggregatedAmountListReturnNothing
               CheckPaysheetLineNewAmountsAfterUpdate
    """
    sequence_list.addSequenceString(sequence_string)
    sequence_list.play(self)

  def test_updateAddMovements(self):
    '''
2093
      Calculate the paySheet using a model, add a model line in the model 
2094 2095 2096 2097 2098 2099 2100 2101 2102 2103 2104 2105 2106 2107 2108 2109 2110 2111 2112 2113 2114 2115 2116 2117 2118 2119 2120
      and check that updateAggregatedAmount add the movements corresponding
      to this model_line
    '''
    sequence_list = SequenceList()
    sequence_string = self.COMMON_BASIC_DOCUMENT_CREATION_SEQUENCE_STRING + """
               CheckUpdateAggregatedAmountListReturn
               PaysheetApplyTransformation
               Tic
               CheckPaysheetLineAreCreated
               CheckPaysheetLineAmounts
               CheckUpdateAggregatedAmountListReturnNothing
               CreateSicknessInsuranceService
               ModelCreateSicknessInsuranceModelLine
               SicknessInsuranceModelLineCreateMovements
               CheckUpdateAggregatedAmountListReturn
               PaysheetApplyTransformation
               Tic
               CheckPaysheetLineAreCreatedUsingWith3Lines
               CheckPaysheetLineAmountsWithSicknessInsuranceAndUrssaf
               CheckUpdateAggregatedAmountListReturnNothing
               CheckPaysheetLineAmountsWithSicknessInsuranceAndUrssaf
    """
    sequence_list.addSequenceString(sequence_string)
    sequence_list.play(self)

  def test_updateRemoveMovements(self):
    '''
2121
      Calculate the paySheet using a model, delete a model line in the model 
2122 2123 2124 2125 2126 2127 2128 2129 2130 2131 2132 2133 2134 2135 2136 2137 2138 2139 2140 2141 2142 2143 2144 2145 2146 2147 2148 2149 2150 2151 2152 2153 2154 2155 2156 2157 2158 2159
      and check that updateAggregatedAmount remove the movements corresponding
      to this model_line
    '''
    sequence_list = SequenceList()
    sequence_string = self.COMMON_BASIC_DOCUMENT_CREATION_SEQUENCE_STRING + """
               CheckUpdateAggregatedAmountListReturn
               PaysheetApplyTransformation
               Tic
               CheckPaysheetLineAreCreated
               CheckPaysheetLineAmounts
               CheckUpdateAggregatedAmountListReturnNothing
               ModelDelUrssafModelLine
               CheckUpdateAggregatedAmountListReturnAfterRemoveLine
               PaysheetApplyTransformation
               Tic
               CheckThereIsOnlyOnePaysheetLine
               CheckPaysheetLineLabourAmountOnly
               CheckUpdateAggregatedAmountListReturnNothing
               CheckPaysheetLineLabourAmountOnly
    """
    sequence_list.addSequenceString(sequence_string)
    sequence_list.play(self)

  def test_sourceSectionIsSetOnMovements(self):
    '''
      check that after apply transformation, source section is set on movment
      (using Business Process)
    '''
    sequence_list = SequenceList()
    sequence_string = self.COMMON_BASIC_DOCUMENT_CREATION_SEQUENCE_STRING + """
               PaysheetApplyTransformation
               Tic
               CheckSourceSectionOnMovements
    """
    sequence_list.addSequenceString(sequence_string)
    sequence_list.play(self)

  def test_sliceOnModelLine(self):
2160
    """
2161 2162
      It's possible to define some slices on model, and use it on model lines.
      Check that works and that after appy transformation, amounts are goods
2163
    """
2164 2165 2166 2167 2168 2169 2170 2171 2172 2173 2174 2175 2176 2177 2178 2179 2180 2181 2182 2183 2184 2185 2186 2187 2188 2189 2190 2191 2192 2193 2194 2195 2196 2197 2198 2199 2200 2201 2202 2203 2204 2205 2206 2207 2208 2209 2210 2211 2212 2213 2214 2215 2216 2217 2218 2219 2220 2221 2222 2223 2224 2225 2226 2227 2228 2229 2230 2231
    sequence_list = SequenceList()
    sequence_string = self.COMMON_BASIC_DOCUMENT_CREATION_SEQUENCE_STRING + """
               CreateUrssafService
               CreateLabourService
               CreateEmployer
               CreateEmployee
               CreateModelWithSlices
               Tic
               ModelCreateUrssafModelLineWithSlices
               Tic
               UrssafModelLineWithSlicesCreateMovements
               CreateBasicPaysheet
               PaysheetCreateLabourPaySheetLine
  """ + self.BUSINESS_PATH_CREATION_SEQUENCE_STRING + """
               CheckUpdateAggregatedAmountListReturnUsingSlices
               PaysheetApplyTransformation
               Tic
               CheckPaysheetLineAreCreatedUsingSlices
               CheckPaysheetLineAmountsUsingSlices
               CheckUpdateAggregatedAmountListReturnNothing
               CheckPaysheetLineAmountsUsingSlices
    """
    sequence_list.addSequenceString(sequence_string)
    sequence_list.play(self)

  def test_complexSliceOnModelLine(self):
    '''
      Check if there is only slice_200_to_400 (without previous
      slice_0_to_200), amount paid for this tax are correct.
    '''
    sequence_list = SequenceList()
    sequence_string = self.COMMON_BASIC_DOCUMENT_CREATION_SEQUENCE_STRING + """
               CreateUrssafService
               CreateLabourService
               CreateEmployer
               CreateEmployee
               CreateModelWithSlices
               Tic
               ModelCreateUrssafModelLineWithComplexSlices
               Tic
               UrssafModelLineWithComplexSlicesCreateMovements
               CreateBasicPaysheet
               PaysheetCreateLabourPaySheetLine
  """ + self.BUSINESS_PATH_CREATION_SEQUENCE_STRING + """
               CheckUpdateAggregatedAmountListReturnUsingComplexSlices
               PaysheetApplyTransformation
               Tic
               CheckPaysheetLineAreCreatedUsingComplexSlices
               CheckPaysheetLineAmountsUsingComplexSlices
               CheckUpdateAggregatedAmountListReturnNothing
               CheckPaysheetLineAmountsUsingComplexSlices
    """
    sequence_list.addSequenceString(sequence_string)
    sequence_list.play(self)

  def test_modelLineWithNonePrice(self):
    '''
      test the creation of lines when the price is not set, but only the
      quantity. This means that no ratio is applied on this line.
    '''
    sequence_list = SequenceList()
    sequence_string = """
               CreateUrssafService
               CreateLabourService
               CreateEmployer
               CreateEmployee
               CreatePriceCurrency
               CreateBasicModel
2232
               Tic
2233 2234 2235 2236 2237 2238 2239 2240 2241 2242 2243 2244 2245 2246 2247 2248 2249 2250 2251 2252 2253 2254 2255 2256 2257 2258
               ModelCreateUrssafModelLine
               UrssafModelLineCreateMovementsWithQuantityOnly
               CreateBasicPaysheet
               PaysheetCreateLabourPaySheetLine
  """ + self.BUSINESS_PATH_CREATION_SEQUENCE_STRING + """
               CheckUpdateAggregatedAmountListReturn
               PaysheetApplyTransformation
               Tic
               CheckPaysheetLineAreCreated
               CheckPaysheetLineAmountsWithQuantityOnly
               CheckUpdateAggregatedAmountListReturnNothing
               CheckPaysheetLineAmountsWithQuantityOnly
    """
    sequence_list.addSequenceString(sequence_string)
    sequence_list.play(self)

  def test_modelLineWithZeroPrice(self):
    '''Test the creation of lines when the price is set to zero: the line should
    not be created.'''
    sequence_list = SequenceList()
    sequence_string = """
               CreateLabourOutputService
               CreateEmployer
               CreateEmployee
               CreatePriceCurrency
               CreateBasicModel
2259
               Tic
2260 2261 2262 2263 2264 2265 2266 2267 2268 2269 2270
               CreateModelLineZeroPrice
               CreateBasicPaysheet
  """ + self.BUSINESS_PATH_CREATION_SEQUENCE_STRING + """
               PaysheetApplyTransformation
               Tic
               CheckNoPaysheetLineAreCreated
    """
    sequence_list.addSequenceString(sequence_string)
    sequence_list.play(self)

  def test_predicateOnModelLine(self):
2271
    """
2272 2273 2274
      Check predicates can be used on model lines to select a line or not.
      1 - employee have married marital status so Sickness Insurance tax
          should be applied, and Old age insurance should not be
2275
      2 - employee marital status is changed to single. So after re-apply 
2276 2277 2278
          the transformation, Sickness Insurance tax sould not be
          applied (and it's movements should be removed) but Old age insurance
          should be applied (and two movements should be created).
2279
    """
2280 2281 2282 2283 2284 2285 2286 2287 2288 2289 2290 2291 2292 2293 2294 2295 2296 2297 2298 2299 2300 2301 2302 2303 2304 2305 2306 2307 2308 2309 2310 2311
    sequence_list = SequenceList()
    sequence_string = self.COMMON_BASIC_DOCUMENT_CREATION_SEQUENCE_STRING + """
               CreateOldAgeInsuranaceService
               ModelCreateOldAgeInsuranceModelLine
               OldAgeInsuranceModelLineCreateMovements
               AddPredicateOnOldAgeInsuranceModelLineForSinglePerson
               CreateSicknessInsuranceService
               ModelCreateSicknessInsuranceModelLine
               SicknessInsuranceModelLineCreateMovements
               AddPredicateOnSicknessInsuranceModelLineForMarriedPerson
               SetMaritalStatusMarriedOnEmployee
               CheckUpdateAggregatedAmountListReturnUsingPredicate
               PaysheetApplyTransformation
               Tic
               CheckPaysheetLineAreCreatedUsingWith3Lines
               CheckPaysheetLineAmountsWithSicknessInsuranceAndUrssaf
               CheckUpdateAggregatedAmountListReturnNothing
               CheckPaysheetLineAmountsWithSicknessInsuranceAndUrssaf
               SetMaritalStatusSingleOnEmployee
               CheckUpdateAggregatedAmountListReturnAfterChangePredicate
               PaysheetApplyTransformation
               Tic
               CheckPaysheetLineAreCreatedUsingWith3Lines
               CheckPaysheetLineAmountsWithOldAgeInsuranceAndUrssaf
               CheckUpdateAggregatedAmountListReturnNothing
    """
    sequence_list.addSequenceString(sequence_string)
    sequence_list.play(self)

  def test_paySheetCalculationWithBonus(self):
    '''
      add one more line in the paysheet that will not be hour count and rate
2312
      (like the salary) but just a normal amount. Check applyTransformation
2313 2314 2315 2316 2317 2318 2319 2320 2321 2322 2323 2324 2325 2326 2327 2328 2329 2330 2331
      method result. It should create new movements applied on the slary + the
      bonnus
    '''
    sequence_list = SequenceList()
    sequence_string = self.COMMON_BASIC_DOCUMENT_CREATION_SEQUENCE_STRING + """
               CreateBonusService
               PaysheetCreateBonusPaySheetLine
               CheckUpdateAggregatedAmountListReturn
               PaysheetApplyTransformation
               Tic
               CheckPaysheetLineAreCreatedUsingBonus
               CheckPaysheetLineWithBonusAmounts
               CheckUpdateAggregatedAmountListReturnNothing
               CheckPaysheetLineWithBonusAmounts
    """
    sequence_list.addSequenceString(sequence_string)
    sequence_list.play(self)

  def test_modelSubObjectInheritance(self):
2332
    """
2333 2334 2335 2336 2337 2338 2339 2340 2341 2342 2343
      check that a model can inherite some datas from another
      the ineritance rules are the following :
       - a DATA could be a model_line, annotation_line, ratio_line
       - a model_line, annotation_line and a ratio_line have a REFERENCE
       - a model can have some DATA's
       - a model can inherite from another, that's mean :
         o At the calculation step, each DATA of the parent model will be
           checked : the DATA with a REFERENCE that's already in the child
           model will not entered in the calcul. The other will.
         o This will be repeated on each parent model and on each parent of
           the parent model,... until there is no parent model to inherite
2344
    """
2345 2346 2347 2348
    sequence_list = SequenceList()
    sequence_string = """
               CreateModelTree
               ModelTreeAddAnnotationLines
2349
               Tic
2350 2351 2352 2353 2354 2355 2356 2357 2358 2359 2360 2361 2362 2363 2364 2365 2366 2367 2368 2369 2370 2371 2372 2373 2374 2375 2376 2377 2378 2379
               CheckInheritanceModelReferenceDict
    """
    sequence_list.addSequenceString(sequence_string)
    sequence_list.play(self)

  def test_PaySheetTransaction_getMovementList(self):
    '''
      PaySheetTransaction_getMovementList is a script used by the listbox to
      display information about lines in the Paysheet. Just test the return of
      it.
    '''
    sequence_list = SequenceList()
    sequence_string = """
               CreateUrssafService
               CreateBasicPaysheet
               CheckPaySheetTransaction_getMovementListReturn
    """
    sequence_list.addSequenceString(sequence_string)
    sequence_list.play(self)

  def test_PayrollTaxesReport(self):
    currency_module = self.getCurrencyModule()
    if not hasattr(currency_module, 'EUR'):
      currency_module.newContent(
          portal_type = 'Currency',
          reference = "EUR", id = "EUR", base_unit_quantity=0.001 )
    eur = self.portal.currency_module.EUR
    service = self.portal.service_module.newContent(
                      portal_type='Service',
                      title='PS1',
2380
                      variation_base_category_list=('contribution_share',),
2381 2382
                      variation_category_list=('contribution_share/employee',
                                               'contribution_share/employer'))
2383
    self.createPayrollBusinesProcess()
2384 2385 2386 2387 2388 2389 2390 2391 2392 2393 2394 2395 2396 2397 2398 2399 2400 2401 2402 2403 2404 2405 2406 2407
    employer = self.portal.organisation_module.newContent(
                      portal_type='Organisation',
                      title='Employer',
                      price_currency_value=eur,
                      group_value=self.portal.portal_categories.group.demo_group)
    employee1 = self.portal.person_module.newContent(
                      portal_type='Person',
                      title='Employee One',
                      career_reference='E1',
                      career_subordination_value=employer)
    employee2 = self.portal.person_module.newContent(
                      portal_type='Person',
                      title='Employee Two',
                      career_reference='E2',
                      career_subordination_value=employer)
    provider = self.portal.organisation_module.newContent(
                      portal_type='Organisation',
                      title='Service Provider')
    other_provider = self.portal.organisation_module.newContent(
                      portal_type='Organisation',
                      title='Another Service Provider')
    ps1 = self.portal.accounting_module.newContent(
                      portal_type='Pay Sheet Transaction',
                      title='Employee 1',
2408
                      specialise_value=self.business_process,
2409 2410 2411 2412 2413 2414 2415 2416
                      destination_section_value=employer,
                      source_section_value=employee1,
                      start_date=DateTime(2006, 1, 1),)
    line = ps1.newContent(portal_type='Pay Sheet Line',
                   resource_value=service,
                   source_section_value=provider,
                # (destination is set by PaySheetTransaction.createPaySheetLine)
                   destination_value=employee1,
2417 2418 2419
                   variation_category_list=('contribution_share/employee',
                                            'contribution_share/employer'))
    cell_employee = line.newCell('contribution_share/employee',
2420 2421 2422 2423
                                portal_type='Pay Sheet Cell',
                                base_id='movement',
                                mapped_value_property_list=('price',
                                                            'quantity'),)
2424 2425
    cell_employee.edit(price=-.50, quantity=2000, contribution_share='employee')
    cell_employer = line.newCell('contribution_share/employer',
2426 2427 2428 2429
                                portal_type='Pay Sheet Cell',
                                base_id='movement',
                                mapped_value_property_list=('price',
                                                            'quantity'),)
2430
    cell_employer.edit(price=-.40, quantity=2000, contribution_share='employer')
2431 2432 2433 2434 2435
    ps1.plan()

    ps2 = self.portal.accounting_module.newContent(
                      portal_type='Pay Sheet Transaction',
                      title='Employee 2',
2436
                      specialise_value=self.business_process,
2437
                      destination_section_value=employer,
2438 2439 2440 2441 2442 2443
                      source_section_value=employee2,
                      start_date=DateTime(2006, 1, 1),)
    line = ps2.newContent(portal_type='Pay Sheet Line',
                   resource_value=service,
                   source_section_value=provider,
                   destination_value=employee2,
2444 2445 2446
                   variation_category_list=('contribution_share/employee',
                                            'contribution_share/employer'))
    cell_employee = line.newCell('contribution_share/employee',
2447 2448 2449 2450
                                portal_type='Pay Sheet Cell',
                                base_id='movement',
                                mapped_value_property_list=('price',
                                                            'quantity'),)
2451 2452
    cell_employee.edit(price=-.50, quantity=3000, contribution_share='employee')
    cell_employer = line.newCell('contribution_share/employer',
2453 2454 2455 2456
                                portal_type='Pay Sheet Cell',
                                base_id='movement',
                                mapped_value_property_list=('price',
                                                            'quantity'),)
2457
    cell_employer.edit(price=-.40, quantity=3000, contribution_share='employer')
2458 2459 2460 2461 2462

    other_line = ps2.newContent(portal_type='Pay Sheet Line',
                   resource_value=service,
                   destination_value=employee2,
                   source_section_value=other_provider,
2463 2464 2465
                   variation_category_list=('contribution_share/employee',
                                            'contribution_share/employer'))
    cell_employee = other_line.newCell('contribution_share/employee',
2466 2467 2468 2469
                                portal_type='Pay Sheet Cell',
                                base_id='movement',
                                mapped_value_property_list=('price',
                                                            'quantity'),)
2470 2471
    cell_employee.edit(price=-.46, quantity=2998, contribution_share='employee')
    cell_employer = other_line.newCell('contribution_share/employer',
2472 2473 2474 2475
                                portal_type='Pay Sheet Cell',
                                base_id='movement',
                                mapped_value_property_list=('price',
                                                            'quantity'),)
2476
    cell_employer.edit(price=-.42, quantity=2998, contribution_share='employer')
2477 2478 2479 2480 2481 2482 2483 2484 2485 2486 2487 2488 2489 2490 2491 2492 2493 2494 2495 2496 2497 2498 2499 2500 2501 2502 2503 2504
    self.stepTic()

    # AccountingTransactionModule_getPaySheetMovementMirrorSectionItemList is
    # used in the report dialog to display possible organisations.
    self.assertEquals(
        [('', ''),
         (other_provider.getTitle(), other_provider.getRelativeUrl()),
         (provider.getTitle(), provider.getRelativeUrl())],
        self.portal.accounting_module\
    .AccountingTransactionModule_getPaySheetMovementMirrorSectionItemList())

    # set request variables and render
    request_form = self.portal.REQUEST
    request_form['at_date'] = DateTime(2006, 2, 2)
    request_form['section_category'] = 'group/demo_group'
    request_form['simulation_state'] = ['draft', 'planned']
    request_form['resource'] = service.getRelativeUrl()
    request_form['mirror_section'] = provider.getRelativeUrl()

    report_section_list = self.getReportSectionList(
                             self.portal.accounting_module,
                             'AccountingTransactionModule_viewPaySheetLineReport')
    self.assertEquals(1, len(report_section_list))

    line_list = self.getListBoxLineList(report_section_list[0])
    data_line_list = [l for l in line_list if l.isDataLine()]
    self.assertEquals(2, len(data_line_list))

Jérome Perrin's avatar
Jérome Perrin committed
2505 2506
    # base_unit_quantity for EUR is set to 0.001 in the created currencies, so the
    # precision is 3. Editable Fields will reuse this precision.
2507
    self.checkPrecisionOfListBox(report_section_list[0], 3)
2508 2509 2510 2511 2512 2513

    self.checkLineProperties(data_line_list[0],
                            id=1,
                            employee_career_reference='E1',
                            employee_title='Employee One',
                            base=2000,
2514 2515
                            employee=2000 * .50,
                            employer=2000 * .40,
2516 2517 2518 2519 2520 2521
                            total=(2000 * .50 + 2000 * .40))
    self.checkLineProperties(data_line_list[1],
                            id=2,
                            employee_career_reference='E2',
                            employee_title='Employee Two',
                            base=3000,
2522 2523
                            employee=3000 * .50,
                            employer=3000 * .40,
2524 2525 2526 2527
                            total=(3000 * .50 + 3000 * .40))
    # stat line
    self.checkLineProperties(line_list[-1],
                            base=3000 + 2000,
2528 2529
                            employee=(3000 + 2000) * .50,
                            employer=(3000 + 2000) * .40,
2530 2531 2532 2533 2534 2535 2536 2537 2538 2539 2540 2541
                            total=((3000 + 2000) * .50 + (3000 + 2000) * .40))

  def test_PayrollTaxesReportDifferentSalaryRange(self):
    currency_module = self.getCurrencyModule()
    if not hasattr(currency_module, 'EUR'):
      currency_module.newContent(
          portal_type = 'Currency',
          reference = "EUR", id = "EUR", base_unit_quantity=0.001 )
    eur = self.portal.currency_module.EUR
    service = self.portal.service_module.newContent(
                      portal_type='Service',
                      title='PS1',
2542
                      variation_base_category_list=('contribution_share',
2543
                                                    'salary_range'),
2544 2545
                      variation_category_list=('contribution_share/employee',
                                               'contribution_share/employer',
2546 2547
                                               'salary_range/france/slice_a',
                                               'salary_range/france/slice_b'))
2548
    self.createPayrollBusinesProcess()
2549 2550 2551 2552 2553 2554 2555 2556 2557 2558 2559 2560 2561 2562 2563 2564 2565 2566 2567 2568 2569 2570 2571 2572
    employer = self.portal.organisation_module.newContent(
                      portal_type='Organisation',
                      title='Employer',
                      price_currency_value=eur,
                      group_value=self.portal.portal_categories.group.demo_group)
    employee1 = self.portal.person_module.newContent(
                      portal_type='Person',
                      title='Employee One',
                      career_reference='E1',
                      career_subordination_value=employer)
    employee2 = self.portal.person_module.newContent(
                      portal_type='Person',
                      title='Employee Two',
                      career_reference='E2',
                      career_subordination_value=employer)
    provider = self.portal.organisation_module.newContent(
                      portal_type='Organisation',
                      title='Service Provider')
    other_provider = self.portal.organisation_module.newContent(
                      portal_type='Organisation',
                      title='Another Service Provider')
    ps1 = self.portal.accounting_module.newContent(
                      portal_type='Pay Sheet Transaction',
                      title='Employee 1',
2573
                      specialise_value=self.business_process,
2574 2575 2576 2577 2578 2579 2580 2581
                      destination_section_value=employer,
                      source_section_value=employee1,
                      start_date=DateTime(2006, 1, 1),)
    line = ps1.newContent(portal_type='Pay Sheet Line',
                   resource_value=service,
                   source_section_value=provider,
                # (destination is set by PaySheetTransaction.createPaySheetLine)
                   destination_value=employee1,
2582 2583
                   variation_category_list=('contribution_share/employee',
                                            'contribution_share/employer',
2584 2585
                                            'salary_range/france/slice_a',
                                            'salary_range/france/slice_b'))
2586
    cell_employee_a = line.newCell('contribution_share/employee',
2587 2588 2589 2590 2591 2592
                                   'salary_range/france/slice_a',
                                   portal_type='Pay Sheet Cell',
                                   base_id='movement',
                                   mapped_value_property_list=('price',
                                                               'quantity'),)
    cell_employee_a.edit(price=-.50, quantity=1000,
2593
                         contribution_share='employee',
2594
                         salary_range='france/slice_a')
2595
    cell_employee_b = line.newCell('contribution_share/employee',
2596 2597 2598 2599 2600 2601
                                   'salary_range/france/slice_b',
                                   portal_type='Pay Sheet Cell',
                                   base_id='movement',
                                   mapped_value_property_list=('price',
                                                               'quantity'),)
    cell_employee_b.edit(price=-.20, quantity=500,
2602
                         contribution_share='employee',
2603 2604
                         salary_range='france/slice_b')

2605
    cell_employer_a = line.newCell('contribution_share/employer',
2606 2607 2608 2609 2610 2611
                                   'salary_range/france/slice_a',
                                   portal_type='Pay Sheet Cell',
                                   base_id='movement',
                                   mapped_value_property_list=('price',
                                                               'quantity'),)
    cell_employer_a.edit(price=-.40, quantity=1000,
2612
                         contribution_share='employer',
2613
                         salary_range='france/slice_a')
2614
    cell_employer_b = line.newCell('contribution_share/employer',
2615 2616 2617 2618 2619 2620
                                   'salary_range/france/slice_b',
                                   portal_type='Pay Sheet Cell',
                                   base_id='movement',
                                   mapped_value_property_list=('price',
                                                               'quantity'),)
    cell_employer_b.edit(price=-.32, quantity=500,
2621
                         contribution_share='employer',
2622 2623 2624 2625 2626 2627 2628
                         salary_range='france/slice_b')

    ps1.plan()

    ps2 = self.portal.accounting_module.newContent(
                      portal_type='Pay Sheet Transaction',
                      title='Employee 2',
2629
                      specialise_value=self.business_process,
2630 2631 2632 2633 2634 2635 2636
                      destination_section_value=employer,
                      source_section_value=employee2,
                      start_date=DateTime(2006, 1, 1),)
    line = ps2.newContent(portal_type='Pay Sheet Line',
                   resource_value=service,
                   source_section_value=provider,
                   destination_value=employee2,
2637 2638
                   variation_category_list=('contribution_share/employee',
                                            'contribution_share/employer',
2639 2640
                                            'salary_range/france/slice_a',
                                            'salary_range/france/slice_b'))
2641
    cell_employee_a = line.newCell('contribution_share/employee',
2642 2643 2644 2645 2646 2647 2648
                                   'salary_range/france/slice_a',
                                   portal_type='Pay Sheet Cell',
                                   base_id='movement',
                                   mapped_value_property_list=('price',
                                                               'quantity'),)
    cell_employee_a.edit(price=-.50, quantity=1000,
                         salary_range='france/slice_a',
2649 2650
                         contribution_share='employee')
    cell_employee_b = line.newCell('contribution_share/employee',
2651 2652 2653 2654 2655 2656 2657
                                   'salary_range/france/slice_b',
                                   portal_type='Pay Sheet Cell',
                                   base_id='movement',
                                   mapped_value_property_list=('price',
                                                               'quantity'),)
    cell_employee_b.edit(price=-.20, quantity=3000,
                         salary_range='france/slice_b',
2658
                         contribution_share='employee')
2659

2660
    cell_employer_a = line.newCell('contribution_share/employer',
2661 2662 2663 2664 2665 2666 2667
                                   'salary_range/france/slice_a',
                                   portal_type='Pay Sheet Cell',
                                   base_id='movement',
                                   mapped_value_property_list=('price',
                                                               'quantity'),)
    cell_employer_a.edit(price=-.40, quantity=1000,
                         salary_range='france/slice_a',
2668 2669
                         contribution_share='employer')
    cell_employer_b = line.newCell('contribution_share/employer',
2670 2671 2672 2673 2674 2675 2676
                                   'salary_range/france/slice_b',
                                   portal_type='Pay Sheet Cell',
                                   base_id='movement',
                                   mapped_value_property_list=('price',
                                                               'quantity'),)
    cell_employer_b.edit(price=-.32, quantity=3000,
                         salary_range='france/slice_b',
2677
                         contribution_share='employer')
2678 2679 2680 2681 2682 2683 2684 2685 2686 2687 2688 2689 2690 2691 2692 2693 2694 2695 2696 2697 2698 2699 2700 2701
    self.stepTic()

    # set request variables and render
    request_form = self.portal.REQUEST
    request_form['at_date'] = DateTime(2006, 2, 2)
    request_form['section_category'] = 'group/demo_group'
    request_form['simulation_state'] = ['draft', 'planned']
    request_form['resource'] = service.getRelativeUrl()
    request_form['mirror_section'] = provider.getRelativeUrl()

    report_section_list = self.getReportSectionList(
                             self.portal.accounting_module,
                             'AccountingTransactionModule_viewPaySheetLineReport')
    self.assertEquals(1, len(report_section_list))

    line_list = self.getListBoxLineList(report_section_list[0])
    data_line_list = [l for l in line_list if l.isDataLine()]
    self.assertEquals(6, len(data_line_list))

    self.checkLineProperties(data_line_list[0],
                            id=1,
                            employee_career_reference='E1',
                            employee_title='Employee One',
                            base=1000,
2702 2703
                            employee=1000 * .50,
                            employer=1000 * .40,
2704 2705 2706 2707 2708 2709
                            total=(1000 * .50 + 1000 * .40))
    self.checkLineProperties(data_line_list[1],
                            id=2,
                            employee_career_reference='E2',
                            employee_title='Employee Two',
                            base=1000,
2710 2711
                            employee=1000 * .50,
                            employer=1000 * .40,
2712 2713 2714 2715
                            total=(1000 * .50 + 1000 * .40))
    self.checkLineProperties(data_line_list[2],
                            employee_title='Total Slice A',
                            base=2000,
2716 2717
                            employee=2000 * .50,
                            employer=2000 * .40,
2718 2719 2720 2721 2722 2723 2724 2725
                            #total=(2000 * .50 + 2000 * .40)
                            )

    self.checkLineProperties(data_line_list[3],
                            id=3,
                            employee_career_reference='E1',
                            employee_title='Employee One',
                            base=500,
2726 2727
                            employee=500 * .20,
                            employer=500 * .32,
2728 2729 2730 2731 2732 2733
                            total=(500 * .20 + 500 * .32))
    self.checkLineProperties(data_line_list[4],
                            id=4,
                            employee_career_reference='E2',
                            employee_title='Employee Two',
                            base=3000,
2734 2735
                            employee=3000 * .20,
                            employer=3000 * .32,
2736 2737 2738 2739
                            total=(3000 * .20 + 3000 * .32))
    self.checkLineProperties(data_line_list[5],
                            employee_title='Total Slice B',
                            base=3500,
2740 2741
                            employee=3500 * .20,
                            employer=3500 * .32,
2742 2743 2744 2745 2746 2747
                            #total=(3500 * .20 + 3500 * .32),
                            )

    # stat line
    self.checkLineProperties(line_list[-1],
                            base=2000 + 3500,
2748 2749
                            employee=(2000 * .50 + 3500 * .20),
                            employer=(2000 * .40 + 3500 * .32),
2750 2751 2752 2753 2754 2755 2756 2757 2758 2759 2760 2761 2762
                            total=((2000 * .50 + 3500 * .20) +
                                   (2000 * .40 + 3500 * .32)))

  def test_NetSalaryReport(self):
    currency_module = self.getCurrencyModule()
    if not hasattr(currency_module, 'EUR'):
      currency_module.newContent(
          portal_type = 'Currency',
          reference = "EUR", id = "EUR", base_unit_quantity=0.001 )
    eur = self.portal.currency_module.EUR
    salary_service = self.portal.service_module.newContent(
                      portal_type='Service',
                      title='Gross Salary',
2763
                      variation_base_category_list=('contribution_share',),
2764 2765
                      variation_category_list=('contribution_share/employee',
                                               'contribution_share/employer'))
2766 2767 2768
    service = self.portal.service_module.newContent(
                      portal_type='Service',
                      title='PS1',
2769
                      variation_base_category_list=('contribution_share',),
2770 2771
                      variation_category_list=('contribution_share/employee',
                                               'contribution_share/employer'))
2772
    self.createPayrollBusinesProcess()
2773 2774 2775 2776 2777 2778 2779 2780 2781 2782 2783 2784 2785 2786 2787 2788 2789 2790 2791 2792 2793 2794 2795 2796 2797 2798 2799 2800
    employer = self.portal.organisation_module.newContent(
                      portal_type='Organisation',
                      title='Employer',
                      price_currency_value=eur,
                      group_value=self.portal.portal_categories.group.demo_group)
    employee1 = self.portal.person_module.newContent(
                      portal_type='Person',
                      title='Employee One',
                      career_reference='E1',
                      career_subordination_value=employer)
    employee1_ba = employee1.newContent(portal_type='Bank Account',
                                        title='Bank 1')
    employee2 = self.portal.person_module.newContent(
                      portal_type='Person',
                      title='Employee Two',
                      career_reference='E2',
                      career_subordination_value=employer)
    employee2_ba = employee2.newContent(portal_type='Bank Account',
                                        title='Bank 2')
    provider = self.portal.organisation_module.newContent(
                      portal_type='Organisation',
                      title='Service Provider')
    other_provider = self.portal.organisation_module.newContent(
                      portal_type='Organisation',
                      title='Another Service Provider')
    ps1 = self.portal.accounting_module.newContent(
                      portal_type='Pay Sheet Transaction',
                      title='Employee 1',
2801
                      specialise_value=self.business_process,
2802 2803 2804 2805 2806 2807 2808
                      destination_section_value=employer,
                      source_section_value=employee1,
                      payment_condition_source_payment_value=employee1_ba,
                      start_date=DateTime(2006, 1, 1),)
    line = ps1.newContent(portal_type='Pay Sheet Line',
                   resource_value=salary_service,
                   destination_value=employee1,
2809
                   base_contribution_list=['base_amount/payroll/report/salary/net',],
2810 2811 2812
                   variation_category_list=('contribution_share/employee',
                                            'contribution_share/employer'))
    cell_employee = line.newCell('contribution_share/employee',
2813 2814 2815 2816
                                portal_type='Pay Sheet Cell',
                                base_id='movement',
                                mapped_value_property_list=('price',
                                                            'quantity'),)
2817
    cell_employee.edit(price=1, quantity=2000, contribution_share='employee')
2818 2819 2820 2821
    line = ps1.newContent(portal_type='Pay Sheet Line',
                   resource_value=service,
                   source_section_value=provider,
                   destination_value=employee1,
2822
                   base_contribution_list=['base_amount/payroll/report/salary/net',],
2823 2824 2825
                   variation_category_list=('contribution_share/employee',
                                            'contribution_share/employer'))
    cell_employee = line.newCell('contribution_share/employee',
2826 2827 2828 2829
                                portal_type='Pay Sheet Cell',
                                base_id='movement',
                                mapped_value_property_list=('price',
                                                            'quantity'),)
2830 2831
    cell_employee.edit(price=-.50, quantity=2000, contribution_share='employee')
    cell_employer = line.newCell('contribution_share/employer',
2832 2833 2834 2835
                                portal_type='Pay Sheet Cell',
                                base_id='movement',
                                mapped_value_property_list=('price',
                                                            'quantity'),)
2836
    cell_employer.edit(price=-.40, quantity=2000, contribution_share='employer')
2837 2838 2839 2840 2841
    ps1.plan()

    ps2 = self.portal.accounting_module.newContent(
                      portal_type='Pay Sheet Transaction',
                      title='Employee 2',
2842
                      specialise_value=self.business_process,
2843 2844 2845 2846 2847 2848 2849
                      destination_section_value=employer,
                      source_section_value=employee2,
                      payment_condition_source_payment_value=employee2_ba,
                      start_date=DateTime(2006, 1, 1),)
    line = ps2.newContent(portal_type='Pay Sheet Line',
                   resource_value=salary_service,
                   destination_value=employee2,
2850
                   base_contribution_list=['base_amount/payroll/report/salary/net',],
2851 2852 2853
                   variation_category_list=('contribution_share/employee',
                                            'contribution_share/employer'))
    cell_employee = line.newCell('contribution_share/employee',
2854 2855 2856 2857
                                portal_type='Pay Sheet Cell',
                                base_id='movement',
                                mapped_value_property_list=('price',
                                                            'quantity'),)
2858
    cell_employee.edit(price=1, quantity=3000, contribution_share='employee')
2859 2860 2861 2862
    line = ps2.newContent(portal_type='Pay Sheet Line',
                   resource_value=service,
                   source_section_value=provider,
                   destination_value=employee2,
2863
                   base_contribution_list=['base_amount/payroll/report/salary/net',],
2864 2865 2866
                   variation_category_list=('contribution_share/employee',
                                            'contribution_share/employer'))
    cell_employee = line.newCell('contribution_share/employee',
2867 2868 2869 2870
                                portal_type='Pay Sheet Cell',
                                base_id='movement',
                                mapped_value_property_list=('price',
                                                            'quantity'),)
2871 2872
    cell_employee.edit(price=-.50, quantity=3000, contribution_share='employee')
    cell_employer = line.newCell('contribution_share/employer',
2873 2874 2875 2876
                                portal_type='Pay Sheet Cell',
                                base_id='movement',
                                mapped_value_property_list=('price',
                                                            'quantity'),)
2877
    cell_employer.edit(price=-.40, quantity=3000, contribution_share='employer')
2878 2879 2880 2881 2882 2883 2884 2885 2886 2887 2888 2889 2890 2891 2892 2893 2894
    self.stepTic()

    # set request variables and render
    request_form = self.portal.REQUEST
    request_form['at_date'] = DateTime(2006, 2, 2)
    request_form['section_category'] = 'group/demo_group'
    request_form['simulation_state'] = ['draft', 'planned']

    report_section_list = self.getReportSectionList(
                             self.portal.accounting_module,
                             'AccountingTransactionModule_viewNetSalaryReport')
    self.assertEquals(1, len(report_section_list))

    line_list = self.getListBoxLineList(report_section_list[0])
    data_line_list = [l for l in line_list if l.isDataLine()]
    self.assertEquals(2, len(data_line_list))

Jérome Perrin's avatar
Jérome Perrin committed
2895 2896
    # base_unit_quantity for EUR is set to 0.001 in the created currencies, so the
    # precision is 3. Editable Fields will reuse this precision.
2897
    self.checkPrecisionOfListBox(report_section_list[0], 3)
2898 2899 2900 2901 2902 2903 2904 2905 2906 2907 2908 2909 2910 2911 2912 2913

    self.checkLineProperties(data_line_list[0],
                            employee_career_reference='E1',
                            employee_title='Employee One',
                            employee_bank_account='Bank 1',
                            total_price=2000 - (2000 * .5),)
    self.checkLineProperties(data_line_list[1],
                            employee_career_reference='E2',
                            employee_title='Employee Two',
                            employee_bank_account='Bank 2',
                            total_price=3000 - (3000 * .5),)
    # stat line
    self.checkLineProperties(
            line_list[-1],
            total_price=3000 + 2000 - (2000 * .5) - (3000 * .5))

2914
  def createPayrollBusinesProcess(self): 
2915 2916 2917 2918 2919
    currency_module = self.getCurrencyModule()
    if not hasattr(currency_module, 'EUR'):
      currency_module.newContent(
          portal_type = 'Currency',
          reference = "EUR", id = "EUR", base_unit_quantity=0.001 )
2920

2921
    # create services
2922
    self.base_salary = self.portal.service_module.newContent(
2923 2924 2925
      portal_type='Service',
      title='Base Salary',
      product_line='base_salary',
2926
      variation_base_category_list=('contribution_share',),
2927 2928
      variation_category_list=('contribution_share/employee',
                               'contribution_share/employer'))
2929
    self.bonus = self.portal.service_module.newContent(
2930 2931 2932
      portal_type='Service',
      title='Bonus',
      product_line='base_salary',
2933
      variation_base_category_list=('contribution_share',),
2934 2935
      variation_category_list=('contribution_share/employee',
                               'contribution_share/employer'))
2936
    self.deductions = self.portal.service_module.newContent(
2937 2938 2939
      portal_type='Service',
      title='Deductions',
      product_line='base_salary',
2940
      variation_base_category_list=('contribution_share',),
2941 2942
      variation_category_list=('contribution_share/employee',
                               'contribution_share/employer'))
2943
    self.tax1 = self.portal.service_module.newContent(
2944 2945 2946
      portal_type='Service',
      title='Tax1',
      product_line='payroll_tax_1',
2947
      variation_base_category_list=('contribution_share',),
2948 2949
      variation_category_list=('contribution_share/employee',
                               'contribution_share/employer'))
2950 2951

    # create accounts
2952
    self.account_payroll_wages_expense = self.portal.account_module.newContent(
2953 2954 2955
                          portal_type='Account',
                          title='Payroll Wages (expense)',
                          account_type='expense',)
2956
    self.account_payroll_taxes_expense = self.portal.account_module.newContent(
2957 2958 2959
                          portal_type='Account',
                          title='Payroll Taxes (expense)',
                          account_type='expense',)
2960
    self.account_net_wages = self.portal.account_module.newContent(
2961 2962 2963
                          portal_type='Account',
                          title='Net Wages',
                          account_type='liability/payable',)
2964
    self.account_payroll_taxes = self.portal.account_module.newContent(
2965 2966 2967 2968
                          portal_type='Account',
                          title='Payroll Taxes',
                          account_type='liability/payable',)

2969
    self.business_process = self.createBusinessProcess()
2970

2971
    kw = dict(business_process=self.business_process,
2972 2973 2974 2975 2976 2977 2978 2979 2980
              trade_phase='default/accounting',
              trade_date='trade_phase/default/invoicing',
              membership_criterion_base_category_list=['contribution_share',
                                                       'product_line'],
              criterion_property_dict={'portal_type': 'Simulation Movement'})

    # Employee Share * Base Salary
    self.createTradeModelPath(reference='payroll_base_1',
       efficiency=1,
2981
       destination_value=self.account_payroll_wages_expense,
2982 2983 2984 2985 2986
       membership_criterion_category_list=['contribution_share/employee',
                                           'product_line/base_salary'],
       **kw)
    self.createTradeModelPath(reference='payroll_base_2',
       efficiency=-1,
2987
       destination_value=self.account_net_wages,
2988 2989 2990 2991 2992 2993
       membership_criterion_category_list=['contribution_share/employee',
                                           'product_line/base_salary'],
       **kw)
    # Employer Share * Payroll Tax 1
    self.createTradeModelPath(reference='payroll_employer_tax1',
       efficiency=1,
2994
       destination_value=self.account_payroll_taxes,
2995 2996 2997 2998 2999
       membership_criterion_category_list=['contribution_share/employer',
                                      'product_line/payroll_tax_1'],
       **kw)
    self.createTradeModelPath(reference='payroll_employer_tax2',
       efficiency=-1,
3000
       destination_value=self.account_payroll_taxes_expense,
3001 3002 3003 3004 3005 3006 3007
       membership_criterion_category_list=['contribution_share/employer',
                                           'product_line/payroll_tax_1'],
       **kw)

    # Employee Share * Payroll Tax 1
    self.createTradeModelPath(reference='payroll_employee_tax1',
       efficiency=1,
3008
       destination_value=self.account_payroll_taxes,
3009 3010 3011 3012 3013
       membership_criterion_category_list=['contribution_share/employee',
                                           'product_line/payroll_tax_1'],
       **kw)
    self.createTradeModelPath(reference='payroll_employee_tax2',
       efficiency=-1,
3014
       destination_value=self.account_net_wages,
3015 3016 3017 3018 3019
       source_method_id=\
        'SimulationMovement_generatePrevisionForEmployeeSharePaySheetMovement',
       membership_criterion_category_list=['contribution_share/employee',
                                           'product_line/payroll_tax_1'],
       **kw)
3020

3021 3022 3023 3024 3025 3026 3027 3028 3029 3030 3031 3032
  def createPayrollAccountingBusinessLink(self):
    invoiced = self.portal.portal_categories.trade_state.invoiced
    completed_state_list = ['delivered']
    frozen_state_list = ['delivered', 'confirmed']
    self.createBusinessLink(
      self.business_process,
      successor_value=invoiced,
      trade_phase='default/accounting',
      completed_state_list=completed_state_list,
      frozen_state_list=frozen_state_list,
      delivery_builder_list=('portal_deliveries/pay_sheet_transaction_builder',))

3033
  def test_AccountingLineGeneration(self):
3034
    # create a pay sheet
3035
    self.createPayrollBusinesProcess()
3036
    self.createPayrollAccountingBusinessLink()
3037 3038 3039 3040 3041 3042 3043 3044 3045 3046 3047 3048 3049 3050 3051 3052 3053
    eur = self.portal.currency_module.EUR
    employer = self.portal.organisation_module.newContent(
                      portal_type='Organisation',
                      title='Employer',
                      price_currency_value=eur,
                      group_value=self.portal.portal_categories.group.demo_group)
    employee = self.portal.person_module.newContent(
                      portal_type='Person',
                      title='Employee',
                      career_reference='E1',
                      career_subordination_value=employer)
    provider = self.portal.organisation_module.newContent(
                      portal_type='Organisation',
                      title='Service Provider')

    ps = self.portal.accounting_module.newContent(
                      portal_type='Pay Sheet Transaction',
3054
                      specialise_value=self.business_process,
3055 3056 3057 3058 3059 3060 3061 3062 3063 3064
                      price_currency_value=eur,
                      resource_value=eur,
                      title='Employee 1',
                      destination_section_value=employer,
                      source_section_value=employee,
                      start_date=DateTime(2006, 1, 1),)

    # base salary = 2000
    line = ps.newContent(portal_type='Pay Sheet Line',
                   title='Base salary',
3065
                   resource_value=self.base_salary,
3066
                   destination_value=employee,
3067 3068 3069
                   variation_category_list=('contribution_share/employee',
                                            'contribution_share/employer'))
    cell_employee = line.newCell('contribution_share/employee',
3070 3071 3072 3073
                                portal_type='Pay Sheet Cell',
                                base_id='movement',
                                mapped_value_property_list=('price',
                                                            'quantity'),)
3074 3075
    cell_employee.edit(price=1, quantity=2000, contribution_share='employee')
    cell_employer = line.newCell('contribution_share/employer',
3076 3077 3078 3079
                                portal_type='Pay Sheet Cell',
                                base_id='movement',
                                mapped_value_property_list=('price',
                                                            'quantity'),)
3080
    cell_employer.edit(price=1, quantity=2000, contribution_share='employer')
3081 3082 3083 3084

    # base_salary += 100 (bonus)
    line = ps.newContent(portal_type='Pay Sheet Line',
                   title='Bonus',
3085
                   resource_value=self.bonus,
3086
                   destination_value=employee,
3087 3088 3089
                   variation_category_list=('contribution_share/employee',
                                            'contribution_share/employer'))
    cell_employee = line.newCell('contribution_share/employee',
3090 3091 3092 3093
                                portal_type='Pay Sheet Cell',
                                base_id='movement',
                                mapped_value_property_list=('price',
                                                            'quantity'),)
3094 3095
    cell_employee.edit(price=1, quantity=100, contribution_share='employee')
    cell_employer = line.newCell('contribution_share/employer',
3096 3097 3098 3099
                                portal_type='Pay Sheet Cell',
                                base_id='movement',
                                mapped_value_property_list=('price',
                                                            'quantity'),)
3100
    cell_employer.edit(price=1, quantity=100, contribution_share='employer')
3101 3102 3103 3104

    # base_salary -= 50 (deductions)   => base_salary == 2050
    line = ps.newContent(portal_type='Pay Sheet Line',
                   title='Deduction',
3105
                   resource_value=self.deductions,
3106
                   destination_value=employee,
3107 3108 3109
                   variation_category_list=('contribution_share/employee',
                                            'contribution_share/employer'))
    cell_employee = line.newCell('contribution_share/employee',
3110 3111 3112 3113
                                portal_type='Pay Sheet Cell',
                                base_id='movement',
                                mapped_value_property_list=('price',
                                                            'quantity'),)
3114 3115
    cell_employee.edit(price=-1, quantity=50, contribution_share='employee')
    cell_employer = line.newCell('contribution_share/employer',
3116 3117 3118 3119
                                portal_type='Pay Sheet Cell',
                                base_id='movement',
                                mapped_value_property_list=('price',
                                                            'quantity'),)
3120
    cell_employer.edit(price=-1, quantity=50, contribution_share='employer')
3121 3122 3123 3124 3125

    # tax1 = 10% for employee ( 205 )
    #        20% for employer ( 410 )
    line = ps.newContent(portal_type='Pay Sheet Line',
                   title='Tax 1',
3126
                   resource_value=self.tax1,
3127 3128
                   source_section_value=provider,
                   destination_value=employee,
3129 3130 3131
                   variation_category_list=('contribution_share/employee',
                                            'contribution_share/employer'))
    cell_employee = line.newCell('contribution_share/employee',
3132 3133 3134 3135
                                portal_type='Pay Sheet Cell',
                                base_id='movement',
                                mapped_value_property_list=('price',
                                                            'quantity'),)
3136 3137
    cell_employee.edit(price=-.1, quantity=2050, contribution_share='employee')
    cell_employer = line.newCell('contribution_share/employer',
3138 3139 3140 3141
                                portal_type='Pay Sheet Cell',
                                base_id='movement',
                                mapped_value_property_list=('price',
                                                            'quantity'),)
3142
    cell_employer.edit(price=-.2, quantity=2050, contribution_share='employer')
3143 3144 3145 3146 3147 3148 3149 3150 3151 3152 3153 3154 3155 3156 3157 3158
    ps.plan()
    self.stepTic()

    related_applied_rule = ps.getCausalityRelatedValue(
                                portal_type='Applied Rule')
    self.assertNotEquals(related_applied_rule, None)

    # build accounting lines
    ps.confirm()
    ps.start()
    self.stepTic()
    accounting_line_list = ps.contentValues(
        portal_type='Pay Sheet Transaction Line')
    self.assertEquals(len(accounting_line_list), 4)

    line = [l for l in accounting_line_list
3159
            if l.getDestinationValue() == self.account_payroll_wages_expense][0]
3160 3161 3162 3163
    self.assertEquals(2050, line.getDestinationDebit())
    self.assertEquals(employer, line.getDestinationSectionValue())

    line = [l for l in accounting_line_list
3164
            if l.getDestinationValue() == self.account_net_wages][0]
3165 3166 3167 3168 3169
    self.assertEquals(2050 - 205, line.getDestinationCredit())
    self.assertEquals(employer, line.getDestinationSectionValue())
    self.assertEquals(employee, line.getSourceSectionValue())

    line = [l for l in accounting_line_list
3170
            if l.getDestinationValue() == self.account_payroll_taxes_expense][0]
3171 3172 3173 3174
    self.assertEquals(410, line.getDestinationDebit())
    self.assertEquals(employer, line.getDestinationSectionValue())

    line = [l for l in accounting_line_list
3175
            if l.getDestinationValue() == self.account_payroll_taxes][0]
3176 3177 3178 3179 3180 3181
    self.assertEquals(410 + 205, line.getDestinationCredit())
    self.assertEquals(employer, line.getDestinationSectionValue())
    self.assertEquals(provider, line.getSourceSectionValue())

  def test_modelWithoutReferenceValidity(self):
    ''' Check that if no REFERENCE are defined on a model, the behavior is
3182
    that this model is always valid. So check a Pay Sheet Transaction Line 
3183 3184 3185 3186 3187 3188 3189
    is created after calling the calculation script
    '''
    sequence_list = SequenceList()
    sequence_string = """
               CreatePriceCurrency
               CreateLabourOutputService
               CreateBusinessProcess
3190
               CreateBusinessLink
3191
               CreateUrssafRoubaixOrganisation
3192
               ModifyBusinessLinkTradePhase
3193 3194 3195 3196 3197 3198 3199
               Tic
               CheckModelWithoutRefValidity
    """
    sequence_list.addSequenceString(sequence_string)
    sequence_list.play(self)

  def test_modelWithoutDateValidity(self):
3200
    """ Check that if no DATE are defined on a model, the behavior is that 
3201 3202 3203
    this model is always valid. (XXX check if it's what we want)
    So check that a line is created after calling calculation script, even if
    there is no start_date or stop_date
3204
    """
3205 3206 3207 3208 3209
    sequence_list = SequenceList()
    sequence_string = """
               CreatePriceCurrency
               CreateLabourOutputService
               CreateBusinessProcess
3210
               CreateBusinessLink
3211
               CreateUrssafRoubaixOrganisation
3212
               ModifyBusinessLinkTradePhase
3213 3214 3215 3216 3217 3218 3219 3220 3221 3222 3223 3224 3225 3226 3227
               Tic
               CheckModelWithoutDateValidity
    """
    sequence_list.addSequenceString(sequence_string)
    sequence_list.play(self)

  def test_modelDateValidity(self):
    ''' check that model effective_date and expiration_date are take into
    account.
    '''
    sequence_list = SequenceList()
    sequence_string = """
               CreatePriceCurrency
               CreateLabourOutputService
               CreateBusinessProcess
3228
               CreateBusinessLink
3229
               CreateUrssafRoubaixOrganisation
3230
               ModifyBusinessLinkTradePhase
3231 3232 3233 3234 3235 3236 3237 3238
               Tic
               CheckModelDateValidity
    """
    sequence_list.addSequenceString(sequence_string)
    sequence_list.play(self)

  def test_modelVersioning(self):
    '''check that latest version is used in case of more thant one model is
3239
    matching using dates 
3240 3241 3242 3243 3244
    '''
    sequence_list = SequenceList()
    sequence_string = """
               CreatePriceCurrency
               CreateBusinessProcess
3245
               CreateBusinessLink
3246
               CreateUrssafRoubaixOrganisation
3247
               ModifyBusinessLinkTradePhase
3248 3249 3250 3251 3252 3253 3254 3255 3256 3257 3258 3259 3260 3261 3262 3263 3264 3265 3266 3267 3268 3269 3270 3271 3272 3273
               Tic
               CheckModelVersioning
    """
    sequence_list.addSequenceString(sequence_string)
    sequence_list.play(self)

  def test_modelSliceInheritance(self):
    '''Check the slice inheritance'''
    base_id = 'cell'
    paysheet_model_module = self.getPortalObject().paysheet_model_module
    model_1 = paysheet_model_module.newContent(
        portal_type='Pay Sheet Model',
        variation_settings_category_list=
              ('salary_range/france',))

    model_2 = paysheet_model_module.newContent(
        portal_type='Pay Sheet Model',
        specialise_value=model_1,)

    cell = model_1.newCell('salary_range/france/slice_a',
        portal_type='Pay Sheet Model Slice',
        base_id='cell')
    cell.setQuantityRangeMin(1)
    cell.setQuantityRangeMax(2)

    # model 2 gets cell values from model 1 (see test_07_model_getCell)
3274 3275 3276 3277
    model_2_cell = model_2.getCell('salary_range/france/slice_a')
    self.failIf(model_2_cell is None)
    self.assertEquals(1, model_2_cell.getQuantityRangeMin())
    self.assertEquals(2, model_2_cell.getQuantityRangeMax())
3278 3279 3280 3281 3282 3283 3284 3285 3286 3287 3288 3289 3290 3291 3292 3293 3294 3295 3296 3297 3298 3299

    # model 2 can override values
    model_2.edit(variation_settings_category_list=('salary_range/france',))
    cell = model_2.newCell('salary_range/france/slice_a',
                    portal_type='Pay Sheet Model Slice',
                    base_id='cell')
    cell.setQuantityRangeMin(3)
    cell.setQuantityRangeMax(4)
    self.assertEquals(3,
        model_2.getCell('salary_range/france/slice_a').getQuantityRangeMin())
    self.assertEquals(4,
        model_2.getCell('salary_range/france/slice_a').getQuantityRangeMax())

    # when unsetting variation settings category on this model will acquire
    # again values from specialised model
    model_2.edit(variation_settings_category_list=())
    self.assertEquals(1,
        model_2.getCell('salary_range/france/slice_a').getQuantityRangeMin())
    self.assertEquals(2,
        model_2.getCell('salary_range/france/slice_a').getQuantityRangeMax())

  def test_complexModelInheritanceScheme(self):
Fabien Morin's avatar
Fabien Morin committed
3300
    '''check inheritance and effective model with a more complex
3301 3302 3303 3304 3305 3306 3307 3308 3309 3310
    inheritance tree'''
    sequence_list = SequenceList()
    sequence_string = """
               CreatePriceCurrency
               CreateLabourOutputService
               Tic
               ComplexModelInheritanceScheme
    """
    sequence_list.addSequenceString(sequence_string)
    sequence_list.play(self)
3311

Fabien Morin's avatar
Fabien Morin committed
3312
  def test_updatePaysheetAfterModelModification(self):
3313 3314 3315
    '''generate a paysheet using a model, modify the model by adding a new
    model line with the same resource as the first one and update the
    paysheet. Check the paysheet values have been updated (new line created)
Fabien Morin's avatar
Fabien Morin committed
3316 3317 3318 3319 3320 3321 3322 3323 3324
    '''
    sequence_list = SequenceList()
    sequence_string = '''
               CreateUrssafService
               CreateLabourService
               CreateEmployer
               CreateEmployee
               CreatePriceCurrency
               CreateModelWithSlices
3325
               Tic
Fabien Morin's avatar
Fabien Morin committed
3326 3327 3328 3329 3330 3331 3332 3333 3334 3335 3336 3337 3338
               ModelCreateUrssafModelLine
               UrssafModelLineCreateMovements
               CreateBasicPaysheet
               PaysheetCreateLabourPaySheetLine
               Tic
  ''' + self.BUSINESS_PATH_CREATION_SEQUENCE_STRING +'''
               CheckUpdateAggregatedAmountListReturn
               PaysheetApplyTransformation
               Tic
               CheckPaysheetLineAreCreated
               CheckPaysheetLineAmounts
               CheckUpdateAggregatedAmountListReturnNothing
               CheckPaysheetLineAmounts
3339
               ModelCreateUrssafModelLineWithSlices
Fabien Morin's avatar
Fabien Morin committed
3340
               Tic
3341
               UrssafModelLineWithSlicesCreateMovements
Fabien Morin's avatar
Fabien Morin committed
3342
               Tic
Fabien Morin's avatar
Fabien Morin committed
3343 3344
               PaysheetApplyTransformation
               Tic
3345 3346
               CheckPaysheetLineAreCreatedAfterUpdateWithLinesWithSameResource
               CheckPaysheetLineAmountsAfterUpdateWithLinesWithSameResource
Fabien Morin's avatar
Fabien Morin committed
3347
               CheckUpdateAggregatedAmountListReturnNothing
3348
               CheckPaysheetLineAmountsAfterUpdateWithLinesWithSameResource
Fabien Morin's avatar
Fabien Morin committed
3349 3350 3351 3352 3353 3354 3355 3356 3357 3358
    '''
    sequence_list.addSequenceString(sequence_string)
    sequence_list.play(self)

  def test_propertiesAreSetOnPaysheetLines(self):
    '''check porperties from model line (like description, int_index,
    title, ...) are copied on the paysheet lines'''
    sequence_list = SequenceList()
    sequence_string = self.COMMON_BASIC_DOCUMENT_CREATION_SEQUENCE_STRING + """
               SetProperiesOnModelLines
Fabien Morin's avatar
Fabien Morin committed
3359
               Tic
Fabien Morin's avatar
Fabien Morin committed
3360 3361 3362 3363 3364 3365 3366 3367 3368 3369
               PaysheetApplyTransformation
               Tic
               CheckPaysheetLineAreCreated
               CheckPaysheetLineAmounts
               CheckPropertiesAreCopiedFromModelLineToPaySheetLine
    """
    sequence_list.addSequenceString(sequence_string)
    sequence_list.play(self)


3370 3371 3372 3373 3374
import unittest
def test_suite():
  suite = unittest.TestSuite()
  suite.addTest(unittest.makeSuite(TestPayroll))
  return suite