diff --git a/tests/handlers/external_test.py b/tests/handlers/external_test.py index af8dd79c..004908a2 100644 --- a/tests/handlers/external_test.py +++ b/tests/handlers/external_test.py @@ -10,13 +10,17 @@ from fastapi import FastAPI from httpx import AsyncClient from lsst.ts.rubintv.background.currentpoller import CurrentPoller -from lsst.ts.rubintv.models.models import Location +from lsst.ts.rubintv.background.historicaldata import HistoricalPoller +from lsst.ts.rubintv.config import config +from lsst.ts.rubintv.models.models import Camera, Location, get_current_day_obs from lsst.ts.rubintv.models.models_helpers import find_first from lsst.ts.rubintv.models.models_init import ModelsInitiator from ..mockdata import RubinDataMocker m = ModelsInitiator() +app_name = config.name +day_obs = get_current_day_obs().isoformat() @pytest.mark.asyncio @@ -25,7 +29,7 @@ async def test_get_home( ) -> None: """Test that home page has links to every location""" client, app, mocker = mocked_client - response = await client.get("/rubintv/") + response = await client.get(f"/{app_name}/") html = await response.aread() parsed = BeautifulSoup(html, "html.parser") locations = m.locations @@ -51,7 +55,7 @@ async def test_get_location( groups = location.camera_groups.values() camera_names = list(chain(*groups)) - response = await client.get(f"/rubintv/{location_name}") + response = await client.get(f"/{app_name}/{location_name}") html = await response.aread() parsed = BeautifulSoup(html, "html.parser") @@ -78,11 +82,12 @@ async def test_current_channels( loc_cam = f"{location.name}/{camera.name}" for seq_chan in camera.seq_channels(): url = ( - f"/rubintv/{location.name}/{camera.name}" + f"/{app_name}/{location.name}/{camera.name}" f"/current/{seq_chan.name}" ) response = await client.get(url) - assert response.status_code == 200 + assert response.is_success + html = await response.aread() parsed = BeautifulSoup(html, "html.parser") if mocker.empty_channel[loc_cam] == seq_chan.name: @@ -91,3 +96,52 @@ async def test_current_channels( else: assert parsed.select(".event-info") assert not parsed.select(".event-error") + + +@pytest.mark.asyncio +async def test_all_endpoints( + mocked_client: tuple[AsyncClient, FastAPI, RubinDataMocker] +) -> None: + + client, app, mocker = mocked_client + + locations = [loc.name for loc in m.locations] + summit: Location = find_first(m.locations, "name", "summit-usdf") + assert isinstance(summit, Location) + + cameras = [f"{summit.name}/{cam.name}" for cam in summit.cameras if cam.online] + camera_historical = [f"{cam}/historical" for cam in cameras] + camera_dates = [f"{cam}/date/{day_obs}" for cam in cameras if cam] + night_reports = [f"{cam}/night_report" for cam in cameras] + night_reports = [f"{cam}/night_report/{day_obs}" for cam in cameras] + + auxtel: Camera = find_first(summit.cameras, "name", "auxtel") + assert isinstance(auxtel, Camera) + + channels = [ + f"{summit.name}/{auxtel.name}/current/{chan.name}" + for chan in auxtel.seq_channels() + ] + + page_rel_urls = [ + "admin", + *locations, + *cameras, + *camera_dates, + *camera_historical, + *night_reports, + *channels, + ] + + for url_frag in page_rel_urls: + url = f"/{app_name}/{url_frag}" + print(f"Looking for: {url}") + res = await client.get(url) + assert res.is_success + + hp: HistoricalPoller = app.state.historical + while hp._have_downloaded is False: + await asyncio.sleep(0.1) + + # make sure we've reached here! + assert True