68 lines
3.2 KiB
Go
68 lines
3.2 KiB
Go
|
package runner
|
||
|
|
|
||
|
|
import (
|
||
|
|
"context"
|
||
|
|
"testing"
|
||
|
|
|
||
|
|
"gitea.maximumdirect.net/eric/audita/v3/internal/core/config"
|
||
|
|
"gitea.maximumdirect.net/eric/audita/v3/internal/core/schema"
|
||
|
|
"gitea.maximumdirect.net/eric/audita/v3/internal/framework/contracts"
|
||
|
|
"gitea.maximumdirect.net/eric/audita/v3/internal/framework/proposals"
|
||
|
|
)
|
||
|
|
|
||
|
|
type orderedFactory struct {
|
||
|
|
seen chan contracts.ProposalRequest
|
||
|
|
}
|
||
|
|
|
||
|
|
func (f orderedFactory) ModuleForSpec(spec contracts.ModuleRunSpec) (contracts.TranscriptModule, error) {
|
||
|
|
return orderedModule{spec: spec, seen: f.seen}, nil
|
||
|
|
}
|
||
|
|
|
||
|
|
type orderedModule struct {
|
||
|
|
spec contracts.ModuleRunSpec
|
||
|
|
seen chan contracts.ProposalRequest
|
||
|
|
}
|
||
|
|
|
||
|
|
func (m orderedModule) Key() string { return m.spec.ModuleKey }
|
||
|
|
func (m orderedModule) ReplacementPolicy() proposals.ReplacementPolicy {
|
||
|
|
return proposals.ReplacementPolicyRequireUnique
|
||
|
|
}
|
||
|
|
func (m orderedModule) Validators() []contracts.Validator { return nil }
|
||
|
|
func (m orderedModule) Propose(_ context.Context, req contracts.ProposalRequest) (contracts.ProposalResult, error) {
|
||
|
|
m.seen <- req
|
||
|
|
original, corrected := "party", "Party"
|
||
|
|
if m.spec.Index == 1 {
|
||
|
|
original, corrected = "spell", "Spell"
|
||
|
|
}
|
||
|
|
if m.spec.Index == 2 {
|
||
|
|
original, corrected = "Party Spell", "Party Spell!"
|
||
|
|
}
|
||
|
|
return contracts.ProposalResult{Proposals: []proposals.CorrectionProposal{{TargetSegmentID: 1, OriginalText: original, CorrectedText: corrected, Confidence: 1}}}, nil
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestOrderedWorkflowUsesIndependentSelectionsAndPriorEdits(t *testing.T) {
|
||
|
|
cfg := &config.Config{GlossaryCollections: map[string]config.GlossaryCollection{
|
||
|
|
"party": {Entries: []config.GlossaryEntry{{ID: "party", Name: "Party", Category: "person"}}},
|
||
|
|
"spells": {Entries: []config.GlossaryEntry{{ID: "spell", Name: "Spell", Category: "spell"}}},
|
||
|
|
}, Pipeline: config.Pipeline{Stages: []config.Stage{
|
||
|
|
{Module: "glossary", Glossaries: []string{"party"}, ProtectedTerms: []string{"party", "spells"}, Context: config.Context{Description: "party context"}},
|
||
|
|
{Module: "glossary", Glossaries: []string{"spells"}, ProtectedTerms: []string{"party", "spells"}, Context: config.Context{Description: "spell context"}},
|
||
|
|
{Module: "grammar", Glossaries: []string{"party", "spells"}, ProtectedTerms: []string{"party", "spells"}},
|
||
|
|
}}}
|
||
|
|
seen := make(chan contracts.ProposalRequest, 3)
|
||
|
|
output, err := New(orderedFactory{seen: seen}).Run(context.Background(), RunInput{Transcript: &schema.Transcript{Segments: []schema.Segment{{ID: 1, Text: "party spell"}}}, ModuleSpecs: contracts.ResolveModuleRunSpecs(cfg)})
|
||
|
|
if err != nil {
|
||
|
|
t.Fatal(err)
|
||
|
|
}
|
||
|
|
if got := output.FinalTranscript.Segments[0].Text; got != "Party Spell!" {
|
||
|
|
t.Fatalf("final text = %q", got)
|
||
|
|
}
|
||
|
|
first, second, third := <-seen, <-seen, <-seen
|
||
|
|
if first.Settings.Context.Description != "party context" || second.Settings.Context.Description != "spell context" || len(first.Glossary) != 1 || first.Glossary[0].Name != "Party" || len(second.Glossary) != 1 || second.Glossary[0].Name != "Spell" || len(third.Glossary) != 2 {
|
||
|
|
t.Fatalf("incorrect context: %+v %+v %+v", first, second, third)
|
||
|
|
}
|
||
|
|
if second.WorkingTranscript.Segments[0].Text != "Party spell" || third.WorkingTranscript.Segments[0].Text != "Party Spell" {
|
||
|
|
t.Fatalf("edits not carried forward: %q, %q", second.WorkingTranscript.Segments[0].Text, third.WorkingTranscript.Segments[0].Text)
|
||
|
|
}
|
||
|
|
}
|