feat: 增加商品目录导入记录 (#134)

admin/handler/integration/auth_test.go

@@ -2,6 +2,7 @@ package integration
 
 import (
 	"bytes"
+	"database/sql"
 	"net/http"
 	"net/http/httptest"
 	"strings"
@@ -10,6 +11,9 @@ import (
 	"github.com/gin-gonic/gin"
 
 	"cmautobuy/admin/config"
+	"cmautobuy/admin/model"
+	"cmautobuy/admin/repository"
+	_ "modernc.org/sqlite"
 )
 
 func TestRequireCatalogToken(t *testing.T) {
@@ -47,6 +51,35 @@ func TestRequireCatalogToken(t *testing.T) {
 	}
 }
 
+func TestCatalogBatchAPI_按批次查询摘要(t *testing.T) {
+	db, err := sql.Open("sqlite", ":memory:")
+	if err != nil {
+		t.Fatal(err)
+	}
+	defer db.Close()
+	_, err = db.Exec(`CREATE TABLE catalog_import_runs(source TEXT,batch_id TEXT,request_hash TEXT,status TEXT,request_count INTEGER,conflict_count INTEGER,observed_at TEXT,last_request_at TEXT,last_conflict_at TEXT,shopee_created INTEGER DEFAULT 0,shopee_updated INTEGER DEFAULT 0,sku_created INTEGER DEFAULT 0,sku_updated INTEGER DEFAULT 0,pdd_created INTEGER DEFAULT 0,pdd_updated INTEGER DEFAULT 0,association_created INTEGER DEFAULT 0,association_unchanged INTEGER DEFAULT 0,failure_count INTEGER DEFAULT 0,error_summary TEXT,response_body TEXT,created_at TEXT,finished_at TEXT,PRIMARY KEY(source,batch_id))`)
+	if err != nil {
+		t.Fatal(err)
+	}
+	run := model.CatalogImportRun{Source: "script", BatchID: "B-1", RequestHash: "abc", Status: model.CatalogImportSucceeded, LastRequestAt: model.NowISO(), CreatedAt: model.NowISO()}
+	if err := repository.InsertCatalogImportRun(db, run); err != nil {
+		t.Fatal(err)
+	}
+	const token = "0123456789abcdef0123456789abcdef"
+	r := gin.New()
+	Register(r, db, config.CatalogIntegrationConfig{Source: "script", Token: token})
+	req := httptest.NewRequest(http.MethodGet, "/api/v1/integrations/catalog/batches/B-1", nil)
+	req.Header.Set("Authorization", "Bearer "+token)
+	resp := httptest.NewRecorder()
+	r.ServeHTTP(resp, req)
+	if resp.Code != 200 || !strings.Contains(resp.Body.String(), `"batch_id":"B-1"`) {
+		t.Fatalf("status=%d body=%s", resp.Code, resp.Body.String())
+	}
+	if strings.Contains(resp.Body.String(), "request_hash") {
+		t.Fatal("查询响应不应暴露请求哈希")
+	}
+}
+
 func TestCatalogBatchAPI_鉴权JSON与体积限制(t *testing.T) {
 	gin.SetMode(gin.TestMode)
 	const token = "0123456789abcdef0123456789abcdef"