tests.py

#!/usr/bin/env python

import types
import six
try:
 import simplejson as json
except ImportError:
 import json
import getpass
import unittest
from six.moves.urllib.parse import urlparse, parse_qs
from instagram import client, oauth2, InstagramAPIError

TEST_AUTH = False
client_id = "DEBUG"
client_secret = "DEBUG"
access_token = "DEBUG"
redirect_uri = "http://example.com"

class MockHttp(object):

 def __init__(self, *args, **kwargs):
 pass

 def request(self, url, method="GET", body=None, headers={}):
 fail_state = {
 'status':'400'
 }, "{}"

 parsed = urlparse(url)
 options = parse_qs(parsed.query)

 fn_name = str(active_call)
 if fn_name == 'get_authorize_login_url':
 return {
 'status': '200',
 'content-location':'http://example.com/redirect/login'
 }, None

 if not 'access_token' in options and not 'client_id' in options:
 fn_name += '_unauthorized'
 if 'self' in url and not 'access_token' in options:
 fn_name += '_no_auth_user'

 fl = open('fixtures/%s.json' % fn_name)
 content = fl.read()
 fl.close()
 json_content = json.loads(content)
 status = json_content['meta']['code']
 return {
 'status': status
 }, content

oauth2.Http = MockHttp

active_call = None
class TestInstagramAPI(client.InstagramAPI):
 def __getattribute__(self, attr):
 global active_call
 actual_val = super(TestInstagramAPI, self).__getattribute__(attr)
 if isinstance(actual_val, types.MethodType):
 active_call = attr
 return actual_val

class InstagramAuthTests(unittest.TestCase):
 def setUp(self):
 self.unauthenticated_api = TestInstagramAPI(client_id=client_id, redirect_uri=redirect_uri, client_secret=client_secret)

 def test_authorize_login_url(self):
 redirect_uri = self.unauthenticated_api.get_authorize_login_url()
 assert redirect_uri
 print("Please visit and authorize at:\n%s" % redirect_uri)
 code = raw_input("Paste received code (blank to skip): ").strip()
 if not code:
 return

 access_token = self.unauthenticated_api.exchange_code_for_access_token(code)
 assert access_token

 def test_xauth_exchange(self):
 """ Your client ID must be authorized for xAuth access; email
 xauth@instagram.com for access"""
 username = raw_input("Enter username for XAuth (blank to skip): ").strip()
 if not username:
 return
 password = getpass.getpass("Enter password for XAuth (blank to skip): ").strip()
 access_token = self.unauthenticated_api.exchange_xauth_login_for_access_token(username, password)
 assert access_token

