def test_emit_failure_connected(self):
""" running tests on socketio.emit """
for test in self.failure_test_params:
with mock.patch('flask_socketio.SocketIO.emit',
self.mock_emit_messages_channel):
try:
app.emit_all_messages(test['connected'])
except ValueError as err:
self.fail(err)
def test_emit_success_channel(self):
""" running tests on socketio.emit """
for test in self.success_test_params:
with mock.patch('flask_socketio.SocketIO.emit',
self.mock_emit_messages_channel):
try:
app.emit_all_messages(test[MESSAGES_RECEIVED_CHANNEL])
except ValueError as err:
self.fail(err)
def test_emit_all_messages(self):
with mock.patch("app.get_messages", mocked_get_messages):
with mock.patch("app.get_user", mocked_get_user):
client = socketio.test_client(app)
emit_all_messages(MESSAGES_RECEIVED_CHANNEL)
received = client.get_received()
self.assertEqual(MESSAGES_RECEIVED_CHANNEL, received[-1]["name"])
allMessages = received[-1]["args"][0]["allMessages"]
self.assertEqual("test-message", allMessages[0][3])
self.assertEqual("bot-test-message", allMessages[1][3])
client.disconnect()
def test_emit_all_messages_success(self, mock_flask):
session = UnifiedAlchemyMagicMock()
for test in self.success_test_params:
mock_flask.request.sid = test[KEY_INPUT][KEY_SID]
with mock.patch("models.DB.session", session):
session.add(models.EnteredRooms(1, 1))
session.add(models.EnteredRooms(2, 1))
session.commit()
with mock.patch("flask_socketio.SocketIO.emit",
self.mock_emit):
with mock.patch("app.get_room", self.mock_get_room):
app.emit_all_messages(test[KEY_INPUT][KEY_SID])
def test_emit_all_messages_testcase_success(self):
for testcase in self.success_test_params:
app.emit_all_messages("messages received")