testERP5Simulation.py 25.9 KB
Newer Older
1
# -*- coding: utf-8 -*-
2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34
##############################################################################
#
# Copyright (c) 2009 Nexedi SA and Contributors. All Rights Reserved.
#
# WARNING: This program as such is intended to be used by professional
# programmers who take the whole responsibility of assessing all potential
# consequences resulting from its eventual inadequacies and bugs
# End users who are looking for a ready-to-use solution with commercial
# guarantees and support are strongly adviced to contract a Free Software
# 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., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
#
##############################################################################
"""
This test is experimental for new simulation implementation.
"""

import unittest
import transaction

35 36
from zLOG import LOG
from Products.CMFCore.utils import getToolByName
37 38
from Products.ERP5Type.tests.ERP5TypeTestCase import ERP5TypeTestCase
from Products.ERP5Type.tests.Sequence import SequenceList
Kazuhiko Shiozaki's avatar
Kazuhiko Shiozaki committed
39 40 41
from Products.ERP5Type.UnrestrictedMethod import UnrestrictedMethod
from testPackingList import TestPackingList
from testInvoice import TestSaleInvoice, TestInvoiceMixin
42

Kazuhiko Shiozaki's avatar
Kazuhiko Shiozaki committed
43
class TestERP5SimulationMixin(TestInvoiceMixin):
44
  def getBusinessTemplateList(self):
Kazuhiko Shiozaki's avatar
Kazuhiko Shiozaki committed
45 46
    return list(TestInvoiceMixin.getBusinessTemplateList(self)) + \
           ['erp5_administration', 'erp5_simulation',]
47 48

  def afterSetUp(self, quiet=1, run=1):
Kazuhiko Shiozaki's avatar
Kazuhiko Shiozaki committed
49
    TestInvoiceMixin.afterSetUp(self)
50
    portal_rules = self.portal.portal_rules
51
    for rule in portal_rules.objectValues(portal_type='Order Root Simulation Rule'):
52 53
      if rule.getValidationState() == 'validated':
        rule.invalidate()
54
    self.validateNewRules()
55 56 57 58 59 60 61
    self.setUpBusinessProcess()

  def setUpBusinessProcess(self):
    business_process = self.portal.business_process_module.erp5_default_business_process
    pay_business_path = business_process['pay']
    pay_business_path.setSource('account_module/bank')
    pay_business_path.setDestination('account_module/bank')
62

Kazuhiko Shiozaki's avatar
Kazuhiko Shiozaki committed
63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78
  @UnrestrictedMethod
  def createInvoiceTransactionRule(self, resource=None):
    """Create a sale invoice transaction rule with only one cell for
    product_line/apparel and default_region
    The accounting rule cell will have the provided resource, but this his more
    or less optional (as long as price currency is set correctly on order)
    """
    portal = self.portal
    account_module = portal.account_module
    for account_id, account_gap, account_type \
               in self.account_definition_list:
      if not account_id in account_module.objectIds():
        account = account_module.newContent(id=account_id)
        account.setGap(account_gap)
        account.setAccountType(account_type)
        portal.portal_workflow.doActionFor(account, 'validate_action')
79 80 81 82 83 84
    invoice_rule = portal.portal_rules.new_invoice_transaction_simulation_rule
    if invoice_rule.getValidationState() == 'validated':
      invoice_rule.invalidate()
    invoice_rule.deleteContent(list(invoice_rule.contentIds(filter={'portal_type':['Predicate', 'Accounting Rule Cell']})))
    transaction.commit()
    self.tic()
Kazuhiko Shiozaki's avatar
Kazuhiko Shiozaki committed
85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119
    region_predicate = invoice_rule.newContent(portal_type = 'Predicate')
    product_line_predicate = invoice_rule.newContent(portal_type = 'Predicate')
    region_predicate.edit(
      membership_criterion_base_category_list = ['destination_region'],
      membership_criterion_category_list =
                   ['destination_region/region/%s' % self.default_region ],
      int_index = 1,
      string_index = 'region'
    )
    product_line_predicate.edit(
      membership_criterion_base_category_list = ['product_line'],
      membership_criterion_category_list =
                            ['product_line/apparel'],
      int_index = 1,
      string_index = 'product'
    )
    product_line_predicate.immediateReindexObject()
    region_predicate.immediateReindexObject()

    invoice_rule.updateMatrix()
    cell_list = invoice_rule.getCellValueList(base_id='movement')
    self.assertEquals(len(cell_list),1)
    cell = cell_list[0]

    for line_id, line_source_id, line_destination_id, line_ratio in \
        self.transaction_line_definition_list:
      line = cell.newContent(id=line_id,
          portal_type='Accounting Transaction Line', quantity=line_ratio,
          resource_value=resource,
          source_value=account_module[line_source_id],
          destination_value=account_module[line_destination_id])

    invoice_rule.validate()
    transaction.commit()
    self.tic()
