forked from bear/python-twitter
-
Notifications
You must be signed in to change notification settings - Fork 0
/
test_twitter_utils.py
76 lines (61 loc) · 2.69 KB
/
test_twitter_utils.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
# encoding: utf-8
import unittest
import twitter
from twitter.twitter_utils import (
calc_expected_status_length,
parse_media_file
)
class ApiTest(unittest.TestCase):
def setUp(self):
self.api = twitter.Api(
consumer_key='test',
consumer_secret='test',
access_token_key='test',
access_token_secret='test',
sleep_on_rate_limit=False)
self.base_url = 'https://api.twitter.com/1.1'
def test_parse_media_file_http(self):
data_file, filename, file_size, media_type = parse_media_file(
'https://raw.githubusercontent.com/bear/python-twitter/master/testdata/168NQ.jpg')
self.assertTrue(hasattr(data_file, 'read'))
self.assertEqual(filename, '168NQ.jpg')
self.assertEqual(file_size, 44772)
self.assertEqual(media_type, 'image/jpeg')
def test_parse_media_file_local_file(self):
data_file, filename, file_size, media_type = parse_media_file(
'testdata/168NQ.jpg')
self.assertTrue(hasattr(data_file, 'read'))
self.assertEqual(filename, '168NQ.jpg')
self.assertEqual(file_size, 44772)
self.assertEqual(media_type, 'image/jpeg')
def test_parse_media_file_fileobj(self):
with open('testdata/168NQ.jpg', 'rb') as f:
data_file, filename, file_size, media_type = parse_media_file(f)
self.assertTrue(hasattr(data_file, 'read'))
self.assertEqual(filename, '168NQ.jpg')
self.assertEqual(file_size, 44772)
self.assertEqual(media_type, 'image/jpeg')
def test_utils_error_checking(self):
with open('testdata/168NQ.jpg', 'r') as f:
self.assertRaises(
twitter.TwitterError,
lambda: parse_media_file(f))
with open('testdata/user_timeline.json', 'rb') as f:
self.assertRaises(
twitter.TwitterError,
lambda: parse_media_file(f))
self.assertRaises(
twitter.TwitterError,
lambda: twitter.twitter_utils.enf_type('test', int, 'hi'))
def test_calc_expected_status_length(self):
status = 'hi a tweet there'
len_status = calc_expected_status_length(status)
self.assertEqual(len_status, 16)
def test_calc_expected_status_length_with_url(self):
status = 'hi a tweet there example.com'
len_status = calc_expected_status_length(status)
self.assertEqual(len_status, 40)
def test_calc_expected_status_length_with_url_and_extra_spaces(self):
status = 'hi a tweet there example.com'
len_status = calc_expected_status_length(status)
self.assertEqual(len_status, 63)