Function TestOutputWriter_JSONL

cmd/output_test.go:121–158

(t *testing.T)

}

func TestOutputWriter_JSONL(t *testing.T) {
	dir := t.TempDir()
	outPath := filepath.Join(dir, "output.jsonl")

	oldJSONL := jsonLines
	jsonLines = true
	defer func() { jsonLines = oldJSONL }()

	jsonResultsMutex.Lock()
	jsonResults = nil
	jsonResultsMutex.Unlock()

	if err := initOutputWriter(outPath); err != nil {
		t.Fatalf("initOutputWriter: %v", err)
	}

	writeResultToOutput(Result{line: "GET /admin", statusCode: 200, contentLength: 1234, score: 88, likelihood: "high"}, "verb-tampering")
	writeResultToOutput(Result{line: "X-Original-URL: /", statusCode: 302, contentLength: 12, score: 67, likelihood: "medium"}, "header-confusion")
	closeOutputWriter()

	data, err := os.ReadFile(outPath)
	if err != nil {
		t.Fatalf("read output: %v", err)
	}

	lines := strings.Split(strings.TrimSpace(string(data)), "\n")
	if len(lines) != 2 {
		t.Fatalf("expected 2 jsonl lines, got %d", len(lines))
	}

	var first JSONResult
	if err := json.Unmarshal([]byte(lines[0]), &first); err != nil {
		t.Fatalf("unmarshal first jsonl line: %v", err)
	}
	if first.Score != 88 || first.Likelihood != "high" {
		t.Fatalf("unexpected first jsonl result: %+v", first)
	}
}