120 121

  def validateNewRules(self):
122
    # create an Order Rule document.
123
    portal_rules = self.portal.portal_rules
124
    new_order_rule = filter(
125 126
      lambda x:x.title == 'New Default Order Root Simulation Rule',
      portal_rules.objectValues(portal_type='Order Root Simulation Rule'))[0]
127 128 129
    if new_order_rule.getValidationState() != 'validated':
      new_order_rule.validate()

Kazuhiko Shiozaki's avatar
Kazuhiko Shiozaki committed
130
  def _acceptDecisionQuantity(self, document):
131
    solver_tool = self.portal.portal_solvers
Kazuhiko Shiozaki's avatar
Kazuhiko Shiozaki committed
132
    solver_process = solver_tool.newSolverProcess(document)
133
    quantity_solver_decision = filter(
134
      lambda x:x.getCausalityValue().getTestedProperty()=='quantity',
135
      solver_process.contentValues())[0]
Kazuhiko Shiozaki's avatar
Kazuhiko Shiozaki committed
136
    # use Quantity Accept Solver.
137
    quantity_solver_decision.setSolverValue(self.portal.portal_types['Accept Solver'])
138 139 140
    solver_process.buildTargetSolverList()
    solver_process.solve()

Kazuhiko Shiozaki's avatar
Kazuhiko Shiozaki committed
141 142 143
  def _acceptDivergenceOnInvoice(self, invoice, divergence_list):
    print invoice, divergence_list
    return self._acceptDecisionQuantity(invoice)
144

Kazuhiko Shiozaki's avatar
Kazuhiko Shiozaki committed
145
  def stepAcceptDecisionQuantity(self,sequence=None, sequence_list=None):
146
    """
Kazuhiko Shiozaki's avatar
Kazuhiko Shiozaki committed
147
    Solve quantity divergence by using solver tool.
148
    """
Kazuhiko Shiozaki's avatar
Kazuhiko Shiozaki committed
149 150
    packing_list = sequence.get('packing_list')
    self._acceptDecisionQuantity(packing_list)
151

Kazuhiko Shiozaki's avatar
Kazuhiko Shiozaki committed
152 153
  def stepAcceptDecisionQuantityInvoice(self, sequence=None,
                                        sequence_list=None):
154 155 156
    """
    Solve quantity divergence by using solver tool.
    """
Kazuhiko Shiozaki's avatar
Kazuhiko Shiozaki committed
157 158
    invoice = sequence.get('invoice')
    self._acceptDecisionQuantity(invoice)
159

Kazuhiko Shiozaki's avatar
Kazuhiko Shiozaki committed
160
  def stepAcceptDecisionResource(self,sequence=None, sequence_list=None):
161 162 163 164
    """
    Solve quantity divergence by using solver tool.
    """
    packing_list = sequence.get('packing_list')
165 166 167 168 169
    solver_tool = self.portal.portal_solvers
    solver_process = solver_tool.newSolverProcess(packing_list)
    resource_solver_decision = filter(
      lambda x:x.getCausalityValue().getTestedProperty()=='resource',
      solver_process.contentValues())[0]
170
    # use Resource Replacement Solver.
171
    resource_solver_decision.setSolverValue(self.portal.portal_types['Accept Solver'])
172
    solver_process.buildTargetSolverList()
173 174
    solver_process.solve()

Kazuhiko Shiozaki's avatar
Kazuhiko Shiozaki committed
175
  def stepSplitAndDeferPackingList(self, sequence=None, sequence_list=None):
176 177 178 179
    """
      Do the split and defer action
    """
    packing_list = sequence.get('packing_list')
