From f23898df9c35b93fea91aaa67c40be3455d902ca Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Micha=C5=82=20Pasternak?= Date: Mon, 1 Feb 2021 12:49:05 +0100 Subject: [PATCH] Fix tests --- src/bpp/tests/test_selenium/test_raporty.py | 14 +++++++++++--- 1 file changed, 11 insertions(+), 3 deletions(-) diff --git a/src/bpp/tests/test_selenium/test_raporty.py b/src/bpp/tests/test_selenium/test_raporty.py index 2f154d78b..fd5efcea7 100644 --- 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)