fix: rebuild MES as compilable go-zero+ent backend (renamed bj_power_mes), restore 3 workstation projects from pristine original, align naming; all Go projects go build clean

2026-08-27 10:56:41 +08:00
parent 380715f8a9
commit 371b494c54
523 changed files with 67923 additions and 24758 deletions
@@ -0,0 +1,132 @@
+package upload
+
+import (
+	"bytes"
+	"database/sql"
+	"encoding/json"
+	"fmt"
+	"mime/multipart"
+	"net/http"
+	"net/http/httptest"
+	"os"
+	"testing"
+
+	"bj_power_mes/ent"
+	"bj_power_mes/internal/config"
+	"bj_power_mes/internal/svc"
+
+	entsql "entgo.io/ent/dialect/sql"
+	_ "github.com/jackc/pgx/v5/stdlib"
+	"resty.dev/v3"
+)
+
+func TestHandleInspectionUpload(t *testing.T) {
+	// 创建临时上传目录
+	//tmpDir := t.TempDir()
+	svcCtx := newTestServiceContext(t, "F:\\Workspace\\Hardman\\back_cover\\uploads")
+
+	t.Run("上传成功", func(t *testing.T) {
+		body := &bytes.Buffer{}
+		writer := multipart.NewWriter(body)
+		_ = writer.WriteField("deviceId", "test")
+		part, _ := writer.CreateFormFile("file", "test.jpg")
+		data, _ := os.ReadFile("F:\\Workspace\\Hardman\\back_cover\\internal\\upload\\rog.png")
+		part.Write(data)
+		writer.Close()
+
+		req := httptest.NewRequest(http.MethodPost, "/document/upload", body)
+		req.Header.Set("Content-Type", writer.FormDataContentType())
+		w := httptest.NewRecorder()
+
+		HandleInspectionUpload(svcCtx, w, req)
+
+		if w.Code != http.StatusOK {
+			t.Errorf("expected 200, got %d: %s", w.Code, w.Body.String())
+		}
+
+		var resp map[string]any
+		json.NewDecoder(w.Body).Decode(&resp)
+		if resp["code"] != 200 {
+			t.Errorf("expected code 200, got %v", resp["code"])
+		}
+	})
+
+	t.Run("缺少文件返回400", func(t *testing.T) {
+		body := &bytes.Buffer{}
+		writer := multipart.NewWriter(body)
+		_ = writer.WriteField("deviceId", "6")
+		writer.Close()
+
+		req := httptest.NewRequest(http.MethodPost, "/document/upload", body)
+		req.Header.Set("Content-Type", writer.FormDataContentType())
+		w := httptest.NewRecorder()
+
+		HandleInspectionUpload(svcCtx, w, req)
+
+		if w.Code != http.StatusBadRequest {
+			t.Errorf("expected 400, got %d", w.Code)
+		}
+	})
+
+	t.Run("超大文件被拒绝", func(t *testing.T) {
+		body := &bytes.Buffer{}
+		writer := multipart.NewWriter(body)
+		part, _ := writer.CreateFormFile("file", "big.jpg")
+		// 写入超过 5MB 的数据
+		chunk := make([]byte, 1024)
+		for i := 0; i < 6*1024; i++ {
+			part.Write(chunk)
+		}
+		writer.Close()
+
+		req := httptest.NewRequest(http.MethodPost, "/document/upload", body)
+		req.Header.Set("Content-Type", writer.FormDataContentType())
+		w := httptest.NewRecorder()
+
+		HandleInspectionUpload(svcCtx, w, req)
+
+		if w.Code == http.StatusOK {
+			t.Error("超大文件应被拒绝")
+		}
+	})
+
+	t.Run("内窥镜URL测试", func(t *testing.T) {
+		client := resty.New()
+		res, err := client.R().
+			SetMultipartFormData(map[string]string{
+				"userId":   "0",
+				"deviceId": "test",
+			}).
+			SetFile("file", "F:\\Workspace\\Hardman\\back_cover\\internal\\upload\\rog.png").
+			//Post("http://47.115.54.30:8097/document/upload")
+			Post("http://localhost:8888/document/upload")
+
+		fmt.Println(err, res)
+
+	})
+}
+
+// newTestServiceContext 创建测试用的 ServiceContext，指向临时目录
+func newTestServiceContext(t *testing.T, uploadDir string) *svc.ServiceContext {
+	t.Helper()
+
+	db, err := sql.Open("pgx", "postgresql://postgres:postgres@127.0.0.1:5432/back_cover?sslmode=disable")
+	if err != nil {
+		t.Skipf("跳过：无法连接数据库 (%v)", err)
+	}
+	drv := entsql.OpenDB("postgres", db)
+	entClient := ent.NewClient(ent.Driver(drv))
+
+	return &svc.ServiceContext{
+		Config: config.Config{
+			Upload: struct {
+				Dir           string `json:",default=./uploads"`
+				RetentionDays int    `json:",default=30"`
+			}{
+				Dir:           uploadDir,
+				RetentionDays: -1,
+			},
+		},
+		EntClient: entClient,
+	}
+}