forked from bear/python-twitter
-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathtest_twitter_utils.py
More file actions
60 lines (48 loc) · 2.04 KB
/
test_twitter_utils.py
File metadata and controls
60 lines (48 loc) · 2.04 KB
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
# encoding: utf-8
import unittest
import twitter
from twitter.twitter_utils import (
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'))