Fix repository_fixture in archive

Author: jameslinnell

archived_epr/src_old/layers/domain/repository/device_repository/tests/utils.py

@@ -1,6 +1,9 @@
 from collections.abc import Generator
 
 from domain.core.device import Device
+from domain.repository.device_reference_data_repository import (
+    DeviceReferenceDataRepository,
+)
 from domain.repository.device_repository import DeviceRepository
 
 from conftest import dynamodb_client_with_sleep as dynamodb_client
@@ -11,7 +14,7 @@
 
 
 def repository_fixture[
-    T: DeviceRepository
+    T: DeviceRepository | DeviceReferenceDataRepository
 ](is_integration_test: bool, repository_class: type[T]) -> Generator[T, None, None]:
     if is_integration_test:
         table_name = read_terraform_output("dynamodb_epr_table_name.value")
@@ -32,7 +35,7 @@ def devices_exactly_equal(device_a: Device, device_b: Device) -> bool:
 
 
 def repository_fixture_cpm[
-    T
+    T: DeviceRepository | DeviceReferenceDataRepository
 ](is_integration_test: bool, repository_class: type[T]) -> Generator[T, None, None]:
     if is_integration_test:
         table_name = read_terraform_output("dynamodb_cpm_table_name.value")