Skip to content

Commit 0bc8c21

Browse files
authored
[dg] fix dg docs test (#29226)
## Summary Fixes dg docs test, which #29216 broke (offending test didn't trigger on branch)
1 parent 7b8d287 commit 0bc8c21

File tree

1 file changed

+19
-10
lines changed

1 file changed

+19
-10
lines changed

python_modules/libraries/dagster-dg/dagster_dg_tests/cli_tests/test_docs_commands.py

+19-10
Original file line numberDiff line numberDiff line change
@@ -160,13 +160,18 @@ def test_build_docs_success_in_published_package():
160160

161161
GET_DOCS_JSON_QUERY = """
162162
query GetDocsJson {
163-
locationDocsJsonOrError(repositorySelector: {repositoryLocationName: "foo-bar", repositoryName: "__repository__"}) {
163+
repositoryOrError(repositorySelector: {repositoryLocationName: "foo-bar", repositoryName: "__repository__"}) {
164164
__typename
165-
... on LocationDocsJson {
166-
json
167-
}
168-
... on PythonError {
169-
message
165+
... on Repository {
166+
locationDocsJsonOrError {
167+
__typename
168+
... on LocationDocsJson {
169+
json
170+
}
171+
... on PythonError {
172+
message
173+
}
174+
}
170175
}
171176
}
172177
}
@@ -208,11 +213,15 @@ def test_build_docs_success_matches_graphql():
208213
try:
209214
gql_client = DagsterGraphQLClient(hostname="localhost", port_number=port)
210215
result = gql_client._execute(GET_DOCS_JSON_QUERY) # noqa: SLF001
211-
assert result["locationDocsJsonOrError"]["__typename"] == "LocationDocsJson", str(
212-
result
213-
)
216+
assert result["repositoryOrError"]["__typename"] == "Repository", str(result)
217+
assert (
218+
result["repositoryOrError"]["locationDocsJsonOrError"]["__typename"]
219+
== "LocationDocsJson"
220+
), str(result)
214221
assert json.dumps(
215-
_sort_sample_yamls(json.loads(result["locationDocsJsonOrError"]["json"])),
222+
_sort_sample_yamls(
223+
json.loads(result["repositoryOrError"]["locationDocsJsonOrError"]["json"])
224+
),
216225
sort_keys=True,
217226
indent=2,
218227
) == json.dumps(_sort_sample_yamls(contents), sort_keys=True, indent=2)

0 commit comments

Comments
 (0)