fix: ensure /github/branches endpoint returns all branches

api/pyproject.toml

@@ -93,6 +93,7 @@ responses = "*"
 pytest = "*"
 pytest-asyncio = "*"
 pytest-cov = "*"
+pytest-mock = "*"
 black = "*"
 check-manifest = "*"
 isort = "*"

api/src/shipit_api/admin/github.py

@@ -114,27 +114,46 @@ def ref_to_commit(owner, repo, ref):
 
 
 def list_github_branches(owner, repo, limit=100):
-    query = """
-    {
-      repository(name: "%(repo)s", owner: "%(owner)s") {
-        refs(first: %(limit)s, refPrefix: "refs/heads/") {
-          nodes {
-            name
-            target {
-              ... on Commit {
-                committedDate
+    page = {
+        "hasNextPage": True,
+        "endCursor": None,
+    }
+
+    branches = []
+    while page["hasNextPage"]:
+        after = ""
+        if page["endCursor"]:
+            after = f", after: \"{page['endCursor']}\""
+
+        query = """
+        {
+          repository(name: "%(repo)s", owner: "%(owner)s") {
+            refs(first: %(limit)s%(after)s, refPrefix: "refs/heads/") {
+              nodes {
+                name
+                target {
+                  ... on Commit {
+                    committedDate
+                  }
+                }
+              }
+              pageInfo {
+                endCursor
+                hasNextPage
               }
             }
           }
         }
-      }
-    }
-    """ % dict(
-        owner=owner, repo=repo, limit=limit
-    )
-    content = query_api(query)
-    nodes = content["data"]["repository"]["refs"]["nodes"]
-    return [{"committer_date": node["target"]["committedDate"], "name": node["name"]} for node in nodes]
+        """ % dict(
+            owner=owner, repo=repo, limit=limit, after=after
+        )
+        content = query_api(query)
+        page = content["data"]["repository"]["refs"]["pageInfo"]
+
+        nodes = content["data"]["repository"]["refs"]["nodes"]
+        branches.extend([{"committer_date": node["target"]["committedDate"], "name": node["name"]} for node in nodes])
+
+    return branches
 
 
 def list_github_commits(owner, repo, branch, limit=10):

api/tests/conftest.py

@@ -6,11 +6,18 @@
 import os
 
 import pytest
+import responses as rsps
 
 import backend_common.testing
 from shipit_api.common.models import XPI, XPIRelease
 
 
+@pytest.fixture
+def responses():
+    with rsps.RequestsMock() as req_mock:
+        yield req_mock
+
+
 @pytest.fixture(scope="session")
 def app():
     """Load shipit_api in test mode"""

api/tests/test_github.py

@@ -0,0 +1,52 @@
+from pprint import pprint
+
+import pytest
+from flask import Flask
+
+from shipit_api.admin import github
+
+
+@pytest.fixture(autouse=True)
+def setup(monkeypatch, mocker):
+    app = Flask(__name__)
+    app.config["GITHUB_TOKEN"] = "token"
+    with app.app_context():
+        monkeypatch.setattr(github, "current_app", app)
+        mocker.patch("shipit_api.admin.github._require_auth", return_value=None)
+        yield
+
+
+def test_list_github_branches(responses):
+    rsp1 = responses.post(
+        "https://api.github.com/graphql",
+        json={
+            "data": {
+                "repository": {
+                    "refs": {
+                        "nodes": [{"name": "foo", "target": {"committedDate": "1"}}, {"name": "bar", "target": {"committedDate": "2"}}],
+                        "pageInfo": {"hasNextPage": True, "endCursor": "abc"},
+                    }
+                }
+            }
+        },
+    )
+    rsp2 = responses.post(
+        "https://api.github.com/graphql",
+        json={
+            "data": {
+                "repository": {
+                    "refs": {
+                        "nodes": [{"name": "baz", "target": {"committedDate": "3"}}],
+                        "pageInfo": {"hasNextPage": False, "endCursor": "def"},
+                    }
+                }
+            }
+        },
+    )
+    repos = github.list_github_branches("org", "repo", limit=2)
+    assert b"after" not in rsp1.calls[0].request.body
+    assert b'after: \\"abc\\",' in rsp2.calls[0].request.body
+
+    print("Dumping for copy/paste:")
+    pprint(repos)
+    assert repos == [{"committer_date": "1", "name": "foo"}, {"committer_date": "2", "name": "bar"}, {"committer_date": "3", "name": "baz"}]