mirror of
https://github.com/rcourtman/Pulse.git
synced 2026-05-08 18:21:55 +00:00
110 lines
3.2 KiB
Go
110 lines
3.2 KiB
Go
package updates
|
|
|
|
import (
|
|
"context"
|
|
"crypto/sha256"
|
|
"encoding/hex"
|
|
"os"
|
|
"path/filepath"
|
|
"strings"
|
|
"testing"
|
|
)
|
|
|
|
func TestInstallShAdapter_DetectServiceName(t *testing.T) {
|
|
dir := t.TempDir()
|
|
systemctl := filepath.Join(dir, "systemctl")
|
|
script := `#!/bin/sh
|
|
if [ "$1" = "is-active" ] && [ "$2" = "pulse-backend" ]; then
|
|
echo "active"
|
|
exit 0
|
|
fi
|
|
echo "inactive"
|
|
exit 0
|
|
`
|
|
if err := os.WriteFile(systemctl, []byte(script), 0755); err != nil {
|
|
t.Fatalf("write systemctl: %v", err)
|
|
}
|
|
|
|
t.Setenv("PATH", dir+string(os.PathListSeparator)+os.Getenv("PATH"))
|
|
|
|
adapter := &InstallShAdapter{}
|
|
name, err := adapter.detectServiceName()
|
|
if err != nil {
|
|
t.Fatalf("detectServiceName error: %v", err)
|
|
}
|
|
if name != "pulse-backend" {
|
|
t.Fatalf("expected pulse-backend, got %q", name)
|
|
}
|
|
}
|
|
|
|
func TestInstallShAdapter_DownloadInstallScript(t *testing.T) {
|
|
content := "echo hi"
|
|
sum := sha256.Sum256([]byte(content))
|
|
checksum := hex.EncodeToString(sum[:])
|
|
|
|
dir := t.TempDir()
|
|
curl := filepath.Join(dir, "curl")
|
|
script := strings.Join([]string{
|
|
"#!/bin/sh",
|
|
`out=""`,
|
|
`url=""`,
|
|
`while [ "$#" -gt 0 ]; do`,
|
|
` if [ "$1" = "-o" ]; then`,
|
|
` out="$2"`,
|
|
` shift 2`,
|
|
` continue`,
|
|
` fi`,
|
|
` url="$1"`,
|
|
` shift`,
|
|
`done`,
|
|
`if echo "$url" | grep -q ".sha256$"; then`,
|
|
` echo "` + checksum + ` install.sh" > "$out"`,
|
|
`else`,
|
|
` printf '%s' "` + content + `" > "$out"`,
|
|
`fi`,
|
|
``,
|
|
}, "\n")
|
|
if err := os.WriteFile(curl, []byte(script), 0755); err != nil {
|
|
t.Fatalf("write curl: %v", err)
|
|
}
|
|
t.Setenv("PATH", dir+string(os.PathListSeparator)+os.Getenv("PATH"))
|
|
|
|
adapter := &InstallShAdapter{installScriptURL: "http://example/install.sh"}
|
|
out, err := adapter.downloadInstallScript(context.Background(), "")
|
|
if err != nil {
|
|
t.Fatalf("downloadInstallScript error: %v", err)
|
|
}
|
|
if out != content {
|
|
t.Fatalf("unexpected script content: %q", out)
|
|
}
|
|
}
|
|
|
|
func TestInstallShAdapter_InstallScriptURLForVersion(t *testing.T) {
|
|
adapter := NewInstallShAdapter(nil)
|
|
got := adapter.installScriptURLForVersion("v1.2.3")
|
|
want := "https://github.com/rcourtman/Pulse/releases/download/v1.2.3/install.sh"
|
|
if got != want {
|
|
t.Fatalf("installScriptURLForVersion() = %q, want %q", got, want)
|
|
}
|
|
|
|
custom := &InstallShAdapter{installScriptURL: "http://example/install.sh"}
|
|
if got := custom.installScriptURLForVersion("v1.2.3"); got != "http://example/install.sh" {
|
|
t.Fatalf("custom installScriptURLForVersion() = %q, want custom URL", got)
|
|
}
|
|
}
|
|
|
|
func TestInstallShReleaseAssetURLUsesConfiguredRepo(t *testing.T) {
|
|
t.Setenv("PULSE_GITHUB_REPO", "example/pulse-fork")
|
|
|
|
if got := defaultInstallScriptLatestURL(); got != "https://github.com/example/pulse-fork/releases/latest/download/install.sh" {
|
|
t.Fatalf("defaultInstallScriptLatestURL() = %q", got)
|
|
}
|
|
|
|
if got := installShReleaseAssetURL("v1.2.3", "install.sh"); got != "https://github.com/example/pulse-fork/releases/download/v1.2.3/install.sh" {
|
|
t.Fatalf("installShReleaseAssetURL() = %q", got)
|
|
}
|
|
|
|
if got := installShReleaseAssetURL("v1.2.3", "pulse-v1.2.3-linux-amd64.tar.gz"); got != "https://github.com/example/pulse-fork/releases/download/v1.2.3/pulse-v1.2.3-linux-amd64.tar.gz" {
|
|
t.Fatalf("installShReleaseAssetURL() tarball = %q", got)
|
|
}
|
|
}
|