diff options
Diffstat (limited to 'vendor/github.com/mitchellh/packer/fix/fixer_sshdisableagent_test.go')
-rw-r--r-- | vendor/github.com/mitchellh/packer/fix/fixer_sshdisableagent_test.go | 83 |
1 files changed, 83 insertions, 0 deletions
diff --git a/vendor/github.com/mitchellh/packer/fix/fixer_sshdisableagent_test.go b/vendor/github.com/mitchellh/packer/fix/fixer_sshdisableagent_test.go new file mode 100644 index 00000000..c38f9c58 --- /dev/null +++ b/vendor/github.com/mitchellh/packer/fix/fixer_sshdisableagent_test.go @@ -0,0 +1,83 @@ +package fix + +import ( + "reflect" + "testing" +) + +func TestFixerSSHDisableAgent_Impl(t *testing.T) { + var _ Fixer = new(FixerSSHDisableAgent) +} + +func TestFixerSSHDisableAgent_Fix(t *testing.T) { + cases := []struct { + Input map[string]interface{} + Expected map[string]interface{} + }{ + // No disable_agent field + { + Input: map[string]interface{}{ + "type": "virtualbox", + }, + + Expected: map[string]interface{}{ + "type": "virtualbox", + }, + }, + + // disable_agent_forwarding without disable_agent + { + Input: map[string]interface{}{ + "ssh_disable_agent_forwarding": true, + }, + + Expected: map[string]interface{}{ + "ssh_disable_agent_forwarding": true, + }, + }, + + // disable_agent without disable_agent_forwarding + { + Input: map[string]interface{}{ + "ssh_disable_agent": true, + }, + + Expected: map[string]interface{}{ + "ssh_disable_agent_forwarding": true, + }, + }, + + // disable_agent and disable_agent_forwarding + { + Input: map[string]interface{}{ + "ssh_disable_agent": true, + "ssh_disable_agent_forwarding": false, + }, + + Expected: map[string]interface{}{ + "ssh_disable_agent_forwarding": false, + }, + }, + } + + for _, tc := range cases { + var f FixerSSHDisableAgent + + input := map[string]interface{}{ + "builders": []map[string]interface{}{tc.Input}, + } + + expected := map[string]interface{}{ + "builders": []map[string]interface{}{tc.Expected}, + } + + output, err := f.Fix(input) + if err != nil { + t.Fatalf("err: %s", err) + } + + if !reflect.DeepEqual(output, expected) { + t.Fatalf("unexpected: %#v\nexpected: %#v\n", output, expected) + } + } +} |