File size: 2,290 Bytes
8d3471e | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 | package rawsample
import (
"encoding/json"
"os"
"path/filepath"
"strings"
"testing"
)
func TestNormalizeSampleID(t *testing.T) {
got := NormalizeSampleID(" Hello, World! ")
if got != "hello-world" {
t.Fatalf("expected hello-world, got %q", got)
}
}
func TestPersistWritesSampleFilesAndMeta(t *testing.T) {
root := t.TempDir()
saved, err := Persist(PersistOptions{
RootDir: root,
SampleID: "My Sample! 01",
Source: "unit-test",
Request: map[string]any{
"model": "deepseek-v4-flash",
"stream": true,
"messages": []any{
map[string]any{"role": "user", "content": "广州天气"},
},
},
Capture: CaptureSummary{
Label: "deepseek_completion",
URL: "https://chat.deepseek.com/api/v0/chat/completion",
StatusCode: 200,
},
UpstreamBody: []byte("data: {\"v\":\"hello [reference:1]\"}\n\n" +
"data: {\"v\":\"FINISHED\",\"p\":\"response/status\"}\n\n"),
})
if err != nil {
t.Fatalf("Persist failed: %v", err)
}
if saved.SampleID != "my-sample-01" {
t.Fatalf("expected normalized sample id, got %q", saved.SampleID)
}
if _, err := os.Stat(saved.Dir); err != nil {
t.Fatalf("sample dir missing: %v", err)
}
if _, err := os.Stat(saved.UpstreamPath); err != nil {
t.Fatalf("upstream file missing: %v", err)
}
if _, err := os.Stat(filepath.Join(saved.Dir, "openai.stream.sse")); !os.IsNotExist(err) {
t.Fatalf("unexpected processed stream file: %v", err)
}
if _, err := os.Stat(filepath.Join(saved.Dir, "openai.output.txt")); !os.IsNotExist(err) {
t.Fatalf("unexpected processed text file: %v", err)
}
metaBytes, err := os.ReadFile(saved.MetaPath)
if err != nil {
t.Fatalf("read meta: %v", err)
}
var meta Meta
if err := json.Unmarshal(metaBytes, &meta); err != nil {
t.Fatalf("decode meta: %v", err)
}
if meta.SampleID != saved.SampleID {
t.Fatalf("expected meta sample id %q, got %q", saved.SampleID, meta.SampleID)
}
if meta.Capture.ReferenceMarkerCount != 1 {
t.Fatalf("expected one reference marker, got %+v", meta.Capture)
}
if meta.Capture.FinishedTokenCount != 1 {
t.Fatalf("expected one finished token, got %+v", meta.Capture)
}
if strings.Contains(string(metaBytes), "\"processed\"") {
t.Fatalf("meta should not include processed payload: %s", string(metaBytes))
}
}
|