all: sync with master; upd chlog

internal/updater/updater_internal_test.go

@@ -0,0 +1,107 @@
+package updater
+
+import (
+	"os"
+	"path/filepath"
+	"testing"
+
+	"github.com/AdguardTeam/AdGuardHome/internal/aghtest"
+	"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,
+			GOOS:     tc.os,
+			Version:  "v0.103.0",
+			ExecPath: exePath,
+			WorkDir:  wd,
+			ConfName: yamlPath,
+		})
+
+		u.newVersion = "v0.103.1"
+		u.packageURL = fakeURL.String()
+
+		require.NoError(t, u.prepare())
+		require.NoError(t, u.downloadPackageFile())
+		require.NoError(t, u.unpack())
+		require.NoError(t, u.backup(false))
+		require.NoError(t, u.replace())
+
+		u.clean()
+
+		// check backup files
+		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))
+
+		// check updated files
+		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))
+	}
+}