def test_should_publish(dagster_modules):
dagster, dagster_k8s = dagster_modules
lakehouse = DagsterModule("lakehouse", is_library=True)
assert dagster.should_publish
assert dagster_k8s.should_publish
assert not lakehouse.should_publish
def test_set_version_info():
new_version = '100.100.0'
new_nightly_version = get_nightly_version()
dmp = DagsterModulePublisher()
# Test setting both version and nightly
version = dmp.set_version_info(new_version=new_version, dry_run=True)
assert version == {
'__version__': new_version,
'__nightly__': new_nightly_version
}
# Test only setting nightly
version = dmp.set_version_info(dry_run=True)
dm = DagsterModule('dagster', is_library=False)
existing_version = dm.get_version_info()['__version__']
assert version == {
'__version__': existing_version,
'__nightly__': new_nightly_version
}
def dagster_modules():
return (
DagsterModule("dagster", is_library=False),
DagsterModule("dagster-k8s", is_library=True),
)
def dagster_modules():
return (
DagsterModule('dagster', is_library=False),
DagsterModule('dagster-k8s', is_library=True),
)