File tree 1 file changed +19
-10
lines changed
python_modules/libraries/dagster-dg/dagster_dg_tests/cli_tests
1 file changed +19
-10
lines changed Original file line number Diff line number Diff line change @@ -160,13 +160,18 @@ def test_build_docs_success_in_published_package():
160
160
161
161
GET_DOCS_JSON_QUERY = """
162
162
query GetDocsJson {
163
- locationDocsJsonOrError (repositorySelector: {repositoryLocationName: "foo-bar", repositoryName: "__repository__"}) {
163
+ repositoryOrError (repositorySelector: {repositoryLocationName: "foo-bar", repositoryName: "__repository__"}) {
164
164
__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
+ }
170
175
}
171
176
}
172
177
}
@@ -208,11 +213,15 @@ def test_build_docs_success_matches_graphql():
208
213
try :
209
214
gql_client = DagsterGraphQLClient (hostname = "localhost" , port_number = port )
210
215
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 )
214
221
assert json .dumps (
215
- _sort_sample_yamls (json .loads (result ["locationDocsJsonOrError" ]["json" ])),
222
+ _sort_sample_yamls (
223
+ json .loads (result ["repositoryOrError" ]["locationDocsJsonOrError" ]["json" ])
224
+ ),
216
225
sort_keys = True ,
217
226
indent = 2 ,
218
227
) == json .dumps (_sort_sample_yamls (contents ), sort_keys = True , indent = 2 )
You can’t perform that action at this time.
0 commit comments