add tests for TerraformApplyCommand and TerraformPlanCommand

Author: treylittle

src/TerraformApplyCommand.groovy

@@ -43,14 +43,14 @@ class TerraformApplyCommand implements TerraformCommand, Resettable {
         return this
     }
 
-    public TerraformPlanCommand withVariableFile(String key, Map values) {
+    public TerraformApplyCommand withVariableFile(String key, Map values) {
         String fileName = "${this.environment}-${key}.tfvars"
         String value = convertMapToCliString(values)
         Jenkinsfile.writeFile(fileName, "${key}=${value}")
         return withVariableFile(fileName)
     }
 
-    public TerraformPlanCommand withVariableFile(String fileName) {
+    public TerraformApplyCommand withVariableFile(String fileName) {
         this.args << "-var-file=./${fileName}"
         return this
     }

test/TerraformApplyCommandTest.groovy

@@ -9,6 +9,7 @@ import static org.mockito.Mockito.mock
 import static org.mockito.Mockito.spy
 import static org.mockito.Mockito.times
 import static org.mockito.Mockito.verify
+import static org.mockito.Mockito.verifyNoMoreInteractions
 
 import org.junit.jupiter.api.Nested
 import org.junit.jupiter.api.Test
@@ -98,6 +99,57 @@ class TerraformApplyCommandTest {
         }
     }
 
+    @Nested
+    public class WithVariableFile {
+        @Test
+        void isFluent1() {
+            def stage = new TerraformApplyCommand('foo')
+            def result = stage.withVariableFile('somekey')
+
+            assertThat(result, equalTo(stage))
+        }
+
+        @Test
+        void testWithFile() {
+            def filename = 'filename'
+            def command = new TerraformApplyCommand().withVariableFile(filename)
+
+            def actualCommand = command.toString()
+            assertThat(actualCommand, containsString("-var-file=./${filename}"))
+        }
+    }
+
+    @Nested
+    public class WithVariableFileAndMap {
+        @Test
+        void isFluent() {
+            def stage = new TerraformApplyCommand('foo')
+            Jenkinsfile.original = mock(MockWorkflowScript.class)
+            def result = stage.withVariableFile('somekey', ['key':'value'])
+
+            assertThat(result, equalTo(stage))
+        }
+
+        @Test
+        void testWithFileAndMap() {
+            def myKey = 'myKey'
+            def expectedValue = 'expValue'
+            def map = [expectedKey:expectedValue]
+            def original = mock(MockWorkflowScript.class)
+            Jenkinsfile.original = original
+            def command = spy(new TerraformApplyCommand("dev")).withVariableFile(myKey, map)
+
+            def actualCommand = command.toString()
+            assertThat(actualCommand, containsString("-var-file=./dev-${myKey}"))
+
+            def filename = "dev-${myKey}.tfvars"
+            verify(command).withVariableFile(filename)
+            def content = "${myKey}={expectedKey=\"${expectedValue}\"}"
+            verify(original).writeFile(file: filename.toString(), text: content.toString())
+            verifyNoMoreInteractions(original)
+        }
+    }
+
     @Nested
     public class WithVariableMap {
         @Test

test/TerraformPlanCommandTest.groovy

@@ -9,6 +9,7 @@ import static org.mockito.Mockito.mock
 import static org.mockito.Mockito.spy
 import static org.mockito.Mockito.times
 import static org.mockito.Mockito.verify
+import static org.mockito.Mockito.verifyNoMoreInteractions
 
 import org.junit.jupiter.api.Nested
 import org.junit.jupiter.api.Test
@@ -167,6 +168,57 @@ class TerraformPlanCommandTest {
         }
     }
 
+    @Nested
+    public class WithVariableFile {
+        @Test
+        void isFluent1() {
+            def stage = new TerraformPlanCommand('foo')
+            def result = stage.withVariableFile('somekey')
+
+            assertThat(result, equalTo(stage))
+        }
+
+        @Test
+        void testWithFile() {
+            def filename = 'filename'
+            def command = new TerraformPlanCommand().withVariableFile(filename)
+
+            def actualCommand = command.toString()
+            assertThat(actualCommand, containsString("-var-file=./${filename}"))
+        }
+    }
+
+    @Nested
+    public class WithVariableFileAndMap {
+        @Test
+        void isFluent() {
+            def stage = new TerraformPlanCommand('foo')
+            Jenkinsfile.original = mock(MockWorkflowScript.class)
+            def result = stage.withVariableFile('somekey', ['key':'value'])
+
+            assertThat(result, equalTo(stage))
+        }
+
+        @Test
+        void testWithFileAndMap() {
+            def myKey = 'myKey'
+            def expectedValue = 'expValue'
+            def map = [expectedKey:expectedValue]
+            def original = mock(MockWorkflowScript.class)
+            Jenkinsfile.original = original
+            def command = spy(new TerraformPlanCommand("dev")).withVariableFile(myKey, map)
+
+            def actualCommand = command.toString()
+            assertThat(actualCommand, containsString("-var-file=./dev-${myKey}"))
+
+            def filename = "dev-${myKey}.tfvars"
+            verify(command).withVariableFile(filename)
+            def content = "${myKey}={expectedKey=\"${expectedValue}\"}"
+            verify(original).writeFile(file: filename.toString(), text: content.toString())
+            verifyNoMoreInteractions(original)
+        }
+    }
+
     @Nested
     public class WithVariableMap {
         @Test