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