Adapt tests

Signed-off-by: Kimmo Lehto <klehto@mirantis.com>
k0sproject · Jan 4, 2024 · 6eeb781 · 6eeb781
1 parent a08415b
commit 6eeb781
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 1 deletion.
diff --git a/configurer/linux/linux_test.go b/configurer/linux/linux_test.go
@@ -2,6 +2,7 @@ package linux
 
 import (
 	"fmt"
+	"io"
 	"testing"
 
 	"github.com/k0sproject/rig/exec"
@@ -35,6 +36,10 @@ func (m mockHost) ExecOutputf(string, ...interface{}) (string, error) {
 	return "", nil
 }
 
+func (m mockHost) ExecStreams(cmd string, stdin io.ReadCloser, stdout io.Writer, stderr io.Writer, opts ...exec.Option) (exec.Waiter, error) {
+	return nil, nil
+}
+
 func (m mockHost) String() string {
 	return ""
 }

diff --git a/pkg/apis/k0sctl.k0sproject.io/v1beta1/cluster/host_test.go b/pkg/apis/k0sctl.k0sproject.io/v1beta1/cluster/host_test.go
@@ -6,6 +6,7 @@ import (
 
 	cfg "github.com/k0sproject/k0sctl/configurer"
 	"github.com/k0sproject/k0sctl/configurer/linux"
+	"github.com/k0sproject/rig"
 	"github.com/k0sproject/rig/exec"
 	"github.com/k0sproject/rig/os"
 	"github.com/stretchr/testify/require"
@@ -69,7 +70,8 @@ func TestUnQE(t *testing.T) {
 }
 
 func TestK0sInstallCommand(t *testing.T) {
-	h := Host{Role: "worker", DataDir: "/tmp/k0s"}
+	h := Host{Role: "worker", DataDir: "/tmp/k0s", Connection: rig.Connection{Localhost: &rig.Localhost{Enabled: true}}}
+	_ = h.Connect()
 	h.Configurer = &mockconfigurer{}
 	h.Configurer.SetPath("K0sConfigPath", "from-configurer")
 	h.Configurer.SetPath("K0sJoinTokenPath", "from-configurer")