66
77import pytest
88
9- from runtime .errors import DatasetNotFoundError
10- from runtime .services .dataset_discovery import (
9+ from domain .errors import DatasetNotFoundError
10+ from domain .services .dataset_discovery import (
1111 DATASET_NS ,
1212 _get_first_image ,
1313 get_first_dataset_path ,
@@ -71,7 +71,7 @@ def test_returns_none_when_template_dataset_path_is_missing(self, tmp_path: Path
7171
7272class TestGetFirstImage :
7373 def test_returns_first_supported_image_in_sorted_order (self , tmp_path : Path , monkeypatch ) -> None :
74- monkeypatch .setattr ("runtime .services.dataset_discovery.settings.supported_extensions" , {".jpg" , ".png" })
74+ monkeypatch .setattr ("domain .services.dataset_discovery.settings.supported_extensions" , {".jpg" , ".png" })
7575
7676 (tmp_path / "z_image.png" ).touch ()
7777 expected = tmp_path / "a_image.jpg"
@@ -83,7 +83,7 @@ def test_returns_first_supported_image_in_sorted_order(self, tmp_path: Path, mon
8383 assert first_image == expected
8484
8585 def test_returns_none_when_no_supported_images_exist (self , tmp_path : Path , monkeypatch ) -> None :
86- monkeypatch .setattr ("runtime .services.dataset_discovery.settings.supported_extensions" , {".jpg" , ".png" })
86+ monkeypatch .setattr ("domain .services.dataset_discovery.settings.supported_extensions" , {".jpg" , ".png" })
8787 (tmp_path / "README.txt" ).touch ()
8888
8989 first_image = _get_first_image (tmp_path )
@@ -95,8 +95,8 @@ class TestScanDatasets:
9595 def test_builds_id_to_path_mapping_and_pagination (self , tmp_path : Path , monkeypatch ) -> None :
9696 dataset_dir = tmp_path / "aquarium"
9797 dataset_dir .mkdir ()
98- monkeypatch .setattr ("runtime .services.dataset_discovery.generate_image_thumbnail" , lambda _path : "thumb" )
99- monkeypatch .setattr ("runtime .services.dataset_discovery.settings.supported_extensions" , {".jpg" , ".png" })
98+ monkeypatch .setattr ("domain .services.dataset_discovery.generate_image_thumbnail" , lambda _path : "thumb" )
99+ monkeypatch .setattr ("domain .services.dataset_discovery.settings.supported_extensions" , {".jpg" , ".png" })
100100 (dataset_dir / "image.jpg" ).touch ()
101101
102102 datasets , dataset_paths = scan_datasets (tmp_path )
@@ -115,9 +115,9 @@ def test_sets_thumbnail_from_first_supported_image(self, tmp_path: Path, monkeyp
115115 (dataset_dir / "README.txt" ).write_text ("info" )
116116 first_image = dataset_dir / "0001.jpg"
117117 first_image .touch ()
118- monkeypatch .setattr ("runtime .services.dataset_discovery.settings.supported_extensions" , {".jpg" , ".png" })
118+ monkeypatch .setattr ("domain .services.dataset_discovery.settings.supported_extensions" , {".jpg" , ".png" })
119119 monkeypatch .setattr (
120- "runtime .services.dataset_discovery.generate_image_thumbnail" ,
120+ "domain .services.dataset_discovery.generate_image_thumbnail" ,
121121 lambda image_path : f"thumb::{ image_path .name } " ,
122122 )
123123
@@ -130,7 +130,7 @@ def test_sets_thumbnail_none_when_no_supported_images(self, tmp_path: Path, monk
130130 dataset_dir = tmp_path / "aquarium"
131131 dataset_dir .mkdir ()
132132 (dataset_dir / "README.txt" ).write_text ("no images" )
133- monkeypatch .setattr ("runtime .services.dataset_discovery.settings.supported_extensions" , {".jpg" , ".png" })
133+ monkeypatch .setattr ("domain .services.dataset_discovery.settings.supported_extensions" , {".jpg" , ".png" })
134134
135135 datasets , _ = scan_datasets (tmp_path )
136136
0 commit comments