mirror of
https://github.com/Rain-kl/OpenFlare.git
synced 2026-09-30 06:16:37 +08:00
deqiying
104 lines
2.9 KiB
Go
104 lines
2.9 KiB
Go
// Copyright 2026 Arctel.net
|
|
// SPDX-License-Identifier: Apache-2.0
|
|
|
|
package pages
|
|
|
|
import (
|
|
"archive/zip"
|
|
"bytes"
|
|
"context"
|
|
"net/http"
|
|
"net/http/httptest"
|
|
"os"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func TestParseAndValidatePagesDownloadURL(t *testing.T) {
|
|
_, err := parseAndValidatePagesDownloadURL("")
|
|
require.Error(t, err)
|
|
assert.Contains(t, err.Error(), "填写")
|
|
|
|
_, err = parseAndValidatePagesDownloadURL("ftp://example.com/a.zip")
|
|
require.Error(t, err)
|
|
|
|
// Private / local hosts are allowed (operator-controlled artifact hosts).
|
|
for _, raw := range []string{
|
|
"http://127.0.0.1/a.zip",
|
|
"https://localhost/a.zip",
|
|
"http://192.168.1.10:8080/site.tar.gz",
|
|
"https://example.com/dist/site.tar.gz?x=1",
|
|
} {
|
|
parsed, parseErr := parseAndValidatePagesDownloadURL(raw)
|
|
require.NoError(t, parseErr, raw)
|
|
assert.NotEmpty(t, parsed.Hostname(), raw)
|
|
}
|
|
}
|
|
|
|
func TestDownloadPagesPackageFromURLAllowsPrivateHost(t *testing.T) {
|
|
var body bytes.Buffer
|
|
zw := zip.NewWriter(&body)
|
|
w, err := zw.Create("index.html")
|
|
require.NoError(t, err)
|
|
_, err = w.Write([]byte("ok"))
|
|
require.NoError(t, err)
|
|
require.NoError(t, zw.Close())
|
|
zipBytes := body.Bytes()
|
|
|
|
var sawProviderUA bool
|
|
server := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
|
if r.Header.Get("User-Agent") == remoteSourceUserAgent {
|
|
sawProviderUA = true
|
|
}
|
|
w.Header().Set("Content-Disposition", `attachment; filename="remote-site.zip"`)
|
|
w.Header().Set("Content-Type", "application/zip")
|
|
_, _ = w.Write(zipBytes)
|
|
}))
|
|
t.Cleanup(server.Close)
|
|
|
|
tempPath, checksum, size, format, fileName, err := downloadPagesPackageFromURL(
|
|
context.Background(),
|
|
server.URL+"/pkg.zip",
|
|
10*1024*1024,
|
|
)
|
|
require.NoError(t, err)
|
|
t.Cleanup(func() { _ = os.Remove(tempPath) })
|
|
assert.True(t, sawProviderUA)
|
|
assert.NotEmpty(t, checksum)
|
|
assert.Positive(t, size)
|
|
assert.Equal(t, "zip", string(format))
|
|
assert.Equal(t, "pkg.zip", fileName)
|
|
}
|
|
|
|
func TestUploadDeploymentFromURLPrivateHost(t *testing.T) {
|
|
cleanup := setupPagesTestDB(t)
|
|
defer cleanup()
|
|
_, disableStorage := setupPagesStorageMock(t)
|
|
defer disableStorage()
|
|
ctx := context.Background()
|
|
|
|
var body bytes.Buffer
|
|
zw := zip.NewWriter(&body)
|
|
w, err := zw.Create("index.html")
|
|
require.NoError(t, err)
|
|
_, err = w.Write([]byte("from-url"))
|
|
require.NoError(t, err)
|
|
require.NoError(t, zw.Close())
|
|
|
|
server := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
|
w.Header().Set("Content-Disposition", `attachment; filename="from-url.zip"`)
|
|
_, _ = w.Write(body.Bytes())
|
|
}))
|
|
t.Cleanup(server.Close)
|
|
|
|
project, err := CreateProject(ctx, Input{Name: "URL Site", Slug: "url-site", Enabled: true})
|
|
require.NoError(t, err)
|
|
|
|
deployment, err := UploadDeploymentFromURL(ctx, project.ID, server.URL+"/from-url.zip", "root")
|
|
require.NoError(t, err)
|
|
assert.NotZero(t, deployment.ID)
|
|
assert.Equal(t, 1, deployment.FileCount)
|
|
}
|