180 181 182 183 184
    solver_tool = self.portal.portal_solvers
    solver_process = solver_tool.newSolverProcess(packing_list)
    quantity_solver_decision = filter(
      lambda x:x.getCausalityValue().getTestedProperty()=='quantity',
      solver_process.contentValues())[0]
185 186 187
    # use Quantity Split Solver.
    quantity_solver_decision.setSolverValue(self.portal.portal_types['Quantity Split Solver'])
    # configure for Quantity Split Solver.
188 189 190
    kw = {'delivery_solver':'FIFO',
          'start_date':self.datetime + 15,
          'stop_date':self.datetime + 25}
191 192
    quantity_solver_decision.updateConfiguration(**kw)
    solver_process.buildTargetSolverList()
193
    solver_process.solve()
194 195 196 197 198 199 200 201 202 203 204 205 206 207
    # build split deliveries manually. XXX ad-hoc
    previous_tag = None
    for delivery_builder in packing_list.getBuilderList():
      this_builder_tag = '%s_split_%s' % (packing_list.getPath(),
                                          delivery_builder.getId())
      after_tag = []
      if previous_tag:
        after_tag.append(previous_tag)
      delivery_builder.activate(
        after_method_id=('solve',
                         'immediateReindexObject',
                         'recursiveImmediateReindexObject',), # XXX too brutal.
        after_tag=after_tag,
        ).build(explanation_uid=packing_list.getCausalityValue().getUid())
208

209 210 211 212 213 214 215 216 217 218
  def _adoptPrevisionQuantity(self, packing_list):
    """
    Solve quantity divergence by using solver tool.
    """
    solver_tool = self.portal.portal_solvers
    solver_process = solver_tool.newSolverProcess(packing_list)
    quantity_solver_decision = filter(
      lambda x:x.getCausalityValue().getTestedProperty()=='quantity',
      solver_process.contentValues())[0]
    # use Quantity Adoption Solver.
219
    quantity_solver_decision.setSolverValue(self.portal.portal_types['Adopt Solver'])
220 221 222
    solver_process.buildTargetSolverList()
    solver_process.solve()

Kazuhiko Shiozaki's avatar
Kazuhiko Shiozaki committed
223 224 225 226 227 228 229 230 231
  def _adoptDivergenceOnInvoice(self, invoice, divergence_list):
    print invoice, divergence_list
    return self._adoptPrevisionQuantity(invoice)

  def _adoptDivergenceOnPackingList(self, packing_list, divergence_list):
    print packing_list, divergence_list
    return self._adoptPrevisionQuantity(packing_list)

  def stepAdoptPrevisionQuantity(self,sequence=None, sequence_list=None):
232 233 234 235 236 237 238
    """
    Solve quantity divergence by using solver tool.
    """
    packing_list = sequence.get('packing_list')
    self._adoptPrevisionQuantity(packing_list)

  def stepNewPackingListAdoptPrevisionQuantity(self, sequence=None,
Kazuhiko Shiozaki's avatar
Kazuhiko Shiozaki committed
239
                                               sequence_list=None):
240 241 242 243 244 245
    """
    Solve quantity divergence by using solver tool.
    """
    packing_list = sequence.get('new_packing_list')
    self._adoptPrevisionQuantity(packing_list)

Kazuhiko Shiozaki's avatar
Kazuhiko Shiozaki committed
246
  def stepAdoptPrevisionResource(self,sequence=None, sequence_list=None):
247 248 249 250 251 252 253 254 255 256
    """
    Solve resource divergence by using solver tool.
    """
    packing_list = sequence.get('packing_list')
    solver_tool = self.portal.portal_solvers
    solver_process = solver_tool.newSolverProcess(packing_list)
    resource_solver_decision = filter(
      lambda x:x.getCausalityValue().getTestedProperty()=='resource',
      solver_process.contentValues())[0]
    # use Resource Adopt Solver.
257
    resource_solver_decision.setSolverValue(self.portal.portal_types['Adopt Solver'])
258 259 260
    solver_process.buildTargetSolverList()
    solver_process.solve()

Kazuhiko Shiozaki's avatar
Kazuhiko Shiozaki committed
261
  def stepCheckPackingListLineWithSameResource(self,sequence=None, sequence_list=None):
