@@ -179,14 +179,51 @@ def test_missing_service(builder, caplog):
179179 ),
180180 ],
181181)
182- def test_gb_extract_entities (builder_with_setup , index , type , desc , pv , macros ): # noqa: N803
182+ def test_gb_extract_entities_ioc_yaml (
183+ builder , techui_support , index , type , desc , pv , macros
184+ ):
185+ # We don't want to use builder_with_setup as that calls _extract_services()
186+ # and in turn that calls _extract_entities()
187+ builder .techui_support = techui_support
183188 prefix = pv .split (":" , maxsplit = 1 )[0 ]
184189
185- builder_with_setup ._extract_entities (
190+ builder ._extract_entities (
186191 "bl01t-mo-ioc-01" ,
187- builder_with_setup ._services_dir .joinpath ("bl01t-mo-ioc-01/config/ioc.yaml" ),
192+ builder ._services_dir .joinpath ("bl01t-mo-ioc-01/config/ioc.yaml" ),
193+ )
194+ entity = builder .entities [prefix ][index ]
195+ assert entity .type == type
196+ assert entity .desc == desc
197+ assert entity .prefix == pv
198+ assert entity .macros == macros
199+
200+
201+ @pytest .mark .parametrize (
202+ "index, type, desc, pv, macros" ,
203+ [
204+ (
205+ 0 ,
206+ "fastcs.TemperatureController" ,
207+ None ,
208+ "BL01T-EA-TEST-01" ,
209+ {"name" : "BL01T-EA-TEST-01" },
210+ ),
211+ ],
212+ )
213+ def test_gb_extract_entities_fastcs_yaml (
214+ builder , techui_support , index , type , desc , pv , macros
215+ ):
216+ # We don't want to use builder_with_setup as that calls _extract_services()
217+ # and in turn that calls _extract_entities()
218+ builder .techui_support = techui_support
219+
220+ prefix = pv .split (":" , maxsplit = 1 )[0 ]
221+
222+ builder ._extract_entities (
223+ "bl01t-ea-ioc-01" ,
224+ builder ._services_dir .joinpath ("bl01t-ea-ioc-01/config/fastcs.yaml" ),
188225 )
189- entity = builder_with_setup .entities [prefix ][index ]
226+ entity = builder .entities [prefix ][index ]
190227 assert entity .type == type
191228 assert entity .desc == desc
192229 assert entity .prefix == pv
0 commit comments