Skip to content

Commit 1cd7a5d

Browse files
committed
tests; updated
1 parent 92a61cc commit 1cd7a5d

File tree

1 file changed

+8
-2
lines changed
  • tests/test_unit/test_cdf_tk/test_commands/test_migration_cmd

1 file changed

+8
-2
lines changed

tests/test_unit/test_cdf_tk/test_commands/test_migration_cmd/test_command.py

Lines changed: 8 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -41,7 +41,10 @@
4141

4242
@pytest.fixture
4343
def 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()

0 commit comments

Comments
 (0)