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")