def test_event_manager_subscription(): """Test that EventManager subscribing works""" with patch.object(Queue, "get") as fake_queue: saved_data = { "user_id": None, "title": "asdf", "content": "asdf", "team_id": None, "user": None, "team": None, "date": "2019-01-28T01:20:46.017649+00:00", "id": 10, } saved_event = {"type": "notification", "data": saved_data} fake_queue.return_value = saved_event event_manager = EventManager() events = event_manager.subscribe() message = next(events) assert isinstance(message, ServerSentEvent) assert message.to_dict() == {"data": "", "type": "ping"} assert message.__str__().startswith("event:ping") assert len(event_manager.clients) == 1 message = next(events) assert isinstance(message, ServerSentEvent) assert message.to_dict() == saved_event assert message.__str__().startswith("event:notification\ndata:") assert len(event_manager.clients) == 1
def test_event_manager_subscription(): """Test that EventManager subscribing works""" with patch.object(Queue, 'get') as fake_queue: saved_data = { 'user_id': None, 'title': 'asdf', 'content': 'asdf', 'team_id': None, 'user': None, 'team': None, 'date': '2019-01-28T01:20:46.017649+00:00', 'id': 10 } saved_event = {'type': 'notification', 'data': saved_data} fake_queue.return_value = saved_event event_manager = EventManager() for message in event_manager.subscribe(): assert message.to_dict() == saved_event assert message.__str__().startswith('event:notification\ndata:') assert len(event_manager.clients) == 1 break