Merge pull request #1110 from ae-utbm/fix-old-promo

fix promo logo older than promo 10

core/models.py

@@ -341,8 +341,8 @@ class User(AbstractUser):
         return reverse("core:user_profile", kwargs={"user_id": self.pk})
 
     def promo_has_logo(self) -> bool:
-        return Path(
-            settings.BASE_DIR / f"core/static/core/img/promo_{self.promo}.png"
+        return (
+            settings.BASE_DIR / f"core/static/core/img/promo_{self.promo:02d}.png"
         ).exists()
 
     @cached_property

core/tests/test_user.py

@@ -335,3 +335,10 @@ class TestRedirectMe:
     def test_anonymous_user(self, client: Client):
         url = reverse("core:user_me_redirect")
         assertRedirects(client.get(url), reverse("core:login", query={"next": url}))
+
+
+@pytest.mark.parametrize("promo", [7, 22])
+@pytest.mark.django_db
+def test_promo_has_logo(promo):
+    user = baker.make(User, promo=promo)
+    assert user.promo_has_logo()