ae840c9c96
Squashed commit of the following:
commit 89c3df471964b674b7ddafeb22566e5be9b56a13
Author: Ainar Garipov <A.Garipov@AdGuard.COM>
Date: Mon May 12 18:59:39 2025 +0300
updater: imp log
commit d78ba4368027ddcbb41c10fbf09d43fe0721dc4c
Merge: 68410954c 187b759fc
Author: Ainar Garipov <A.Garipov@AdGuard.COM>
Date: Mon May 12 18:53:33 2025 +0300
Merge branch 'master' into AGDNS-2374-updater-slog
commit 68410954c80d76b2adafe4ed28fafdd6b6b6daae
Author: Ainar Garipov <A.Garipov@AdGuard.COM>
Date: Wed Apr 30 15:54:30 2025 +0300
updater: imp docs
commit 99a705218fb849bb59dee5b801c5279a501bcf98
Author: Ainar Garipov <A.Garipov@AdGuard.COM>
Date: Wed Apr 30 15:40:30 2025 +0300
updater: imp docs, logs
commit 2a83ee3ebf9610a2703d99ec6a6b327a315f6cce
Author: Ainar Garipov <A.Garipov@AdGuard.COM>
Date: Tue Apr 29 21:01:02 2025 +0300
updater: use slog
126 lines
3.3 KiB
Go
126 lines
3.3 KiB
Go
package updater
|
|
|
|
import (
|
|
"context"
|
|
"net/url"
|
|
"os"
|
|
"path/filepath"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/AdguardTeam/AdGuardHome/internal/aghtest"
|
|
"github.com/AdguardTeam/golibs/logutil/slogutil"
|
|
"github.com/AdguardTeam/golibs/testutil"
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func TestUpdater_internal(t *testing.T) {
|
|
wd := t.TempDir()
|
|
|
|
exePathUnix := filepath.Join(wd, "AdGuardHome.exe")
|
|
exePathWindows := filepath.Join(wd, "AdGuardHome")
|
|
yamlPath := filepath.Join(wd, "AdGuardHome.yaml")
|
|
readmePath := filepath.Join(wd, "README.md")
|
|
licensePath := filepath.Join(wd, "LICENSE.txt")
|
|
|
|
require.NoError(t, os.WriteFile(exePathUnix, []byte("AdGuardHome.exe"), 0o755))
|
|
require.NoError(t, os.WriteFile(exePathWindows, []byte("AdGuardHome"), 0o755))
|
|
require.NoError(t, os.WriteFile(yamlPath, []byte("AdGuardHome.yaml"), 0o644))
|
|
require.NoError(t, os.WriteFile(readmePath, []byte("README.md"), 0o644))
|
|
require.NoError(t, os.WriteFile(licensePath, []byte("LICENSE.txt"), 0o644))
|
|
|
|
testCases := []struct {
|
|
name string
|
|
exeName string
|
|
os string
|
|
archiveName string
|
|
}{{
|
|
name: "unix",
|
|
os: "linux",
|
|
exeName: "AdGuardHome",
|
|
archiveName: "AdGuardHome.tar.gz",
|
|
}, {
|
|
name: "windows",
|
|
os: "windows",
|
|
exeName: "AdGuardHome.exe",
|
|
archiveName: "AdGuardHome.zip",
|
|
}}
|
|
|
|
for _, tc := range testCases {
|
|
exePath := filepath.Join(wd, tc.exeName)
|
|
|
|
// Start server for returning package file.
|
|
pkgData, err := os.ReadFile(filepath.Join("testdata", tc.archiveName))
|
|
require.NoError(t, err)
|
|
|
|
fakeClient, fakeURL := aghtest.StartHTTPServer(t, pkgData)
|
|
fakeURL = fakeURL.JoinPath(tc.archiveName)
|
|
|
|
u := NewUpdater(&Config{
|
|
Client: fakeClient,
|
|
Logger: slogutil.NewDiscardLogger(),
|
|
GOOS: tc.os,
|
|
Version: "v0.103.0",
|
|
ExecPath: exePath,
|
|
WorkDir: wd,
|
|
ConfName: yamlPath,
|
|
// TODO(e.burkov): Rewrite the test to use a fake version check
|
|
// URL with a fake URLs for the package files.
|
|
VersionCheckURL: &url.URL{},
|
|
})
|
|
|
|
u.newVersion = "v0.103.1"
|
|
u.packageURL = fakeURL.String()
|
|
|
|
require.NoError(t, u.prepare(newCtx(t)))
|
|
require.NoError(t, u.downloadPackageFile(newCtx(t)))
|
|
require.NoError(t, u.unpack(newCtx(t)))
|
|
require.NoError(t, u.backup(newCtx(t), false))
|
|
require.NoError(t, u.replace(newCtx(t)))
|
|
|
|
u.clean(newCtx(t))
|
|
|
|
require.True(t, t.Run("backup", func(t *testing.T) {
|
|
var d []byte
|
|
d, err = os.ReadFile(filepath.Join(wd, "agh-backup", "AdGuardHome.yaml"))
|
|
require.NoError(t, err)
|
|
|
|
assert.Equal(t, "AdGuardHome.yaml", string(d))
|
|
|
|
d, err = os.ReadFile(filepath.Join(wd, "agh-backup", tc.exeName))
|
|
require.NoError(t, err)
|
|
|
|
assert.Equal(t, tc.exeName, string(d))
|
|
}))
|
|
|
|
require.True(t, t.Run("updated", func(t *testing.T) {
|
|
var d []byte
|
|
d, err = os.ReadFile(exePath)
|
|
require.NoError(t, err)
|
|
|
|
assert.Equal(t, "1", string(d))
|
|
|
|
d, err = os.ReadFile(readmePath)
|
|
require.NoError(t, err)
|
|
|
|
assert.Equal(t, "2", string(d))
|
|
|
|
d, err = os.ReadFile(licensePath)
|
|
require.NoError(t, err)
|
|
|
|
assert.Equal(t, "3", string(d))
|
|
|
|
d, err = os.ReadFile(yamlPath)
|
|
require.NoError(t, err)
|
|
|
|
assert.Equal(t, "AdGuardHome.yaml", string(d))
|
|
}))
|
|
}
|
|
}
|
|
|
|
// newCtx is a helper that returns a new context with a timeout.
|
|
func newCtx(tb testing.TB) (ctx context.Context) {
|
|
return testutil.ContextWithTimeout(tb, 1*time.Second)
|
|
}
|