def test_one_page_followed_by_empty_page(self):
self.connection.expect._call_.withArguments(
self.ActionChecker("ListTables", {})).andReturn(
_lv.ListTablesResponse(TableNames=["A", "B", "C"],
LastEvaluatedTableName="D"))
self.connection.expect._call_.withArguments(
self.ActionChecker("ListTables",
{"ExclusiveStartTableName": "D"})).andReturn(
_lv.ListTablesResponse(TableNames=[]))
self.assertEqual(list(iterate_list_tables(self.connection.object)),
["A", "B", "C"])
def test_one_page(self):
self.connection.expect._call_.withArguments(
self.ActionChecker("ListTables", {})).andReturn(
_lv.ListTablesResponse(TableNames=["A", "B", "C"]))
self.assertEqual(list(iterate_list_tables(self.connection.object)),
["A", "B", "C"])
def test_several_pages(self):
self.connection.expect._call_.withArguments(
self.ActionChecker("ListTables", {})).andReturn(
_lv.ListTablesResponse(TableNames=["A", "B", "C"],
LastEvaluatedTableName="D"))
self.connection.expect._call_.withArguments(
self.ActionChecker(
"ListTables", {"ExclusiveStartTableName": "D"})).andReturn(
_lv.ListTablesResponse(TableNames=["E", "F", "G"],
LastEvaluatedTableName="H"))
self.connection.expect._call_.withArguments(
self.ActionChecker(
"ListTables", {"ExclusiveStartTableName": "H"})).andReturn(
_lv.ListTablesResponse(TableNames=["I", "J", "K"]))
self.assertEqual(list(iterate_list_tables(self.connection.object)),
["A", "B", "C", "E", "F", "G", "I", "J", "K"])