def test_run_mocked_check_output(dib, DIB, capsys):
with mock.patch.object(dib.subprocess, "Popen"):
dib = dib.DIB("filename42", ["element42"])
dib.run()
out = capsys.readouterr()[0]
assert 'filename42' in out
assert 'element42' in out
def test_run_echoed(dib, DIB):
dib = dib.DIB("filename", ["element1", "element2"], exec_path="echo")
assert dib.run() == 0
def test_prep_env_empy(dib):
with mock.patch.object(dib.os, "environ", {"key1": "value1"}):
dib = dib.DIB(sentinel.filename, [sentinel.element])
new_env = dib._prep_env()
assert new_env["key1"] == "value1"
def test_dib_cmdline_additional_options(dib):
opts = [sentinel.opt1, sentinel.opt2]
dib = dib.DIB(sentinel.filename, [sentinel.element], additional_options=opts)
assert opts[0] in dib.cmdline
assert opts[1] in dib.cmdline
def test_dib_cmdline_no_elements(dib):
with pytest.raises(dib.NoElementsError):
dib = dib.DIB(sentinel.filename, [])
def test_dib_cmdline_offline(dib):
dib = dib.DIB(sentinel.filename, [sentinel.element], offline=True, tracing=False)
assert '--offline' in dib.cmdline
def test_dib_cmdline_tracing(dib):
dib = dib.DIB(sentinel.filename, [sentinel.element], tracing=True)
assert '-x' in dib.cmdline
def test_dib_cmdline_no_tracing(dib):
dib = dib.DIB(sentinel.filename, [sentinel.element], tracing=False)
assert '-x' not in dib.cmdline
def test_dib_cmdline_all_defaults(dib):
dib = dib.DIB(sentinel.filename, [sentinel.element])
assert dib.cmdline == ['disk-image-create', '-a', 'amd64', '-o', sentinel.filename, sentinel.element]
def DIB(dib):
DIB = dib.DIB(sentinel.filename, [sentinel.element])
return DIB