82 lines
1.7 KiB
Go
82 lines
1.7 KiB
Go
package core
|
|
|
|
import (
|
|
"fmt"
|
|
"io"
|
|
"net/http"
|
|
"net/http/httptest"
|
|
"os"
|
|
"path/filepath"
|
|
|
|
"github.com/pkg/errors"
|
|
"go.uber.org/zap"
|
|
"gopkg.d7z.net/gitea-pages/pkg"
|
|
)
|
|
|
|
type TestServer struct {
|
|
server *pkg.Server
|
|
dummy *ProviderDummy
|
|
}
|
|
|
|
type SvcOpts func(options *pkg.ServerOptions)
|
|
|
|
func NewDefaultTestServer() *TestServer {
|
|
return NewTestServer("example.com", func(options *pkg.ServerOptions) {
|
|
options.CacheMetaTTL = 0
|
|
})
|
|
}
|
|
|
|
func NewTestServer(domain string, opts ...SvcOpts) *TestServer {
|
|
atom := zap.NewAtomicLevel()
|
|
atom.SetLevel(zap.DebugLevel)
|
|
cfg := zap.NewProductionConfig()
|
|
cfg.Level = atom
|
|
logger, _ := cfg.Build()
|
|
zap.ReplaceGlobals(logger)
|
|
options := pkg.DefaultOptions(domain)
|
|
for _, opt := range opts {
|
|
opt(&options)
|
|
}
|
|
dummy, err := NewDummy()
|
|
if err != nil {
|
|
zap.S().Fatal(err)
|
|
}
|
|
|
|
server := pkg.NewPageServer(dummy, options)
|
|
|
|
return &TestServer{
|
|
dummy: dummy,
|
|
server: server,
|
|
}
|
|
}
|
|
|
|
func (t *TestServer) AddFile(path, data string, args ...interface{}) {
|
|
join := filepath.Join(t.dummy.BaseDir, path)
|
|
err := os.MkdirAll(filepath.Dir(join), 0o755)
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
err = os.WriteFile(join, []byte(fmt.Sprintf(data, args...)), 0o644)
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
}
|
|
|
|
func (t *TestServer) OpenFile(url string) ([]byte, *http.Response, error) {
|
|
recorder := httptest.NewRecorder()
|
|
t.server.ServeHTTP(recorder, httptest.NewRequest(http.MethodGet, url, nil))
|
|
response := recorder.Result()
|
|
if response.Body != nil {
|
|
defer response.Body.Close()
|
|
}
|
|
if response.StatusCode != http.StatusOK {
|
|
return nil, response, errors.New(response.Status)
|
|
}
|
|
all, _ := io.ReadAll(response.Body)
|
|
return all, response, nil
|
|
}
|
|
|
|
func (t *TestServer) Close() error {
|
|
return t.server.Close()
|
|
}
|