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

Removing optiosn and args pass from state stuff, state stuff doesn't take options or args.

Browse source
This commit is contained in:
DJDavisson 2019-05-07 12:11:28 -07:00
parent 3ddab331f0
commit 01e3dbb05a
2 changed files with 11 additions and 17 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

21
python_terraform/__init__.py
View file

@ -389,40 +389,31 @@ class Terraform(object):
self.tfstate = Tfstate.load_file(file_path)
def set_workspace(self, workspace=None, dir_or_plan=None, **kwargs):
def set_workspace(self, workspace):
"""
set workspace
:param workspace: the desired workspace.
:return: status
"""
options = kwargs
options = self._generate_default_options(options)
args = self._generate_default_args(dir_or_plan)
return self.cmd('workspace select ' + workspace, *args, **options)
return self.cmd('workspace select ' + workspace)
def create_workspace(self, workspace=None, dir_or_plan=None, **kwargs):
def create_workspace(self, workspace):
"""
create workspace
:param workspace: the desired workspace.
:return: status
"""
options = kwargs
options = self._generate_default_options(options)
args = self._generate_default_args(dir_or_plan)
return self.cmd('workspace new ' + workspace, *args, **options)
return self.cmd('workspace new ' + workspace)
def show_workspace(self, dir_or_plan=None, **kwargs):
def show_workspace(self):
"""
show workspace
:return: workspace
"""
options = kwargs
options = self._generate_default_options(options)
args = self._generate_default_args(dir_or_plan)
return self.cmd('workspace show', *args, **options)
return self.cmd('workspace show')
def __exit__(self, exc_type, exc_value, traceback):
self.temp_var_files.clean_up()

5
test/test_terraform.py
View file

@ -323,12 +323,14 @@ class TestTerraform(object):
def test_create_workspace(self):
tf = Terraform(working_dir=current_path)
tf.init()
ret, out, err = tf.create_workspace('test')
assert ret == 0
assert err == ''
def test_set_workspace(self):
tf = Terraform(working_dir=current_path)
tf.init()
tf.create_workspace('test')
ret, out, err = tf.set_workspace('test')
assert ret == 0
@ -336,7 +338,8 @@ class TestTerraform(object):
def test_show_workspace(self):
tf = Terraform(working_dir=current_path)
tf.init()
tf.create_workspace('test')
ret, out, err = tf.show_workspace('test')
ret, out, err = tf.show_workspace()
assert ret == 0
assert err == ''