sideshow/tests/test_orders.py

# -*- coding: utf-8; -*-

from wuttjamaican.testing import DataTestCase

from sideshow import orders as mod


class TestOrderHandler(DataTestCase):

    def make_config(self, **kwargs):
        config = super().make_config(**kwargs)
        config.setdefault('wutta.model_spec', 'sideshow.db.model')
        config.setdefault('wutta.enum_spec', 'sideshow.enum')
        return config

    def make_handler(self):
        return mod.OrderHandler(self.config)

    def test_expose_store_id(self):
        handler = self.make_handler()

        # false by default
        self.assertFalse(handler.expose_store_id())

        # config can enable
        self.config.setdefault('sideshow.orders.expose_store_id', 'true')
        self.assertTrue(handler.expose_store_id())

    def test_get_order_qty_uom_text(self):
        enum = self.app.enum
        handler = self.make_handler()

        # typical, plain text
        text = handler.get_order_qty_uom_text(2, enum.ORDER_UOM_CASE, case_size=12)
        self.assertEqual(text, "2 Cases (x 12 = 24 Units)")

        # typical w/ html
        text = handler.get_order_qty_uom_text(2, enum.ORDER_UOM_CASE, case_size=12, html=True)
        self.assertEqual(text, "2 Cases (× 12 = 24 Units)")

        # unknown case size
        text = handler.get_order_qty_uom_text(2, enum.ORDER_UOM_CASE)
        self.assertEqual(text, "2 Cases (x ?? = ?? Units)")
        text = handler.get_order_qty_uom_text(2, enum.ORDER_UOM_CASE, html=True)
        self.assertEqual(text, "2 Cases (× ?? = ?? Units)")

        # units only
        text = handler.get_order_qty_uom_text(2, enum.ORDER_UOM_UNIT)
        self.assertEqual(text, "2 Units")
        text = handler.get_order_qty_uom_text(2, enum.ORDER_UOM_UNIT, html=True)
        self.assertEqual(text, "2 Units")

    def test_item_status_to_variant(self):
        enum = self.app.enum
        handler = self.make_handler()

        # typical
        self.assertIsNone(handler.item_status_to_variant(enum.ORDER_ITEM_STATUS_INITIATED))
        self.assertIsNone(handler.item_status_to_variant(enum.ORDER_ITEM_STATUS_READY))
        self.assertIsNone(handler.item_status_to_variant(enum.ORDER_ITEM_STATUS_PLACED))
        self.assertIsNone(handler.item_status_to_variant(enum.ORDER_ITEM_STATUS_RECEIVED))
        self.assertIsNone(handler.item_status_to_variant(enum.ORDER_ITEM_STATUS_CONTACTED))
        self.assertIsNone(handler.item_status_to_variant(enum.ORDER_ITEM_STATUS_PAID))

        # warning
        self.assertEqual(handler.item_status_to_variant(enum.ORDER_ITEM_STATUS_CANCELED), 'warning')
        self.assertEqual(handler.item_status_to_variant(enum.ORDER_ITEM_STATUS_REFUND_PENDING), 'warning')
        self.assertEqual(handler.item_status_to_variant(enum.ORDER_ITEM_STATUS_REFUNDED), 'warning')
        self.assertEqual(handler.item_status_to_variant(enum.ORDER_ITEM_STATUS_RESTOCKED), 'warning')
        self.assertEqual(handler.item_status_to_variant(enum.ORDER_ITEM_STATUS_EXPIRED), 'warning')
        self.assertEqual(handler.item_status_to_variant(enum.ORDER_ITEM_STATUS_INACTIVE), 'warning')

    def test_resolve_pending_product(self):
        model = self.app.model
        enum = self.app.enum
        handler = self.make_handler()

        # sample data
        user = model.User(username='barney')
        self.session.add(user)
        pending = model.PendingProduct(description='vinegar', unit_price_reg=5.99,
                                       status=enum.PendingProductStatus.PENDING,
                                       created_by=user)
        self.session.add(pending)
        order = model.Order(order_id=100, customer_name="Fred Flintstone", created_by=user)
        item = model.OrderItem(pending_product=pending,
                               order_qty=1, order_uom=enum.ORDER_UOM_UNIT,
                               status_code=enum.ORDER_ITEM_STATUS_READY)
        order.items.append(item)
        self.session.add(order)
        self.session.flush()

        info = {
            'product_id': '07430500132',
            'scancode': '07430500132',
            'brand_name': "Bragg's",
            'description': "Apple Cider Vinegar",
            'size': "32oz",
            'weighed': False,
            'department_id': None,
            'department_name': None,
            'special_order': False,
            'vendor_name': None,
            'vendor_item_code': None,
            'case_size': 12,
            'unit_cost': 2.99,
            'unit_price_reg': 5.99,
        }

        # first try fails b/c pending status
        self.assertEqual(len(item.events), 0)
        self.assertRaises(ValueError, handler.resolve_pending_product, pending, info, user)

        # resolves okay if ready status
        pending.status = enum.PendingProductStatus.READY
        handler.resolve_pending_product(pending, info, user)
        self.assertEqual(len(item.events), 1)
        self.assertEqual(item.events[0].type_code, enum.ORDER_ITEM_EVENT_PRODUCT_RESOLVED)
        self.assertIsNone(item.events[0].note)

        # more sample data
        pending2 = model.PendingProduct(description='vinegar', unit_price_reg=5.99,
                                        status=enum.PendingProductStatus.READY,
                                        created_by=user)
        self.session.add(pending2)
        order2 = model.Order(order_id=101, customer_name="Wilma Flintstone", created_by=user)
        item2 = model.OrderItem(pending_product=pending2,
                                order_qty=1, order_uom=enum.ORDER_UOM_UNIT,
                                status_code=enum.ORDER_ITEM_STATUS_READY)
        order2.items.append(item2)
        self.session.add(order2)
        self.session.flush()

        # resolve with extra note
        handler.resolve_pending_product(pending2, info, user, note='hello world')
        self.assertEqual(len(item2.events), 2)
        self.assertEqual(item2.events[0].type_code, enum.ORDER_ITEM_EVENT_PRODUCT_RESOLVED)
        self.assertIsNone(item2.events[0].note)
        self.assertEqual(item2.events[1].type_code, enum.ORDER_ITEM_EVENT_NOTE_ADDED)
        self.assertEqual(item2.events[1].note, "hello world")

    def test_process_placement(self):
        model = self.app.model
        enum = self.app.enum
        handler = self.make_handler()

        # sample data
        user = model.User(username='barney')
        self.session.add(user)
        order = model.Order(order_id=42, customer_name="Fred Flintstone", created_by=user)
        item1 = model.OrderItem(order_qty=1, order_uom=enum.ORDER_UOM_UNIT,
                                status_code=enum.ORDER_ITEM_STATUS_READY)
        order.items.append(item1)
        item2 = model.OrderItem(order_qty=1, order_uom=enum.ORDER_UOM_UNIT,
                                status_code=enum.ORDER_ITEM_STATUS_READY)
        order.items.append(item2)
        item3 = model.OrderItem(order_qty=1, order_uom=enum.ORDER_UOM_UNIT,
                                status_code=enum.ORDER_ITEM_STATUS_READY)
        order.items.append(item3)
        self.session.add(order)
        self.session.flush()

        # two items are updated
        self.assertEqual(item1.status_code, enum.ORDER_ITEM_STATUS_READY)
        self.assertEqual(item2.status_code, enum.ORDER_ITEM_STATUS_READY)
        self.assertEqual(len(item1.events), 0)
        self.assertEqual(len(item2.events), 0)
        handler.process_placement([item1, item2], user,
                                      vendor_name="Acme Dist", po_number='ACME123')
        self.assertEqual(item1.status_code, enum.ORDER_ITEM_STATUS_PLACED)
        self.assertEqual(item2.status_code, enum.ORDER_ITEM_STATUS_PLACED)
        self.assertEqual(len(item1.events), 1)
        self.assertEqual(len(item2.events), 1)
        self.assertEqual(item1.events[0].note, "PO ACME123 for vendor Acme Dist")
        self.assertEqual(item2.events[0].note, "PO ACME123 for vendor Acme Dist")
        self.assertEqual(item1.events[0].type_code, enum.ORDER_ITEM_EVENT_PLACED)
        self.assertEqual(item2.events[0].type_code, enum.ORDER_ITEM_EVENT_PLACED)

        # update last item, without vendor name but extra note
        self.assertEqual(item3.status_code, enum.ORDER_ITEM_STATUS_READY)
        self.assertEqual(len(item3.events), 0)
        handler.process_placement([item3], user, po_number="939234", note="extra note")
        self.assertEqual(item3.status_code, enum.ORDER_ITEM_STATUS_PLACED)
        self.assertEqual(len(item3.events), 2)
        self.assertEqual(item3.events[0].note, "PO 939234")
        self.assertEqual(item3.events[1].note, "extra note")
        self.assertEqual(item3.events[0].type_code, enum.ORDER_ITEM_EVENT_PLACED)
        self.assertEqual(item3.events[1].type_code, enum.ORDER_ITEM_EVENT_NOTE_ADDED)

    def test_process_receiving(self):
        model = self.app.model
        enum = self.app.enum
        handler = self.make_handler()

        # sample data
        user = model.User(username='barney')
        self.session.add(user)
        order = model.Order(order_id=42, customer_name="Fred Flintstone", created_by=user)
        item1 = model.OrderItem(order_qty=1, order_uom=enum.ORDER_UOM_UNIT,
                                status_code=enum.ORDER_ITEM_STATUS_PLACED)
        order.items.append(item1)
        item2 = model.OrderItem(order_qty=1, order_uom=enum.ORDER_UOM_UNIT,
                                status_code=enum.ORDER_ITEM_STATUS_PLACED)
        order.items.append(item2)
        item3 = model.OrderItem(order_qty=1, order_uom=enum.ORDER_UOM_UNIT,
                                status_code=enum.ORDER_ITEM_STATUS_PLACED)
        order.items.append(item3)
        item4 = model.OrderItem(order_qty=1, order_uom=enum.ORDER_UOM_UNIT,
                                status_code=enum.ORDER_ITEM_STATUS_PLACED)
        order.items.append(item4)
        item5 = model.OrderItem(order_qty=1, order_uom=enum.ORDER_UOM_UNIT,
                                status_code=enum.ORDER_ITEM_STATUS_PLACED)
        order.items.append(item5)
        self.session.add(order)
        self.session.flush()

        # all info provided
        self.assertEqual(item1.status_code, enum.ORDER_ITEM_STATUS_PLACED)
        self.assertEqual(len(item1.events), 0)
        handler.process_receiving([item1], user, vendor_name="Acme Dist",
                                  invoice_number='INV123', po_number='123')
        self.assertEqual(item1.status_code, enum.ORDER_ITEM_STATUS_RECEIVED)
        self.assertEqual(len(item1.events), 1)
        self.assertEqual(item1.events[0].note, "invoice INV123 (PO 123) from vendor Acme Dist")
        self.assertEqual(item1.events[0].type_code, enum.ORDER_ITEM_EVENT_RECEIVED)

        # missing PO number
        self.assertEqual(item2.status_code, enum.ORDER_ITEM_STATUS_PLACED)
        self.assertEqual(len(item2.events), 0)
        handler.process_receiving([item2], user, vendor_name="Acme Dist", invoice_number='INV123')
        self.assertEqual(item2.status_code, enum.ORDER_ITEM_STATUS_RECEIVED)
        self.assertEqual(len(item2.events), 1)
        self.assertEqual(item2.events[0].note, "invoice INV123 from vendor Acme Dist")
        self.assertEqual(item2.events[0].type_code, enum.ORDER_ITEM_EVENT_RECEIVED)

        # missing invoice number
        self.assertEqual(item3.status_code, enum.ORDER_ITEM_STATUS_PLACED)
        self.assertEqual(len(item3.events), 0)
        handler.process_receiving([item3], user, vendor_name="Acme Dist", po_number='123')
        self.assertEqual(item3.status_code, enum.ORDER_ITEM_STATUS_RECEIVED)
        self.assertEqual(len(item3.events), 1)
        self.assertEqual(item3.events[0].note, "PO 123 from vendor Acme Dist")
        self.assertEqual(item3.events[0].type_code, enum.ORDER_ITEM_EVENT_RECEIVED)

        # vendor name only
        self.assertEqual(item4.status_code, enum.ORDER_ITEM_STATUS_PLACED)
        self.assertEqual(len(item4.events), 0)
        handler.process_receiving([item4], user, vendor_name="Acme Dist")
        self.assertEqual(item4.status_code, enum.ORDER_ITEM_STATUS_RECEIVED)
        self.assertEqual(len(item4.events), 1)
        self.assertEqual(item4.events[0].note, "from vendor Acme Dist")
        self.assertEqual(item4.events[0].type_code, enum.ORDER_ITEM_EVENT_RECEIVED)

        # no info; extra note
        self.assertEqual(item5.status_code, enum.ORDER_ITEM_STATUS_PLACED)
        self.assertEqual(len(item5.events), 0)
        handler.process_receiving([item5], user, note="extra note")
        self.assertEqual(item5.status_code, enum.ORDER_ITEM_STATUS_RECEIVED)
        self.assertEqual(len(item5.events), 2)
        self.assertIsNone(item5.events[0].note)
        self.assertEqual(item5.events[0].type_code, enum.ORDER_ITEM_EVENT_RECEIVED)
        self.assertEqual(item5.events[1].note, "extra note")
        self.assertEqual(item5.events[1].type_code, enum.ORDER_ITEM_EVENT_NOTE_ADDED)

    def test_process_reorder(self):
        model = self.app.model
        enum = self.app.enum
        handler = self.make_handler()

        # sample data
        user = model.User(username='barney')
        self.session.add(user)
        order = model.Order(order_id=42, customer_name="Fred Flintstone", created_by=user)
        item1 = model.OrderItem(order_qty=1, order_uom=enum.ORDER_UOM_UNIT,
                                status_code=enum.ORDER_ITEM_STATUS_PLACED)
        order.items.append(item1)
        item2 = model.OrderItem(order_qty=1, order_uom=enum.ORDER_UOM_UNIT,
                                status_code=enum.ORDER_ITEM_STATUS_PLACED)
        order.items.append(item2)
        item3 = model.OrderItem(order_qty=1, order_uom=enum.ORDER_UOM_UNIT,
                                status_code=enum.ORDER_ITEM_STATUS_PLACED)
        order.items.append(item3)
        self.session.add(order)
        self.session.flush()

        # two items are updated
        self.assertEqual(item1.status_code, enum.ORDER_ITEM_STATUS_PLACED)
        self.assertEqual(item2.status_code, enum.ORDER_ITEM_STATUS_PLACED)
        self.assertEqual(len(item1.events), 0)
        self.assertEqual(len(item2.events), 0)
        handler.process_reorder([item1, item2], user)
        self.assertEqual(item1.status_code, enum.ORDER_ITEM_STATUS_READY)
        self.assertEqual(item2.status_code, enum.ORDER_ITEM_STATUS_READY)
        self.assertEqual(len(item1.events), 1)
        self.assertEqual(len(item2.events), 1)
        self.assertIsNone(item1.events[0].note)
        self.assertIsNone(item2.events[0].note)
        self.assertEqual(item1.events[0].type_code, enum.ORDER_ITEM_EVENT_REORDER)
        self.assertEqual(item2.events[0].type_code, enum.ORDER_ITEM_EVENT_REORDER)

        # update last item, with extra note
        self.assertEqual(item3.status_code, enum.ORDER_ITEM_STATUS_PLACED)
        self.assertEqual(len(item3.events), 0)
        handler.process_reorder([item3], user, note="extra note")
        self.assertEqual(item3.status_code, enum.ORDER_ITEM_STATUS_READY)
        self.assertEqual(len(item3.events), 2)
        self.assertIsNone(item3.events[0].note)
        self.assertEqual(item3.events[1].note, "extra note")
        self.assertEqual(item3.events[0].type_code, enum.ORDER_ITEM_EVENT_REORDER)
        self.assertEqual(item3.events[1].type_code, enum.ORDER_ITEM_EVENT_NOTE_ADDED)

    def test_process_contact_success(self):
        model = self.app.model
        enum = self.app.enum
        handler = self.make_handler()

        # sample data
        user = model.User(username='barney')
        self.session.add(user)
        order = model.Order(order_id=42, customer_name="Fred Flintstone", created_by=user)
        item1 = model.OrderItem(order_qty=1, order_uom=enum.ORDER_UOM_UNIT,
                                status_code=enum.ORDER_ITEM_STATUS_RECEIVED)
        order.items.append(item1)
        item2 = model.OrderItem(order_qty=1, order_uom=enum.ORDER_UOM_UNIT,
                                status_code=enum.ORDER_ITEM_STATUS_RECEIVED)
        order.items.append(item2)
        item3 = model.OrderItem(order_qty=1, order_uom=enum.ORDER_UOM_UNIT,
                                status_code=enum.ORDER_ITEM_STATUS_RECEIVED)
        order.items.append(item3)
        self.session.add(order)
        self.session.flush()

        # two items are updated
        self.assertEqual(item1.status_code, enum.ORDER_ITEM_STATUS_RECEIVED)
        self.assertEqual(item2.status_code, enum.ORDER_ITEM_STATUS_RECEIVED)
        self.assertEqual(len(item1.events), 0)
        self.assertEqual(len(item2.events), 0)
        handler.process_contact_success([item1, item2], user)
        self.assertEqual(item1.status_code, enum.ORDER_ITEM_STATUS_CONTACTED)
        self.assertEqual(item2.status_code, enum.ORDER_ITEM_STATUS_CONTACTED)
        self.assertEqual(len(item1.events), 1)
        self.assertEqual(len(item2.events), 1)
        self.assertIsNone(item1.events[0].note)
        self.assertIsNone(item2.events[0].note)
        self.assertEqual(item1.events[0].type_code, enum.ORDER_ITEM_EVENT_CONTACTED)
        self.assertEqual(item2.events[0].type_code, enum.ORDER_ITEM_EVENT_CONTACTED)

        # update last item, with extra note
        self.assertEqual(item3.status_code, enum.ORDER_ITEM_STATUS_RECEIVED)
        self.assertEqual(len(item3.events), 0)
        handler.process_contact_success([item3], user, note="extra note")
        self.assertEqual(item3.status_code, enum.ORDER_ITEM_STATUS_CONTACTED)
        self.assertEqual(len(item3.events), 2)
        self.assertIsNone(item3.events[0].note)
        self.assertEqual(item3.events[1].note, "extra note")
        self.assertEqual(item3.events[0].type_code, enum.ORDER_ITEM_EVENT_CONTACTED)
        self.assertEqual(item3.events[1].type_code, enum.ORDER_ITEM_EVENT_NOTE_ADDED)

    def test_process_contact_failure(self):
        model = self.app.model
        enum = self.app.enum
        handler = self.make_handler()

        # sample data
        user = model.User(username='barney')
        self.session.add(user)
        order = model.Order(order_id=42, customer_name="Fred Flintstone", created_by=user)
        item1 = model.OrderItem(order_qty=1, order_uom=enum.ORDER_UOM_UNIT,
                                status_code=enum.ORDER_ITEM_STATUS_RECEIVED)
        order.items.append(item1)
        item2 = model.OrderItem(order_qty=1, order_uom=enum.ORDER_UOM_UNIT,
                                status_code=enum.ORDER_ITEM_STATUS_RECEIVED)
        order.items.append(item2)
        item3 = model.OrderItem(order_qty=1, order_uom=enum.ORDER_UOM_UNIT,
                                status_code=enum.ORDER_ITEM_STATUS_RECEIVED)
        order.items.append(item3)
        self.session.add(order)
        self.session.flush()

        # two items are updated
        self.assertEqual(item1.status_code, enum.ORDER_ITEM_STATUS_RECEIVED)
        self.assertEqual(item2.status_code, enum.ORDER_ITEM_STATUS_RECEIVED)
        self.assertEqual(len(item1.events), 0)
        self.assertEqual(len(item2.events), 0)
        handler.process_contact_failure([item1, item2], user)
        self.assertEqual(item1.status_code, enum.ORDER_ITEM_STATUS_CONTACT_FAILED)
        self.assertEqual(item2.status_code, enum.ORDER_ITEM_STATUS_CONTACT_FAILED)
        self.assertEqual(len(item1.events), 1)
        self.assertEqual(len(item2.events), 1)
        self.assertIsNone(item1.events[0].note)
        self.assertIsNone(item2.events[0].note)
        self.assertEqual(item1.events[0].type_code, enum.ORDER_ITEM_EVENT_CONTACT_FAILED)
        self.assertEqual(item2.events[0].type_code, enum.ORDER_ITEM_EVENT_CONTACT_FAILED)

        # update last item, with extra note
        self.assertEqual(item3.status_code, enum.ORDER_ITEM_STATUS_RECEIVED)
        self.assertEqual(len(item3.events), 0)
        handler.process_contact_failure([item3], user, note="extra note")
        self.assertEqual(item3.status_code, enum.ORDER_ITEM_STATUS_CONTACT_FAILED)
        self.assertEqual(len(item3.events), 2)
        self.assertIsNone(item3.events[0].note)
        self.assertEqual(item3.events[1].note, "extra note")
        self.assertEqual(item3.events[0].type_code, enum.ORDER_ITEM_EVENT_CONTACT_FAILED)
        self.assertEqual(item3.events[1].type_code, enum.ORDER_ITEM_EVENT_NOTE_ADDED)

    def test_process_delivery(self):
        model = self.app.model
        enum = self.app.enum
        handler = self.make_handler()

        # sample data
        user = model.User(username='barney')
        self.session.add(user)
        order = model.Order(order_id=42, customer_name="Fred Flintstone", created_by=user)
        item1 = model.OrderItem(order_qty=1, order_uom=enum.ORDER_UOM_UNIT,
                                status_code=enum.ORDER_ITEM_STATUS_RECEIVED)
        order.items.append(item1)
        item2 = model.OrderItem(order_qty=1, order_uom=enum.ORDER_UOM_UNIT,
                                status_code=enum.ORDER_ITEM_STATUS_RECEIVED)
        order.items.append(item2)
        item3 = model.OrderItem(order_qty=1, order_uom=enum.ORDER_UOM_UNIT,
                                status_code=enum.ORDER_ITEM_STATUS_RECEIVED)
        order.items.append(item3)
        self.session.add(order)
        self.session.flush()

        # two items are updated
        self.assertEqual(item1.status_code, enum.ORDER_ITEM_STATUS_RECEIVED)
        self.assertEqual(item2.status_code, enum.ORDER_ITEM_STATUS_RECEIVED)
        self.assertEqual(len(item1.events), 0)
        self.assertEqual(len(item2.events), 0)
        handler.process_delivery([item1, item2], user)
        self.assertEqual(item1.status_code, enum.ORDER_ITEM_STATUS_DELIVERED)
        self.assertEqual(item2.status_code, enum.ORDER_ITEM_STATUS_DELIVERED)
        self.assertEqual(len(item1.events), 1)
        self.assertEqual(len(item2.events), 1)
        self.assertIsNone(item1.events[0].note)
        self.assertIsNone(item2.events[0].note)
        self.assertEqual(item1.events[0].type_code, enum.ORDER_ITEM_EVENT_DELIVERED)
        self.assertEqual(item2.events[0].type_code, enum.ORDER_ITEM_EVENT_DELIVERED)

        # update last item, with extra note
        self.assertEqual(item3.status_code, enum.ORDER_ITEM_STATUS_RECEIVED)
        self.assertEqual(len(item3.events), 0)
        handler.process_delivery([item3], user, note="extra note")
        self.assertEqual(item3.status_code, enum.ORDER_ITEM_STATUS_DELIVERED)
        self.assertEqual(len(item3.events), 2)
        self.assertIsNone(item3.events[0].note)
        self.assertEqual(item3.events[1].note, "extra note")
        self.assertEqual(item3.events[0].type_code, enum.ORDER_ITEM_EVENT_DELIVERED)
        self.assertEqual(item3.events[1].type_code, enum.ORDER_ITEM_EVENT_NOTE_ADDED)

    def test_process_restock(self):
        model = self.app.model
        enum = self.app.enum
        handler = self.make_handler()

        # sample data
        user = model.User(username='barney')
        self.session.add(user)
        order = model.Order(order_id=42, customer_name="Fred Flintstone", created_by=user)
        item1 = model.OrderItem(order_qty=1, order_uom=enum.ORDER_UOM_UNIT,
                                status_code=enum.ORDER_ITEM_STATUS_RECEIVED)
        order.items.append(item1)
        item2 = model.OrderItem(order_qty=1, order_uom=enum.ORDER_UOM_UNIT,
                                status_code=enum.ORDER_ITEM_STATUS_RECEIVED)
        order.items.append(item2)
        item3 = model.OrderItem(order_qty=1, order_uom=enum.ORDER_UOM_UNIT,
                                status_code=enum.ORDER_ITEM_STATUS_RECEIVED)
        order.items.append(item3)
        self.session.add(order)
        self.session.flush()

        # two items are updated
        self.assertEqual(item1.status_code, enum.ORDER_ITEM_STATUS_RECEIVED)
        self.assertEqual(item2.status_code, enum.ORDER_ITEM_STATUS_RECEIVED)
        self.assertEqual(len(item1.events), 0)
        self.assertEqual(len(item2.events), 0)
        handler.process_restock([item1, item2], user)
        self.assertEqual(item1.status_code, enum.ORDER_ITEM_STATUS_RESTOCKED)
        self.assertEqual(item2.status_code, enum.ORDER_ITEM_STATUS_RESTOCKED)
        self.assertEqual(len(item1.events), 1)
        self.assertEqual(len(item2.events), 1)
        self.assertIsNone(item1.events[0].note)
        self.assertIsNone(item2.events[0].note)
        self.assertEqual(item1.events[0].type_code, enum.ORDER_ITEM_EVENT_RESTOCKED)
        self.assertEqual(item2.events[0].type_code, enum.ORDER_ITEM_EVENT_RESTOCKED)

        # update last item, with extra note
        self.assertEqual(item3.status_code, enum.ORDER_ITEM_STATUS_RECEIVED)
        self.assertEqual(len(item3.events), 0)
        handler.process_restock([item3], user, note="extra note")
        self.assertEqual(item3.status_code, enum.ORDER_ITEM_STATUS_RESTOCKED)
        self.assertEqual(len(item3.events), 2)
        self.assertIsNone(item3.events[0].note)
        self.assertEqual(item3.events[1].note, "extra note")
        self.assertEqual(item3.events[0].type_code, enum.ORDER_ITEM_EVENT_RESTOCKED)
        self.assertEqual(item3.events[1].type_code, enum.ORDER_ITEM_EVENT_NOTE_ADDED)