def testConnection() -> None: """tests the Connection object """ connection = DummyConnection() assert connection.commands connection.userJoinedRoom(core.BotUser("user1", connection), connection.getRoom("tE ST r]OOm")) assert connection.userList[connection.getUser("user1")] == {"testroom"} assert connection.getUserRooms(connection.getUser("user1")) == {"testroom"} connection.userLeftRoom(connection.getUser("user1"), connection.getRoom("testroom")) assert connection.userList[connection.getUser("user1")] == set() assert connection.getUserRooms(connection.getUser("user1")) == set() assert connection.getRoom("testroom").id == "testroom" assert connection.getRoom("T e s tROO &%# m").id == "testroom" assert isinstance(str(connection), str)
def testMessageQueryResponse() -> None: """Tests the ability of Message objects to handle query responses """ connection = DummyConnection() message = core.BotMessage( """|queryresponse|roominfo|{"id":"testroom","roomid":"testroom","title":"Magic & Mayhem","type":"chat","visibility":"hidden","modchat":null,"auth":{"#":["annika","awa","cleo","meicoo"],"%":["dawnofares","instruct","ratisweep","pirateprincess","watfor","oaklynnthylacine"],"@":["gwynt","darth","profsapling","ravioliqueen","miapi"],"+":["madmonty","birdy","captanpasta","iwouldprefernotto","xprienzo","nui","toxtricityamped"],"*":["expectobotronum","kida"]}, "users":["user1","user2"]}""", connection ) assert message.type == "queryresponse" assert "testroom" in connection.userList[connection.getUser('user1')] assert "testroom" in connection.userList[connection.getUser('user2')] allUserIDs = [user.id for user in connection.userList] assert 'user1' in allUserIDs assert 'user2' in allUserIDs auth = connection.getRoom("testroom").auth assert auth['#'] == {"annika", "awa", "cleo", "meicoo"} assert auth['*'] == {"expectobotronum", "kida"} assert auth['@'] == {"gwynt", "darth", "profsapling", "ravioliqueen", "miapi"} assert auth['%'] == {"dawnofares", "instruct", "ratisweep", "pirateprincess", "watfor", "oaklynnthylacine"} assert auth['+'] == {"madmonty", "birdy", "captanpasta", "iwouldprefernotto", "xprienzo", "nui", "toxtricityamped"}