diff --git a/core/models.py b/core/models.py index ce4b2102..b71f5408 100644 --- a/core/models.py +++ b/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 diff --git a/core/tests/test_user.py b/core/tests/test_user.py index e19c98f4..5d4cb734 100644 --- a/core/tests/test_user.py +++ b/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()