def test_repo_entrypoints():
module = importlib.import_module('dagster_examples.intro_tutorial.repos')
expected = LoaderEntrypoint(module, 'dagster_examples.intro_tutorial.repos', 'define_repo')
handle = ExecutionTargetHandle.for_repo_yaml(script_relative_path('repository.yaml'))
assert handle.entrypoint.module == expected.module
assert handle.entrypoint.module_name == expected.module_name
assert handle.entrypoint.fn_name == expected.fn_name
assert handle.entrypoint.from_handle == handle
module = importlib.import_module('dagster')
expected = LoaderEntrypoint(module, 'dagster', 'define_bar_repo')
handle = ExecutionTargetHandle.for_repo_module(module_name='dagster', fn_name='define_bar_repo')
assert handle.entrypoint.module == expected.module
assert handle.entrypoint.module_name == expected.module_name
assert handle.entrypoint.fn_name == expected.fn_name
assert handle.entrypoint.from_handle == handle
python_file = script_relative_path('bar_repo.py')
module = imp.load_source('bar_repo', python_file)
expected = LoaderEntrypoint(module, 'bar_repo', 'define_bar_repo')
handle = ExecutionTargetHandle.for_repo_python_file(
python_file=python_file, fn_name='define_bar_repo'
)
assert handle.entrypoint.module == expected.module
assert handle.entrypoint.module_name == expected.module_name
assert handle.entrypoint.fn_name == expected.fn_name
assert handle.entrypoint.from_handle == handle
def define_examples_context():
return DagsterGraphQLContext(
handle=ExecutionTargetHandle.for_repo_module('dagster_examples',
'define_demo_repo'),
instance=DagsterInstance.ephemeral(),
execution_manager=SynchronousExecutionManager(),
)
def define_examples_context(raise_on_error=True):
return DagsterGraphQLContext(
handle=ExecutionTargetHandle.for_repo_module('dagster_examples', 'define_demo_repo'),
pipeline_runs=PipelineRunStorage(),
execution_manager=SynchronousExecutionManager(),
raise_on_error=raise_on_error,
)
def test_smoke_app():
flask_app = app.create_app_with_execution_handle(
ExecutionTargetHandle.for_repo_module(
module_name='dagster_examples.intro_tutorial.repos',
fn_name='define_repo'),
DagsterInstance.ephemeral(),
)
client = flask_app.test_client()
result = client.post(
'/graphql', data={'query': 'query { pipelines { nodes { name }}}'})
data = json.loads(result.data.decode('utf-8'))
assert len(data['data']['pipelines']['nodes']) == 2
assert {
node_data['name']
for node_data in data['data']['pipelines']['nodes']
} == set(['hello_cereal_pipeline', 'complex_pipeline'])
result = client.get('/graphql')
assert result.status_code == 400
data = json.loads(result.data.decode('utf-8'))
assert len(data['errors']) == 1
assert data['errors'][0]['message'] == 'Must provide query string.'
result = client.get('/dagit/notebook?path=foo.bar')
assert result.status_code == 400
assert result.data.decode('utf-8') == 'Invalid Path'
result = client.post('/graphql',
data={'query': 'query { version { slkjd } }'})
data = json.loads(result.data.decode('utf-8'))
assert 'errors' in data
assert len(data['errors']) == 1
assert 'must not have a sub selection' in data['errors'][0]['message']
# Missing routes return the index.html file of the Dagit react app, so the user
# gets our UI when they navigate to "synthetic" react router URLs.
result = client.get('static/foo/bar')
assert result.status_code == 200
assert "You need to enable JavaScript to run this app." in result.data.decode(
'utf-8')
result = client.get('pipelines/foo')
assert result.status_code == 200
assert "You need to enable JavaScript to run this app." in result.data.decode(
'utf-8')
def test_smoke_app():
pipeline_run_storage = PipelineRunStorage()
flask_app = app.create_app(
ExecutionTargetHandle.for_repo_module(
module_name='dagster_examples.intro_tutorial.repos',
fn_name='define_repo'),
pipeline_run_storage,
)
client = flask_app.test_client()
result = client.post(
'/graphql', data={'query': 'query { pipelines { nodes { name }}}'})
data = json.loads(result.data.decode('utf-8'))
assert len(data['data']['pipelines']['nodes']) == 1
assert {
node_data['name']
for node_data in data['data']['pipelines']['nodes']
} == set(['repo_demo_pipeline'])
result = client.get('/graphql')
assert result.status_code == 400
data = json.loads(result.data.decode('utf-8'))
assert len(data['errors']) == 1
assert data['errors'][0]['message'] == 'Must provide query string.'
result = client.get('/dagit/notebook?path=foo.bar')
assert result.status_code == 400
assert result.data.decode('utf-8') == 'Invalid Path'
result = client.post('/graphql',
data={'query': 'query { version { slkjd } }'})
data = json.loads(result.data.decode('utf-8'))
assert 'errors' in data
assert len(data['errors']) == 1
assert 'must not have a sub selection' in data['errors'][0]['message']
result = client.get('static/foo/bar')
assert result.status_code == 404
result = client.get('vendor/foo/bar')
assert result.status_code == 404
