projects / z_archive / twitter.git / blame
| Commit | Line | Data |
|---|---|---|
| 066c34e0 | 1 | # encoding: utf8 |
| e541e268 MV |
2 | |
| 3 | from random import choice | |
| c77b5e4b | 4 | import time |
| e541e268 | 5 | |
| 32a4811d | 6 | from twitter import Twitter, NoAuth, OAuth, read_token_file, TwitterHTTPError |
| e541e268 MV |
7 | from twitter.cmdline import CONSUMER_KEY, CONSUMER_SECRET |
| 8 | ||
| 9 | noauth = NoAuth() | |
| 10 | oauth = OAuth(*read_token_file('tests/oauth_creds') | |
| 11 | + (CONSUMER_KEY, CONSUMER_SECRET)) | |
| 12 | ||
| 13 | twitter = Twitter(domain='api.twitter.com', | |
| 14 | auth=oauth, | |
| 15 | api_version='1') | |
| 920528cd MV |
16 | twitter11 = Twitter(domain='api.twitter.com', |
| 17 | auth=oauth, | |
| 18 | api_version='1.1') | |
| e541e268 | 19 | twitter_na = Twitter(domain='api.twitter.com', auth=noauth, api_version='1') |
| 32a4811d | 20 | twitter11_na = Twitter(domain='api.twitter.com', auth=noauth, api_version='1.1') |
| e541e268 MV |
21 | |
| 22 | ||
| 23 | AZaz = "abcdefghijklmnopqrstuvwxyz1234567890ABCDEFGHIJKLMNOPQRSTUVWXYZ" | |
| 24 | ||
| 25 | def get_random_str(): | |
| 26 | return ''.join(choice(AZaz) for _ in range(10)) | |
| 27 | ||
| 28 | ||
| e541e268 MV |
29 | def test_API_set_tweet(): |
| 30 | random_tweet = "A random tweet " + get_random_str() | |
| 31 | twitter.statuses.update(status=random_tweet) | |
| c77b5e4b | 32 | time.sleep(2) |
| e541e268 MV |
33 | recent = twitter.statuses.user_timeline() |
| 34 | assert recent | |
| c77b5e4b SK |
35 | assert isinstance(recent.rate_limit_remaining, int) |
| 36 | assert isinstance(recent.rate_limit_reset, int) | |
| e541e268 MV |
37 | assert random_tweet == recent[0]['text'] |
| 38 | ||
| 39 | ||
| 066c34e0 | 40 | def test_API_set_unicode_tweet(): |
| 920528cd | 41 | random_tweet = u"A random tweet with unicode üøπ" + get_random_str() |
| 066c34e0 | 42 | twitter.statuses.update(status=random_tweet) |
| 43 | ||
| 44 | recent = twitter.statuses.user_timeline() | |
| 45 | assert recent | |
| 46 | assert random_tweet == recent[0]['text'] | |
| 47 | ||
| 48 | ||
| 652c5402 MV |
49 | def test_search(): |
| 50 | t_search = Twitter(domain='search.twitter.com') | |
| 51 | results = t_search.search(q='foo') | |
| 52 | assert results | |
| e748eed8 | 53 | |
| 54 | ||
| 55 | def test_get_trends(): | |
| 56 | # This is one method of inserting parameters, using named | |
| 57 | # underscore params. | |
| 58 | world_trends = twitter.trends._woeid(_woeid=1) | |
| 59 | assert world_trends | |
| 60 | ||
| 61 | ||
| 62 | def test_get_trends_2(): | |
| 63 | # This is a nicer variation of the same call as above. | |
| 64 | world_trends = twitter.trends._(1) | |
| 65 | assert world_trends | |
| 920528cd MV |
66 | |
| 67 | ||
| 68 | def test_get_trends_3(): | |
| 69 | # Of course they broke it all again in 1.1... | |
| 70 | assert twitter11.trends.place(_id=1) | |
| 71 | ||
| 32a4811d HN |
72 | def test_TwitterHTTPError_raised_for_invalid_oauth(): |
| 73 | test_passed = False | |
| 74 | try: | |
| 75 | twitter11_na.statuses.mentions_timeline() | |
| 76 | except TwitterHTTPError: | |
| 77 | # this is the error we are looking for :) | |
| 78 | test_passed = True | |
| 79 | assert test_passed | |
| 80 |