karaolidis/fbchat
Archived
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Added an extra test

Browse Source
This commit is contained in:
Mads Marquart
2017-05-05 18:59:38 +02:00
parent fa026021b2
commit 2280d31cbb
1 changed files with 7 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

15
tests.py
View File

@@ -28,10 +28,6 @@ If you've made any changes to the 2FA functionality, test it with a 2FA enabled
""" """
class CustomClient(fbchat.Client):
def on_message_new(self, *args, **kwargs):
self.listening = False
class TestFbchat(unittest.TestCase): class TestFbchat(unittest.TestCase):
def test_login_functions(self): def test_login_functions(self):
self.assertTrue(client.is_logged_in()) self.assertTrue(client.is_logged_in())
@@ -50,7 +46,7 @@ class TestFbchat(unittest.TestCase):
def test_sessions(self): def test_sessions(self):
global client global client
session_cookies = client.getSession() session_cookies = client.getSession()
client = CustomClient(email, password, session_cookies=session_cookies) client = fbchat.Client(email, password, session_cookies=session_cookies)
self.assertTrue(client.is_logged_in()) self.assertTrue(client.is_logged_in())
@@ -116,8 +112,11 @@ class TestFbchat(unittest.TestCase):
self.assertEquals(info['name'], 'Mark Zuckerberg') self.assertEquals(info['name'], 'Mark Zuckerberg')
def test_remove_add_from_chat(self): def test_remove_add_from_chat(self):
client.remove_user_from_chat(group_uid, user_uid) self.assertTrue(client.remove_user_from_chat(group_uid, user_uid))
client.add_users_to_chat(group_uid, user_uid) self.assertTrue(client.add_users_to_chat(group_uid, user_uid))
def test_changeThreadTitle(self):
self.assertTrue(client.changeThreadTitle(group_uid, 'test_changeThreadTitle'))
def start_test(param_client, param_group_uid, param_user_uid, tests=[]): def start_test(param_client, param_group_uid, param_user_uid, tests=[]):
@@ -160,7 +159,7 @@ if __name__ == '__main__':
user_uid = input('Please enter a user uid (To test kicking/adding functionality): ') user_uid = input('Please enter a user uid (To test kicking/adding functionality): ')
print ('Logging in') print ('Logging in')
client = CustomClient(email, password) client = fbchat.Client(email, password)
# Warning! Taking user input directly like this could be dangerous! Use only for testing purposes! # Warning! Taking user input directly like this could be dangerous! Use only for testing purposes!
start_test(client, group_uid, user_uid, sys.argv[1:]) start_test(client, group_uid, user_uid, sys.argv[1:])