mirror of
https://github.com/Jguer/yay
synced 2024-10-31 04:12:51 +00:00
223 lines
5.7 KiB
Go
223 lines
5.7 KiB
Go
package settings
|
|
|
|
import (
|
|
"encoding/json"
|
|
"os"
|
|
"path/filepath"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
)
|
|
|
|
// GIVEN a non existing build dir in the config
|
|
// WHEN the config is loaded
|
|
// THEN the directory should be created
|
|
func TestNewConfig(t *testing.T) {
|
|
configDir, err := os.MkdirTemp(os.TempDir(), "yay-config-home")
|
|
assert.NoError(t, err)
|
|
err = os.MkdirAll(filepath.Join(configDir, "yay"), 0o755)
|
|
assert.NoError(t, err)
|
|
|
|
os.Setenv("XDG_CONFIG_HOME", configDir)
|
|
|
|
cacheDir, err := os.MkdirTemp(os.TempDir(), "yay-cache-home")
|
|
assert.NoError(t, err)
|
|
|
|
config := map[string]string{"BuildDir": filepath.Join(cacheDir, "test-build-dir")}
|
|
|
|
f, err := os.Create(filepath.Join(configDir, "yay", "config.json"))
|
|
assert.NoError(t, err)
|
|
|
|
defer f.Close()
|
|
|
|
configJSON, _ := json.Marshal(config)
|
|
_, err = f.WriteString(string(configJSON))
|
|
assert.NoError(t, err)
|
|
|
|
newConfig, err := NewConfig("v1.0.0")
|
|
assert.NoError(t, err)
|
|
|
|
assert.Equal(t, filepath.Join(cacheDir, "test-build-dir"), newConfig.BuildDir)
|
|
|
|
_, err = os.Stat(filepath.Join(cacheDir, "test-build-dir"))
|
|
assert.NoError(t, err)
|
|
}
|
|
|
|
// GIVEN a non existing build dir in the config and AURDEST set to a non-existing folder
|
|
// WHEN the config is loaded
|
|
// THEN the directory of AURDEST should be created and selected
|
|
func TestNewConfigAURDEST(t *testing.T) {
|
|
configDir, err := os.MkdirTemp(os.TempDir(), "yay-config-home")
|
|
assert.NoError(t, err)
|
|
err = os.MkdirAll(filepath.Join(configDir, "yay"), 0o755)
|
|
assert.NoError(t, err)
|
|
|
|
os.Setenv("XDG_CONFIG_HOME", configDir)
|
|
|
|
cacheDir, err := os.MkdirTemp(os.TempDir(), "yay-cache-home")
|
|
assert.NoError(t, err)
|
|
|
|
config := map[string]string{"BuildDir": filepath.Join(cacheDir, "test-other-dir")}
|
|
os.Setenv("AURDEST", filepath.Join(cacheDir, "test-build-dir"))
|
|
|
|
f, err := os.Create(filepath.Join(configDir, "yay", "config.json"))
|
|
assert.NoError(t, err)
|
|
|
|
defer f.Close()
|
|
|
|
configJSON, _ := json.Marshal(config)
|
|
_, err = f.WriteString(string(configJSON))
|
|
assert.NoError(t, err)
|
|
|
|
newConfig, err := NewConfig("v1.0.0")
|
|
assert.NoError(t, err)
|
|
|
|
assert.Equal(t, filepath.Join(cacheDir, "test-build-dir"), newConfig.BuildDir)
|
|
|
|
_, err = os.Stat(filepath.Join(cacheDir, "test-build-dir"))
|
|
assert.NoError(t, err)
|
|
}
|
|
|
|
// GIVEN default config
|
|
// WHEN setPrivilegeElevator gets called
|
|
// THEN sudobin should stay as "sudo" (given sudo exists)
|
|
func TestConfiguration_setPrivilegeElevator(t *testing.T) {
|
|
oldPath := os.Getenv("PATH")
|
|
|
|
path, err := os.MkdirTemp("", "yay-test")
|
|
assert.NoError(t, err)
|
|
|
|
doas := filepath.Join(path, "sudo")
|
|
_, err = os.Create(doas)
|
|
os.Chmod(doas, 0o755)
|
|
assert.NoError(t, err)
|
|
|
|
defer os.RemoveAll(path)
|
|
|
|
config := DefaultConfig()
|
|
config.SudoLoop = true
|
|
config.SudoFlags = "-v"
|
|
|
|
os.Setenv("PATH", path)
|
|
err = config.setPrivilegeElevator()
|
|
os.Setenv("PATH", oldPath)
|
|
assert.NoError(t, err)
|
|
|
|
assert.Equal(t, "sudo", config.SudoBin)
|
|
assert.Equal(t, "-v", config.SudoFlags)
|
|
assert.True(t, config.SudoLoop)
|
|
}
|
|
|
|
// GIVEN default config and sudo loop enabled
|
|
// GIVEN only su in path
|
|
// WHEN setPrivilegeElevator gets called
|
|
// THEN sudobin should be changed to "su"
|
|
func TestConfiguration_setPrivilegeElevator_su(t *testing.T) {
|
|
oldPath := os.Getenv("PATH")
|
|
|
|
path, err := os.MkdirTemp("", "yay-test")
|
|
assert.NoError(t, err)
|
|
|
|
doas := filepath.Join(path, "su")
|
|
_, err = os.Create(doas)
|
|
os.Chmod(doas, 0o755)
|
|
assert.NoError(t, err)
|
|
|
|
defer os.RemoveAll(path)
|
|
|
|
config := DefaultConfig()
|
|
config.SudoLoop = true
|
|
config.SudoFlags = "-v"
|
|
|
|
os.Setenv("PATH", path)
|
|
err = config.setPrivilegeElevator()
|
|
os.Setenv("PATH", oldPath)
|
|
|
|
assert.NoError(t, err)
|
|
assert.Equal(t, "su", config.SudoBin)
|
|
assert.Equal(t, "", config.SudoFlags)
|
|
assert.False(t, config.SudoLoop)
|
|
}
|
|
|
|
// GIVEN default config and sudo loop enabled
|
|
// GIVEN no sudo in path
|
|
// WHEN setPrivilegeElevator gets called
|
|
// THEN sudobin should be changed to "su"
|
|
func TestConfiguration_setPrivilegeElevator_no_path(t *testing.T) {
|
|
oldPath := os.Getenv("PATH")
|
|
|
|
os.Setenv("PATH", "")
|
|
config := DefaultConfig()
|
|
config.SudoLoop = true
|
|
config.SudoFlags = "-v"
|
|
|
|
err := config.setPrivilegeElevator()
|
|
os.Setenv("PATH", oldPath)
|
|
|
|
assert.Error(t, err)
|
|
assert.Equal(t, "sudo", config.SudoBin)
|
|
assert.Equal(t, "", config.SudoFlags)
|
|
assert.False(t, config.SudoLoop)
|
|
}
|
|
|
|
// GIVEN default config and sudo loop enabled
|
|
// GIVEN doas in path
|
|
// WHEN setPrivilegeElevator gets called
|
|
// THEN sudobin should be changed to "doas"
|
|
func TestConfiguration_setPrivilegeElevator_doas(t *testing.T) {
|
|
oldPath := os.Getenv("PATH")
|
|
|
|
path, err := os.MkdirTemp("", "yay-test")
|
|
assert.NoError(t, err)
|
|
|
|
doas := filepath.Join(path, "doas")
|
|
_, err = os.Create(doas)
|
|
os.Chmod(doas, 0o755)
|
|
assert.NoError(t, err)
|
|
|
|
defer os.RemoveAll(path)
|
|
|
|
config := DefaultConfig()
|
|
config.SudoLoop = true
|
|
config.SudoFlags = "-v"
|
|
|
|
os.Setenv("PATH", path)
|
|
err = config.setPrivilegeElevator()
|
|
os.Setenv("PATH", oldPath)
|
|
assert.NoError(t, err)
|
|
assert.Equal(t, "doas", config.SudoBin)
|
|
assert.Equal(t, "", config.SudoFlags)
|
|
assert.False(t, config.SudoLoop)
|
|
}
|
|
|
|
// GIVEN config with wrapper and sudo loop enabled
|
|
// GIVEN wrapper is in path
|
|
// WHEN setPrivilegeElevator gets called
|
|
// THEN sudobin should be kept as the wrapper
|
|
func TestConfiguration_setPrivilegeElevator_custom_script(t *testing.T) {
|
|
oldPath := os.Getenv("PATH")
|
|
|
|
path, err := os.MkdirTemp("", "yay-test")
|
|
assert.NoError(t, err)
|
|
|
|
wrapper := filepath.Join(path, "custom-wrapper")
|
|
_, err = os.Create(wrapper)
|
|
os.Chmod(wrapper, 0o755)
|
|
assert.NoError(t, err)
|
|
|
|
defer os.RemoveAll(path)
|
|
|
|
config := DefaultConfig()
|
|
config.SudoLoop = true
|
|
config.SudoBin = wrapper
|
|
config.SudoFlags = "-v"
|
|
|
|
os.Setenv("PATH", path)
|
|
err = config.setPrivilegeElevator()
|
|
os.Setenv("PATH", oldPath)
|
|
|
|
assert.NoError(t, err)
|
|
assert.Equal(t, wrapper, config.SudoBin)
|
|
assert.Equal(t, "-v", config.SudoFlags)
|
|
assert.True(t, config.SudoLoop)
|
|
}
|