From 3b03d402795a9e900190665fce0830e9afb44720 Mon Sep 17 00:00:00 2001 From: Carlosokumu Date: Mon, 10 Feb 2025 12:52:01 +0300 Subject: [PATCH] return the persister and flag parser --- testutil/engine.go | 12 ++++++++++-- testutil/engine_test.go | 2 +- 2 files changed, 11 insertions(+), 3 deletions(-) diff --git a/testutil/engine.go b/testutil/engine.go index a179b5d..0fc0122 100644 --- a/testutil/engine.go +++ b/testutil/engine.go @@ -9,7 +9,9 @@ import ( "path/filepath" "time" + "git.defalsify.org/vise.git/asm" "git.defalsify.org/vise.git/engine" + "git.defalsify.org/vise.git/persist" "git.defalsify.org/vise.git/resource" "git.grassecon.net/grassrootseconomics/sarafu-api/remote" httpremote "git.grassecon.net/grassrootseconomics/sarafu-api/remote/http" @@ -60,7 +62,7 @@ func CleanDatabase() { } } -func TestEngine(sessionId string) (engine.Engine, func(), chan bool) { +func TestEngine(sessionId string) (engine.Engine, func(), chan bool, *persist.Persister, *asm.FlagParser) { config.LoadConfig() err := config.Apply(override) if err != nil { @@ -75,6 +77,12 @@ func TestEngine(sessionId string) (engine.Engine, func(), chan bool) { logg.InfoCtxf(ctx, "loaded engine setup", "conns", conns) pfp := path.Join(scriptDir, "pp.csv") + parser := asm.NewFlagParser() + _, err = parser.Load(pfp) + if err != nil { + os.Exit(1) + } + var eventChannel = make(chan bool) cfg := engine.Config{ @@ -157,5 +165,5 @@ func TestEngine(sessionId string) (engine.Engine, func(), chan bool) { } logg.Infof("testengine storage closed") } - return en, cleanFn, eventChannel + return en, cleanFn, eventChannel, pe, parser } diff --git a/testutil/engine_test.go b/testutil/engine_test.go index f747468..dc22806 100644 --- a/testutil/engine_test.go +++ b/testutil/engine_test.go @@ -5,7 +5,7 @@ import ( ) func TestCreateEngine(t *testing.T) { - o, clean, eventC := TestEngine("foo") + o, clean, eventC, _, _ := TestEngine("foo") defer clean() defer func() { <-eventC