testERP5Simulation.py 25.6 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 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
    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'])
83

Kazuhiko Shiozaki's avatar
Kazuhiko Shiozaki committed
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 112 113 114 115 116 117 118
    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()
119 120

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

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

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

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

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

Kazuhiko Shiozaki's avatar
Kazuhiko Shiozaki committed
159
  def stepAcceptDecisionResource(self,sequence=None, sequence_list=None):
160 161 162 163
    """
    Solve quantity divergence by using solver tool.
    """
    packing_list = sequence.get('packing_list')
164 165 166 167 168
    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]
169
    # use Resource Replacement Solver.
170
    resource_solver_decision.setSolverValue(self.portal.portal_types['Accept Solver'])
171
    solver_process.buildTargetSolverList()
172 173
    solver_process.solve()

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

208 209 210 211 212 213 214 215 216 217
  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.
218
    quantity_solver_decision.setSolverValue(self.portal.portal_types['Adopt Solver'])
219 220 221
    solver_process.buildTargetSolverList()
    solver_process.solve()

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

Kazuhiko Shiozaki's avatar
Kazuhiko Shiozaki committed
260
  def stepCheckPackingListLineWithSameResource(self,sequence=None, sequence_list=None):
261 262 263 264 265 266 267 268 269 270 271 272 273
    """
      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
274
  def stepUnifyDestinationWithDecision(self,sequence=None, sequence_list=None):
275 276 277 278 279 280 281 282 283 284
    """
      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.
285
      destination_solver_decision.setSolverValue(self.portal.portal_types['Accept Solver'])
286 287 288
    solver_process.buildTargetSolverList()
    solver_process.solve()

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

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

Kazuhiko Shiozaki's avatar
Kazuhiko Shiozaki committed
332
  def checkOrderRuleSimulation(self, rule_reference, sequence=None, sequence_list=None):
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 391 392 393 394 395 396 397
    """
      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
398
  def stepModifySimulationLineQuantityForMergedLine(self,sequence=None, sequence_list=None):
399 400 401 402 403 404 405 406 407 408
    """
      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
409
  def stepCheckSimulationQuantityUpdatedForMergedLine(self,sequence=None, sequence_list=None):
410 411 412 413 414 415 416 417 418 419 420
    """
      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
421
  def stepCheckPackingListLineWithDifferentResource(self,sequence=None, sequence_list=None):
422 423 424 425 426 427 428 429 430 431 432 433
    """
      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
434 435 436 437
class TestERP5Simulation(TestERP5SimulationMixin, ERP5TypeTestCase):
  run_all_test = 1
  quiet = 0

438 439 440 441 442 443 444 445 446
  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']
447 448 449
    new_order_rule['quantity_tester'].edit(quantity=0,
                                           quantity_range_max=None,
                                           quantity_range_min=None)
450
    transaction.commit()
451
    TestERP5SimulationMixin.beforeTearDown(self)
Kazuhiko Shiozaki's avatar
Kazuhiko Shiozaki committed
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 592 593 594 595 596 597 598

  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

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