84 lines
2.3 KiB
Go
84 lines
2.3 KiB
Go
package updatecdn
|
|
|
|
import (
|
|
"crypto/sha256"
|
|
"encoding/hex"
|
|
"io"
|
|
"net/http"
|
|
"net/http/httptest"
|
|
"os"
|
|
"path/filepath"
|
|
"strings"
|
|
"testing"
|
|
"time"
|
|
)
|
|
|
|
func TestHandlerServesCurrentResolveAndRange(t *testing.T) {
|
|
dir := t.TempDir()
|
|
filesDir := filepath.Join(dir, "files")
|
|
if err := os.Mkdir(filesDir, 0o755); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
packageData := []byte("0123456789abcdef")
|
|
packageName := "tx64upd7007000"
|
|
if err := os.WriteFile(filepath.Join(filesDir, packageName), packageData, 0o600); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
hash := sha256.Sum256(packageData)
|
|
manifestPath := writeTestManifest(t, dir, Manifest{
|
|
SchemaVersion: ManifestSchemaVersion,
|
|
Desktop: map[string]map[string]DesktopRelease{
|
|
"win64": {"stable": {Build: 7007000, File: packageName, SHA256: hex.EncodeToString(hash[:])}},
|
|
},
|
|
Apps: map[string]map[string]ApplicationRelease{
|
|
"ios": {"stable": {ID: 8, Version: "12.9.1", URL: "https://apps.apple.com/app/id1", Notes: map[string]string{"en": "Update available"}}},
|
|
},
|
|
})
|
|
store, err := NewStore(manifestPath, filesDir)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
handler, err := NewHandler(store)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
server := httptest.NewServer(handler)
|
|
defer server.Close()
|
|
|
|
resp, err := http.Get(server.URL + "/current4")
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
currentBody, _ := io.ReadAll(resp.Body)
|
|
resp.Body.Close()
|
|
if resp.StatusCode != http.StatusOK || !strings.Contains(string(currentBody), `"released":7007000`) {
|
|
t.Fatalf("current4 = %d %s", resp.StatusCode, currentBody)
|
|
}
|
|
|
|
client, err := NewClient(server.URL, time.Second)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
resolved, err := client.Resolve(t.Context(), ResolveRequest{Platform: "ios", Version: "12.9.0", LangCode: "en"})
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if resolved == nil || resolved.ID != 8 || resolved.Version != "12.9.1" {
|
|
t.Fatalf("resolved = %#v", resolved)
|
|
}
|
|
|
|
req, err := http.NewRequest(http.MethodGet, server.URL+"/files/"+packageName, nil)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
req.Header.Set("Range", "bytes=2-5")
|
|
rangeResp, err := http.DefaultClient.Do(req)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
rangeBody, _ := io.ReadAll(rangeResp.Body)
|
|
rangeResp.Body.Close()
|
|
if rangeResp.StatusCode != http.StatusPartialContent || string(rangeBody) != "2345" {
|
|
t.Fatalf("range = %d %q", rangeResp.StatusCode, rangeBody)
|
|
}
|
|
}
|