def test_clear(self): user = PicklableMock() items = [{"ts": 35343820800, "message": {"test": 567}, "direction": "incoming"}] description = PicklableMock() description.max_message_count = 1 description.lifetime = 10 preprocessing_messages = PreprocessingScenariosMessages(items, description, user) preprocessing_messages.clear() self.assertListEqual([item for item in preprocessing_messages.processed_items], [])
def test_add_1(self): user = PicklableMock() items = None description = PicklableMock() description.max_message_count = 3 description.lifetime = 10 preprocessing_messages = PreprocessingScenariosMessages(items, description, user) text_preprocessing_result = PicklableMock() text_preprocessing_result.raw = {"test": 123} preprocessing_messages.add(text_preprocessing_result) self.assertEqual([item for item in preprocessing_messages.processed_items], [{"test": 123}])
def test_add_3(self): user = PicklableMock() items = [{"ts": 35343820800, "message": {"test": 567}, "direction": "incoming"}] description = PicklableMock() description.max_message_count = 1 description.lifetime = 10 preprocessing_messages = PreprocessingScenariosMessages(items, description, user) text_preprocessing_result = PicklableMock() text_preprocessing_result.raw = {"test": 123} preprocessing_messages.add(text_preprocessing_result) self.assertListEqual([item for item in preprocessing_messages.processed_items], [{"test": 123}])