def testWithPagingAndSlicing(self):
responses = [
old_mock_api.MockRequest({
'kind': 'numbers',
'items': [1, 2, 3],
'nextPageToken': 'abc'
}),
old_mock_api.MockRequest({
'kind': 'numbers',
'items': [4, 5, 6]
})
]
# pylint: disable=unused-argument,redefined-builtin,g-bad-name
def MockFunc(project=None,
maxResults=None,
filter=None,
pageToken=None):
self._page += 1
return responses[self._page - 1]
self.assertEqual(utils.All(MockFunc, 'my-project', max_results=5), {
'kind': 'numbers',
'items': [1, 2, 3, 4, 5]
})
def MockFunc(project=None,
maxResults=None,
filter=None,
pageToken=None):
if project == 'project-1':
return old_mock_api.MockRequest({
'kind': 'numbers',
'items': {
'numbers': {
'items': [1, 2, 3]
}
}
})
elif project == 'project-2':
raise apiclient.errors.HttpError(
httplib2.Response({
'status': 404,
'reason': 'project not found'
}), 'content')
elif project == 'project-3':
return old_mock_api.MockRequest({
'kind': 'numbers',
'items': {
'numbers': {
'items': [4, 5, 6]
}
}
})
def MockFunc(project=None,
maxResults=None,
filter=None,
pageToken=None):
return old_mock_api.MockRequest({
'kind': 'numbers',
'items': [1, 2, 3]
})
def MockFunc(project=None,
maxResults=None,
filter=None,
pageToken=None,
zone=None):
self.assertEqual('some-zone', zone)
return old_mock_api.MockRequest({
'kind': 'numbers',
'items': [1, 2, 3]
})
def MockFunc(project=None,
maxResults=None,
filter=None,
pageToken=None):
self.assertEqual(project, 'my-project')
self.assertEqual(maxResults, 500)
self.assertEqual(filter, 'name eq my-instance')
self.assertEqual(pageToken, None)
return old_mock_api.MockRequest({
'kind': 'numbers',
'items': [1, 2, 3]
})
def MockFunc(project=None,
maxResults=None,
filter=None,
pageToken=None):
max_result_values.append(maxResults)
return old_mock_api.MockRequest({
'kind':
'numbers',
'items':
range(returned_counts[len(max_result_values) - 1]),
'nextPageToken':
'next'
})
def MockAggregatedFunc(project=None,
maxResults=None,
filter=None,
pageToken=None):
self.assertEqual(maxResults, 37)
return old_mock_api.MockRequest({
'items': {
'danger-zone': {
'numbers': [{
'kind': 'numbers',
'items': [1, 2, 3]
}]
}
}
})
def list(self, **unused_kwargs): return old_mock_api.MockRequest(disks)
def aggregatedList(self, **unused_kwargs): return old_mock_api.MockRequest(disks)
def MockFunc(project=None,
maxResults=None,
filter=None,
pageToken=None):
self.assertEquals(500, maxResults)
return old_mock_api.MockRequest({})
