Merge pull request #199 from lstocchi/testVirtionet

virtionet: testing connectUnixPath

pkg/vf/virtionet_test.go

@@ -0,0 +1,84 @@
+package vf
+
+import (
+ "bytes"
+ "fmt"
+ "math/rand"
+ "net"
+ "os"
+ "path/filepath"
+ "strings"
+ "testing"
+
+ "github.com/crc-org/vfkit/pkg/config"
+ "github.com/stretchr/testify/require"
+)
+
+func testConnectUnixgram(t *testing.T) error {
+ unixSocketPath := filepath.Join("/tmp", fmt.Sprintf("vnet-test-%x.sock", rand.Int31n(0xffff))) //#nosec G404 -- no need for crypto/rand here
+ addr, err := net.ResolveUnixAddr("unixgram", unixSocketPath)
+ require.NoError(t, err)
+
+ l, err := net.ListenUnixgram("unixgram", addr)
+ require.NoError(t, err)
+
+ defer l.Close()
+ defer os.Remove(unixSocketPath)
+
+ dev := &VirtioNet{
+ &config.VirtioNet{
+ UnixSocketPath: unixSocketPath,
+ },
+ &net.UnixAddr{},
+ }
+
+ return dev.connectUnixPath()
+}
+
+func TestConnectUnixPath(t *testing.T) {
+ t.Run("Successful connection - no error", func(t *testing.T) {
+ err := testConnectUnixgram(t)
+ require.NoError(t, err)
+ })
+
+ t.Run("Failed connection - End socket longer than 104 bytes", func(t *testing.T) {
+ // Retrieve HOME env variable (used by the os.UserHomeDir)
+ origUserHome := os.Getenv("HOME")
+ defer func() {
+ os.Setenv("HOME", origUserHome)
+ }()
+
+ // Create a string of 100 bytes to update the user home to be sure to create a socket path > 104 bytes
+ b := bytes.Repeat([]byte("a"), 100)
+ subDir := string(b)
+
+ // Update HOME env so os.UserHomeDir returns the update path with subfolder
+ updatedUserHome := filepath.Join(origUserHome, subDir)
+ os.Setenv("HOME", updatedUserHome)
+ defer os.RemoveAll(updatedUserHome)
+
+ err := testConnectUnixgram(t)
+ // It should return an error
+ require.Error(t, err)
+ require.ErrorContains(t, err, "invalid argument")
+ })
+}
+
+func TestLocalUnixSocketPath(t *testing.T) {
+ t.Run("Success case - Creates temporary socket path", func(t *testing.T) {
+ // Retrieve HOME env variable (used by the os.UserHomeDir)
+ userHome := os.Getenv("HOME")
+
+ path, err := localUnixSocketPath()
+
+ // Assert successful execution
+ require.NoError(t, err)
+
+ // Check if path starts with the expected prefix
+ expectedPrefix := filepath.Join(userHome, "Library", "Application Support", "vfkit")
+ require.Truef(t, strings.HasPrefix(path, expectedPrefix), "Path doesn't start with expected prefix: %v", path)
+
+ // Check if path ends with a socket extension
+ require.Equalf(t, ".sock", filepath.Ext(path), "Path doesn't end with .sock extension: %v, ext is %v", path, filepath.Ext(path))
+ })
+}