89 lines
2.4 KiB
Go
89 lines
2.4 KiB
Go
package frp
|
|
|
|
import (
|
|
"archive/zip"
|
|
"bytes"
|
|
"os"
|
|
"path/filepath"
|
|
"runtime"
|
|
"testing"
|
|
"time"
|
|
)
|
|
|
|
func TestSupportedReleasesUsesOfficialDigest(t *testing.T) {
|
|
version := "v0.99.0"
|
|
extension := ".tar.gz"
|
|
if runtime.GOOS == "windows" {
|
|
extension = ".zip"
|
|
}
|
|
assetName := "frp_0.99.0_" + runtime.GOOS + "_" + runtime.GOARCH + extension
|
|
releases := supportedReleases([]githubRelease{{
|
|
TagName: version,
|
|
Assets: []githubAsset{{ID: 123, Name: assetName, Size: 42, Digest: "sha256:abc123"}},
|
|
PublishedAt: time.Date(2026, 9, 15, 0, 0, 0, 0, time.UTC),
|
|
}})
|
|
if len(releases) != 1 || releases[0].Checksum != "abc123" || releases[0].AssetID != 123 {
|
|
t.Fatalf("未读取官方资产摘要：%+v", releases)
|
|
}
|
|
}
|
|
|
|
func TestExtractZipBinary(t *testing.T) {
|
|
var buffer bytes.Buffer
|
|
writer := zip.NewWriter(&buffer)
|
|
file, err := writer.Create("frp/" + clientBinaryName())
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if _, err := file.Write([]byte("client")); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if err := writer.Close(); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
data, err := extractZipBinary(buffer.Bytes())
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if string(data) != "client" {
|
|
t.Fatalf("提取内容错误：%q", data)
|
|
}
|
|
}
|
|
|
|
func TestValidDownloadProxy(t *testing.T) {
|
|
for _, proxy := range []string{"", "https://gh-proxy.org/", "https://v4.gh-proxy.org/", "https://cdn.gh-proxy.org/"} {
|
|
if !validDownloadProxy(proxy) {
|
|
t.Fatalf("合法下载代理被拒绝：%s", proxy)
|
|
}
|
|
}
|
|
if validDownloadProxy("https://untrusted.example/") {
|
|
t.Fatal("非白名单下载代理不应被接受")
|
|
}
|
|
}
|
|
|
|
func TestReleaseListURLsPrioritizesSelectedProxy(t *testing.T) {
|
|
urls := releaseListURLs("https://v4.gh-proxy.org/")
|
|
if urls[0] != "https://v4.gh-proxy.org/"+githubReleasesURL {
|
|
t.Fatalf("未优先使用已选代理：%v", urls)
|
|
}
|
|
if urls[1] != githubReleasesURL {
|
|
t.Fatalf("直连地址缺失：%v", urls)
|
|
}
|
|
}
|
|
|
|
func TestRemoveBinaryVersion(t *testing.T) {
|
|
root := t.TempDir()
|
|
path := filepath.Join(root, "bin", "v0.71.0", clientBinaryName())
|
|
if err := os.MkdirAll(filepath.Dir(path), 0o700); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if err := os.WriteFile(path, []byte("client"), 0o700); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if err := removeBinaryVersion(root, "v0.71.0", path); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if _, err := os.Stat(filepath.Dir(path)); !os.IsNotExist(err) {
|
|
t.Fatalf("版本目录未删除：%v", err)
|
|
}
|
|
}
|