def test_to_yaml_defaults():
ret = GenerateOptions(
skip_default_metrics=False,
metric_package_names=[],
repo='.',
database='database.db',
).to_yaml()
assert ret == {'repo': '.', 'database': 'database.db'}
def test_minimal_defaults():
ret = GenerateOptions.from_yaml({'repo': './', 'database': 'database.db'})
assert ret == GenerateOptions(
skip_default_metrics=False,
metric_package_names=[],
repo='./',
database='database.db',
)
def test_minimal_defaults():
ret = GenerateOptions.from_yaml({'repo': './', 'database': 'database.db'})
assert ret == GenerateOptions(
skip_default_metrics=False,
metric_package_names=[],
repo='./',
repo_type='git',
database='database.db',
exclude=re.compile(b'^$'),
)
def test_none_for_tempdir_allowed():
ret = GenerateOptions.from_yaml({
'repo': 'repo',
'database': 'database.db',
})
assert ret == GenerateOptions(
skip_default_metrics=False,
metric_package_names=[],
repo='repo',
database='database.db',
)
def test_with_all_options_specified():
ret = GenerateOptions.from_yaml({
'skip_default_metrics': True,
'metric_package_names': ['my_package'],
'repo': '.',
'database': 'database.db',
})
assert ret == GenerateOptions(
skip_default_metrics=True,
metric_package_names=['my_package'],
repo='.',
database='database.db',
)
def test_to_yaml_all_specified():
ret = GenerateOptions(
skip_default_metrics=True,
tempdir_location='/foo/tmp',
metric_package_names=['my_package'],
repo='.',
database='database.db',
).to_yaml()
assert ret == {
'skip_default_metrics': True,
'tempdir_location': '/foo/tmp',
'metric_package_names': ['my_package'],
'repo': '.',
'database': 'database.db',
}
def test_with_all_options_specified():
ret = GenerateOptions.from_yaml({
'skip_default_metrics': True,
'metric_package_names': ['my_package'],
'repo': '.',
'repo_type': 'git',
'database': 'database.db',
'exclude': '^vendor/',
})
assert ret == GenerateOptions(
skip_default_metrics=True,
metric_package_names=['my_package'],
repo='.',
repo_type='git',
database='database.db',
exclude=re.compile(b'^vendor/'),
)
def test_get_options_from_config_create_config(tempdir_factory):
tmpdir = tempdir_factory.get()
with cwd(tmpdir):
ret = get_options_from_config([
'--create-config',
'.',
'database.db',
])
assert os.path.exists('generate_config.yaml')
assert yaml.load(io.open('generate_config.yaml').read()) == {
'repo': '.',
'database': 'database.db',
}
assert ret == GenerateOptions(
skip_default_metrics=False,
metric_package_names=[],
repo='.',
database='database.db',
)
