mirror of
https://github.com/AvengeMedia/DankMaterialShell.git
synced 2026-10-14 15:24:00 -04:00
70 lines
2.0 KiB
Go
70 lines
2.0 KiB
Go
package utils
|
|
|
|
import (
|
|
"os"
|
|
"path/filepath"
|
|
"strings"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func TestCommandExistsFallsBackToLocalBin(t *testing.T) {
|
|
home := t.TempDir()
|
|
binDir := filepath.Join(home, ".local", "bin")
|
|
require.NoError(t, os.MkdirAll(binDir, 0o755))
|
|
require.NoError(t, os.WriteFile(filepath.Join(binDir, "pywalfox"), []byte("#!/bin/sh\n"), 0o755))
|
|
|
|
t.Setenv("HOME", home)
|
|
t.Setenv("PATH", t.TempDir())
|
|
|
|
assert.True(t, CommandExists("pywalfox"))
|
|
}
|
|
|
|
func TestCommandExistsFallsBackToNixProfileBin(t *testing.T) {
|
|
home := t.TempDir()
|
|
binDir := filepath.Join(home, ".nix-profile", "bin")
|
|
require.NoError(t, os.MkdirAll(binDir, 0o755))
|
|
require.NoError(t, os.WriteFile(filepath.Join(binDir, "pywalfox"), []byte("#!/bin/sh\n"), 0o755))
|
|
|
|
t.Setenv("HOME", home)
|
|
t.Setenv("PATH", t.TempDir())
|
|
|
|
assert.True(t, CommandExists("pywalfox"))
|
|
}
|
|
|
|
func TestCommandExistsIgnoresNonExecutableLocalBinFile(t *testing.T) {
|
|
home := t.TempDir()
|
|
binDir := filepath.Join(home, ".local", "bin")
|
|
require.NoError(t, os.MkdirAll(binDir, 0o755))
|
|
require.NoError(t, os.WriteFile(filepath.Join(binDir, "pywalfox"), []byte("not executable"), 0o644))
|
|
|
|
t.Setenv("HOME", home)
|
|
t.Setenv("PATH", t.TempDir())
|
|
|
|
assert.False(t, CommandExists("pywalfox"))
|
|
}
|
|
|
|
func TestEnvWithUserBinPathPrependsLocalBin(t *testing.T) {
|
|
home := t.TempDir()
|
|
t.Setenv("HOME", home)
|
|
|
|
env := EnvWithUserBinPath([]string{"PATH=/usr/bin", "OTHER=value"})
|
|
var pathValue string
|
|
for _, entry := range env {
|
|
if after, ok := strings.CutPrefix(entry, "PATH="); ok {
|
|
pathValue = after
|
|
break
|
|
}
|
|
}
|
|
|
|
parts := filepath.SplitList(pathValue)
|
|
require.NotEmpty(t, parts)
|
|
assert.Equal(t, filepath.Join(home, ".local", "bin"), parts[0])
|
|
assert.Equal(t, filepath.Join(home, ".nix-profile", "bin"), parts[1])
|
|
assert.Contains(t, parts, filepath.Join("/etc/profiles/per-user", filepath.Base(home), "bin"))
|
|
assert.Contains(t, parts, "/usr/local/bin")
|
|
assert.Contains(t, env, "OTHER=value")
|
|
}
|