diff --git a/integration-tests/testfile.py b/integration-tests/testfile.py deleted file mode 100755 index a9b37264..00000000 --- a/integration-tests/testfile.py +++ /dev/null @@ -1,214 +0,0 @@ -#! /usr/bin/env python3 -"""Start a server subprocess for integration tests. - -These tests are separated from unit tests to avoid inflating test coverage. -For now, this file should be run directly rather than through a test framework. - -They are designed to run on CI and should work on Linux or WSL for local debugging. -""" - -import os -import shutil -import subprocess -from time import sleep, time -from typing import Optional - -from PIL import Image - -import labthings_fastapi as lt - -THIS_DIR: str = os.path.dirname(os.path.realpath(__file__)) -WORKING_DIR: str = os.path.join(THIS_DIR, "working_dir") -REPO_DIR: str = os.path.dirname(THIS_DIR) -CONFIG_FILE: str = os.path.join(REPO_DIR, "ofm_config_simulation.json") -SERVER_CMD: list[str] = [ - "openflexure-microscope-server", - "--fallback", - "-c", - CONFIG_FILE, -] - - -def main() -> None: - """Set up the server, run basic checks, shutdown, check for graceful exit. - - The basic checks include checks that: - - The server boots - - The server reports the expected IP and port - - A ThingClient can connect to the camera - - The client grab a frame from the stream, and it is the expected size - - A background process can subscribe to the stream - """ - set_up_working_dir() - os.chdir(WORKING_DIR) - - server_process: Optional[subprocess.Popen] = None - subscriber_process: Optional[subprocess.Popen] = None - try: - print("Starting server") - server_process = start_server() - error_if_server_not_started(server_process, timeout=15) - - test_client_connection() - - # This also sleeps for 2s and checks it is still connected - subscriber_process = subscribe_to_mjpeg_stream() - - server_process.terminate() - sleep(3) - - check_for_graceful_shutdown(server_process) - - finally: - # Ensure the server and subscriber processes are really dead - if subscriber_process is not None and subscriber_process.poll() is None: - subscriber_process.kill() - if server_process is not None and server_process.poll() is None: - server_process.kill() - raise RuntimeError("Server needed killing at end of test.") - - -def test_client_connection() -> None: - """Check a ThingClient can interact with the simulation microscope camera.""" - print("Connecting Python client to microscope, and capturing image") - cam_client = lt.ThingClient.from_url("http://localhost:5000/camera/") - img = Image.open(cam_client.grab_jpeg().open()) - print(f"Successfully grabbed image of size {img.size}") - assert img.size == (820, 616) - print("Successfully grabbed image from camera mjpeg stream") - - -def subscribe_to_mjpeg_stream() -> subprocess.Popen: - """Start a background process subscribed to the mjpeg stream. - - :returns: The Popen object for the ongoing process. - - :raises: RuntimeError if the stream is not still connected after 2s - """ - # Use nohup to stop process hanging up unexpectedly. Explicitly forward stream - # to /dev/null to keep curl connected. - - curl_command = [ - "nohup", - "curl", - "-s", - "http://localhost:5000/camera/mjpeg_stream", - ">", - "/dev/null", - "2>&1", - ] - - process = subprocess.Popen( - curl_command, - stdout=subprocess.PIPE, - stderr=subprocess.STDOUT, - bufsize=1, - universal_newlines=True, - ) - os.set_blocking(process.stdout.fileno(), False) - - sleep(2) - if process.poll() is not None: - raise RuntimeError( - "MJPEG Subscriber process is not running. This likely means it could not" - " stay connected to the stream.\n" - ) - return process - - -def set_up_working_dir() -> None: - """If working dir exists, delete it and make a new one.""" - if os.path.exists(WORKING_DIR): - shutil.rmtree(WORKING_DIR) - os.makedirs(WORKING_DIR) - - -def start_server() -> subprocess.Popen: - """Start the server in a subprocess. - - The server is started in a subprocess and all outputs are buffered. - - :returns: Popen object for the ongoing process - """ - process = subprocess.Popen( - SERVER_CMD, - stdout=subprocess.PIPE, - stderr=subprocess.STDOUT, - bufsize=1, - universal_newlines=True, - ) - os.set_blocking(process.stdout.fileno(), False) - return process - - -def error_if_server_not_started( - server_process: subprocess.Popen, timeout: float = 15.0 -) -> None: - """Check the server started up as expected. - - Check the subprocess is running - Check the logs have no errors - Read the Logs to check uvicorn is running on http://127.0.0.1:5000 - - :param server_process: The Popen object for the the server process. - - :raises RuntimeError: If the server is not running as expected. - """ - confirmed_uvicorn_is_running = False - - t_start = time() - while not confirmed_uvicorn_is_running and time() - t_start < timeout: - sleep(0.2) - if stdout := read_process_buffers(server_process): - print("".join(stdout), end="") - if server_process.poll() is not None: - raise RuntimeError("Server process is not running!") - - for line in stdout: - if line.startswith("ERROR:"): - raise RuntimeError(f"Server started up with error\n{line}") - if "Uvicorn running on http://127.0.0.1:5000" in line: - confirmed_uvicorn_is_running = True - - if not confirmed_uvicorn_is_running: - raise RuntimeError("Cannot confirm Uvicorn is running on http://127.0.0.1:5000") - - # If we reached here Everything is fine! - print("Server is running as expected\n\n") - - -def check_for_graceful_shutdown(server_process: subprocess.Popen) -> None: - """Check the server shutdown gracefully. - - Check the subprocess is not running - Check the logs have no errors - Read the Logs to check the shutdown was graceful, rather than killed on Uvicorn - timeout - - :param server_process: The Popen object for the the server process. - - :raises RuntimeError: If the server didn't shutdown gracefully. - """ - stdout = read_process_buffers(server_process) - print("".join(stdout)) - print("\n\n") - for line in stdout: - if line.startswith("ERROR:"): - raise RuntimeError(f"Server encountered error\n{line}") - if "graceful shutdown exceeded" in line: - # This should be logged as an ERROR. This check is belts and braces. - raise RuntimeError("Server failed to shutdown gracefully.") - - -def read_process_buffers(process: subprocess.Popen) -> list[str]: - """Return STDOUT from a process.""" - stdout = [] - - while line := process.stdout.readline(): - stdout.append(line) - - return stdout - - -if __name__ == "__main__": - main() diff --git a/picamera_tests.py b/picamera_tests.py index 29097b60..8f4c4cdd 100755 --- a/picamera_tests.py +++ b/picamera_tests.py @@ -52,7 +52,7 @@ def _get_hashes(include_coverage: bool = False) -> dict[str, str]: def run_tests() -> None: """Run the picamera tests, zip the coverage database, hash relevant source files.""" - subprocess.run(["pytest", "hardware-specific-tests"], check=True) + subprocess.run(["pytest", "tests/hardware_specific_tests"], check=True) shutil.copyfile(".coverage", COVERAGE_FILE) hash_dict = _get_hashes(include_coverage=True) with open(HASH_FILE, "w", encoding="utf-8") as json_file: diff --git a/pyproject.toml b/pyproject.toml index 8f065ad5..f2b3f8c1 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -144,7 +144,7 @@ ignore = [ [tool.ruff.lint.per-file-ignores] # All testing dirs -"{tests,integration-tests,hardware-specific-tests}/**" = [ +"{tests,integration-tests}/**" = [ "B018", # Complaining about useless attribute access in tests, but we need them to check errors are raised "ANN", # Tests are not typehinted for fixtures etc "S101", # Allow asserts in tests diff --git a/hardware-specific-tests/README.md b/tests/hardware_specific_tests/README.md similarity index 97% rename from hardware-specific-tests/README.md rename to tests/hardware_specific_tests/README.md index c55ecc20..ec86b9d2 100644 --- a/hardware-specific-tests/README.md +++ b/tests/hardware_specific_tests/README.md @@ -34,13 +34,13 @@ As before, the server must be stopped before running these tests. The camera test are very slow as they run on hardware. They can be run with: - pytest hardware-specific-tests + pytest tests/hardware_specific_tests However, this will not archive the tests in the Git repository for reporting the coverage. For this, see the section above on reporting the coverage. When writing and debugging these unit tests, it is often best to run a specific test and to use the `-s` flag to see the print statements. It is also often useful to use `--pdb` to drop you into a python debug session on any failure. For example, you might run: - pytest hardware-specific-tests/picamera2/test_exposure_time_drift.py::test_exposure_time_saves_and_loads -s --pdb + pytest tests/hardware_specific_tests/picamera2/test_exposure_time_drift.py::test_exposure_time_saves_and_loads -s --pdb ### CI explanation diff --git a/hardware-specific-tests/picamera2/cam_test_utils.py b/tests/hardware_specific_tests/picamera2/cam_test_utils.py similarity index 95% rename from hardware-specific-tests/picamera2/cam_test_utils.py rename to tests/hardware_specific_tests/picamera2/cam_test_utils.py index 25e5f020..ebe86980 100644 --- a/hardware-specific-tests/picamera2/cam_test_utils.py +++ b/tests/hardware_specific_tests/picamera2/cam_test_utils.py @@ -5,7 +5,7 @@ from typing import Optional from openflexure_microscope_server.things.camera.picamera import StreamingPiCamera2 -from ..tests.utilities.lt_test_utils import LabThingsTestEnv +from ...shared_utils.lt_test_utils import LabThingsTestEnv @contextmanager diff --git a/hardware-specific-tests/picamera2/conftest.py b/tests/hardware_specific_tests/picamera2/conftest.py similarity index 100% rename from hardware-specific-tests/picamera2/conftest.py rename to tests/hardware_specific_tests/picamera2/conftest.py diff --git a/hardware-specific-tests/picamera2/test_acquisition.py b/tests/hardware_specific_tests/picamera2/test_acquisition.py similarity index 100% rename from hardware-specific-tests/picamera2/test_acquisition.py rename to tests/hardware_specific_tests/picamera2/test_acquisition.py diff --git a/hardware-specific-tests/picamera2/test_calibration.py b/tests/hardware_specific_tests/picamera2/test_calibration.py similarity index 100% rename from hardware-specific-tests/picamera2/test_calibration.py rename to tests/hardware_specific_tests/picamera2/test_calibration.py diff --git a/hardware-specific-tests/picamera2/test_exposure_time_drift.py b/tests/hardware_specific_tests/picamera2/test_exposure_time_drift.py similarity index 100% rename from hardware-specific-tests/picamera2/test_exposure_time_drift.py rename to tests/hardware_specific_tests/picamera2/test_exposure_time_drift.py diff --git a/hardware-specific-tests/picamera2/test_sensor_mode.py b/tests/hardware_specific_tests/picamera2/test_sensor_mode.py similarity index 100% rename from hardware-specific-tests/picamera2/test_sensor_mode.py rename to tests/hardware_specific_tests/picamera2/test_sensor_mode.py diff --git a/hardware-specific-tests/picamera2/test_tuning.py b/tests/hardware_specific_tests/picamera2/test_tuning.py similarity index 100% rename from hardware-specific-tests/picamera2/test_tuning.py rename to tests/hardware_specific_tests/picamera2/test_tuning.py diff --git a/tests/unit_tests/__init__.py b/tests/unit_tests/__init__.py index b58ffa86..6383f17d 100644 --- a/tests/unit_tests/__init__.py +++ b/tests/unit_tests/__init__.py @@ -1,6 +1,6 @@ """The unit-test suite for the OpenFlexure Microscope Server. This package contains all of the unit tests that can be run without specific -hardware. See also the `hardware-specific-tests` directory and the +hardware. See also the `hardware_specific_tests` directory and the `integration-tests` directory for more testing!. """ diff --git a/tests/unit_tests/test_cameras.py b/tests/unit_tests/test_cameras.py index 3f90689f..15d42dc0 100644 --- a/tests/unit_tests/test_cameras.py +++ b/tests/unit_tests/test_cameras.py @@ -1,6 +1,6 @@ """Tests for camera classes. These tests focus on checking the camera APIs are equivalent. -Tests for specific camera hardware are in the hardware-specific-tests directory. Tests +Tests for specific camera hardware are in the hardware_specific_tests directory. Tests on camera functionality using the simulation camera are in "test_camera". """