FilesExpand file tree

 master

/

test_twitter_utils.py

Copy path

Blame

More file actions

Blame

More file actions

Latest commit

 

History

History

History

60 lines (48 loc) · 2.04 KB

 master

/

test_twitter_utils.py

Copy path

Top

File metadata and controls

• Code
• Blame

60 lines (48 loc) · 2.04 KB

Raw

Copy raw file

Download raw file

Open symbols panel

Edit and raw actions

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'))