test.py

# Sample code to use the Netflix python client
import unittest, os
import simplejson
from netflix import *

APP_NAME   = ''
API_KEY    = ''
API_SECRET = ''
CALLBACK   = ''

MOVIE_TITLE = "Foo Fighters"

EXAMPLE_USER = {
        'request': {
                'key': '',
                'secret': ''
        },
        'access': {
                'key': '',
                'secret': ''
        }
}

EMPTY_USER = {
        'request': {
                'key': '',
                'secret': ''
        },
        'access': {
                'key': '',
                'secret': ''
        }
}


class TestQuery(unittest.TestCase):
	def test_base(self):
		netflixClient = NetflixClient(APP_NAME, API_KEY, API_SECRET, CALLBACK)
		
#	def test_token_functions(self):
#		netflixClient = NetflixClient(APP_NAME, API_KEY, API_SECRET, CALLBACK)
#		netflixUser = NetflixUser(EMPTY_USER,netflixClient)
		# I'd love to test the token functions, but unfortunately running these
		# invalidates the existing tokens.  Foo.

		
	def test_catalog_functions(self):
		netflixClient = NetflixClient(APP_NAME, API_KEY, API_SECRET, CALLBACK)
		data = netflixClient.catalog.autocomplete('Foo')
		for info in data:
			assert re.search('Foo',info)
			
		data = netflixClient.catalog.search_titles(MOVIE_TITLE)
		assert isinstance(data[0].title,str)
		
        # API reads that it returns "all titles in the catalog"
        # meaing every movie that Netflix has?  Let's not call 
        # this every time then
#		movie = netflixClient.catalog.index()
#		assert isinstance(movie['catalog_title']['title']['regular'],str)
		
		people = netflixClient.catalog.search_people('Flip Wilson',maxResults=1)
		assert isinstance(people[0],NetflixPerson)
		
	# DISC TESTS
	def test_disc_functions(self):
		netflixClient = NetflixClient(APP_NAME, API_KEY, API_SECRET, CALLBACK)
		titles = netflixClient.catalog.search_titles('Cocoon',1,2)
		formats = titles[0].formats()
		assert isinstance(formats,list)
		synopsis = titles[0].synopsis
		assert isinstance(synopsis,str)
				
	def test_user_functions(self):
		netflixClient = NetflixClient(APP_NAME, API_KEY, API_SECRET, CALLBACK)
		netflixUser = NetflixUser(EXAMPLE_USER,netflixClient)
		username = netflixUser.name
		assert isinstance(username,str)
		data = netflixClient.catalog.search_titles('Cocoon',1,2)
		ratings = netflixUser.getRatings(data)
		history = netflixUser.getRentalHistory('shipped',updatedMin=1219775019,maxResults=4)
		# 45 is how many times it had been rented when I wrote the test
		assert int(history['rental_history']['number_of_results']) >= 45 
		
		# yeah, these fail with Unauthorized, imagine that
		queue = NetflixUserQueue(netflixUser)
		response = queue.addTitle( urls=["http://api.netflix.com/catalog/titles/movies/60010242"] )
		response = queue.addTitle( urls=["http://api.netflix.com/catalog/titles/movies/60020242"], position=1)
		response = queue.removeTitle( id="60010242")

		discAvailable = queue.getAvailable('disc')
		instantAvailable =  queue.getAvailable('instant')
		discSaved =  queue.getSaved('disc')
		instantSaved = queue.getSaved('instant')
		

if __name__ == '__main__':
    EXAMPLE_USER = simplejson.load(open("user.json"))
    appinfo = simplejson.load(open("app.json"))
    APP_NAME   = appinfo["app_name"]
    API_KEY    = appinfo["api_key"]
    API_SECRET = appinfo["api_secret"]

    unittest.main()