def test_presence_message_w_users(self, test_uuids): server_state = get_state() user = User("test_user") user.state_from_dict({"key": "1", "key2": "2"}) user.state_public_keys = ["key2"] server_state.users[user.username] = user connection = Connection("test_user", conn_id=test_uuids[1]) user.add_connection(connection) user2 = User("test_user2") user2.state_from_dict({"key": "1", "key2": "2"}) server_state.users[user2.username] = user2 connection2 = Connection("test_user2", conn_id=test_uuids[2]) user2.add_connection(connection2) config = { "notify_presence": True, "broadcast_presence_with_user_lists": True } channel = Channel("test", channel_config=config) channel.add_connection(connection) channel.add_connection(connection2) payload = channel.send_notify_presence_info("test_user", "join") assert len(payload["users"]) == 2 sorted_users = sorted(payload["users"], key=lambda x: x["user"]) assert sorted_users == [ { "state": { "key2": "2" }, "user": "******" }, { "state": {}, "user": "******" }, ]
def connect(self): """return the id of connected users - will be secured with password string for webapp to internally call the server - we combine conn string with user id, and we tell which channels the user is allowed to subscribe to""" user_name = self.request.json_body.get('user') def_status = self.request.registry.server_config['status_codes'][ 'online'] user_status = int(self.request.json_body.get('status', def_status)) conn_id = self.request.json_body.get('conn_id') subscribe_to_channels = self.request.json_body.get('channels') if user_name is None: self.request.response.status = 400 return {'error': "No username specified"} if not subscribe_to_channels: self.request.response.status = 400 return {'error': "No channels specified"} # everything is ok so lets add new connection to channel and connection list if not user_name in users: user = User(user_name, def_status) users[user_name] = user else: user = users[user_name] connection = Connection(user_name, conn_id) if not connection.id in connections: connections[connection.id] = connection user.add_connection(connection) for channel_name in subscribe_to_channels: # user gets assigned to a channel if channel_name not in channels: channel = Channel(channel_name) channels[channel_name] = channel channels[channel_name].add_connection(connection) log.info('connecting %s with uuid %s' % (user_name, connection.id)) return {'conn_id': connection.id, 'status': user.status}
def test_user_single_assignment(self, test_uuids): server_state = get_state() user = User("test_user") connection = Connection("test_user", conn_id=test_uuids[1]) user.add_connection(connection) channel = Channel("test") server_state.channels[channel.name] = channel channel.add_connection(connection) assert [channel] == user.get_channels()
def test_user_single_assignment(self, test_uuids): server_state = get_state() user = User("test_user") connection = Connection("test_user", conn_id=test_uuids[1]) user.add_connection(connection) channel = Channel("test") server_state.channels[channel.name] = channel channel.add_connection(connection) assert [channel] == user.get_channels()
def test_remove_connection_w_presence(self, test_uuids): server_state = get_state() user = User("test_user") server_state.users[user.username] = user connection = Connection("test_user", conn_id=test_uuids[1]) user.add_connection(connection) config = {"notify_presence": True, "broadcast_presence_with_user_lists": True} channel = Channel("test", channel_config=config) channel.add_connection(connection) channel.remove_connection(connection)
def test_presence_message(self, test_uuids): user = User("test_user") connection = Connection("test_user", conn_id=test_uuids[1]) user.add_connection(connection) channel = Channel("test") channel.add_connection(connection) payload = channel.send_notify_presence_info("test_user", "join") assert payload["user"] == "test_user" assert payload["message"]["action"] == "join" assert payload["type"] == "presence" assert payload["channel"] == "test" assert len(payload["users"]) == 0
def test_presence_message(self, test_uuids): user = User("test_user") connection = Connection("test_user", conn_id=test_uuids[1]) user.add_connection(connection) channel = Channel("test") channel.add_connection(connection) payload = channel.send_notify_presence_info("test_user", "join") assert payload["user"] == "test_user" assert payload["message"]["action"] == "join" assert payload["type"] == "presence" assert payload["channel"] == "test" assert len(payload["users"]) == 0
def test_remove_connection_w_presence(self, test_uuids): server_state = get_state() user = User("test_user") server_state.users[user.username] = user connection = Connection("test_user", conn_id=test_uuids[1]) user.add_connection(connection) config = { "notify_presence": True, "broadcast_presence_with_user_lists": True } channel = Channel("test", channel_config=config) channel.add_connection(connection) channel.remove_connection(connection)
def test_user_state(self, test_uuids): user = User("test_user") changed = user.state_from_dict({"key": "1", "key2": "2"}) user.state_public_keys = ["key2"] connection = Connection("test_user", conn_id=test_uuids[1]) user.add_connection(connection) channel = Channel("test") channel.add_connection(connection) payload = channel.send_user_state(user, changed) assert payload["user"] == "test_user" assert payload["message"]["state"] == {"key2": "2"} assert payload["message"]["changed"] == [{"key": "key2", "value": "2"}] assert payload["type"] == "user_state_change" assert payload["channel"] == "test"
def test_user_state(self, test_uuids): user = User("test_user") changed = user.state_from_dict({"key": "1", "key2": "2"}) user.state_public_keys = ["key2"] connection = Connection("test_user", conn_id=test_uuids[1]) user.add_connection(connection) channel = Channel("test") channel.add_connection(connection) payload = channel.send_user_state(user, changed) assert payload["user"] == "test_user" assert payload["message"]["state"] == {"key2": "2"} assert payload["message"]["changed"] == [{"key": "key2", "value": "2"}] assert payload["type"] == "user_state_change" assert payload["channel"] == "test"
def test_add_connection_w_presence(self): server_state = get_state() user = User("test_user") server_state.users[user.username] = user connection = Connection("test_user", conn_id=test_uuids[1]) user.add_connection(connection) config = { "notify_presence": True, "broadcast_presence_with_user_lists": True } channel = Channel("test", channel_config=config) channel.add_connection(connection) assert len(channel.connections["test_user"]) == 1 assert "test_user" in channel.connections assert connection in channel.connections["test_user"]
def test_user_multi_assignment(self, test_uuids): server_state = get_state() user = User("test_user") connection = Connection("test_user", conn_id=test_uuids[1]) connection2 = Connection("test_user", conn_id=test_uuids[2]) connection3 = Connection("test_user", conn_id=test_uuids[3]) user.add_connection(connection) user.add_connection(connection2) user.add_connection(connection3) channel = Channel("test") channel2 = Channel("test2") server_state.channels[channel.name] = channel server_state.channels[channel2.name] = channel2 channel.add_connection(connection) channel.add_connection(connection2) channel2.add_connection(connection3) assert ["test", "test2"] == sorted([c.name for c in user.get_channels()])
def test_messages(self, test_uuids): user = User("test_user") connection = Connection("test_user", conn_id=test_uuids[1]) connection.queue = Queue() connection2 = Connection("test_user", conn_id=test_uuids[2]) connection2.queue = Queue() user.add_connection(connection) user.add_connection(connection2) user.add_message({ "type": "message", "no_history": False, "pm_users": [], "exclude_users": [], }) assert len(user.connections) == 2 assert len(user.connections[0].queue.get()) == 1 assert len(user.connections[1].queue.get()) == 1
def connect( username=None, fresh_user_state=None, state_public_keys=None, update_user_state=None, conn_id=None, channels=None, channel_configs=None, ): """ :param username: :param fresh_user_state: :param state_public_keys: :param update_user_state: :param conn_id: :param channels: :param channel_configs: :return: """ server_state = get_state() with server_state.lock: if username not in server_state.users: user = User(username) user.state_from_dict(fresh_user_state) server_state.users[username] = user else: user = server_state.users[username] if state_public_keys is not None: user.state_public_keys = state_public_keys user.state_from_dict(update_user_state) connection = Connection(username, conn_id) if connection.id not in server_state.connections: server_state.connections[connection.id] = connection user.add_connection(connection) for channel_name in channels: # user gets assigned to a channel if channel_name not in server_state.channels: channel = Channel( channel_name, channel_config=channel_configs.get(channel_name) ) server_state.channels[channel_name] = channel server_state.channels[channel_name].add_connection(connection) log.info("connecting %s with uuid %s" % (username, connection.id)) return connection, user
def test_user_multi_assignment(self, test_uuids): server_state = get_state() user = User("test_user") connection = Connection("test_user", conn_id=test_uuids[1]) connection2 = Connection("test_user", conn_id=test_uuids[2]) connection3 = Connection("test_user", conn_id=test_uuids[3]) user.add_connection(connection) user.add_connection(connection2) user.add_connection(connection3) channel = Channel("test") channel2 = Channel("test2") server_state.channels[channel.name] = channel server_state.channels[channel2.name] = channel2 channel.add_connection(connection) channel.add_connection(connection2) channel2.add_connection(connection3) assert ["test", "test2"] == sorted([c.name for c in user.get_channels()])
def test_messages(self, test_uuids): user = User("test_user") connection = Connection("test_user", conn_id=test_uuids[1]) connection.queue = Queue() connection2 = Connection("test_user", conn_id=test_uuids[2]) connection2.queue = Queue() user.add_connection(connection) user.add_connection(connection2) user.add_message( { "type": "message", "no_history": False, "pm_users": [], "exclude_users": [], } ) assert len(user.connections) == 2 assert len(user.connections[0].queue.get()) == 1 assert len(user.connections[1].queue.get()) == 1
def test_presence_message_w_users(self, test_uuids): server_state = get_state() user = User("test_user") user.state_from_dict({"key": "1", "key2": "2"}) user.state_public_keys = ["key2"] server_state.users[user.username] = user connection = Connection("test_user", conn_id=test_uuids[1]) user.add_connection(connection) user2 = User("test_user2") user2.state_from_dict({"key": "1", "key2": "2"}) server_state.users[user2.username] = user2 connection2 = Connection("test_user2", conn_id=test_uuids[2]) user2.add_connection(connection2) config = {"notify_presence": True, "broadcast_presence_with_user_lists": True} channel = Channel("test", channel_config=config) channel.add_connection(connection) channel.add_connection(connection2) payload = channel.send_notify_presence_info("test_user", "join") assert len(payload["users"]) == 2 sorted_users = sorted(payload["users"], key=lambda x: x["user"]) assert sorted_users == [ {"state": {"key2": "2"}, "user": "******"}, {"state": {}, "user": "******"}, ]
def test_gc_connections_collecting(self, test_uuids): server_state = get_state() channel = Channel("test") server_state.channels[channel.name] = channel channel2 = Channel("test2") server_state.channels[channel2.name] = channel2 user = User("test_user") server_state.users[user.username] = user user2 = User("test_user2") server_state.users[user2.username] = user2 connection = Connection("test_user", test_uuids[1]) server_state.connections[connection.id] = connection connection2 = Connection("test_user", test_uuids[2]) connection2.mark_for_gc() server_state.connections[connection2.id] = connection2 connection3 = Connection("test_user2", test_uuids[3]) connection3.mark_for_gc() server_state.connections[connection3.id] = connection3 connection4 = Connection("test_user2", test_uuids[4]) server_state.connections[connection4.id] = connection4 user.add_connection(connection) user.add_connection(connection2) channel.add_connection(connection) channel.add_connection(connection2) user2.add_connection(connection3) user2.add_connection(connection4) channel2.add_connection(connection3) channel2.add_connection(connection4) channelstream.gc.gc_conns() assert len(server_state.connections.items()) == 2 conns = server_state.channels["test"].connections["test_user"] assert len(conns) == 1 assert conns == [connection] conns = server_state.channels["test2"].connections["test_user2"] assert len(conns) == 1 assert conns == [connection4] assert len(user.connections) == 1 assert len(user2.connections) == 1 connection.mark_for_gc() connection4.mark_for_gc() channelstream.gc.gc_conns() assert "test_user" not in server_state.channels["test"].connections assert "test_user2" not in server_state.channels["test2"].connections assert len(server_state.channels["test"].connections.items()) == 0 assert len(server_state.channels["test2"].connections.items()) == 0
def test_gc_connections_collecting(self, test_uuids): server_state = get_state() channel = Channel("test") server_state.channels[channel.name] = channel channel2 = Channel("test2") server_state.channels[channel2.name] = channel2 user = User("test_user") server_state.users[user.username] = user user2 = User("test_user2") server_state.users[user2.username] = user2 connection = Connection("test_user", test_uuids[1]) server_state.connections[connection.id] = connection connection2 = Connection("test_user", test_uuids[2]) connection2.mark_for_gc() server_state.connections[connection2.id] = connection2 connection3 = Connection("test_user2", test_uuids[3]) connection3.mark_for_gc() server_state.connections[connection3.id] = connection3 connection4 = Connection("test_user2", test_uuids[4]) server_state.connections[connection4.id] = connection4 user.add_connection(connection) user.add_connection(connection2) channel.add_connection(connection) channel.add_connection(connection2) user2.add_connection(connection3) user2.add_connection(connection4) channel2.add_connection(connection3) channel2.add_connection(connection4) channelstream.gc.gc_conns() assert len(server_state.connections.items()) == 2 conns = server_state.channels["test"].connections["test_user"] assert len(conns) == 1 assert conns == [connection] conns = server_state.channels["test2"].connections["test_user2"] assert len(conns) == 1 assert conns == [connection4] assert len(user.connections) == 1 assert len(user2.connections) == 1 connection.mark_for_gc() connection4.mark_for_gc() channelstream.gc.gc_conns() assert "test_user" not in server_state.channels["test"].connections assert "test_user2" not in server_state.channels["test2"].connections assert len(server_state.channels["test"].connections.items()) == 0 assert len(server_state.channels["test2"].connections.items()) == 0
def test_gc_connections_active(self, test_uuids): server_state = get_state() channel = Channel("test") server_state.channels[channel.name] = channel channel2 = Channel("test2") server_state.channels[channel2.name] = channel2 user = User("test_user") server_state.users[user.username] = user user2 = User("test_user2") server_state.users[user2.username] = user2 connection = Connection("test_user", test_uuids[1]) server_state.connections[connection.id] = connection connection2 = Connection("test_user", test_uuids[2]) server_state.connections[connection2.id] = connection2 connection3 = Connection("test_user2", test_uuids[3]) server_state.connections[connection3.id] = connection3 connection4 = Connection("test_user2", test_uuids[4]) server_state.connections[connection4.id] = connection4 user.add_connection(connection) user.add_connection(connection2) channel.add_connection(connection) channel.add_connection(connection2) user2.add_connection(connection3) user2.add_connection(connection4) channel2.add_connection(connection3) channel2.add_connection(connection4) channelstream.gc.gc_conns() conns = server_state.channels["test"].connections["test_user"] assert len(conns) == 2 assert len(server_state.connections.items()) == 4 conns = server_state.channels["test2"].connections["test_user2"] assert len(conns) == 2 assert len(user.connections) == 2 assert len(user2.connections) == 2 assert sorted(channel.connections.keys()) == ["test_user"] assert sorted(channel2.connections.keys()) == ["test_user2"]
def test_gc_connections_active(self, test_uuids): server_state = get_state() channel = Channel("test") server_state.channels[channel.name] = channel channel2 = Channel("test2") server_state.channels[channel2.name] = channel2 user = User("test_user") server_state.users[user.username] = user user2 = User("test_user2") server_state.users[user2.username] = user2 connection = Connection("test_user", test_uuids[1]) server_state.connections[connection.id] = connection connection2 = Connection("test_user", test_uuids[2]) server_state.connections[connection2.id] = connection2 connection3 = Connection("test_user2", test_uuids[3]) server_state.connections[connection3.id] = connection3 connection4 = Connection("test_user2", test_uuids[4]) server_state.connections[connection4.id] = connection4 user.add_connection(connection) user.add_connection(connection2) channel.add_connection(connection) channel.add_connection(connection2) user2.add_connection(connection3) user2.add_connection(connection4) channel2.add_connection(connection3) channel2.add_connection(connection4) channelstream.gc.gc_conns() conns = server_state.channels["test"].connections["test_user"] assert len(conns) == 2 assert len(server_state.connections.items()) == 4 conns = server_state.channels["test2"].connections["test_user2"] assert len(conns) == 2 assert len(user.connections) == 2 assert len(user2.connections) == 2 assert sorted(channel.connections.keys()) == ["test_user"] assert sorted(channel2.connections.keys()) == ["test_user2"]