Add tasks for activating access codes on failure

tests/test_tasks/test_create_missing_pindora_reservations.py

@@ -0,0 +1,265 @@
+from __future__ import annotations
+
+import datetime
+
+import pytest
+from freezegun import freeze_time
+
+from tilavarauspalvelu.enums import AccessType, ReservationStateChoice, ReservationTypeChoice
+from tilavarauspalvelu.integrations.keyless_entry import PindoraClient
+from tilavarauspalvelu.integrations.keyless_entry.exceptions import PindoraAPIError
+from tilavarauspalvelu.tasks import create_missing_pindora_reservations
+from utils.date_utils import DEFAULT_TIMEZONE, local_datetime
+
+from tests.factories import ReservationFactory
+from tests.helpers import patch_method
+
+
+@pytest.mark.django_db
+@freeze_time("2023-01-01")
+def test_create_missing_pindora_reservations__create_missing():
+    now = local_datetime()
+
+    reservation = ReservationFactory.create(
+        state=ReservationStateChoice.CONFIRMED,
+        type=ReservationTypeChoice.NORMAL,
+        access_type=AccessType.ACCESS_CODE,
+        access_code_generated_at=None,
+        access_code_is_active=False,
+        begin=now + datetime.timedelta(days=1),
+        end=now + datetime.timedelta(days=1, hours=1),
+    )
+
+    data = {
+        "access_code_generated_at": datetime.datetime(2023, 1, 1, tzinfo=DEFAULT_TIMEZONE),
+        "access_code_is_active": True,
+    }
+    with patch_method(PindoraClient.create_reservation, return_value=data) as patch:
+        create_missing_pindora_reservations()
+
+    assert patch.called is True
+    assert patch.call_args.kwargs["is_active"] is True
+
+    reservation.refresh_from_db()
+    assert reservation.access_code_generated_at == datetime.datetime(2023, 1, 1, tzinfo=DEFAULT_TIMEZONE)
+    assert reservation.access_code_is_active is True
+
+
+@pytest.mark.django_db
+@freeze_time("2023-01-01")
+def test_create_missing_pindora_reservations__blocked():
+    now = local_datetime()
+
+    reservation = ReservationFactory.create(
+        state=ReservationStateChoice.CONFIRMED,
+        type=ReservationTypeChoice.BLOCKED,
+        access_type=AccessType.ACCESS_CODE,
+        access_code_generated_at=None,
+        access_code_is_active=False,
+        begin=now + datetime.timedelta(days=1),
+        end=now + datetime.timedelta(days=1, hours=1),
+    )
+
+    data = {
+        "access_code_generated_at": datetime.datetime(2023, 1, 1, tzinfo=DEFAULT_TIMEZONE),
+        "access_code_is_active": False,
+    }
+    with patch_method(PindoraClient.create_reservation, return_value=data) as patch:
+        create_missing_pindora_reservations()
+
+    assert patch.called is True
+    assert patch.call_args.kwargs["is_active"] is False
+
+    reservation.refresh_from_db()
+    assert reservation.access_code_generated_at == datetime.datetime(2023, 1, 1, tzinfo=DEFAULT_TIMEZONE)
+    assert reservation.access_code_is_active is False
+
+
+@pytest.mark.django_db
+@freeze_time("2023-01-01")
+def test_create_missing_pindora_reservations__in_the_past():
+    now = local_datetime()
+
+    reservation = ReservationFactory.create(
+        state=ReservationStateChoice.CONFIRMED,
+        type=ReservationTypeChoice.NORMAL,
+        access_type=AccessType.ACCESS_CODE,
+        access_code_generated_at=None,
+        access_code_is_active=False,
+        begin=now - datetime.timedelta(days=1),
+        end=now - datetime.timedelta(days=1, hours=1),
+    )
+
+    with patch_method(PindoraClient.create_reservation) as patch:
+        create_missing_pindora_reservations()
+
+    assert patch.called is False
+
+    reservation.refresh_from_db()
+    assert reservation.access_code_generated_at is None
+    assert reservation.access_code_is_active is False
+
+
+@pytest.mark.django_db
+@freeze_time("2023-01-01")
+def test_create_missing_pindora_reservations__ongoing():
+    now = local_datetime()
+
+    reservation = ReservationFactory.create(
+        state=ReservationStateChoice.CONFIRMED,
+        type=ReservationTypeChoice.NORMAL,
+        access_type=AccessType.ACCESS_CODE,
+        access_code_generated_at=None,
+        access_code_is_active=False,
+        begin=now - datetime.timedelta(days=1),
+        end=now + datetime.timedelta(days=1, hours=1),
+    )
+
+    data = {
+        "access_code_generated_at": datetime.datetime(2023, 1, 1, tzinfo=DEFAULT_TIMEZONE),
+        "access_code_is_active": True,
+    }
+    with patch_method(PindoraClient.create_reservation, return_value=data) as patch:
+        create_missing_pindora_reservations()
+
+    assert patch.called is True
+    assert patch.call_args.kwargs["is_active"] is True
+
+    reservation.refresh_from_db()
+    assert reservation.access_code_generated_at == datetime.datetime(2023, 1, 1, tzinfo=DEFAULT_TIMEZONE)
+    assert reservation.access_code_is_active is True
+
+
+@pytest.mark.django_db
+@freeze_time("2023-01-01")
+def test_create_missing_pindora_reservations__not_confirmed():
+    now = local_datetime()
+
+    reservation = ReservationFactory.create(
+        state=ReservationStateChoice.WAITING_FOR_PAYMENT,
+        type=ReservationTypeChoice.NORMAL,
+        access_type=AccessType.ACCESS_CODE,
+        access_code_generated_at=None,
+        access_code_is_active=False,
+        begin=now + datetime.timedelta(days=1),
+        end=now + datetime.timedelta(days=1, hours=1),
+    )
+
+    with patch_method(PindoraClient.create_reservation) as patch:
+        create_missing_pindora_reservations()
+
+    assert patch.called is False
+
+    reservation.refresh_from_db()
+    assert reservation.access_code_generated_at is None
+    assert reservation.access_code_is_active is False
+
+
+@pytest.mark.django_db
+@freeze_time("2023-01-01")
+def test_create_missing_pindora_reservations__not_access_code():
+    now = local_datetime()
+
+    reservation = ReservationFactory.create(
+        state=ReservationStateChoice.CONFIRMED,
+        type=ReservationTypeChoice.NORMAL,
+        access_type=AccessType.UNRESTRICTED,
+        access_code_generated_at=None,
+        access_code_is_active=False,
+        begin=now + datetime.timedelta(days=1),
+        end=now + datetime.timedelta(days=1, hours=1),
+    )
+
+    with patch_method(PindoraClient.create_reservation) as patch:
+        create_missing_pindora_reservations()
+
+    assert patch.called is False
+
+    reservation.refresh_from_db()
+    assert reservation.access_code_generated_at is None
+    assert reservation.access_code_is_active is False
+
+
+@pytest.mark.django_db
+@freeze_time("2023-01-01")
+def test_create_missing_pindora_reservations__already_generated():
+    now = local_datetime()
+
+    reservation = ReservationFactory.create(
+        state=ReservationStateChoice.CONFIRMED,
+        type=ReservationTypeChoice.NORMAL,
+        access_type=AccessType.ACCESS_CODE,
+        access_code_generated_at=now,
+        access_code_is_active=False,
+        begin=now + datetime.timedelta(days=1),
+        end=now + datetime.timedelta(days=1, hours=1),
+    )
+
+    with patch_method(PindoraClient.create_reservation) as patch:
+        create_missing_pindora_reservations()
+
+    assert patch.called is False
+
+    reservation.refresh_from_db()
+    assert reservation.access_code_generated_at == now
+    assert reservation.access_code_is_active is False
+
+
+@pytest.mark.django_db
+@freeze_time("2023-01-01")
+def test_create_missing_pindora_reservations__multiple():
+    now = local_datetime()
+
+    ReservationFactory.create(
+        state=ReservationStateChoice.CONFIRMED,
+        type=ReservationTypeChoice.NORMAL,
+        access_type=AccessType.ACCESS_CODE,
+        access_code_generated_at=None,
+        access_code_is_active=False,
+        begin=now + datetime.timedelta(days=1),
+        end=now + datetime.timedelta(days=1, hours=1),
+    )
+
+    ReservationFactory.create(
+        state=ReservationStateChoice.CONFIRMED,
+        type=ReservationTypeChoice.NORMAL,
+        access_type=AccessType.ACCESS_CODE,
+        access_code_generated_at=None,
+        access_code_is_active=False,
+        begin=now + datetime.timedelta(days=1),
+        end=now + datetime.timedelta(days=1, hours=1),
+    )
+
+    data = {
+        "access_code_generated_at": datetime.datetime(2023, 1, 1, tzinfo=DEFAULT_TIMEZONE),
+        "access_code_is_active": False,
+    }
+    with patch_method(PindoraClient.create_reservation, return_value=data) as patch:
+        create_missing_pindora_reservations()
+
+    assert patch.call_count == 2
+
+
+@pytest.mark.django_db
+@freeze_time("2023-01-01")
+def test_create_missing_pindora_reservations__pindora_error():
+    now = local_datetime()
+
+    reservation = ReservationFactory.create(
+        state=ReservationStateChoice.CONFIRMED,
+        type=ReservationTypeChoice.NORMAL,
+        access_type=AccessType.ACCESS_CODE,
+        access_code_generated_at=None,
+        access_code_is_active=False,
+        begin=now + datetime.timedelta(days=1),
+        end=now + datetime.timedelta(days=1, hours=1),
+    )
+
+    with patch_method(PindoraClient.create_reservation, side_effect=PindoraAPIError()) as patch:
+        create_missing_pindora_reservations()
+
+    assert patch.called is True
+
+    reservation.refresh_from_db()
+    assert reservation.access_code_generated_at is None
+    assert reservation.access_code_is_active is False