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"])