meissa/dda-python-terraform
6
3
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

add more parameter test for output

Browse source
This commit is contained in:
beelit94 2017-08-04 14:14:57 -07:00
parent 8214c6ac61
commit 12d57862ed
1 changed files with 18 additions and 7 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

25
test/test_terraform.py
View file

@ -71,9 +71,10 @@ def string_logger(request):
def td(): def td():
root_logger.removeHandler(handler) root_logger.removeHandler(handler)
log_stream.close()
request.addfinalizer(td) request.addfinalizer(td)
return log_stream return lambda: str(log_stream.getvalue())
class TestTerraform(object): class TestTerraform(object):
@ -105,7 +106,7 @@ class TestTerraform(object):
def test_cmd(self, method, expected_output, expected_ret_code, expected_logs, string_logger): def test_cmd(self, method, expected_output, expected_ret_code, expected_logs, string_logger):
tf = Terraform(working_dir=current_path) tf = Terraform(working_dir=current_path)
ret, out, err = method(tf) ret, out, err = method(tf)
logs = str(string_logger.getvalue()) logs = string_logger()
logs = logs.replace('\n', '') logs = logs.replace('\n', '')
assert expected_output in out assert expected_output in out
assert expected_ret_code == ret assert expected_ret_code == ret
@ -167,10 +168,22 @@ class TestTerraform(object):
out = tf.output('test_output') out = tf.output('test_output')
assert 'test2' in out assert 'test2' in out
def test_get_output(self): @pytest.mark.parametrize(
("param", "param_str"),
[
({}, ""),
({'module': 'test2'}, "-module=test2"),
({'module': 'test2', 'no_color': IsFlagged}, "-no-color -module=test2")
]
)
def test_output(self, param, param_str, string_logger):
tf = Terraform(working_dir=current_path, variables={'test_var': 'test'}) tf = Terraform(working_dir=current_path, variables={'test_var': 'test'})
tf.apply('var_to_output') tf.apply('var_to_output')
assert tf.output('test_output') == 'test' result = tf.output('test_output', **param)
if param:
assert 'terraform output -json {0} test_output'.format(param_str) in string_logger()
else:
assert result == 'test'
def test_destroy(self): def test_destroy(self):
tf = Terraform(working_dir=current_path, variables={'test_var': 'test'}) tf = Terraform(working_dir=current_path, variables={'test_var': 'test'})
@ -197,6 +210,4 @@ class TestTerraform(object):
def test_import(self, string_logger): def test_import(self, string_logger):
tf = Terraform(working_dir=current_path) tf = Terraform(working_dir=current_path)
tf.import_cmd('aws_instance.foo', 'i-abc1234', no_color=IsFlagged) tf.import_cmd('aws_instance.foo', 'i-abc1234', no_color=IsFlagged)
logs = string_logger.getvalue() assert 'command: terraform import -no-color aws_instance.foo i-abc1234' in string_logger()
print(logs)
assert 'command: terraform import -no-color aws_instance.foo i-abc1234' in logs