def _create_or_update_delivery(self): delivery_service = sysparam.get_object(self.store, 'DELIVERY_SERVICE') delivery_sellable = delivery_service.sellable items = [ item for item in self.slave.klist if item.sellable.product is not None ] if self._delivery is not None: model = self._delivery else: model = CreateDeliveryModel(price=delivery_sellable.price, recipient=self.model.person) rv = run_dialog(CreateDeliveryEditor, self.get_toplevel().get_toplevel(), self.store, model=model, items=items, person_type=Person) if not rv: return self._delivery = rv if self._delivery_item: self.slave.klist.update(self._delivery_item) else: self._delivery_item = self.get_order_item(delivery_sellable, self._delivery.price, 1) self.slave.klist.append(None, self._delivery_item)
def _create_or_update_delivery(self): delivery_service = sysparam.get_object(self.store, 'DELIVERY_SERVICE') delivery_sellable = delivery_service.sellable items = [ item for item in self.slave.klist if item.sellable.product is not None ] if self._delivery is not None: model = self._delivery else: model = CreateDeliveryModel(price=delivery_sellable.price, client=self.model.client) rv = run_dialog(CreateDeliveryEditor, self.get_toplevel().get_toplevel(), self.store, model, sale_items=items) if not rv: return self._delivery = rv if self._delivery_item: self._delivery_item.price = self._delivery.price self._delivery_item.notes = self._delivery.notes self._delivery_item.estimated_fix_date = self._delivery.estimated_fix_date self.slave.klist.update(self._delivery_item) else: self._delivery_item = self.get_order_item(delivery_sellable, self._delivery.price, 1) self.slave.klist.append(None, self._delivery_item)
def test_wizard_remove_delivery(self, yesno): yesno.return_value = True branch = api.get_current_branch(self.store) storable = self.create_storable(branch=branch, stock=1) sellable = storable.product.sellable # Run the wizard wizard = StockDecreaseWizard(self.store) step = wizard.get_current_step() step.reason.update('test') self.click(wizard.next_button) step = wizard.get_current_step() self.assertNotSensitive(step, ['delivery_button']) step.sellable_selected(sellable) step.quantity.update(1) self.click(step.add_sellable_button) self.assertSensitive(step, ['delivery_button']) delivery_sellable = sysparam.get_object(self.store, 'DELIVERY_SERVICE').sellable delivery = CreateDeliveryModel(price=delivery_sellable.price, recipient=wizard.model.person) module = 'stoqlib.gui.wizards.stockdecreasewizard.run_dialog' with mock.patch(module) as run_dialog: # Delivery set run_dialog.return_value = delivery self.click(step.delivery_button) self.assertEqual(step._delivery, delivery) self.assertTrue(isinstance(step._delivery_item, StockDecreaseItem)) # Remove the delivery item run_dialog.return_value = delivery step.slave.klist.select(step.slave.klist[1]) self.click(step.slave.delete_button) self.assertIsNone(step._delivery) self.assertIsNone(step._delivery_item)
def setup_slaves(self): SellableItemStep.setup_slaves(self) self.hide_add_button() self.cost_label.set_label(_('Price:')) self.cost.set_editable(True) delivery = self._find_delivery() if delivery is not None: self._delivery_item = delivery.service_item self._delivery = CreateDeliveryModel( price=self._delivery_item.price, notes=self._delivery_item.notes, transporter=delivery.transporter, address=delivery.address, estimated_fix_date=self._delivery_item.estimated_fix_date) else: self._delivery = None self._delivery_item = None if isinstance(self.model, Sale): self.delivery_btn = self.slave.add_extra_button( label=_("Add delivery")) self.delivery_btn.set_sensitive(bool(len(self.slave.klist))) else: self.delivery_btn = None self.discount_btn = self.slave.add_extra_button( label=_("Apply discount")) self.discount_btn.set_sensitive(bool(len(self.slave.klist))) self.slave.klist.connect('has-rows', self._on_klist__has_rows) self.slave.klist.connect('selection-changed', self._on_klist__selection_changed)
def setup_slaves(self): SellableItemStep.setup_slaves(self) self.hide_add_button() self.cost_label.set_label(_('Price:')) self.cost.set_editable(True) delivery = self._find_delivery() if delivery is not None: self._delivery_item = delivery.service_item self._delivery = CreateDeliveryModel.from_delivery(delivery) else: self._delivery = None self._delivery_item = None if isinstance(self.model, Sale): self.delivery_btn = self.slave.add_extra_button(label=_("Add delivery")) self.delivery_btn.set_sensitive(bool(len(self.slave.klist))) else: self.delivery_btn = None self.discount_btn = self.slave.add_extra_button(label=_("Apply discount")) self.discount_btn.set_sensitive(bool(len(self.slave.klist))) self.slave.klist.connect('has-rows', self._on_klist__has_rows) self.slave.klist.connect('selection-changed', self._on_klist__selection_changed)
def test_wizard_with_delivery(self): branch = api.get_current_branch(self.store) storable = self.create_storable(branch=branch, stock=1) sellable = storable.product.sellable # Run the wizard wizard = StockDecreaseWizard(self.store) step = wizard.get_current_step() step.reason.update('test') self.click(wizard.next_button) step = wizard.get_current_step() self.assertNotSensitive(step, ['delivery_button']) step.sellable_selected(sellable) step.quantity.update(1) self.click(step.add_sellable_button) item = step.slave.klist[0] self.assertSensitive(step, ['delivery_button']) delivery_sellable = sysparam.get_object(self.store, 'DELIVERY_SERVICE').sellable delivery = CreateDeliveryModel(price=delivery_sellable.price, recipient=wizard.model.person) module = 'stoqlib.gui.wizards.stockdecreasewizard.run_dialog' with mock.patch(module) as run_dialog: # Nothing done with the editor, no delivery returned run_dialog.return_value = None self.click(step.delivery_button) self.assertIsNone(step._delivery) self.assertIsNone(step._delivery_item) # Delivery set run_dialog.return_value = delivery self.click(step.delivery_button) self.assertEqual(step._delivery, delivery) self.assertTrue(isinstance(step._delivery_item, StockDecreaseItem)) # Edit the delivery item run_dialog.return_value = delivery step.slave.klist.select(step.slave.klist[1]) self.click(step.slave.edit_button) self.assertEqual(step._delivery, delivery) self.assertTrue(isinstance(step._delivery_item, StockDecreaseItem)) # Finishing the wizard must create a Delivery object module = 'stoqlib.gui.wizards.stockdecreasewizard.yesno' with mock.patch(module) as yesno: with mock.patch.object(self.store, 'commit'): yesno.return_value = False item.deliver = True self.click(wizard.next_button) self.assertTrue(isinstance(item.delivery, Delivery))
def test_create_delivery(self, run_dialog): delivery = CreateDeliveryModel(price=Decimal('150')) delivery.notes = u'notes about the delivery' delivery.client = self.create_client() delivery.transporter = self.create_transporter() delivery.address = self.create_address() run_dialog.return_value = delivery pos = self._get_pos_with_open_till() sale_item = self._add_product(pos, self.create_sellable()) olist = pos.sale_items olist.select(olist[0]) self.activate(pos.NewDelivery) self.assertEqual(run_dialog.call_count, 1) args, kwargs = run_dialog.call_args editor, store, delivery = args self.assertEqual(editor, CreateDeliveryEditor) self.assertTrue(store is not None) self.assertEqual(delivery, None) self.assertEqual(kwargs[u'items'], [sale_item])
def test_create_delivery(self, run_dialog): delivery = CreateDeliveryModel(price=Decimal('150')) delivery.notes = u'notes about the delivery' delivery.client = self.create_client() delivery.transporter = self.create_transporter() delivery.address = self.create_address() run_dialog.return_value = delivery pos = self._get_pos_with_open_till() sale_item = self._add_product(pos, self.create_sellable()) olist = pos.sale_items olist.select(olist[0]) self.activate(pos.NewDelivery) self.assertEqual(run_dialog.call_count, 1) args, kwargs = run_dialog.call_args editor, store, delivery = args self.assertEqual(editor, CreateDeliveryEditor) self.assertTrue(store is not None) self.assertEqual(delivery, None) self.assertEqual(kwargs[u'items'], [sale_item])