statuses update
This commit is contained in:
58
workers/transcribe/internal/models/models_test.go
Normal file
58
workers/transcribe/internal/models/models_test.go
Normal file
@@ -0,0 +1,58 @@
|
||||
package models
|
||||
|
||||
import (
|
||||
"encoding/json"
|
||||
"testing"
|
||||
)
|
||||
|
||||
func TestAudioTaskRoundTrip(t *testing.T) {
|
||||
src := AudioTask{
|
||||
TaskID: "01HX",
|
||||
FilePath: "/data/processing/01HX.wav",
|
||||
Filename: "call.wav",
|
||||
Size: 2048,
|
||||
CreatedAt: 1717843200,
|
||||
}
|
||||
body, err := json.Marshal(src)
|
||||
if err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
var got AudioTask
|
||||
if err := json.Unmarshal(body, &got); err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
if got != src {
|
||||
t.Fatalf("round-trip mismatch: %+v vs %+v", got, src)
|
||||
}
|
||||
}
|
||||
|
||||
func TestTranscriptionResultRoundTrip(t *testing.T) {
|
||||
src := TranscriptionResult{
|
||||
TaskID: "01HX",
|
||||
Filename: "call.wav",
|
||||
FilePath: "/data/processing/01HX.wav",
|
||||
Transcription: "текст звонка",
|
||||
Language: "ru",
|
||||
Segments: []Segment{
|
||||
{Start: 0, End: 1.2, Text: "текст"},
|
||||
},
|
||||
Prompts: []Prompt{
|
||||
{ID: 1, IDSection: 1, Name: "behavioral", Prompt: "analyze", DtCreate: "2026-01-01"},
|
||||
},
|
||||
TranscribedAt: 1717843200,
|
||||
}
|
||||
body, err := json.Marshal(src)
|
||||
if err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
var got TranscriptionResult
|
||||
if err := json.Unmarshal(body, &got); err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
if got.TaskID != src.TaskID || got.Transcription != src.Transcription {
|
||||
t.Fatalf("mismatch: %+v", got)
|
||||
}
|
||||
if len(got.Segments) != 1 || len(got.Prompts) != 1 {
|
||||
t.Fatalf("nested fields lost: %+v", got)
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user