lazydocker/pkg/commands/socket_detection_windows_test.go
DeiAsPie 81e4e44309
Enhance Podman support with improved socket detection and error handling across platforms
Signed-off-by: DeiAsPie <93835541+DeiAsPie@users.noreply.github.com>
2025-12-27 14:56:14 +05:30

102 lines
3.1 KiB
Go

//go:build windows
package commands
import (
"context"
"errors"
"os"
"path/filepath"
"testing"
"github.com/sirupsen/logrus"
"github.com/stretchr/testify/assert"
)
func TestDetectPlatformCandidates_Windows_DockerSuccess(t *testing.T) {
oldValidate := validateSocketFunc
defer func() { validateSocketFunc = oldValidate }()
validateSocketFunc = func(ctx context.Context, host string, useEnv bool) error {
if host == "npipe:////./pipe/docker_engine" {
return nil
}
return errors.New("mock failure")
}
log := logrus.NewEntry(logrus.New())
host, runtime, err := detectPlatformCandidates(log)
assert.NoError(t, err)
assert.Equal(t, "npipe:////./pipe/docker_engine", host)
assert.Equal(t, RuntimeDocker, runtime)
}
func TestDetectPlatformCandidates_Windows_PodmanSuccess(t *testing.T) {
oldValidate := validateSocketFunc
defer func() { validateSocketFunc = oldValidate }()
validateSocketFunc = func(ctx context.Context, host string, useEnv bool) error {
if host == "npipe:////./pipe/podman-machine-default" {
return nil
}
return errors.New("mock failure")
}
log := logrus.NewEntry(logrus.New())
host, runtime, err := detectPlatformCandidates(log)
assert.NoError(t, err)
assert.Equal(t, "npipe:////./pipe/podman-machine-default", host)
assert.Equal(t, RuntimePodman, runtime)
}
func TestDetectPlatformCandidates_Windows_Failure(t *testing.T) {
oldValidate := validateSocketFunc
defer func() { validateSocketFunc = oldValidate }()
validateSocketFunc = func(ctx context.Context, host string, useEnv bool) error {
return errors.New("mock failure")
}
log := logrus.NewEntry(logrus.New())
_, _, err := detectPlatformCandidates(log)
assert.Error(t, err)
assert.Equal(t, ErrNoDockerSocket, err)
}
func TestGetPodmanPipes(t *testing.T) {
log := logrus.NewEntry(logrus.New())
// Test fallback when home dir is missing or empty
// We can't easily mock UserHomeDir without refactoring, but we can test the logic
// by creating the expected directory structure in a temp dir and setting USERPROFILE.
tmpDir, err := os.MkdirTemp("", "podman-test")
assert.NoError(t, err)
defer os.RemoveAll(tmpDir)
// Set USERPROFILE to our temp dir
oldProfile := os.Getenv("USERPROFILE")
os.Setenv("USERPROFILE", tmpDir)
defer os.Setenv("USERPROFILE", oldProfile)
// 1. Test fallback when directory doesn't exist
pipes := getPodmanPipes(log)
assert.Equal(t, []string{"npipe:////./pipe/podman-machine-default"}, pipes)
// 2. Test with actual config files
configDir := filepath.Join(tmpDir, ".config", "containers", "podman", "machine", "wsl")
err = os.MkdirAll(configDir, 0755)
assert.NoError(t, err)
err = os.WriteFile(filepath.Join(configDir, "machine1.json"), []byte("{}"), 0644)
assert.NoError(t, err)
err = os.WriteFile(filepath.Join(configDir, "machine2.json"), []byte("{}"), 0644)
assert.NoError(t, err)
err = os.WriteFile(filepath.Join(configDir, "not-a-config.txt"), []byte("{}"), 0644)
assert.NoError(t, err)
pipes = getPodmanPipes(log)
assert.Len(t, pipes, 2)
assert.Contains(t, pipes, "npipe:////./pipe/machine1")
assert.Contains(t, pipes, "npipe:////./pipe/machine2")
}