test(hypervisors): add unit tests for SPT

pkg/unikontainers/hypervisors/spt_test.go

@@ -0,0 +1,147 @@
+// Copyright (c) 2023-2026, Nubificus LTD
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+//     http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+
+package hypervisors
+
+import (
+	"strings"
+	"testing"
+
+	"github.com/stretchr/testify/assert"
+	"github.com/urunc-dev/urunc/pkg/unikontainers/types"
+)
+
+type mockUnikernel struct {
+	netCli   string
+	blockCli []types.MonitorBlockArgs
+	monCli   types.MonitorCliArgs
+}
+
+func (m *mockUnikernel) Init(_ types.UnikernelParams) error        { return nil }
+func (m *mockUnikernel) CommandString() (string, error)            { return "", nil }
+func (m *mockUnikernel) SupportsBlock() bool                       { return false }
+func (m *mockUnikernel) SupportsFS(_ string) bool                  { return false }
+func (m *mockUnikernel) MonitorNetCli(_, _ string) string          { return m.netCli }
+func (m *mockUnikernel) MonitorBlockCli() []types.MonitorBlockArgs { return m.blockCli }
+func (m *mockUnikernel) MonitorCli() types.MonitorCliArgs          { return m.monCli }
+
+func newTestSPT() *SPT {
+	return &SPT{
+		binary:     SptBinary,
+		binaryPath: "/usr/bin/solo5-spt",
+	}
+}
+
+func TestSPTBuildExecCmd(t *testing.T) {
+	t.Parallel()
+
+	tests := []struct {
+		name        string
+		args        types.ExecArgs
+		unikernel   *mockUnikernel
+		wantContain []string
+		wantAbsent  []string
+	}{
+		{
+			name: "default memory when MemSizeB is zero",
+			args: types.ExecArgs{
+				UnikernelPath: "/unikernel.bin",
+			},
+			unikernel:   &mockUnikernel{},
+			wantContain: []string{"--mem=256"},
+		},
+		{
+			name: "memory set from MemSizeB",
+			args: types.ExecArgs{
+				UnikernelPath: "/unikernel.bin",
+				MemSizeB:      512 * 1000 * 1000,
+			},
+			unikernel:   &mockUnikernel{},
+			wantContain: []string{"--mem=512"},
+		},
+		{
+			name: "net cli from unikernel when TapDev set",
+			args: types.ExecArgs{
+				UnikernelPath: "/unikernel.bin",
+				Net: types.NetDevParams{
+					TapDev: "tap0",
+					MAC:    "aa:bb:cc:dd:ee:ff",
+				},
+			},
+			unikernel:   &mockUnikernel{netCli: "--net:service0=tap0"},
+			wantContain: []string{"--net:service0=tap0"},
+		},
+		{
+			name: "no net args when TapDev is empty",
+			args: types.ExecArgs{
+				UnikernelPath: "/unikernel.bin",
+			},
+			unikernel:  &mockUnikernel{},
+			wantAbsent: []string{"--net"},
+		},
+		{
+			name: "block device appended",
+			args: types.ExecArgs{
+				UnikernelPath: "/unikernel.bin",
+			},
+			unikernel: &mockUnikernel{
+				blockCli: []types.MonitorBlockArgs{
+					{ID: "storage0", Path: "/dev/vda"},
+				},
+			},
+			wantContain: []string{"--block:storage0=/dev/vda"},
+		},
+		{
+			name: "extra args from MonitorCli appended",
+			args: types.ExecArgs{
+				UnikernelPath: "/unikernel.bin",
+			},
+			unikernel: &mockUnikernel{
+				monCli: types.MonitorCliArgs{OtherArgs: "--dumpcore"},
+			},
+			wantContain: []string{"--dumpcore"},
+		},
+	}
+
+	for _, tc := range tests {
+		t.Run(tc.name, func(t *testing.T) {
+			t.Parallel()
+			s := newTestSPT()
+			got, err := s.BuildExecCmd(tc.args, tc.unikernel)
+			assert.NoError(t, err)
+			assert.Equal(t, "/usr/bin/solo5-spt", got[0])
+
+			joined := strings.Join(got, " ")
+			for _, want := range tc.wantContain {
+				assert.Contains(t, joined, want)
+			}
+			for _, absent := range tc.wantAbsent {
+				assert.NotContains(t, joined, absent)
+			}
+		})
+	}
+}
+
+func TestSPTBuildExecCmdOrdering(t *testing.T) {
+	t.Parallel()
+	s := newTestSPT()
+	got, err := s.BuildExecCmd(types.ExecArgs{
+		UnikernelPath: "/unikernel.bin",
+		Command:       "console=ttyS0",
+	}, &mockUnikernel{})
+	assert.NoError(t, err)
+	assert.Equal(t, "/usr/bin/solo5-spt", got[0])
+	assert.Equal(t, "/unikernel.bin", got[len(got)-2])
+	assert.Equal(t, "console=ttyS0", got[len(got)-1])
+}