def test_search_documents(self):
response = {'items': [{'id': 'abcdef'}, {'id': '1234'}]}
requestBuilder = RequestMockBuilder({
'drive.files.list': (None, json.dumps(response)),
})
GoogleDriveHelper.get_docs_client = Mock()
GoogleDriveHelper.get_docs_client.return_value = apiclient.discovery.build(
"drive", "v2", requestBuilder=requestBuilder)
creator = User.objects.create_user('test-user')
Training.objects.create(resource_id='resource_id_test',
title='toto',
creator=creator,
is_displayed=True,
is_deleted=False)
Training.objects.create(resource_id='resource_id_test2',
title='tata',
creator=creator,
is_displayed=True,
is_deleted=False)
search_results = GoogleDriveHelper.search_documents('toto')
self.assertGreater(len(search_results), 0)
self.assertIn('abcdef', search_results)
self.assertIn('1234', search_results)
self.assertIn('resource_id_test', search_results)
self.assertNotIn('resource_id_test2', search_results)
def test_simple_response(self):
requestBuilder = RequestMockBuilder({
'plus.activities.get': (None, '{"data": {"foo": "bar"}}')
})
plus = build('plus', 'v1', http=self.http, requestBuilder=requestBuilder)
activity = plus.activities().get(activityId='tag:blah').execute()
self.assertEqual({"foo": "bar"}, activity)
def test_default_response(self):
requestBuilder = RequestMockBuilder({})
plus = build('plus',
'v1',
http=self.http,
requestBuilder=requestBuilder)
activity = plus.activities().get(activityId='tag:blah').execute()
self.assertEqual({}, activity)
def test_default_response(self):
requestBuilder = RequestMockBuilder({})
buzz = build('buzz',
'v1',
http=self.http,
requestBuilder=requestBuilder)
activity = buzz.activities().get(postId='tag:blah',
userId='@me').execute()
self.assertEqual({}, activity)
def test_unexpected_call(self):
requestBuilder = RequestMockBuilder({}, check_unexpected=True)
plus = build('plus', 'v1', http=self.http, requestBuilder=requestBuilder)
try:
plus.activities().get(activityId='tag:blah').execute()
self.fail('UnexpectedMethodError should have been raised')
except UnexpectedMethodError:
pass
def test_simple_matching_dict_body(self):
requestBuilder = RequestMockBuilder({
'zoo.animals.insert': (None, '{"data": {"foo": "bar"}}',
{'data': {'foo': 'bar'}})
})
zoo = build('zoo', 'v1', http=self.zoo_http, requestBuilder=requestBuilder)
activity = zoo.animals().insert(
body={'data': {'foo': 'bar'}}).execute()
self.assertEqual({'foo': 'bar'}, activity)
def test_simple_expected_body(self):
requestBuilder = RequestMockBuilder({
'zoo.animals.insert': (None, '{"data": {"foo": "bar"}}', '{}')
})
zoo = build('zoo', 'v1', http=self.zoo_http, requestBuilder=requestBuilder)
try:
zoo.animals().insert(body='').execute()
self.fail('UnexpectedBodyError should have been raised')
except UnexpectedBodyError:
pass
def test_simple_response(self):
requestBuilder = RequestMockBuilder(
{'chili.activities.get': (None, '{"data": {"foo": "bar"}}')})
buzz = build('buzz',
'v1',
http=self.http,
requestBuilder=requestBuilder)
activity = buzz.activities().get(postId='tag:blah',
userId='@me').execute()
self.assertEqual({"foo": "bar"}, activity)
def setUp(self):
super(ActivityViewTest, self).setUp()
self.logger = LoggingHelper.getDebugLogger()
response = {'items': [{'id': 'abcdef'}, {'id': '1234'}]}
requestBuilder = RequestMockBuilder({
'drive.files.list': (None, json.dumps(response)),
})
GoogleDriveHelper.get_docs_client = Mock()
GoogleDriveHelper.get_docs_client.return_value = apiclient.discovery.build(
"drive", "v2", requestBuilder=requestBuilder)
def test_errors(self):
errorResponse = httplib2.Response({'status': 500, 'reason': 'Server Error'})
requestBuilder = RequestMockBuilder({
'plus.activities.list': (errorResponse, '{}')
})
plus = build('plus', 'v1', http=self.http, requestBuilder=requestBuilder)
try:
activity = plus.activities().list(collection='public', userId='me').execute()
self.fail('An exception should have been thrown')
except HttpError, e:
self.assertEqual('{}', e.content)
self.assertEqual(500, e.resp.status)
self.assertEqual('Server Error', e.resp.reason)
def test_create_unistar_folder(self):
response = {'id': 'abcdef'}
requestBuilder = RequestMockBuilder({
'drive.files.insert': (None, json.dumps(response)),
})
GoogleDriveHelper.get_docs_client = Mock()
GoogleDriveHelper.get_docs_client.return_value = apiclient.discovery.build(
"drive", "v2", requestBuilder=requestBuilder)
user = User.objects.create_user('arnaud.breton', '*****@*****.**')
self.assertEqual(GoogleDriveHelper.create_unistar_folder(user),
'abcdef')
def setUpClass(cls):
# The apiclient library has support for mocking requests for
# testing, but it doesn't extend to the discovery document
# itself. For now, bring up an API server that will serve
# a discovery document.
# FIXME: Figure out a better way to stub this out.
run_test_server.run()
mock_responses = {
'arvados.humans.delete': (
fake_httplib2_response(500, **cls.ERROR_HEADERS),
""),
'arvados.humans.get': cls.api_error_response(
422, "Bad UUID format", "Bad output format"),
'arvados.humans.list': (None, json.dumps(
{'items_available': 0, 'items': []})),
}
req_builder = RequestMockBuilder(mock_responses)
cls.api = arvados.api('v1', cache=False,
host=os.environ['ARVADOS_API_HOST'],
token='discovery-doc-only-no-token-needed',
insecure=True,
requestBuilder=req_builder)
