File tree Expand file tree Collapse file tree 1 file changed +8
-2
lines changed
tests/test_unit/test_cdf_tk/test_commands/test_migration_cmd Expand file tree Collapse file tree 1 file changed +8
-2
lines changed Original file line number Diff line number Diff line change 4141
4242@pytest .fixture
4343def cognite_migration_model (
44- toolkit_config : ToolkitClientConfig , rsps : responses .RequestsMock , cognite_core_no_3D : DataModel [View ]
44+ toolkit_config : ToolkitClientConfig ,
45+ rsps : responses .RequestsMock ,
46+ cognite_core_no_3D : DataModel [View ],
47+ cognite_extractor_views : list [View ],
4548) -> Iterator [responses .RequestsMock ]:
4649 config = toolkit_config
4750 mapping_by_id = {mapping .external_id : mapping for mapping in create_default_mappings ()}
@@ -66,7 +69,10 @@ def cognite_migration_model(
6669 )
6770 rsps .post (
6871 config .create_api_url ("models/views/byids" ),
69- json = {"items" : [view .dump () for view in cognite_core_no_3D .views ]},
72+ json = {
73+ "items" : [view .dump () for view in cognite_core_no_3D .views ]
74+ + [view .dump () for view in cognite_extractor_views ]
75+ },
7076 )
7177 # Migration model
7278 migration_model = COGNITE_MIGRATION_MODEL .dump ()
You can’t perform that action at this time.
0 commit comments