From ee3646594b8fcb6e6d569b97a5e3e3fd1bd456d1 Mon Sep 17 00:00:00 2001 From: imperosol Date: Thu, 5 Jun 2025 16:25:08 +0200 Subject: [PATCH] fix promo logo older than promo 10 --- core/models.py | 4 ++-- core/tests/test_user.py | 7 +++++++ 2 files changed, 9 insertions(+), 2 deletions(-) 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()