Fix tests

iplweb · Feb 1, 2021 · f23898d · f23898d
1 parent a860d20
commit f23898d
Showing 1 changed file with 11 additions and 3 deletions.
diff --git a/src/bpp/tests/test_selenium/test_raporty.py b/src/bpp/tests/test_selenium/test_raporty.py
@@ -6,7 +6,6 @@
     from django.urls import reverse
 
 import pytest
-from celeryui.models import Report
 
 from bpp.tests.util import (
     CURRENT_YEAR,
@@ -15,7 +14,7 @@
     any_jednostka,
     select_select2_autocomplete,
 )
-
+from celeryui.models import Report
 from django_bpp.selenium_util import wait_for, wait_for_page_load
 
 
@@ -64,7 +63,16 @@ def test_ranking_autorow(raporty_browser, jednostka_raportow, asgi_live_server):
     raporty_browser.visit(
         asgi_live_server.url + reverse("bpp:ranking_autorow_formularz")
     )
-    assert 'value="%s"' % (CURRENT_YEAR - 1) in raporty_browser.html
+    from django.utils import timezone
+
+    now = timezone.now()
+
+    if now.month < 2:
+        val = CURRENT_YEAR - 1
+    else:
+        val = CURRENT_YEAR
+
+    assert 'value="%s"' % val in raporty_browser.html
 
 
 @pytest.mark.django_db(transaction=True)