Use Project.badge_key for generating badge URLs

hc/accounts/tests/test_badges.py

@@ -5,10 +5,10 @@ from hc.api.models import Check
 class BadgesTestCase(BaseTestCase):
 
     def test_it_shows_badges(self):
-        self.client.login(username="[email protected]", password="password")
         Check.objects.create(project=self.project, tags="foo a-B_1  baz@")
         Check.objects.create(project=self.bobs_project, tags="bobs-tag")
 
+        self.client.login(username="[email protected]", password="password")
         r = self.client.get("/accounts/profile/badges/")
         self.assertContains(r, "foo.svg")
         self.assertContains(r, "a-B_1.svg")
@@ -18,3 +18,16 @@ class BadgesTestCase(BaseTestCase):
 
         # Expect only Alice's tags
         self.assertNotContains(r, "bobs-tag.svg")
+
+    def test_it_uses_badge_key(self):
+        Check.objects.create(project=self.project, tags="foo bar")
+        Check.objects.create(project=self.bobs_project, tags="bobs-tag")
+
+        self.project.badge_key = "alices-badge-key"
+        self.project.save()
+
+        self.client.login(username="[email protected]", password="password")
+        r = self.client.get("/accounts/profile/badges/")
+        print(r.content)
+        self.assertContains(r, "badge/alices-badge-key/")
+        self.assertContains(r, "badge/alices-badge-key/")

hc/accounts/views.py

@@ -366,14 +366,13 @@ def badges(request):
         sorted_tags.append("*")  # For the "overall status" badge
 
         urls = []
-        username = project.owner.username
         for tag in sorted_tags:
             if not re.match("^[\w-]+$", tag) and tag != "*":
                 continue
 
             urls.append({
-                "svg": get_badge_url(username, tag),
-                "json": get_badge_url(username, tag, format="json"),
+                "svg": get_badge_url(project.badge_key, tag),
+                "json": get_badge_url(project.badge_key, tag, format="json"),
             })
 
         badge_sets.append({"project": project, "urls": urls})