def test_set_version_info():
    new_version = "100.100.0"
    dmp = DagsterModulePublisher()

    # Test setting version
    version = dmp.set_version_info(new_version=new_version, dry_run=True)
    assert version == {"__version__": new_version}
Esempio n. 2
0
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
    }