144 lines
3.7 KiB
Go
144 lines
3.7 KiB
Go
package testweb
|
||
|
||
import (
|
||
"os"
|
||
"path/filepath"
|
||
"strings"
|
||
"testing"
|
||
)
|
||
|
||
const sampleRecipeLog = `=== META ===
|
||
Time: 2026-07-13T08:00:00+08:00
|
||
ClientIP: 127.0.0.1
|
||
Method: POST
|
||
Path: /province/supervise/data
|
||
Kind: supervise
|
||
DurationMs: 250
|
||
Status: 200
|
||
|
||
=== 1. 入站请求（transit → forward）===
|
||
URL: POST 127.0.0.1:16001/province/supervise/data
|
||
|
||
Headers:
|
||
Content-Type: application/json
|
||
X-Service-Method: uploadRecipeIndicators
|
||
X-Forward-Token: secret
|
||
|
||
Body:
|
||
{"encrypted":"x"}
|
||
|
||
=== 2. 出站转发（forward → 政务云）===
|
||
TargetURL: https://59.202.52.129:28212/province/supervise/data
|
||
Method: POST
|
||
|
||
Headers:
|
||
Content-Type: application/json
|
||
X-Service-Method: uploadRecipeIndicators
|
||
|
||
Body:
|
||
{"encrypted":"x"}
|
||
|
||
=== 3. 回复（政务云 → forward → transit）===
|
||
UpstreamStatus: 200
|
||
|
||
UpstreamHeaders:
|
||
Content-Type: application/json
|
||
X-Ca-Request-Id: trace-recipe-1
|
||
|
||
Body:
|
||
{"msgCode":0,"msg":"ok"}
|
||
|
||
BodyJSON:
|
||
{
|
||
"msgCode": 0,
|
||
"msg": "ok"
|
||
}
|
||
`
|
||
|
||
func verificationLogBytes() []byte {
|
||
return []byte(strings.ReplaceAll(sampleRecipeLog, "uploadRecipeIndicators", "uploadRecipeVerificationIndicators"))
|
||
}
|
||
|
||
func TestClassifyPrescriptionLog_recipe(t *testing.T) {
|
||
rec, ok := classifyPrescriptionLog("2026-07-13 08/20260713080000_supervise_abcd.log", []byte(sampleRecipeLog))
|
||
if !ok {
|
||
t.Fatal("expected classified")
|
||
}
|
||
if rec.BizType != "recipe" || rec.Channel != "supervise" {
|
||
t.Fatalf("rec=%+v", rec)
|
||
}
|
||
if !rec.OK || rec.MsgCode != 0 || rec.TraceID != "trace-recipe-1" {
|
||
t.Fatalf("response fields=%+v", rec)
|
||
}
|
||
}
|
||
|
||
func TestClassifyPrescriptionLog_verification(t *testing.T) {
|
||
rec, ok := classifyPrescriptionLog("2026-07-13 08/20260713080000_supervise_abcd.log", verificationLogBytes())
|
||
if !ok || rec.BizType != "verification" {
|
||
t.Fatalf("rec=%+v ok=%v", rec, ok)
|
||
}
|
||
}
|
||
|
||
func TestClassifyPrescriptionLog_pdf(t *testing.T) {
|
||
rec, ok := classifyPrescriptionLog("2026-07-13 08/20260713080000_file_abcd.log", []byte(sampleLog))
|
||
if !ok || rec.BizType != "pdf" {
|
||
t.Fatalf("rec=%+v", rec)
|
||
}
|
||
if rec.FileID != "id1" || !rec.OK {
|
||
t.Fatalf("pdf response=%+v", rec)
|
||
}
|
||
}
|
||
|
||
func TestRebuildPrescriptionIndex(t *testing.T) {
|
||
tmp := t.TempDir()
|
||
hour := filepath.Join(tmp, "2026-07-13 08")
|
||
if err := os.MkdirAll(hour, 0o755); err != nil {
|
||
t.Fatal(err)
|
||
}
|
||
if err := os.WriteFile(filepath.Join(hour, "20260713080000_file_abcd.log"), []byte(sampleLog), 0o644); err != nil {
|
||
t.Fatal(err)
|
||
}
|
||
if err := os.WriteFile(filepath.Join(hour, "20260713080100_supervise_abcd.log"), []byte(sampleRecipeLog), 0o644); err != nil {
|
||
t.Fatal(err)
|
||
}
|
||
n, err := RebuildPrescriptionIndex(tmp)
|
||
if err != nil || n != 2 {
|
||
t.Fatalf("count=%d err=%v", n, err)
|
||
}
|
||
pdfList, err := ListPrescriptionRecords(tmp, "pdf", 10)
|
||
if err != nil || len(pdfList) != 1 {
|
||
t.Fatalf("pdf=%v err=%v", pdfList, err)
|
||
}
|
||
recipeList, err := ListPrescriptionRecords(tmp, "recipe", 10)
|
||
if err != nil || len(recipeList) != 1 {
|
||
t.Fatalf("recipe=%v err=%v", recipeList, err)
|
||
}
|
||
}
|
||
|
||
func TestListSuperviseLogs_and_resolve(t *testing.T) {
|
||
tmp := t.TempDir()
|
||
hour := filepath.Join(tmp, "2026-07-13 08")
|
||
if err := os.MkdirAll(hour, 0o755); err != nil {
|
||
t.Fatal(err)
|
||
}
|
||
name := "20260713080100_supervise_abcd.log"
|
||
if err := os.WriteFile(filepath.Join(hour, name), []byte(sampleRecipeLog), 0o644); err != nil {
|
||
t.Fatal(err)
|
||
}
|
||
list, err := ListSuperviseLogs(tmp)
|
||
if err != nil || len(list) != 1 || list[0].Kind != "supervise" {
|
||
t.Fatalf("list=%v err=%v", list, err)
|
||
}
|
||
rel := list[0].Rel
|
||
if _, err := resolveLogRelPath(tmp, rel); err != nil {
|
||
t.Fatal(err)
|
||
}
|
||
p, err := ParseAPILogFile(tmp, rel)
|
||
if err != nil {
|
||
t.Fatal(err)
|
||
}
|
||
if p.Inbound.Headers["X-Service-Method"] != MethodRecipe {
|
||
t.Fatalf("method=%q", p.Inbound.Headers["X-Service-Method"])
|
||
}
|
||
}
|