79 lines
2.1 KiB
Python
79 lines
2.1 KiB
Python
#!/usr/bin/python2.4
|
|
#
|
|
# Copyright 2010 Google Inc. All Rights Reserved.
|
|
|
|
"""Discovery document tests
|
|
|
|
Unit tests for objects created from discovery documents.
|
|
"""
|
|
|
|
__author__ = 'jcgregorio@google.com (Joe Gregorio)'
|
|
|
|
from apiclient.discovery import build
|
|
import httplib2
|
|
import os
|
|
import unittest
|
|
|
|
DATA_DIR = os.path.join(os.path.dirname(__file__), 'data')
|
|
|
|
class HttpMock(object):
|
|
|
|
def __init__(self, filename, headers):
|
|
f = file(os.path.join(DATA_DIR, filename), 'r')
|
|
self.data = f.read()
|
|
f.close()
|
|
self.headers = headers
|
|
|
|
def request(self, uri, method="GET", body=None, headers=None, redirections=1, connection_type=None):
|
|
return httplib2.Response(self.headers), self.data
|
|
|
|
|
|
class Discovery(unittest.TestCase):
|
|
def test_method_error_checking(self):
|
|
self.http = HttpMock('buzz.json', {'status': '200'})
|
|
buzz = build('buzz', 'v1', self.http)
|
|
|
|
# Missing required parameters
|
|
try:
|
|
buzz.activities().list()
|
|
self.fail()
|
|
except TypeError, e:
|
|
self.assertTrue('Missing' in str(e))
|
|
|
|
# Parameter doesn't match regex
|
|
try:
|
|
buzz.activities().list(scope='@self', userId='')
|
|
self.fail()
|
|
except TypeError, e:
|
|
self.assertTrue('does not match' in str(e))
|
|
|
|
# Parameter doesn't match regex
|
|
try:
|
|
buzz.activities().list(scope='not@', userId='foo')
|
|
self.fail()
|
|
except TypeError, e:
|
|
self.assertTrue('does not match' in str(e))
|
|
|
|
# Unexpected parameter
|
|
try:
|
|
buzz.activities().list(flubber=12)
|
|
self.fail()
|
|
except TypeError, e:
|
|
self.assertTrue('unexpected' in str(e))
|
|
|
|
def test_buzz_resources(self):
|
|
self.http = HttpMock('buzz.json', {'status': '200'})
|
|
buzz = build('buzz', 'v1', self.http)
|
|
self.assertTrue(getattr(buzz, 'activities'))
|
|
self.assertTrue(getattr(buzz, 'search'))
|
|
self.assertTrue(getattr(buzz, 'feeds'))
|
|
self.assertTrue(getattr(buzz, 'photos'))
|
|
self.assertTrue(getattr(buzz, 'people'))
|
|
self.assertTrue(getattr(buzz, 'groups'))
|
|
self.assertTrue(getattr(buzz, 'comments'))
|
|
self.assertTrue(getattr(buzz, 'related'))
|
|
|
|
|
|
if __name__ == '__main__':
|
|
unittest.main()
|