1package provisiontest 2 3import ( 4 "testing" 5 6 "github.com/stretchr/testify/assert" 7) 8 9func TestCreateFakeSSHCommander(t *testing.T) { 10 sshCmder := NewFakeSSHCommander(FakeSSHCommanderOptions{FilesystemType: "btrfs"}) 11 output, err := sshCmder.SSHCommand("stat -f -c %T /var/lib") 12 if err != nil || output != "btrfs\n" { 13 t.Fatal("FakeSSHCommander should have returned btrfs and no error but returned '", output, "' and error", err) 14 } 15} 16 17func TestStatSSHCommand(t *testing.T) { 18 sshCmder := FakeSSHCommander{ 19 Responses: map[string]string{"sshcommand": "sshcommandresponse"}, 20 } 21 22 output, err := sshCmder.SSHCommand("sshcommand") 23 assert.NoError(t, err) 24 assert.Equal(t, "sshcommandresponse", output) 25 26 output, err = sshCmder.SSHCommand("errorcommand") 27 assert.Error(t, err) 28} 29