142 lines
3.2 KiB
Go
142 lines
3.2 KiB
Go
|
package testweb
|
|||
|
|
|
|||
|
|
import (
|
|||
|
|
"encoding/base64"
|
|||
|
|
"os"
|
|||
|
|
"path/filepath"
|
|||
|
|
"strings"
|
|||
|
|
"testing"
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
const sampleLog = `=== META ===
|
|||
|
|
Time: 2026-05-28T08:00:00+08:00
|
|||
|
|
ClientIP: 127.0.0.1
|
|||
|
|
Method: POST
|
|||
|
|
Path: /mng/file/auth/upload
|
|||
|
|
Kind: file
|
|||
|
|
DurationMs: 100
|
|||
|
|
Status: 200
|
|||
|
|
|
|||
|
|
=== 1. 入站请求（transit → forward）===
|
|||
|
|
URL: POST 127.0.0.1:16001/mng/file/auth/upload
|
|||
|
|
|
|||
|
|
Headers:
|
|||
|
|
Content-Type: multipart/form-data; boundary=abc
|
|||
|
|
X-Authorization: tok-in
|
|||
|
|
X-Forward-Token: secret
|
|||
|
|
|
|||
|
|
Body:
|
|||
|
|
body_len=3
|
|||
|
|
body_base64=Zm9v
|
|||
|
|
|
|||
|
|
=== 2. 出站转发（forward → 政务云）===
|
|||
|
|
TargetURL: https://59.202.52.129:28211/mng/file/auth/upload
|
|||
|
|
Method: POST
|
|||
|
|
|
|||
|
|
Headers:
|
|||
|
|
Content-Type: multipart/form-data; boundary=abc
|
|||
|
|
X-Authorization: tok-out
|
|||
|
|
|
|||
|
|
Body:
|
|||
|
|
body_len=3
|
|||
|
|
body_base64=Zm9v
|
|||
|
|
|
|||
|
|
=== 3. 回复（政务云 → forward → transit）===
|
|||
|
|
UpstreamStatus: 200
|
|||
|
|
|
|||
|
|
UpstreamHeaders:
|
|||
|
|
Content-Type: application/json
|
|||
|
|
|
|||
|
|
Body:
|
|||
|
|
{"success":true,"record":{"fileId":"id1"}}
|
|||
|
|
|
|||
|
|
BodyJSON:
|
|||
|
|
{
|
|||
|
|
"success": true,
|
|||
|
|
"record": {
|
|||
|
|
"fileId": "id1"
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
`
|
|||
|
|
|
|||
|
|
func TestParseAPILogBytes_outboundBody(t *testing.T) {
|
|||
|
|
p, err := ParseAPILogBytes("2026-05-28 08/20260528080000_file_abcd.log", []byte(sampleLog))
|
|||
|
|
if err != nil {
|
|||
|
|
t.Fatal(err)
|
|||
|
|
}
|
|||
|
|
if p.Outbound.URL != "https://59.202.52.129:28211/mng/file/auth/upload" {
|
|||
|
|
t.Fatalf("outbound url=%q", p.Outbound.URL)
|
|||
|
|
}
|
|||
|
|
if p.Outbound.Headers["X-Authorization"] != "tok-out" {
|
|||
|
|
t.Fatalf("auth=%q", p.Outbound.Headers["X-Authorization"])
|
|||
|
|
}
|
|||
|
|
if string(p.Outbound.BodyBytes) != "foo" {
|
|||
|
|
t.Fatalf("body=%q", p.Outbound.BodyBytes)
|
|||
|
|
}
|
|||
|
|
if p.Response.Status != 200 || !strings.Contains(p.Response.Body, "fileId") {
|
|||
|
|
t.Fatalf("response=%+v", p.Response)
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
func TestListFileLogs_and_resolve(t *testing.T) {
|
|||
|
|
tmp := t.TempDir()
|
|||
|
|
hour := filepath.Join(tmp, "2026-05-28 08")
|
|||
|
|
if err := os.MkdirAll(hour, 0o755); err != nil {
|
|||
|
|
t.Fatal(err)
|
|||
|
|
}
|
|||
|
|
name := "20260528080000_file_abcd.log"
|
|||
|
|
if err := os.WriteFile(filepath.Join(hour, name), []byte(sampleLog), 0o644); err != nil {
|
|||
|
|
t.Fatal(err)
|
|||
|
|
}
|
|||
|
|
list, err := ListFileLogs(tmp)
|
|||
|
|
if err != nil || len(list) != 1 {
|
|||
|
|
t.Fatalf("list=%v err=%v", list, err)
|
|||
|
|
}
|
|||
|
|
rel := list[0].Rel
|
|||
|
|
p, err := ParseAPILogFile(tmp, rel)
|
|||
|
|
if err != nil {
|
|||
|
|
t.Fatal(err)
|
|||
|
|
}
|
|||
|
|
if p.Meta["Kind"] != "file" {
|
|||
|
|
t.Fatalf("kind=%v", p.Meta)
|
|||
|
|
}
|
|||
|
|
if _, err := resolveLogRelPath(tmp, "../etc/passwd"); err == nil {
|
|||
|
|
t.Fatal("expected reject ..")
|
|||
|
|
}
|
|||
|
|
if _, err := resolveLogRelPath(tmp, "bad/name.log"); err == nil {
|
|||
|
|
t.Fatal("expected reject pattern")
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
func TestToPostmanCollection_multipart(t *testing.T) {
|
|||
|
|
p, err := ParseAPILogBytes("x.log", []byte(sampleLog))
|
|||
|
|
if err != nil {
|
|||
|
|
t.Fatal(err)
|
|||
|
|
}
|
|||
|
|
data, err := ToPostmanCollection(p, false)
|
|||
|
|
if err != nil {
|
|||
|
|
t.Fatal(err)
|
|||
|
|
}
|
|||
|
|
s := string(data)
|
|||
|
|
if !strings.Contains(s, "postman.com/json/collection/v2.1.0") {
|
|||
|
|
t.Fatal("missing schema")
|
|||
|
|
}
|
|||
|
|
if !strings.Contains(s, "formdata") {
|
|||
|
|
t.Fatal("expected formdata")
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
func Test_applyBody_plain(t *testing.T) {
|
|||
|
|
var rs RequestSection
|
|||
|
|
rs.Headers = map[string]string{"Content-Type": "application/json"}
|
|||
|
|
applyBodyToSection(&rs, []string{`{"a":1}`})
|
|||
|
|
if !rs.BodyPlain || string(rs.BodyBytes) != `{"a":1}` {
|
|||
|
|
t.Fatalf("%+v", rs)
|
|||
|
|
}
|
|||
|
|
b64 := base64.StdEncoding.EncodeToString([]byte("bin"))
|
|||
|
|
applyBodyToSection(&rs, []string{"body_len=3", "body_base64=" + b64})
|
|||
|
|
if string(rs.BodyBytes) != "bin" {
|
|||
|
|
t.Fatalf("got %q", rs.BodyBytes)
|
|||
|
|
}
|
|||
|
|
}
|