healthchecks/hc/accounts/tests/test_badges.py

from hc.test import BaseTestCase
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(user=self.alice, tags="foo a-B_1  baz@",
                             project=self.project)
        Check.objects.create(user=self.bob, tags="bobs-tag",
                             project=self.bobs_project)

        r = self.client.get("/accounts/profile/badges/")
        self.assertContains(r, "foo.svg")
        self.assertContains(r, "a-B_1.svg")

        # Expect badge URLs only for tags that match \w+
        self.assertNotContains(r, "[email protected]")

        # Expect only Alice's tags
        self.assertNotContains(r, "bobs-tag.svg")