重构项目

tests/core/test.go

@@ -0,0 +1,78 @@
+package core
+
+import (
+	"fmt"
+	"io"
+	"net/http"
+	"net/http/httptest"
+	"os"
+	"path/filepath"
+
+	"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 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) {
+	join := filepath.Join(t.dummy.BaseDir, path)
+	err := os.MkdirAll(filepath.Dir(join), 0o755)
+	if err != nil {
+		panic(err)
+	}
+	err = os.WriteFile(join, []byte(data), 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("GET", url, nil))
+	response := recorder.Result()
+	if response.Body != nil {
+		defer response.Body.Close()
+	}
+	if response.StatusCode != http.StatusOK {
+		return nil, response, fmt.Errorf(response.Status)
+	}
+	all, _ := io.ReadAll(response.Body)
+	return all, response, nil
+}
+
+func (t *TestServer) Close() error {
+	return t.server.Close()
+}