testERP5Simulation.py 25.4 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 Rule'):
52 53
      if rule.getValidationState() == 'validated':
        rule.invalidate()
54
    self.validateNewRules()
55

Kazuhiko Shiozaki's avatar
Kazuhiko Shiozaki committed
56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71
  @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')
72 73 74 75
    portal_rules = portal.portal_rules
    clipboard = portal_rules.manage_copyObjects(ids=['new_invoice_transaction_rule'])
    pasted = portal_rules.manage_pasteObjects(clipboard)
    invoice_rule = getattr(portal_rules, pasted[0]['new_id'])
76

Kazuhiko Shiozaki's avatar
Kazuhiko Shiozaki committed
77 78 79 80 81 82 83 84 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
    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()
112 113

  def validateNewRules(self):
114
    # create an Order Rule document.
115
    portal_rules = self.portal.portal_rules
116
    new_order_rule = filter(
Kazuhiko Shiozaki's avatar
Kazuhiko Shiozaki committed
117
      lambda x:x.title == 'New Default Order Rule',
118
      portal_rules.objectValues(portal_type='Order Rule'))[0]
119 120 121
    if new_order_rule.getValidationState() != 'validated':
      new_order_rule.validate()

Kazuhiko Shiozaki's avatar
Kazuhiko Shiozaki committed
122
  def _acceptDecisionQuantity(self, document):
123
    solver_tool = self.portal.portal_solvers
Kazuhiko Shiozaki's avatar
Kazuhiko Shiozaki committed
124
    solver_process = solver_tool.newSolverProcess(document)
125
    quantity_solver_decision = filter(
126
      lambda x:x.getCausalityValue().getTestedProperty()=='quantity',
127
      solver_process.contentValues())[0]
Kazuhiko Shiozaki's avatar
Kazuhiko Shiozaki committed
128 129
    # use Quantity Accept Solver.
    quantity_solver_decision.setSolverValue(self.portal.portal_types['Quantity Accept Solver'])
130 131 132
    solver_process.buildTargetSolverList()
    solver_process.solve()

Kazuhiko Shiozaki's avatar
Kazuhiko Shiozaki committed
133 134 135
  def _acceptDivergenceOnInvoice(self, invoice, divergence_list):
    print invoice, divergence_list
    return self._acceptDecisionQuantity(invoice)
136

Kazuhiko Shiozaki's avatar
Kazuhiko Shiozaki committed
137
  def stepAcceptDecisionQuantity(self,sequence=None, sequence_list=None):
138
    """
Kazuhiko Shiozaki's avatar
Kazuhiko Shiozaki committed
139
    Solve quantity divergence by using solver tool.
140
    """
Kazuhiko Shiozaki's avatar
Kazuhiko Shiozaki committed
141 142
    packing_list = sequence.get('packing_list')
    self._acceptDecisionQuantity(packing_list)
143

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

Kazuhiko Shiozaki's avatar
Kazuhiko Shiozaki committed
152
  def stepAcceptDecisionResource(self,sequence=None, sequence_list=None):
153 154 155 156
    """
    Solve quantity divergence by using solver tool.
    """
    packing_list = sequence.get('packing_list')
157 158 159 160 161
    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]
162 163
    # use Resource Replacement Solver.
    resource_solver_decision.setSolverValue(self.portal.portal_types['Resource Replacement Solver'])
164
    solver_process.buildTargetSolverList()
165 166
    solver_process.solve()

Kazuhiko Shiozaki's avatar
Kazuhiko Shiozaki committed
167
  def stepSplitAndDeferPackingList(self, sequence=None, sequence_list=None):
168 169 170 171
    """
      Do the split and defer action
    """
    packing_list = sequence.get('packing_list')
172 173 174 175 176
    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]
177 178 179
    # use Quantity Split Solver.
    quantity_solver_decision.setSolverValue(self.portal.portal_types['Quantity Split Solver'])
    # configure for Quantity Split Solver.
180 181 182
    kw = {'delivery_solver':'FIFO',
          'start_date':self.datetime + 15,
          'stop_date':self.datetime + 25}
183 184
    quantity_solver_decision.updateConfiguration(**kw)
    solver_process.buildTargetSolverList()
185
    solver_process.solve()
186 187 188 189 190 191 192 193 194 195 196 197 198 199
    # 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())
200

201 202 203 204 205 206 207 208 209 210 211 212 213 214
  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.
    quantity_solver_decision.setSolverValue(self.portal.portal_types['Quantity Adoption Solver'])
    solver_process.buildTargetSolverList()
    solver_process.solve()

Kazuhiko Shiozaki's avatar
Kazuhiko Shiozaki committed
215 216 217 218 219 220 221 222 223
  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):
224 225 226 227 228 229 230
    """
    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
231
                                               sequence_list=None):
232 233 234 235 236 237
    """
    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
238
  def stepAdoptPrevisionResource(self,sequence=None, sequence_list=None):
239 240 241 242 243 244 245 246 247 248 249 250 251 252
    """
    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.
    resource_solver_decision.setSolverValue(self.portal.portal_types['Resource Adoption Solver'])
    solver_process.buildTargetSolverList()
    solver_process.solve()

Kazuhiko Shiozaki's avatar
Kazuhiko Shiozaki committed
253
  def stepCheckPackingListLineWithSameResource(self,sequence=None, sequence_list=None):
254 255 256 257 258 259 260 261 262 263 264 265 266
    """
      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(),
                        [x.getOrder() for x in \
                         line.getDeliveryRelatedValueList()])

