def test_get_batch_item(self): storable = self.create_storable() batch = self.create_storable_batch(storable, batch_number=u'123456') dialog = BatchIncreaseSelectionDialog(self.store, storable, 10) self.assertEqual(dialog.get_batch_item(batch), u'123456') self.assertEqual(dialog.get_batch_item(batch.batch_number), u'123456') self.assertEqual(dialog.get_batch_item(None), None)