262 263 264 265 266 267 268 269 270 271
    """
      Look if the packing list has new previsions
    """
    old_packing_list_line = sequence.get('packing_list_line')
    packing_list_line = old_packing_list_line.aq_parent[str(int(old_packing_list_line.getId())-1)]
    resource = sequence.get('resource')
    for line in sequence.get('packing_list').getMovementList():
      self.assertEquals(line.getResourceValue(), resource)
      self.assertEquals(line.getQuantity(), self.default_quantity)
      self.assertEquals(line.getCausalityList(),
272
                        [x.getParentValue().getParentValue().getDelivery() for x in \
273 274
                         line.getDeliveryRelatedValueList()])

Kazuhiko Shiozaki's avatar
Kazuhiko Shiozaki committed
275
  def stepUnifyDestinationWithDecision(self,sequence=None, sequence_list=None):
276 277 278 279 280 281 282 283 284 285
    """
      Check if simulation movement are disconnected
    """
    packing_list = sequence.get('packing_list')
    solver_tool = self.portal.portal_solvers
    solver_process = solver_tool.newSolverProcess(packing_list)
    for destination_solver_decision in filter(
      lambda x:x.getCausalityValue().getTestedProperty()=='destination',
      solver_process.contentValues()):
      # use Destination Replacement Solver.
286
      destination_solver_decision.setSolverValue(self.portal.portal_types['Accept Solver'])
287 288 289
    solver_process.buildTargetSolverList()
    solver_process.solve()

Kazuhiko Shiozaki's avatar
Kazuhiko Shiozaki committed
290
  def _unifyStartDateWithDecision(self, document):
291
    solver_tool = self.portal.portal_solvers
Kazuhiko Shiozaki's avatar
Kazuhiko Shiozaki committed
292
    solver_process = solver_tool.newSolverProcess(document)
293 294 295 296
    for start_date_solver_decision in filter(
      lambda x:x.getCausalityValue().getTestedProperty()=='start_date',
      solver_process.contentValues()):
      # use StartDate Replacement Solver.
297
      start_date_solver_decision.setSolverValue(self.portal.portal_types['Unify Solver'])
298
      # configure for Quantity Split Solver.
Kazuhiko Shiozaki's avatar
Kazuhiko Shiozaki committed
299
      kw = {'value':document.getStartDate()}
300 301 302 303
      start_date_solver_decision.updateConfiguration(**kw)
    solver_process.buildTargetSolverList()
    solver_process.solve()

Kazuhiko Shiozaki's avatar
Kazuhiko Shiozaki committed
304 305 306 307 308 309 310 311 312
  def stepUnifyStartDateWithDecision(self,sequence=None, sequence_list=None):
    packing_list = sequence.get('packing_list')
    self._unifyStartDateWithDecision(packing_list)

  def stepUnifyStartDateWithDecisionInvoice(self,sequence=None, sequence_list=None):
    invoice = sequence.get('invoice')
    self._unifyStartDateWithDecision(invoice)

  def stepUnifyStartDateWithPrevision(self,sequence=None, sequence_list=None):
313 314 315 316
    """
      Check if simulation movement are disconnected
    """
    packing_list = sequence.get('packing_list')
317 318 319
    applied_rule = sequence.get('applied_rule')
    simulation_line_list = applied_rule.objectValues()
    start_date = simulation_line_list[-1].getStartDate()
320 321 322 323 324 325
    solver_tool = self.portal.portal_solvers
    solver_process = solver_tool.newSolverProcess(packing_list)
    for start_date_solver_decision in filter(
      lambda x:x.getCausalityValue().getTestedProperty()=='start_date',
      solver_process.contentValues()):
      # use StartDate Replacement Solver.
326
      start_date_solver_decision.setSolverValue(self.portal.portal_types['Unify Solver'])
327
      # configure for Quantity Split Solver.
328
      kw = {'value':start_date}
329
      start_date_solver_decision.updateConfiguration(**kw)
330 331 332
    solver_process.buildTargetSolverList()
    solver_process.solve()

Kazuhiko Shiozaki's avatar
Kazuhiko Shiozaki committed
333
  def checkOrderRuleSimulation(self, rule_reference, sequence=None, sequence_list=None):