Kazuhiko Shiozaki's avatar
Kazuhiko Shiozaki committed
267
  def stepUnifyDestinationWithDecision(self,sequence=None, sequence_list=None):
268 269 270 271 272 273 274 275 276 277 278 279 280 281
    """
      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.
      destination_solver_decision.setSolverValue(self.portal.portal_types['Destination Replacement Solver'])
    solver_process.buildTargetSolverList()
    solver_process.solve()

Kazuhiko Shiozaki's avatar
Kazuhiko Shiozaki committed
282
  def _unifyStartDateWithDecision(self, document):
283
    solver_tool = self.portal.portal_solvers
Kazuhiko Shiozaki's avatar
Kazuhiko Shiozaki committed
284
    solver_process = solver_tool.newSolverProcess(document)
285 286 287 288 289
    for start_date_solver_decision in filter(
      lambda x:x.getCausalityValue().getTestedProperty()=='start_date',
      solver_process.contentValues()):
      # use StartDate Replacement Solver.
      start_date_solver_decision.setSolverValue(self.portal.portal_types['Start Date Replacement Solver'])
290
      # configure for Quantity Split Solver.
Kazuhiko Shiozaki's avatar
Kazuhiko Shiozaki committed
291
      kw = {'value':document.getStartDate()}
292 293 294 295
      start_date_solver_decision.updateConfiguration(**kw)
    solver_process.buildTargetSolverList()
    solver_process.solve()

Kazuhiko Shiozaki's avatar
Kazuhiko Shiozaki committed
296 297 298 299 300 301 302 303 304
  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):
305 306 307 308
    """
      Check if simulation movement are disconnected
    """
    packing_list = sequence.get('packing_list')
309 310 311
    applied_rule = sequence.get('applied_rule')
    simulation_line_list = applied_rule.objectValues()
    start_date = simulation_line_list[-1].getStartDate()
312 313 314 315 316 317 318 319
    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.
      start_date_solver_decision.setSolverValue(self.portal.portal_types['Start Date Replacement Solver'])
      # configure for Quantity Split Solver.
320
      kw = {'value':start_date}
321
      start_date_solver_decision.updateConfiguration(**kw)
322 323 324
    solver_process.buildTargetSolverList()
    solver_process.solve()

Kazuhiko Shiozaki's avatar
Kazuhiko Shiozaki committed
325
  def checkOrderRuleSimulation(self, rule_reference, sequence=None, sequence_list=None):
326 327 328 329 330 331 332 333 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 374 375 376 377 378 379 380 381 382 383 384 385 386 387 388 389 390
    """
      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
      order_movement_list = sum([x.getOrderValueList() for x in \
                                 simulation_movement_list], [])
      self.failIfDifferentSet(movement_list, order_movement_list)

      # Check each simulation movement
      for simulation_movement in simulation_movement_list:
        order_movement_list = simulation_movement.getOrderValueList()
        # 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
391
  def stepModifySimulationLineQuantityForMergedLine(self,sequence=None, sequence_list=None):
392 393 394 395 396 397 398 399 400 401
    """
      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)
      simulation_line.getOrderValue().edit(quantity=self.default_quantity-2)

Kazuhiko Shiozaki's avatar
Kazuhiko Shiozaki committed
402
  def stepCheckSimulationQuantityUpdatedForMergedLine(self,sequence=None, sequence_list=None):
403 404 405 406 407 408 409 410 411 412 413
    """
      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
414
  def stepCheckPackingListLineWithDifferentResource(self,sequence=None, sequence_list=None):
415 416 417 418 419 420 421 422 423 424 425 426
    """
      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()
    order_line_list = sum([x.getOrderList() for x in simulation_line_list], [])
    self.assertEquals(sorted(packing_list_line.getCausalityList()),
                      sorted(order_line_list))

Kazuhiko Shiozaki's avatar
Kazuhiko Shiozaki committed
427 428 429 430
class TestERP5Simulation(TestERP5SimulationMixin, ERP5TypeTestCase):
  run_all_test = 1
  quiet = 0

431 432 433 434 435 436 437 438 439
  def afterSetUp(self):
    TestERP5SimulationMixin.afterSetUp(self)
    new_order_rule = self.portal.portal_rules['new_order_rule']
    new_order_rule['quantity_tester'].edit(quantity=None,
                                           quantity_range_max=2,
                                           quantity_range_min=-1)

  def beforeTearDown(self):
    new_order_rule = self.portal.portal_rules['new_order_rule']
440 441 442
    new_order_rule['quantity_tester'].edit(quantity=0,
                                           quantity_range_max=None,
                                           quantity_range_min=None)
443
    transaction.commit()
444
    TestERP5SimulationMixin.beforeTearDown(self)
Kazuhiko Shiozaki's avatar
Kazuhiko Shiozaki committed
445 446 447 448 449 450 451 452 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

  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

592 593 594
def test_suite():
  suite = unittest.TestSuite()
  suite.addTest(unittest.makeSuite(TestERP5Simulation))
595
  suite.addTest(unittest.makeSuite(TestERP5SimulationPackingList))
Kazuhiko Shiozaki's avatar
Kazuhiko Shiozaki committed
596
  suite.addTest(unittest.makeSuite(TestERP5SimulationInvoice))
597
  return suite