class InstagramAPITests(unittest.TestCase):

 def setUp(self):
 super(InstagramAPITests, self).setUp()
 self.client_only_api = TestInstagramAPI(client_id=client_id)
 self.api = TestInstagramAPI(access_token=access_token)

 def test_media_popular(self):
 self.api.media_popular(count=10)

 def test_media_search(self):
 self.client_only_api.media_search(lat=37.7,lng=-122.22)
 self.api.media_search(lat=37.7,lng=-122.22)

 def test_media_shortcode(self):
 self.client_only_api.media_shortcode('os1NQjxtvF')
 self.api.media_shortcode('os1NQjxtvF')

 def test_media_likes(self):
 self.client_only_api.media_likes(media_id=4)

 def test_like_media(self):
 self.api.like_media(media_id=4)
 self.api.unlike_media(media_id=4)

 """
 TEMP; disabled this test while we add
 a proper response to create_media_comment
 def test_comment_media(self):
 comment = self.api.create_media_comment(media_id=4, text='test')
 self.api.delete_comment(media_id=4, comment_id=comment.id)
 """

 def test_user_feed(self):
 self.api.user_media_feed(count=50)

 def test_generator_user_feed(self):
 generator = self.api.user_media_feed(as_generator=True, max_pages=3, count=2)
 for page in generator:
 str(generator)

 def test_generator_user_feed_all(self):
 generator = self.api.user_media_feed(as_generator=True, max_pages=None)
 for i in range(10):
 page = six.advance_iterator(generator)
 str(generator)

 generator = self.api.user_media_feed(as_generator=True, max_pages=0)
 for page in generator:
 assert False

 def test_user_liked_media(self):
 self.api.user_liked_media(count=10)

 def test_user_recent_media(self):
 media, url = self.api.user_recent_media(count=10)

 self.assertTrue( all( [hasattr(obj, 'type') for obj in media] ) )

 image = media[0]
 self.assertEqual(
 image.get_standard_resolution_url(),
 "http://distillery-dev.s3.amazonaws.com/media/2011/02/02/1ce5f3f490a640ca9068e6000c91adc5_7.jpg")

 self.assertEqual(
 image.get_low_resolution_url(),
 "http://distillery-dev.s3.amazonaws.com/media/2011/02/02/1ce5f3f490a640ca9068e6000c91adc5_6.jpg")

 self.assertEqual(
 image.get_thumbnail_url(),
 "http://distillery-dev.s3.amazonaws.com/media/2011/02/02/1ce5f3f490a640ca9068e6000c91adc5_5.jpg")

 self.assertEqual( False, hasattr(image, 'videos') )

 video = media[1]
 self.assertEqual(
 video.get_standard_resolution_url(),
 video.videos['standard_resolution'].url)

 self.assertEqual(
 video.get_standard_resolution_url(),
 "http://distilleryvesper9-13.ak.instagram.com/090d06dad9cd11e2aa0912313817975d_101.mp4")

 self.assertEqual(
 video.get_low_resolution_url(),
 "http://distilleryvesper9-13.ak.instagram.com/090d06dad9cd11e2aa0912313817975d_102.mp4")

 self.assertEqual(
 video.get_thumbnail_url(),
 "http://distilleryimage2.ak.instagram.com/11f75f1cd9cc11e2a0fd22000aa8039a_5.jpg")





 def test_user_search(self):
 self.api.user_search('mikeyk', 10)

 def test_user_follows(self):
 for page in self.api.user_followed_by(as_generator=True):
 str(page)

 def test_user_followed_by(self):
 for page in self.api.user_followed_by(as_generator=True):
 str(page)

 def test_other_user_followed_by(self):
 self.api.user_followed_by(user_id=3)

 def test_self_info(self):
 self.api.user()
 self.assertRaises(InstagramAPIError, self.client_only_api.user)

 def test_location_recent_media(self):
 self.api.location_recent_media(location_id=1)

 def test_location_search(self):
 self.api.location_search(lat=37.7,lng=-122.22, distance=2500)

 def test_location(self):
 self.api.location(1)

 def test_tag_recent_media(self):
 self.api.tag_recent_media(tag_name='1', count=5, max_tag_id='12345')

 def test_tag_recent_media_paginated(self):
 for page in self.api.tag_recent_media(tag_name='1', count=5, as_generator=True, max_pages=2):
 str(page)

 def test_tag_search(self):
 self.api.tag_search("coff")

 def test_tag(self):
 self.api.tag("coffee")

 def test_user_follows(self):
 self.api.user_follows()

 def test_user_followed_by(self):
 self.api.user_followed_by()

 def test_user_followed_by(self):
 self.api.user_followed_by()

 def test_user_requested_by(self):
 self.api.user_followed_by()

 def test_user_incoming_requests(self):
 self.api.user_incoming_requests()

 def test_change_relationship(self):
 self.api.change_user_relationship(user_id=10, action="follow")
 # test shortcuts as well
 self.api.follow_user(user_id='10')
 self.api.unfollow_user(user_id='10')

 def test_geography_recent_media(self):
 self.api.geography_recent_media(geography_id=1)

if __name__ == '__main__':
 if not TEST_AUTH:
 del InstagramAuthTests

 unittest.main()