Fix pagination of public repos, make more efficient and add test

test/test_api_usage.py

@@ -24,7 +24,7 @@ from app import app, config_provider
 from buildtrigger.basehandler import BuildTriggerHandler
 from initdb import setup_database_for_testing, finished_database_for_testing
 from data import database, model
-from data.database import RepositoryActionCount
+from data.database import RepositoryActionCount, Repository as RepositoryTable
 from test.helpers import assert_action_logged
 
 from endpoints.api.team import TeamMember, TeamMemberList, TeamMemberInvite, OrganizationTeam
@@ -1453,19 +1453,38 @@ class TestListRepos(ApiTestCase):
     self.assertEquals(len(json['repositories']), 1)
 
   def test_listrepos_asguest_withpages(self):
-    # Add public repos until we have enough for over 1 page.
     public_user = model.user.get_user('public')
-    for i in range(0, REPOS_PER_PAGE):
-      model.repository.create_repository('public', 'publicrepo%s' % i, public_user,
-                                         visibility='public')
-    # Request the first page of results.
-    json = self.getJsonResponse(RepositoryList, params=dict(public=True))
-    self.assertEquals(len(json['repositories']), REPOS_PER_PAGE)
 
-    # Request the second page of results.
-    json = self.getJsonResponse(RepositoryList, params=dict(public=True,
-                                                            next_page=json['next_page']))
-    self.assertGreater(len(json['repositories']), 0)
+    # Delete all existing repos under the namespace.
+    for repo in list(RepositoryTable.select().where(RepositoryTable.namespace_user == public_user)):
+      model.repository.purge_repository(public_user.username, repo.name)
+
+    # Add public repos until we have enough for a few pages.
+    required = set()
+    for i in range(0, REPOS_PER_PAGE * 2):
+      name = 'publicrepo%s' % i
+      model.repository.create_repository('public', name, public_user,
+                                         visibility='public')
+      required.add(name)
+
+    # Request results until we no longer have any.
+    next_page = None
+    while True:
+      json = self.getJsonResponse(RepositoryList, params=dict(public=True, next_page=next_page))
+      for repo in json['repositories']:
+        name = repo['name']
+        self.assertTrue(name in required)
+        required.remove(name)
+
+      if 'next_page' in json:
+        self.assertEquals(len(json['repositories']), REPOS_PER_PAGE)
+      else:
+        break
+
+      next_page = json['next_page']
+
+    # Ensure we found all the repositories.
+    self.assertEquals(0, len(required))
 
   def test_listrepos_asorgmember(self):
     self.login(READ_ACCESS_USER)