mirror of
https://github.com/ae-utbm/sith.git
synced 2025-01-09 16:41:12 +00:00
55 lines
1.7 KiB
Python
55 lines
1.7 KiB
Python
|
from dataclasses import dataclass
|
||
|
|
from pathlib import Path
|
||
|
|
from unittest.mock import MagicMock, patch
|
||
|
|
|
||
|
|
import pytest
|
||
|
|
from django.conf import settings
|
||
|
|
from django.test.client import Client
|
||
|
|
from django.urls import reverse
|
||
|
|
|
||
|
|
from com.calendar import IcsCalendar
|
||
|
|
|
||
|
|
|
||
|
|
@dataclass
|
||
|
|
class MockResponse:
|
||
|
|
status: int
|
||
|
|
value: str
|
||
|
|
|
||
|
|
@property
|
||
|
|
def data(self):
|
||
|
|
return self.value.encode("utf8")
|
||
|
|
|
||
|
|
|
||
|
|
@pytest.mark.django_db
|
||
|
|
class TestExternalCalendar:
|
||
|
|
@pytest.fixture
|
||
|
|
def mock_request(self):
|
||
|
|
request = MagicMock()
|
||
|
|
with patch("urllib3.request", request):
|
||
|
|
yield request
|
||
|
|
|
||
|
|
@pytest.fixture(autouse=True)
|
||
|
|
def clear_cache(self):
|
||
|
|
IcsCalendar._EXTERNAL_CALENDAR.unlink(missing_ok=True)
|
||
|
|
|
||
|
|
@pytest.mark.parametrize("error_code", [403, 404, 500])
|
||
|
|
def test_fetch_error(
|
||
|
|
self, client: Client, mock_request: MagicMock, error_code: int
|
||
|
|
):
|
||
|
|
mock_request.return_value = MockResponse(error_code, "not allowed")
|
||
|
|
assert client.get(reverse("api:calendar_external")).status_code == 404
|
||
|
|
|
||
|
|
def test_fetch_success(self, client: Client, mock_request: MagicMock):
|
||
|
|
external_response = MockResponse(200, "Definitely an ICS")
|
||
|
|
mock_request.return_value = external_response
|
||
|
|
response = client.get(reverse("api:calendar_external"))
|
||
|
|
assert response.status_code == 200
|
||
|
|
redirect = Path(response.headers.get("X-Accel-Redirect", ""))
|
||
|
|
assert redirect.is_relative_to(Path("/") / settings.MEDIA_ROOT.stem)
|
||
|
|
out_file = settings.MEDIA_ROOT / redirect.relative_to(
|
||
|
|
Path("/") / settings.MEDIA_ROOT.stem
|
||
|
|
)
|
||
|
|
assert out_file.exists()
|
||
|
|
with open(out_file, "r") as f:
|
||
|
|
assert f.read() == external_response.value
|