def testYieldFromListPartial(self):
self.mocked_client.column.List.Expect(
fusiontables.FusiontablesColumnListRequest(
maxResults=100,
pageToken=None,
tableId='mytable',
),
fusiontables.ColumnList(
items=[
fusiontables.Column(name='c0'),
fusiontables.Column(name='c1'),
fusiontables.Column(name='c2'),
fusiontables.Column(name='c3'),
],
nextPageToken='x',
))
self.mocked_client.column.List.Expect(
fusiontables.FusiontablesColumnListRequest(
maxResults=100,
pageToken='x',
tableId='mytable',
),
fusiontables.ColumnList(items=[
fusiontables.Column(name='c4'),
fusiontables.Column(name='c5'),
fusiontables.Column(name='c6'),
fusiontables.Column(name='c7'),
], ))
client = fusiontables.FusiontablesV1(get_credentials=False)
request = fusiontables.FusiontablesColumnListRequest(tableId='mytable')
results = list_pager.YieldFromList(client.column, request, limit=6)
self._AssertInstanceSequence(results, 6)
def testYieldFromListWithPredicate(self):
self.mocked_client.column.List.Expect(
fusiontables.FusiontablesColumnListRequest(
maxResults=100,
pageToken=None,
tableId='mytable',
),
fusiontables.ColumnList(
items=[
fusiontables.Column(name='c0'),
fusiontables.Column(name='bad0'),
fusiontables.Column(name='c1'),
fusiontables.Column(name='bad1'),
],
nextPageToken='x',
))
self.mocked_client.column.List.Expect(
fusiontables.FusiontablesColumnListRequest(
maxResults=100,
pageToken='x',
tableId='mytable',
),
fusiontables.ColumnList(items=[
fusiontables.Column(name='c2'),
], ))
client = fusiontables.FusiontablesV1(get_credentials=False)
request = fusiontables.FusiontablesColumnListRequest(tableId='mytable')
results = list_pager.YieldFromList(client.column,
request,
predicate=lambda x: 'c' in x.name)
self._AssertInstanceSequence(results, 3)
def testYieldFromListEmpty(self):
self.mocked_client.column.List.Expect(
fusiontables.FusiontablesColumnListRequest(
maxResults=100,
pageToken=None,
tableId='mytable',
), fusiontables.ColumnList())
client = fusiontables.FusiontablesV1(get_credentials=False)
request = fusiontables.FusiontablesColumnListRequest(tableId='mytable')
results = list_pager.YieldFromList(client.column, request, limit=6)
self._AssertInstanceSequence(results, 0)
def testYieldNoRecords(self):
client = fusiontables.FusiontablesV1(get_credentials=False)
request = fusiontables.FusiontablesColumnListRequest(tableId='mytable')
results = list_pager.YieldFromList(client.column, request, limit=False)
self.assertEqual(0, len(list(results)))