334 335 336 337 338 339 340 341 342 343 344 345 346 347 348 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
    """
      Test if simulation is matching order, be sure that rule_reference is used
      to expand simulation for order
    """
    order = sequence.get('order')
    related_applied_rule_list = order.getCausalityRelatedValueList( \
                                   portal_type=self.applied_rule_portal_type)
    no_applied_rule_state = ('draft', 'auto_planned')
    order_state = order.getSimulationState()

    if order_state in no_applied_rule_state:
      self.assertEquals(0, len(related_applied_rule_list))
    else:
      LOG('stepCheckOrderRuleSimulation', 0, 'related_applied_rule_list: %s' %
                   str([x.getObject() for x in related_applied_rule_list]))
      self.assertEquals(1, len(related_applied_rule_list))
      applied_rule = related_applied_rule_list[0].getObject()
      sequence.edit(applied_rule=applied_rule)
      self.failUnless(applied_rule is not None)
      self.failUnless(order_state, \
                      applied_rule.getLastExpandSimulationState())

      # Test if applied rule has a specialise value with passed rule_reference
      portal_rules = getToolByName(order, 'portal_rules')
      self.assertEquals(rule_reference,
                        applied_rule.getSpecialiseReference())

      simulation_movement_list = applied_rule.objectValues()
      sequence.edit(simulation_movement_list=simulation_movement_list)

      # Count the number of movement in order
      movement_list = order.getMovementList()
      # Check if number of unique movement is equal to number of
      # simulation movement
      unique_movement_list = dict(
        [('%r,%r,%r' % (x.getResource(), x.getVariationCategoryList(),
          x.getVariationPropertyDict())), x] for x in movement_list).values()
      self.assertEquals(len(unique_movement_list),
                        len(simulation_movement_list))
      # Check if all movements are related to simulation movements
Kazuhiko Shiozaki's avatar
Kazuhiko Shiozaki committed
374
      order_movement_list = sum([x.getDeliveryValueList() for x in \
375 376 377 378 379
                                 simulation_movement_list], [])
      self.failIfDifferentSet(movement_list, order_movement_list)

      # Check each simulation movement
      for simulation_movement in simulation_movement_list:
380
        order_movement_list = simulation_movement.getDeliveryValueList()
381 382 383 384 385 386 387 388 389 390 391 392 393 394 395 396 397 398
        # Test quantity
        self.assertEquals(sum([x.getQuantity() for x in order_movement_list]),
                          simulation_movement.getQuantity())
        for order_movement in order_movement_list:
          # Test price
          self.assertEquals(order_movement.getPrice(), \
                            simulation_movement.getPrice())
          # Test resource
          self.assertEquals(order_movement.getResource(), \
                            simulation_movement.getResource())
          # Test resource variation
          self.assertEquals(order_movement.getVariationText(), \
                            simulation_movement.getVariationText())
          self.assertEquals(order_movement.getVariationCategoryList(), \
                            simulation_movement.getVariationCategoryList())
          # XXX Test acquisition
          self.checkAcquisition(simulation_movement, order_movement)

Kazuhiko Shiozaki's avatar
Kazuhiko Shiozaki committed
399
  def stepModifySimulationLineQuantityForMergedLine(self,sequence=None, sequence_list=None):
400 401 402 403 404 405 406 407
    """
      Check if simulation movement are disconnected
    """
    applied_rule = sequence.get('applied_rule')
    simulation_line_list = applied_rule.objectValues()
    self.assertEquals(len(simulation_line_list), 1)
    for simulation_line in simulation_line_list:
      simulation_line.edit(quantity=self.default_quantity-2)
408
      simulation_line.getDeliveryValue().edit(quantity=self.default_quantity-2)
409

Kazuhiko Shiozaki's avatar
Kazuhiko Shiozaki committed
410
  def stepCheckSimulationQuantityUpdatedForMergedLine(self,sequence=None, sequence_list=None):
411 412 413 414 415 416 417 418 419 420 421
    """
      Test if the quantity of the simulation movement was changed
    """
    applied_rule = sequence.get('applied_rule')
    simulation_line_list = applied_rule.objectValues()
    self.assertEquals(len(simulation_line_list), 1)
    for simulation_line in simulation_line_list:
      self.assertEquals(simulation_line.getQuantity() + \
                        simulation_line.getDeliveryError(),
                        self.default_quantity * 2)

