def test_regular(self): context_data = MagicMock() attribute_product_data = MagicMock() target = Target(context_data, attribute_product_data) in_messages = target.extract_user_messages([{"direction": 1, "_id": 1}, {"direction": 0, "_id": 2}]) self.assertListEqual([{"_id": 1, "direction": 1}], list(in_messages))
def test_items(self): context_data = MagicMock() attribute_product_data = MagicMock() target = Target(context_data, attribute_product_data) target.extract_user_messages = MagicMock(return_value=["value_1", "value_2"].__iter__()) actual = target.extract_last_user_message("messages_value") self.assertEqual("value_1", actual)
def test_items(self): context_data = MagicMock() attribute_product_data = MagicMock() target = Target(context_data, attribute_product_data) target.extract_user_messages = MagicMock( return_value=["value_1", "value_2"].__iter__()) actual = target.extract_last_user_message("messages_value") self.assertEqual("value_1", actual)
def test_regular(self): context_data = MagicMock() attribute_product_data = MagicMock() target = Target(context_data, attribute_product_data) in_messages = target.extract_user_messages([{ "direction": 1, "_id": 1 }, { "direction": 0, "_id": 2 }]) self.assertListEqual([{'_id': 1, 'direction': 1}], list(in_messages))