test (machine) : add vsock interaction methods to VirtualMachine interface

Make VirtualMachine implement vsock methods so that vsock interaction
is also done via VirtualMachine interface. This would help in writing
tests, we can override the behavior of expose/unexpose in fake vm
implementation.

Signed-off-by: Rohan Kumar <[email protected]>

Author: rohanKanojia

pkg/crc/machine/delete.go

@@ -22,7 +22,7 @@ func (client *client) Delete() error {
 
 	// In case usermode networking make sure all the port bind on host should be released
 	if client.useVSock() {
-		if err := unexposePorts(); err != nil {
+		if err := vm.UnExposePorts(); err != nil {
 			return err
 		}
 	}

pkg/crc/machine/fakemachine/virtualmachine.go

@@ -5,6 +5,7 @@ import (
 
 	"github.com/crc-org/crc/v2/pkg/crc/machine/bundle"
 	"github.com/crc-org/crc/v2/pkg/crc/machine/state"
+	crcPreset "github.com/crc-org/crc/v2/pkg/crc/preset"
 	"github.com/crc-org/crc/v2/pkg/crc/ssh"
 	"github.com/crc-org/crc/v2/pkg/libmachine"
 	libmachinehost "github.com/crc-org/crc/v2/pkg/libmachine/host"
@@ -39,6 +40,7 @@ type FakeVirtualMachine struct {
 	FailingStop   bool
 	FailingState  bool
 	FakeSSHClient *FakeSSHClient
+	PortsExposed  bool
 }
 
 func (vm *FakeVirtualMachine) Close() error {
@@ -98,6 +100,16 @@ func (vm *FakeVirtualMachine) GetHost() *libmachinehost.Host {
 	panic("not implemented")
 }
 
+func (vm *FakeVirtualMachine) ExposePorts(_ crcPreset.Preset, _, _ uint) error {
+	vm.PortsExposed = true
+	return nil
+}
+
+func (vm *FakeVirtualMachine) UnExposePorts() error {
+	vm.PortsExposed = false
+	return nil
+}
+
 func NewFakeVirtualMachine(failingStop bool, failingState bool) *FakeVirtualMachine {
 	return &FakeVirtualMachine{
 		FakeSSHClient: NewFakeSSHClient(),

pkg/crc/machine/start.go

@@ -368,7 +368,7 @@ func (client *client) Start(ctx context.Context, startConfig types.StartConfig)
 	logging.Infof("Starting CRC VM for %s %s...", startConfig.Preset, vm.Bundle().GetVersion())
 
 	if client.useVSock() {
-		if err := exposePorts(startConfig.Preset, startConfig.IngressHTTPPort, startConfig.IngressHTTPSPort); err != nil {
+		if err := vm.ExposePorts(startConfig.Preset, startConfig.IngressHTTPPort, startConfig.IngressHTTPSPort); err != nil {
 			return nil, err
 		}
 	}

pkg/crc/machine/stop.go

@@ -45,7 +45,7 @@ func (client *client) Stop() (state.State, error) {
 	}
 	// In case usermode networking make sure all the port bind on host should be released
 	if client.useVSock() {
-		return status, unexposePorts()
+		return status, vm.UnExposePorts()
 	}
 	return status, nil
 }

pkg/crc/machine/stop_test.go

@@ -19,18 +19,19 @@ func TestStop_WhenVMRunning_ThenShouldStopVirtualMachine(t *testing.T) {
 		"network-mode")
 	_, err := crcConfigStorage.Set(crcConfig.NetworkMode, "true")
 	assert.NoError(t, err)
-	virtualMachine := fakemachine.NewFakeVirtualMachine(false, false)
-	client := newClientWithVirtualMachine("fake-virtual-machine", false, crcConfigStorage, virtualMachine)
+	fakeVirtualMachine := fakemachine.NewFakeVirtualMachine(false, false)
+	client := newClientWithVirtualMachine("fake-virtual-machine", false, crcConfigStorage, fakeVirtualMachine)
 
 	// When
 	clusterState, stopErr := client.Stop()
 
 	// Then
 	assert.NoError(t, stopErr)
 	assert.Equal(t, clusterState, state.Stopped)
-	assert.Equal(t, virtualMachine.IsStopped, true)
-	assert.Equal(t, virtualMachine.FakeSSHClient.LastExecutedCommand, "sudo -- sh -c 'crictl stop $(crictl ps -q)'")
-	assert.Equal(t, virtualMachine.FakeSSHClient.IsSSHClientClosed, true)
+	assert.Equal(t, fakeVirtualMachine.IsStopped, true)
+	assert.Equal(t, fakeVirtualMachine.FakeSSHClient.LastExecutedCommand, "sudo -- sh -c 'crictl stop $(crictl ps -q)'")
+	assert.Equal(t, fakeVirtualMachine.FakeSSHClient.IsSSHClientClosed, true)
+	assert.Equal(t, fakeVirtualMachine.PortsExposed, false)
 }
 
 func TestStop_WhenStopVmFailed_ThenErrorThrown(t *testing.T) {
@@ -40,8 +41,8 @@ func TestStop_WhenStopVmFailed_ThenErrorThrown(t *testing.T) {
 		"network-mode")
 	_, err := crcConfigStorage.Set(crcConfig.NetworkMode, "true")
 	assert.NoError(t, err)
-	virtualMachine := fakemachine.NewFakeVirtualMachine(true, false)
-	client := newClientWithVirtualMachine("fake-virtual-machine", false, crcConfigStorage, virtualMachine)
+	fakeVirtualMachine := fakemachine.NewFakeVirtualMachine(true, false)
+	client := newClientWithVirtualMachine("fake-virtual-machine", false, crcConfigStorage, fakeVirtualMachine)
 
 	// When
 	_, stopErr := client.Stop()
@@ -57,18 +58,18 @@ func TestStop_WhenVMAlreadyStopped_ThenThrowError(t *testing.T) {
 		"network-mode")
 	_, err := crcConfigStorage.Set(crcConfig.NetworkMode, "true")
 	assert.NoError(t, err)
-	virtualMachine := fakemachine.NewFakeVirtualMachine(false, false)
-	err = virtualMachine.Stop()
+	fakeVirtualMachine := fakemachine.NewFakeVirtualMachine(false, false)
+	err = fakeVirtualMachine.Stop()
 	assert.NoError(t, err)
-	client := newClientWithVirtualMachine("fake-virtual-machine", false, crcConfigStorage, virtualMachine)
+	client := newClientWithVirtualMachine("fake-virtual-machine", false, crcConfigStorage, fakeVirtualMachine)
 
 	// When
 	clusterState, stopErr := client.Stop()
 
 	// Then
 	assert.EqualError(t, stopErr, "Instance is already stopped")
 	assert.Equal(t, clusterState, state.Error)
-	assert.Equal(t, virtualMachine.IsStopped, true)
+	assert.Equal(t, fakeVirtualMachine.IsStopped, true)
 }
 
 func TestClient_WhenStopInvokedWithNonExistentVM_ThenThrowError(t *testing.T) {

pkg/crc/machine/virtualmachine.go

@@ -3,6 +3,8 @@ package machine
 import (
 	"fmt"
 
+	crcPreset "github.com/crc-org/crc/v2/pkg/crc/preset"
+
 	"github.com/crc-org/crc/v2/pkg/crc/constants"
 	"github.com/crc-org/crc/v2/pkg/crc/machine/bundle"
 	"github.com/crc-org/crc/v2/pkg/crc/machine/state"
@@ -26,6 +28,8 @@ type VirtualMachine interface {
 	Driver() drivers.Driver
 	API() libmachine.API
 	Host() *libmachinehost.Host
+	ExposePorts(preset crcPreset.Preset, ingressHTTPPort, ingressHTTPSPort uint) error
+	UnExposePorts() error
 }
 
 type virtualMachine struct {

pkg/crc/machine/vsock.go

@@ -16,7 +16,7 @@ import (
 	"github.com/pkg/errors"
 )
 
-func exposePorts(preset crcPreset.Preset, ingressHTTPPort, ingressHTTPSPort uint) error {
+func (vm *virtualMachine) ExposePorts(preset crcPreset.Preset, ingressHTTPPort, ingressHTTPSPort uint) error {
 	portsToExpose := vsockPorts(preset, ingressHTTPPort, ingressHTTPSPort)
 	daemonClient := daemonclient.New()
 	alreadyOpenedPorts, err := listOpenPorts(daemonClient)
@@ -47,7 +47,7 @@ func isOpened(exposed []types.ExposeRequest, port types.ExposeRequest) bool {
 	return false
 }
 
-func unexposePorts() error {
+func (vm *virtualMachine) UnExposePorts() error {
 	var mErr crcErrors.MultiError
 	daemonClient := daemonclient.New()
 	alreadyOpenedPorts, err := listOpenPorts(daemonClient)