Kazuhiko Shiozaki's avatar
Kazuhiko Shiozaki committed
422
  def stepCheckPackingListLineWithDifferentResource(self,sequence=None, sequence_list=None):
423 424 425 426 427 428 429 430
    """
      Look if the packing list has new previsions
    """
    packing_list_line = sequence.get('packing_list_line')
    new_resource = sequence.get('resource')
    self.assertEquals(packing_list_line.getQuantity(), self.default_quantity*2)
    self.assertEquals(packing_list_line.getResourceValue(), new_resource)
    simulation_line_list = packing_list_line.getDeliveryRelatedValueList()
431
    order_line_list = sum([x.getParentValue().getParentValue().getDeliveryList() for x in simulation_line_list], [])
432 433 434
    self.assertEquals(sorted(packing_list_line.getCausalityList()),
                      sorted(order_line_list))

Kazuhiko Shiozaki's avatar
Kazuhiko Shiozaki committed
435 436 437 438
class TestERP5Simulation(TestERP5SimulationMixin, ERP5TypeTestCase):
  run_all_test = 1
  quiet = 0

439 440
  def afterSetUp(self):
    TestERP5SimulationMixin.afterSetUp(self)
441 442
    new_delivery_rule = self.portal.portal_rules.new_delivery_simulation_rule
    new_delivery_rule['quantity_tester'].edit(quantity=None,
443 444 445 446
                                           quantity_range_max=2,
                                           quantity_range_min=-1)

  def beforeTearDown(self):
447 448
    new_delivery_rule = self.portal.portal_rules.new_delivery_simulation_rule
    new_delivery_rule['quantity_tester'].edit(quantity=0,
449 450
                                           quantity_range_max=None,
                                           quantity_range_min=None)
451
    transaction.commit()
452
    TestERP5SimulationMixin.beforeTearDown(self)
