from python_terraform import Terraform import pytest import os import logging import re logging.basicConfig(level=logging.DEBUG) ACCESS_KEY = os.environ['PACKER_access_key'] SECRET_KEY = os.environ['PACKER_secret_key'] STRING_CASES = [ [ lambda x: x.generate_cmd_string('apply', 'the_folder', no_color='', var={'a': 'b', 'c': 'd'}), "terraform apply -var='a=b' -var='c=d' -no-color the_folder" ], [ lambda x: x.generate_cmd_string('push', 'path', var={'a': 'b'}, vcs=True, token='token', atlas_address='url'), "terraform push -var='a=b' -vcs=true -token=token -atlas-address=url path" ], ] CMD_CASES = [ ['method', 'expected_output'], [ [ lambda x: x.cmd('plan', 'aws_tf', no_color='', var={'access_key': ACCESS_KEY, 'secret_key': SECRET_KEY}) , 'Plan: 1 to add, 0 to change, 0 to destroy' ] ] ] class TestTerraform: def teardown_method(self, method): """ teardown any state that was previously setup with a setup_method call. """ def purge(dir, pattern): for f in os.listdir(dir): if re.search(pattern, f): if os.path.isfile(f): os.remove(os.path.join(dir, f)) purge('.', '.tfstate') @pytest.mark.parametrize([ "method", "expected" ], STRING_CASES) def test_generate_cmd_string(self, method, expected): tf = Terraform() result = method(tf) strs = expected.split() for s in strs: assert s in result @pytest.mark.parametrize(*CMD_CASES) def test_cmd(self, method, expected_output): tf = Terraform() ret, out, err = method(tf) assert expected_output in out assert ret == 0 def test_state_data(self): tf = Terraform(working_dir='test_tfstate_file') tf.read_state_file() assert tf.tfstate.modules[0]['path'] == ['root'] def test_apply(self): tf = Terraform(working_dir='apply_tf', variables={'test_var': 'test'}) tf.apply(var={'test_var': 'test2'}) def test_get_output(self): tf = Terraform(working_dir='apply_tf', variables={'test_var': 'test'}) tf.apply() assert tf.output('test_output') == 'test'