def test_add_message_for_chatroom(self): cr = chatter_classes.Chatroom(2, db) message_count = cr.get_message_count() cr.add_message("Added by test_add_message_for_chatroom()", cr.get_all_owners()[0].userid) self.assertEqual(message_count + 1, chatter_classes.Chatroom(2, db).get_message_count())
def test_user_send_message(self): cr = chatter_classes.Chatroom(1, db) message_count = cr.get_message_count() u = chatter_classes.User(1, db) u.send_message("Sent by test_user_send_message()", cr.chatroomid) self.assertEqual(message_count + 1, chatter_classes.Chatroom(1, db).get_message_count())
def test_check_user_is_member(self): cr = chatter_classes.Chatroom(2, db) u1 = chatter_classes.User(3, db) u2 = chatter_classes.User(4, db) self.assertTrue(cr.user_is_member(u1)) self.assertTrue(cr.user_is_member(u2))
def test_get_all_members(self): cr = chatter_classes.Chatroom(1, db) members = cr.get_all_members() self.assertEqual(2, len(members)) self.assertEqual(2, members[0].userid) self.assertEqual(3, members[1].userid)
def test_contructor_existing_message(self): m = chatter_classes.Message(1, db) self.assertEqual( m.content, "This is the first message in TestRoom1, sent by TestUser1. It has two attachments (a picture of Donald Trump and another of Gary Barlow)." ) self.assertEqual(m.chatroom.name, chatter_classes.Chatroom(1, db).name) self.assertEqual(m.sender.username, chatter_classes.User(1, db).username) self.assertEqual(m.timestamp.year, 2021)
def test_add_new_message(self): text = "This is a test message generated in the unit test test_add_new_message()" message = chatter_classes.Message.add(text, 1, 1, db) self.assertIsInstance(message, chatter_classes.Message) self.assertEqual(message.content, text) self.assertEqual(message.sender.username, chatter_classes.User(1, db).username) self.assertEqual(message.chatroom.name, chatter_classes.Chatroom(1, db).name)
def test_get_all_messages_since(self): cr = chatter_classes.Chatroom(1, db) start_time = datetime.datetime.now() message_content = "Added by test_get_all_message_since()" print("Waiting two seconds before adding checking for recent messages") time.sleep(2) chatter_classes.Message.add(message_content, cr.chatroomid, cr.get_all_owners()[0].userid, db) messages = cr.get_messages(start_time) self.assertEqual(1, len(messages)) self.assertEqual(message_content, messages[0].content)
def test_get_message_count_since(self): cr = chatter_classes.Chatroom(1, db) now = datetime.datetime.now() message_content = "Added by test_get_message_count_since()" print( "Waiting two seconds before adding and checking for recent messages" ) time.sleep(2) chatter_classes.Message.add(message_content, cr.chatroomid, cr.get_all_owners()[0].userid, db) message_count = cr.get_message_count(now) self.assertEqual(1, message_count)
def test_update_chatroom(self): # Create a chatroom for testing cr = chatter_classes.Chatroom.add( "UnitTestChatroomForUpdate", "Created by TestChatroom.test_update_chatroom", db) cr_id = cr.chatroomid cr.update_join_code() cr.update(name="UnitTestChatroomForUpdateB", description="Created by TestChatroom.test_update_chatroomB") # Instantiate a second instance of the same chatroom and check that its properties match those of the updated # one crB = chatter_classes.Chatroom(cr_id, db) self.assertEqual(cr.name, crB.name) self.assertEqual(cr.description, crB.description) self.assertEqual(cr.joincode, crB.joincode) # Delete chatrooms now that we have finished with them cr.delete() crB.delete()
def test_chatroom_json_with_messages(self): c = chatter_classes.Chatroom(1, db) js = c.json_with_messages print(js) self.assertNotEqual(0, len(js))
def test_chatroom_json(self): c = chatter_classes.Chatroom(1, db) js = c.json print(js) self.assertNotEqual(0, len(js))
def test_get_message_count(self): cr = chatter_classes.Chatroom(3, db) self.assertEqual(9, cr.get_message_count())
def test_get_all_messages(self): cr = chatter_classes.Chatroom(3, db) messages = cr.get_messages() self.assertEqual(9, len(messages))
def test_check_user_is_owner(self): cr = chatter_classes.Chatroom(2, db) u = chatter_classes.User(2, db) self.assertTrue(cr.user_is_owner(u))
def test_get_all_owners(self): cr = chatter_classes.Chatroom(1, db) owners = cr.get_all_owners() self.assertEqual(1, len(owners)) self.assertEqual(1, owners[0].userid)
def test_constructor_existing_chatroom(self): cr = chatter_classes.Chatroom(1, db) self.assertEqual(cr.name, "TestRoom1")