a76ebbb22a
- Reworked events, so now they support python 2.7 (I had to remove some
functionality though, but that was a little unnecessary anyway)
- Events now support the old style of writing, for people who's more
comfortable with that: ```python
class EchoBot(fbchat.Client):
def onMessage(self, *args, **kwargs):
self.something(*args, **kwargs)
```
While still supporting the new method:
```python
class EchoBot(fbchat.Client):
def __init__(self, *args, **kwargs):
super(EchoBot, self).__init__(*args, **kwargs)
self.onMessage += lamda *args, **kwargs: self.something(*args,
**kwargs)
```
- Included `msg` as a parameter in every event function, since it's
useful if you want to extract some of the other data
- Moved test data to the folder `tests`
- Fixed various other functions, and improved stability
214 lines
8.3 KiB
Python
214 lines
8.3 KiB
Python
#!/usr/bin/env python
|
|
# -*- coding: UTF-8 -*-
|
|
|
|
from __future__ import unicode_literals
|
|
import json
|
|
import logging
|
|
import unittest
|
|
from getpass import getpass
|
|
from sys import argv
|
|
from os import path
|
|
from fbchat import Client
|
|
from fbchat.models import *
|
|
|
|
logging_level = logging.ERROR
|
|
|
|
"""
|
|
|
|
Tests for fbchat
|
|
~~~~~~~~~~~~~~~~
|
|
|
|
To use these tests copy `tests/data.json` to `tests/my_data.json` or type this information manually in the terminal prompts.
|
|
|
|
- email: Your (or a test user's) email / phone number
|
|
- password: Your (or a test user's) password
|
|
- group_thread_id: A test group that will be used to test group functionality
|
|
- user_thread_id: A person that will be used to test kick/add functionality (This user should be in the group)
|
|
|
|
Please remember to test both python v. 2.7 and python v. 3.6!
|
|
|
|
If you've made any changes to the 2FA functionality, test it with a 2FA enabled account
|
|
If you only want to execute specific tests, pass the function names in the commandline
|
|
|
|
WARNING:
|
|
Do not execute the full set of tests in too quick succession. This can get you temporarily blocked for spam!
|
|
(You should execute the script at max about 10 times a day)
|
|
|
|
"""
|
|
|
|
class TestFbchat(unittest.TestCase):
|
|
def setUp(self):
|
|
pass
|
|
|
|
def tearDown(self):
|
|
pass
|
|
|
|
def test_loginFunctions(self):
|
|
self.assertTrue(client.isLoggedIn())
|
|
|
|
client.logout()
|
|
|
|
self.assertFalse(client.isLoggedIn())
|
|
|
|
with self.assertRaises(Exception):
|
|
client.login('not@email.com', 'not_password', max_retries=1)
|
|
|
|
client.login(email, password)
|
|
|
|
self.assertTrue(client.isLoggedIn())
|
|
|
|
def test_sessions(self):
|
|
global client
|
|
session_cookies = client.getSession()
|
|
client = Client(email, password, session_cookies=session_cookies, logging_level=logging_level)
|
|
|
|
self.assertTrue(client.isLoggedIn())
|
|
|
|
def test_defaultThread(self):
|
|
# setDefaultThread
|
|
client.setDefaultThread(client.uid, ThreadType.USER)
|
|
self.assertTrue(client.sendMessage('test_default_recipient★'))
|
|
|
|
# resetDefaultThread
|
|
client.resetDefaultThread()
|
|
with self.assertRaises(ValueError):
|
|
client.sendMessage('should_not_send')
|
|
|
|
def test_getAllUsers(self):
|
|
users = client.getAllUsers()
|
|
self.assertGreater(len(users), 0)
|
|
|
|
def test_getUsers(self):
|
|
users = client.getUsers("Mark Zuckerberg")
|
|
self.assertGreater(len(users), 0)
|
|
|
|
u = users[0]
|
|
|
|
# Test if values are set correctly
|
|
self.assertIsInstance(u.uid, int)
|
|
self.assertEqual(u.type, 'user')
|
|
self.assertEqual(u.photo[:4], 'http')
|
|
self.assertEqual(u.url[:4], 'http')
|
|
self.assertEqual(u.name, 'Mark Zuckerberg')
|
|
self.assertGreater(u.score, 0)
|
|
|
|
def test_sendEmoji(self):
|
|
self.assertTrue(client.sendEmoji(size=EmojiSize.SMALL, thread_id=user_uid, thread_type=ThreadType.USER))
|
|
self.assertTrue(client.sendEmoji(size=EmojiSize.MEDIUM, thread_id=user_uid, thread_type=ThreadType.USER))
|
|
self.assertTrue(client.sendEmoji("😆", EmojiSize.LARGE, user_uid, ThreadType.USER))
|
|
|
|
self.assertTrue(client.sendEmoji(size=EmojiSize.SMALL, thread_id=group_uid, thread_type=ThreadType.GROUP))
|
|
self.assertTrue(client.sendEmoji(size=EmojiSize.MEDIUM, thread_id=group_uid, thread_type=ThreadType.GROUP))
|
|
self.assertTrue(client.sendEmoji("😆", EmojiSize.LARGE, group_uid, ThreadType.GROUP))
|
|
|
|
def test_sendMessage(self):
|
|
self.assertIsNotNone(client.sendMessage('test_send_user★', user_uid, ThreadType.USER))
|
|
self.assertIsNotNone(client.sendMessage('test_send_group★', group_uid, ThreadType.GROUP))
|
|
self.assertIsNone(client.sendMessage('test_send_user_should_fail★', user_uid, ThreadType.GROUP))
|
|
self.assertIsNone(client.sendMessage('test_send_group_should_fail★', group_uid, ThreadType.USER))
|
|
|
|
def test_sendImages(self):
|
|
image_url = 'https://cdn4.iconfinder.com/data/icons/ionicons/512/icon-image-128.png'
|
|
image_local_url = path.join(path.dirname(__file__), 'tests/image.png')
|
|
#self.assertTrue(client.sendRemoteImage(image_url, 'test_send_user_images_remote', user_uid, ThreadType.USER))
|
|
self.assertTrue(client.sendRemoteImage(image_url, 'test_send_group_images_remote★', group_uid, ThreadType.GROUP))
|
|
# Idk why but doesnt work, payload is null
|
|
#self.assertTrue(client.sendLocalImage(image_local_url, 'test_send_group_images_local', user_uid, ThreadType.USER))
|
|
self.assertTrue(client.sendLocalImage(image_local_url, 'test_send_group_images_local★', group_uid, ThreadType.GROUP))
|
|
|
|
def test_getThreadInfo(self):
|
|
client.sendMessage('test_user_getThreadInfo★', user_uid, ThreadType.USER)
|
|
|
|
info = client.getThreadInfo(20, user_uid, ThreadType.USER)
|
|
self.assertEqual(info[0].author, 'fbid:' + client.uid)
|
|
self.assertEqual(info[0].body, 'test_user_getThreadInfo★')
|
|
|
|
client.sendMessage('test_group_getThreadInfo★', group_uid, ThreadType.GROUP)
|
|
|
|
info = client.getThreadInfo(20, group_uid, ThreadType.GROUP)
|
|
self.assertEqual(info[0].author, 'fbid:' + client.uid)
|
|
self.assertEqual(info[0].body, 'test_group_getThreadInfo★')
|
|
|
|
def test_markAs(self):
|
|
# To be implemented (requires some form of manual watching)
|
|
pass
|
|
|
|
def test_listen(self):
|
|
client.doOneListen()
|
|
|
|
def test_getUserInfo(self):
|
|
info = client.getUserInfo(4)
|
|
self.assertEqual(info['name'], 'Mark Zuckerberg')
|
|
|
|
def test_removeAddFromGroup(self):
|
|
self.assertTrue(client.removeUserFromGroup(user_uid, group_uid))
|
|
self.assertTrue(client.addUsersToGroup([user_uid], group_uid))
|
|
|
|
def test_changeGroupTitle(self):
|
|
self.assertTrue(client.changeGroupTitle('test_changeGroupTitle★', group_uid))
|
|
|
|
def test_changeThreadColor(self):
|
|
self.assertTrue(client.changeThreadColor(ThreadColor.BRILLIANT_ROSE, group_uid))
|
|
self.assertTrue(client.changeThreadColor(ThreadColor.MESSENGER_BLUE, group_uid))
|
|
self.assertTrue(client.changeThreadColor(ThreadColor.BRILLIANT_ROSE, user_uid))
|
|
self.assertTrue(client.changeThreadColor(ThreadColor.MESSENGER_BLUE, user_uid))
|
|
|
|
def test_reactToMessage(self):
|
|
mid = client.sendMessage('react_to_message', user_uid, ThreadType.USER)
|
|
self.assertTrue(client.reactToMessage(mid, MessageReaction.LOVE))
|
|
|
|
def test_setTypingStatus(self):
|
|
self.assertTrue(client.setTypingStatus(TypingStatus.TYPING, thread_id=user_uid, thread_type=ThreadType.USER))
|
|
self.assertTrue(client.setTypingStatus(TypingStatus.STOPPED, thread_id=user_uid, thread_type=ThreadType.USER))
|
|
self.assertTrue(client.setTypingStatus(TypingStatus.TYPING, thread_id=group_uid, thread_type=ThreadType.GROUP))
|
|
self.assertTrue(client.setTypingStatus(TypingStatus.STOPPED, thread_id=group_uid, thread_type=ThreadType.GROUP))
|
|
|
|
|
|
def start_test(param_client, param_group_uid, param_user_uid, tests=[]):
|
|
global client
|
|
global group_uid
|
|
global user_uid
|
|
|
|
client = param_client
|
|
group_uid = param_group_uid
|
|
user_uid = param_user_uid
|
|
|
|
tests = ['test_' + test for test in tests]
|
|
|
|
if len(tests) == 0:
|
|
suite = unittest.TestLoader().loadTestsFromTestCase(TestFbchat)
|
|
else:
|
|
suite = unittest.TestSuite(map(TestFbchat, tests))
|
|
print('Starting test(s)')
|
|
unittest.TextTestRunner(verbosity=2).run(suite)
|
|
|
|
|
|
client = None
|
|
|
|
if __name__ == '__main__':
|
|
# Python 3 does not use raw_input, whereas Python 2 does
|
|
try:
|
|
input = raw_input
|
|
except Exception as e:
|
|
pass
|
|
|
|
try:
|
|
with open(path.join(path.dirname(__file__), 'tests/my_data.json'), 'r') as f:
|
|
json = json.load(f)
|
|
email = json['email']
|
|
password = json['password']
|
|
user_uid = json['user_thread_id']
|
|
group_uid = json['group_thread_id']
|
|
except (IOError, IndexError) as e:
|
|
email = input('Email: ')
|
|
password = getpass()
|
|
group_uid = input('Please enter a group thread id (To test group functionality): ')
|
|
user_uid = input('Please enter a user thread id (To test kicking/adding functionality): ')
|
|
|
|
print('Logging in...')
|
|
client = Client(email, password, logging_level=logging_level)
|
|
|
|
# Warning! Taking user input directly like this could be dangerous! Use only for testing purposes!
|
|
start_test(client, group_uid, user_uid, argv[1:])
|
|
|