Pull request: * all: move internal Go packages to internal/
Merge in DNS/adguard-home from 2234-move-to-internal to master Squashed commit of the following: commit d26a288cabeac86f9483fab307677b1027c78524 Author: Eugene Burkov <e.burkov@adguard.com> Date: Fri Oct 30 12:44:18 2020 +0300 * all: move internal Go packages to internal/ Closes #2234.
This commit is contained in:
Ainar Garipov
210
internal/update/update_test.go
Normal file
210
internal/update/update_test.go
Normal file
@@ -0,0 +1,210 @@
|
||||
package update
|
||||
|
||||
import (
|
||||
"fmt"
|
||||
"io/ioutil"
|
||||
"net"
|
||||
"net/http"
|
||||
"os"
|
||||
"testing"
|
||||
|
||||
"github.com/stretchr/testify/assert"
|
||||
)
|
||||
|
||||
func startHTTPServer(data string) (net.Listener, uint16) {
|
||||
mux := http.NewServeMux()
|
||||
mux.HandleFunc("/", func(w http.ResponseWriter, r *http.Request) {
|
||||
_, _ = w.Write([]byte(data))
|
||||
})
|
||||
|
||||
listener, err := net.Listen("tcp", ":0")
|
||||
if err != nil {
|
||||
panic(err)
|
||||
}
|
||||
|
||||
go func() { _ = http.Serve(listener, mux) }()
|
||||
return listener, uint16(listener.Addr().(*net.TCPAddr).Port)
|
||||
}
|
||||
|
||||
func TestUpdateGetVersion(t *testing.T) {
|
||||
const jsonData = `{
|
||||
"version": "v0.103.0-beta2",
|
||||
"announcement": "AdGuard Home v0.103.0-beta2 is now available!",
|
||||
"announcement_url": "https://github.com/AdguardTeam/AdGuardHome/internal/releases",
|
||||
"selfupdate_min_version": "v0.0",
|
||||
"download_windows_amd64": "https://static.adguard.com/adguardhome/beta/AdGuardHome_windows_amd64.zip",
|
||||
"download_windows_386": "https://static.adguard.com/adguardhome/beta/AdGuardHome_windows_386.zip",
|
||||
"download_darwin_amd64": "https://static.adguard.com/adguardhome/beta/AdGuardHome_darwin_amd64.zip",
|
||||
"download_darwin_386": "https://static.adguard.com/adguardhome/beta/AdGuardHome_darwin_386.zip",
|
||||
"download_linux_amd64": "https://static.adguard.com/adguardhome/beta/AdGuardHome_linux_amd64.tar.gz",
|
||||
"download_linux_386": "https://static.adguard.com/adguardhome/beta/AdGuardHome_linux_386.tar.gz",
|
||||
"download_linux_arm": "https://static.adguard.com/adguardhome/beta/AdGuardHome_linux_armv6.tar.gz",
|
||||
"download_linux_armv5": "https://static.adguard.com/adguardhome/beta/AdGuardHome_linux_armv5.tar.gz",
|
||||
"download_linux_armv6": "https://static.adguard.com/adguardhome/beta/AdGuardHome_linux_armv6.tar.gz",
|
||||
"download_linux_armv7": "https://static.adguard.com/adguardhome/beta/AdGuardHome_linux_armv7.tar.gz",
|
||||
"download_linux_arm64": "https://static.adguard.com/adguardhome/beta/AdGuardHome_linux_arm64.tar.gz",
|
||||
"download_linux_mips": "https://static.adguard.com/adguardhome/beta/AdGuardHome_linux_mips_softfloat.tar.gz",
|
||||
"download_linux_mipsle": "https://static.adguard.com/adguardhome/beta/AdGuardHome_linux_mipsle_softfloat.tar.gz",
|
||||
"download_linux_mips64": "https://static.adguard.com/adguardhome/beta/AdGuardHome_linux_mips64_softfloat.tar.gz",
|
||||
"download_linux_mips64le": "https://static.adguard.com/adguardhome/beta/AdGuardHome_linux_mips64le_softfloat.tar.gz",
|
||||
"download_freebsd_386": "https://static.adguard.com/adguardhome/beta/AdGuardHome_freebsd_386.tar.gz",
|
||||
"download_freebsd_amd64": "https://static.adguard.com/adguardhome/beta/AdGuardHome_freebsd_amd64.tar.gz",
|
||||
"download_freebsd_arm": "https://static.adguard.com/adguardhome/beta/AdGuardHome_freebsd_armv6.tar.gz",
|
||||
"download_freebsd_armv5": "https://static.adguard.com/adguardhome/beta/AdGuardHome_freebsd_armv5.tar.gz",
|
||||
"download_freebsd_armv6": "https://static.adguard.com/adguardhome/beta/AdGuardHome_freebsd_armv6.tar.gz",
|
||||
"download_freebsd_armv7": "https://static.adguard.com/adguardhome/beta/AdGuardHome_freebsd_armv7.tar.gz",
|
||||
"download_freebsd_arm64": "https://static.adguard.com/adguardhome/beta/AdGuardHome_freebsd_arm64.tar.gz"
|
||||
}`
|
||||
|
||||
l, lport := startHTTPServer(jsonData)
|
||||
defer func() { _ = l.Close() }()
|
||||
|
||||
u := NewUpdater(Config{
|
||||
Client: &http.Client{},
|
||||
VersionURL: fmt.Sprintf("http://127.0.0.1:%d/", lport),
|
||||
OS: "linux",
|
||||
Arch: "arm",
|
||||
VersionString: "v0.103.0-beta1",
|
||||
})
|
||||
|
||||
info, err := u.GetVersionResponse(false)
|
||||
assert.Nil(t, err)
|
||||
assert.Equal(t, "v0.103.0-beta2", info.NewVersion)
|
||||
assert.Equal(t, "AdGuard Home v0.103.0-beta2 is now available!", info.Announcement)
|
||||
assert.Equal(t, "https://github.com/AdguardTeam/AdGuardHome/internal/releases", info.AnnouncementURL)
|
||||
assert.Equal(t, "v0.0", info.SelfUpdateMinVersion)
|
||||
assert.True(t, info.CanAutoUpdate)
|
||||
|
||||
_ = l.Close()
|
||||
|
||||
// check cached
|
||||
_, err = u.GetVersionResponse(false)
|
||||
assert.Nil(t, err)
|
||||
}
|
||||
|
||||
func TestUpdate(t *testing.T) {
|
||||
_ = os.Mkdir("aghtest", 0755)
|
||||
defer func() {
|
||||
_ = os.RemoveAll("aghtest")
|
||||
}()
|
||||
|
||||
// create "current" files
|
||||
assert.Nil(t, ioutil.WriteFile("aghtest/AdGuardHome", []byte("AdGuardHome"), 0755))
|
||||
assert.Nil(t, ioutil.WriteFile("aghtest/README.md", []byte("README.md"), 0644))
|
||||
assert.Nil(t, ioutil.WriteFile("aghtest/LICENSE.txt", []byte("LICENSE.txt"), 0644))
|
||||
assert.Nil(t, ioutil.WriteFile("aghtest/AdGuardHome.yaml", []byte("AdGuardHome.yaml"), 0644))
|
||||
|
||||
// start server for returning package file
|
||||
pkgData, err := ioutil.ReadFile("test/AdGuardHome.tar.gz")
|
||||
assert.Nil(t, err)
|
||||
l, lport := startHTTPServer(string(pkgData))
|
||||
defer func() { _ = l.Close() }()
|
||||
|
||||
u := NewUpdater(Config{
|
||||
Client: &http.Client{},
|
||||
PackageURL: fmt.Sprintf("http://127.0.0.1:%d/AdGuardHome.tar.gz", lport),
|
||||
VersionString: "v0.103.0",
|
||||
NewVersion: "v0.103.1",
|
||||
ConfigName: "aghtest/AdGuardHome.yaml",
|
||||
WorkDir: "aghtest",
|
||||
})
|
||||
|
||||
assert.Nil(t, u.prepare())
|
||||
u.currentExeName = "aghtest/AdGuardHome"
|
||||
assert.Nil(t, u.downloadPackageFile(u.PackageURL, u.packageName))
|
||||
assert.Nil(t, u.unpack())
|
||||
// assert.Nil(t, u.check())
|
||||
assert.Nil(t, u.backup())
|
||||
assert.Nil(t, u.replace())
|
||||
u.clean()
|
||||
|
||||
// check backup files
|
||||
d, err := ioutil.ReadFile("aghtest/agh-backup/AdGuardHome.yaml")
|
||||
assert.Nil(t, err)
|
||||
assert.Equal(t, "AdGuardHome.yaml", string(d))
|
||||
|
||||
d, err = ioutil.ReadFile("aghtest/agh-backup/AdGuardHome")
|
||||
assert.Nil(t, err)
|
||||
assert.Equal(t, "AdGuardHome", string(d))
|
||||
|
||||
// check updated files
|
||||
d, err = ioutil.ReadFile("aghtest/AdGuardHome")
|
||||
assert.Nil(t, err)
|
||||
assert.Equal(t, "1", string(d))
|
||||
|
||||
d, err = ioutil.ReadFile("aghtest/README.md")
|
||||
assert.Nil(t, err)
|
||||
assert.Equal(t, "2", string(d))
|
||||
|
||||
d, err = ioutil.ReadFile("aghtest/LICENSE.txt")
|
||||
assert.Nil(t, err)
|
||||
assert.Equal(t, "3", string(d))
|
||||
|
||||
d, err = ioutil.ReadFile("aghtest/AdGuardHome.yaml")
|
||||
assert.Nil(t, err)
|
||||
assert.Equal(t, "AdGuardHome.yaml", string(d))
|
||||
}
|
||||
|
||||
func TestUpdateWindows(t *testing.T) {
|
||||
_ = os.Mkdir("aghtest", 0755)
|
||||
defer func() {
|
||||
_ = os.RemoveAll("aghtest")
|
||||
}()
|
||||
|
||||
// create "current" files
|
||||
assert.Nil(t, ioutil.WriteFile("aghtest/AdGuardHome.exe", []byte("AdGuardHome.exe"), 0755))
|
||||
assert.Nil(t, ioutil.WriteFile("aghtest/README.md", []byte("README.md"), 0644))
|
||||
assert.Nil(t, ioutil.WriteFile("aghtest/LICENSE.txt", []byte("LICENSE.txt"), 0644))
|
||||
assert.Nil(t, ioutil.WriteFile("aghtest/AdGuardHome.yaml", []byte("AdGuardHome.yaml"), 0644))
|
||||
|
||||
// start server for returning package file
|
||||
pkgData, err := ioutil.ReadFile("test/AdGuardHome.zip")
|
||||
assert.Nil(t, err)
|
||||
l, lport := startHTTPServer(string(pkgData))
|
||||
defer func() { _ = l.Close() }()
|
||||
|
||||
u := NewUpdater(Config{
|
||||
WorkDir: "aghtest",
|
||||
Client: &http.Client{},
|
||||
PackageURL: fmt.Sprintf("http://127.0.0.1:%d/AdGuardHome.zip", lport),
|
||||
OS: "windows",
|
||||
VersionString: "v0.103.0",
|
||||
NewVersion: "v0.103.1",
|
||||
ConfigName: "aghtest/AdGuardHome.yaml",
|
||||
})
|
||||
|
||||
assert.Nil(t, u.prepare())
|
||||
u.currentExeName = "aghtest/AdGuardHome.exe"
|
||||
assert.Nil(t, u.downloadPackageFile(u.PackageURL, u.packageName))
|
||||
assert.Nil(t, u.unpack())
|
||||
// assert.Nil(t, u.check())
|
||||
assert.Nil(t, u.backup())
|
||||
assert.Nil(t, u.replace())
|
||||
u.clean()
|
||||
|
||||
// check backup files
|
||||
d, err := ioutil.ReadFile("aghtest/agh-backup/AdGuardHome.yaml")
|
||||
assert.Nil(t, err)
|
||||
assert.Equal(t, "AdGuardHome.yaml", string(d))
|
||||
|
||||
d, err = ioutil.ReadFile("aghtest/agh-backup/AdGuardHome.exe")
|
||||
assert.Nil(t, err)
|
||||
assert.Equal(t, "AdGuardHome.exe", string(d))
|
||||
|
||||
// check updated files
|
||||
d, err = ioutil.ReadFile("aghtest/AdGuardHome.exe")
|
||||
assert.Nil(t, err)
|
||||
assert.Equal(t, "1", string(d))
|
||||
|
||||
d, err = ioutil.ReadFile("aghtest/README.md")
|
||||
assert.Nil(t, err)
|
||||
assert.Equal(t, "2", string(d))
|
||||
|
||||
d, err = ioutil.ReadFile("aghtest/LICENSE.txt")
|
||||
assert.Nil(t, err)
|
||||
assert.Equal(t, "3", string(d))
|
||||
|
||||
d, err = ioutil.ReadFile("aghtest/AdGuardHome.yaml")
|
||||
assert.Nil(t, err)
|
||||
assert.Equal(t, "AdGuardHome.yaml", string(d))
|
||||
}
|
||||
Reference in New Issue
Block a user