Kazuhiko Shiozaki's avatar
Kazuhiko Shiozaki committed
453 454 455 456 457 458 459 460 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 499 500 501 502 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 554 555 556 557 558 559 560 561 562 563 564 565 566 567 568 569 570 571 572 573 574 575 576 577 578 579 580 581 582 583 584 585 586 587 588 589 590 591 592 593 594 595 596 597 598 599

  def _modifyPackingListLineQuantity(self, sequence=None,
      sequence_list=None, delta=0.0):
    """
    Set a increased quantity on packing list lines
    """
    packing_list = sequence.get('packing_list')
    quantity = self.default_quantity + delta
    sequence.edit(line_quantity=quantity)
    for packing_list_line in packing_list.objectValues(
        portal_type=self.packing_list_line_portal_type):
      packing_list_line.edit(quantity=quantity)
    sequence.edit(last_delta=delta)

  def stepIncreasePackingListLineQuantity2(self, sequence=None,
      sequence_list=None, **kw):
    return self._modifyPackingListLineQuantity(sequence, sequence_list, 2.0)

  def stepDecreasePackingListLineQuantity1(self, sequence=None,
      sequence_list=None, **kw):
    return self._modifyPackingListLineQuantity(sequence, sequence_list, -1.0)

  def stepDecreasePackingListLineQuantity10(self, sequence=None,
      sequence_list=None, **kw):
    return self._modifyPackingListLineQuantity(sequence, sequence_list, -10.0)

  def stepSplitAndDeferPackingList(self, sequence=None, sequence_list=None, **kw):
    """
      Do the split and defer action
    """
    packing_list = sequence.get('packing_list')
    solver_tool = self.portal.portal_solvers
    solver_process = solver_tool.newSolverProcess(packing_list)
    sequence.edit(solver_process=solver_process)
    quantity_solver_decision = filter(
      lambda x:x.getCausalityValue().getTestedProperty()=='quantity',
      solver_process.contentValues())[0]
    # use Quantity Split Solver.
    quantity_solver_decision.setSolverValue(self.portal.portal_types['Quantity Split Solver'])
    # configure for Quantity Split Solver.
    kw = {'delivery_solver':'FIFO',
          'start_date':packing_list.getStartDate() + 10}
    quantity_solver_decision.updateConfiguration(**kw)
    solver_process.buildTargetSolverList()
    solver_process.solve()
    # build split deliveries manually. XXX ad-hoc
    previous_tag = None
    for delivery_builder in packing_list.getBuilderList():
      this_builder_tag = '%s_split_%s' % (packing_list.getPath(),
                                          delivery_builder.getId())
      after_tag = []
      if previous_tag:
        after_tag.append(previous_tag)
      delivery_builder.activate(
        after_method_id=('solve',
                         'immediateReindexObject',
                         'recursiveImmediateReindexObject',), # XXX too brutal.
        after_tag=after_tag,
        ).build(explanation_uid=packing_list.getCausalityValue().getUid())

  def stepCheckPackingListSplitted(self, sequence=None, sequence_list=None, **kw):
    """
      Test if packing list was splitted
    """
    order = sequence.get('order')
    packing_list_list = order.getCausalityRelatedValueList(
                               portal_type=self.packing_list_portal_type)
    self.assertEquals(2,len(packing_list_list))
    packing_list1 = None
    packing_list2 = None
    for packing_list in packing_list_list:
      if packing_list.getUid() == sequence.get('packing_list').getUid():
        packing_list1 = packing_list
      else:
        packing_list2 = packing_list
    sequence.edit(new_packing_list=packing_list2)
    for line in packing_list1.objectValues(
          portal_type= self.packing_list_line_portal_type):
      self.assertEquals(self.default_quantity-10,line.getQuantity())
    for line in packing_list2.objectValues(
          portal_type= self.packing_list_line_portal_type):
      self.assertEquals(10,line.getQuantity())

  def _checkSolverState(self, sequence=None, sequence_list=None,
                        state='solved'):
    """
      Check if target solvers' state.
    """
    solver_process = sequence.get('solver_process')
    for solver in solver_process.objectValues(
      portal_type=self.portal.getPortalTargetSolverTypeList()):
      self.assertEquals(state, solver.getSolverState())

  def stepCheckSolverIsSolving(self, sequence=None, sequence_list=None, **kw):
    """
      Check if all target solvers have 'solving' state.
    """
    self._checkSolverState(sequence, sequence_list, 'solving')

  def stepCheckSolverIsSolved(self, sequence=None, sequence_list=None, **kw):
    """
      Check if all target solvers have 'solved' state.
    """
    self._checkSolverState(sequence, sequence_list, 'solved')

  def test_01_splitAndDefer(self, quiet=quiet, run=run_all_test):
    """
      Change the quantity on an delivery line, then
      see if the packing list is divergent and then
      split and defer the packing list
    """
    if not run: return
    sequence_list = SequenceList()

    # Test with a simply order without cell
    sequence_string = self.default_sequence + '\
                      stepIncreasePackingListLineQuantity2 \
                      stepCheckPackingListIsCalculating \
                      stepTic \
                      stepCheckPackingListIsNotDivergent \
                      stepCheckPackingListIsSolved \
                      stepDecreasePackingListLineQuantity1 \
                      stepCheckPackingListIsCalculating \
                      stepTic \
                      stepCheckPackingListIsNotDivergent \
                      stepCheckPackingListIsSolved \
                      stepDecreasePackingListLineQuantity10 \
                      stepCheckPackingListIsCalculating \
                      stepTic \
                      stepCheckPackingListIsDiverged \
                      stepSplitAndDeferPackingList \
                      stepCheckSolverIsSolving \
                      stepTic \
                      stepCheckPackingListSplitted \
                      stepCheckPackingListIsSolved \
                      stepCheckSolverIsSolved \
                      '
    sequence_list.addSequenceString(sequence_string)

    sequence_list.play(self, quiet=quiet)

class TestERP5SimulationPackingList(TestERP5SimulationMixin, TestPackingList):
  pass

class TestERP5SimulationInvoice(TestERP5SimulationMixin, TestSaleInvoice):
  pass

600 601 602
def test_suite():
  suite = unittest.TestSuite()
  suite.addTest(unittest.makeSuite(TestERP5Simulation))
603
  suite.addTest(unittest.makeSuite(TestERP5SimulationPackingList))
Kazuhiko Shiozaki's avatar
Kazuhiko Shiozaki committed
604
  suite.addTest(unittest.makeSuite(TestERP5SimulationInvoice))
605
  return suite