def test_app_on_get_whiteboard(self, mocked_flask):
session = UnifiedAlchemyMagicMock()
for test in self.test_on_get_whiteboard_params:
mocked_flask.request.sid = test[KEY_INPUT][KEY_SID]
with mock.patch("flask_socketio.SocketIO.emit", self.mock_emit):
with mock.patch("models.DB.session", session):
session.add(models.Whiteboards(*test[KEY_INPUT][BOARD_DATA]))
session.commit()
app.on_get_whiteboard()
self.assertDictEqual(self.hold, test[KEY_EXPECTED])
def test_app_on_forced_save(self, mocked_flask):
session = UnifiedAlchemyMagicMock()
for test in self.test_on_forced_save_params:
mocked_flask.request.sid = test[KEY_INPUT][KEY_SID]
with mock.patch("flask_socketio.SocketIO.emit", self.mock_emit):
with mock.patch("models.DB.session", session):
temp_row=models.Whiteboards(*test[KEY_INPUT][KEY_BOARD][:2])
temp_row.save_num = test[KEY_INPUT][KEY_BOARD][2]
session.add(temp_row)
session.add(models.WhiteboardConnections(*test[KEY_INPUT][KEY_CONNECTION]))
session.commit()
app.on_forced_save({"blob":bytes([1,5,2,5,3,2,1])})
self.assertDictEqual(self.hold, test[KEY_EXPECTED])
def test_on_join_whiteboard(self, mocked_flask):
session = UnifiedAlchemyMagicMock()
for test in self.test_on_join_whiteboard_params:
mocked_flask.request.sid = test[KEY_INPUT][KEY_SID]
with mock.patch("flask_socketio.SocketIO.emit", self.mock_emit):
with mock.patch("flask_socketio.join_room", self.mock_join_room):
with mock.patch("models.DB.session", session):
temp_row=models.Whiteboards(test[KEY_INPUT][BOARD_DATA]["room"],test[KEY_INPUT][BOARD_DATA]["name"])
temp_row.id=test[KEY_INPUT][BOARD_DATA]["id"]
temp_row.save_num=test[KEY_INPUT][BOARD_DATA]["save_num"]
session.add(temp_row)
session.commit()
app.on_join_whiteboard(test[KEY_INPUT][KEY_DATA])
self.assertDictEqual(self.hold, test[KEY_EXPECTED])
def on_make_whiteboard(data):
room = get_room(flask.request.sid)
models.DB.session.add(models.Whiteboards(room, data["name"]))
models.DB.session.commit()
emit_boards(flask.request.sid)