From ef4846b11f2564a9f4f056aab266c87ea01dbee4 Mon Sep 17 00:00:00 2001
From: Carlosokumu <carlosokumu254@gmail.com>
Date: Tue, 8 Oct 2024 22:18:44 +0300
Subject: [PATCH] move online,offline and test engine into the testutil package

---
 internal/testutil/TestEngine.go  | 109 +++++++++++++++++++++++++++++++
 internal/testutil/tag_offline.go |   5 ++
 internal/testutil/tag_online.go  |   5 ++
 3 files changed, 119 insertions(+)
 create mode 100644 internal/testutil/TestEngine.go
 create mode 100644 internal/testutil/tag_offline.go
 create mode 100644 internal/testutil/tag_online.go

diff --git a/internal/testutil/TestEngine.go b/internal/testutil/TestEngine.go
new file mode 100644
index 0000000..3241f7c
--- /dev/null
+++ b/internal/testutil/TestEngine.go
@@ -0,0 +1,109 @@
+package testutil
+
+import (
+	"context"
+	"fmt"
+	"os"
+	"path"
+
+	"git.defalsify.org/vise.git/engine"
+	"git.defalsify.org/vise.git/logging"
+	"git.defalsify.org/vise.git/resource"
+	"git.grassecon.net/urdt/ussd/internal/handlers"
+	"git.grassecon.net/urdt/ussd/internal/handlers/server"
+	"git.grassecon.net/urdt/ussd/internal/storage"
+	testdataloader "github.com/peteole/testdata-loader"
+)
+
+var (
+	baseDir   = testdataloader.GetBasePath()
+	logg      = logging.NewVanilla()
+	scriptDir = path.Join(baseDir, "services", "registration")
+)
+
+func TestEngine(sessionId string) (engine.Engine, func()) {
+	var accountService server.AccountServiceInterface
+	ctx := context.Background()
+	ctx = context.WithValue(ctx, "SessionId", sessionId)
+	pfp := path.Join(scriptDir, "pp.csv")
+
+	cfg := engine.Config{
+		Root:       "root",
+		SessionId:  sessionId,
+		OutputSize: uint32(160),
+		FlagCount:  uint32(16),
+	}
+
+	dbDir := ".test_state"
+	resourceDir := scriptDir
+	menuStorageService := storage.NewMenuStorageService(dbDir, resourceDir)
+
+	err := menuStorageService.EnsureDbDir()
+	if err != nil {
+		fmt.Fprintf(os.Stderr, err.Error())
+		os.Exit(1)
+	}
+
+	rs, err := menuStorageService.GetResource(ctx)
+	if err != nil {
+		fmt.Fprintf(os.Stderr, err.Error())
+		os.Exit(1)
+	}
+
+	pe, err := menuStorageService.GetPersister(ctx)
+	if err != nil {
+		fmt.Fprintf(os.Stderr, err.Error())
+		os.Exit(1)
+	}
+
+	userDataStore, err := menuStorageService.GetUserdataDb(ctx)
+	if err != nil {
+		fmt.Fprintf(os.Stderr, err.Error())
+		os.Exit(1)
+	}
+
+	dbResource, ok := rs.(*resource.DbResource)
+	if !ok {
+		fmt.Fprintf(os.Stderr, err.Error())
+		os.Exit(1)
+	}
+
+	lhs, err := handlers.NewLocalHandlerService(pfp, true, dbResource, cfg, rs)
+	lhs.SetDataStore(&userDataStore)
+	lhs.SetPersister(pe)
+
+	if err != nil {
+		fmt.Fprintf(os.Stderr, err.Error())
+		os.Exit(1)
+	}
+
+	if OnlineTestEnabled {
+		accountService = &server.AccountService{}
+	} else {
+		accountService = &server.MockAccountService{}
+	}
+	hl, err := lhs.GetHandler(accountService)
+	if err != nil {
+		fmt.Fprintf(os.Stderr, err.Error())
+		os.Exit(1)
+	}
+
+	en := lhs.GetEngine()
+	en = en.WithFirst(hl.Init)
+
+	cleanFn := func() {
+		err := en.Finish()
+		if err != nil {
+			logg.Errorf(err.Error())
+		}
+
+		err = menuStorageService.Close()
+		if err != nil {
+			logg.Errorf(err.Error())
+		}
+		logg.Infof("testengine storage closed")
+	}
+
+	en = en.WithDebug(nil)
+	return en, cleanFn
+}
diff --git a/internal/testutil/tag_offline.go b/internal/testutil/tag_offline.go
new file mode 100644
index 0000000..ea97f52
--- /dev/null
+++ b/internal/testutil/tag_offline.go
@@ -0,0 +1,5 @@
+// +build !online
+
+package testutil
+
+const OnlineTestEnabled = false
\ No newline at end of file
diff --git a/internal/testutil/tag_online.go b/internal/testutil/tag_online.go
new file mode 100644
index 0000000..6a953fa
--- /dev/null
+++ b/internal/testutil/tag_online.go
@@ -0,0 +1,5 @@
+// +build online
+
+package testutil
+
+const OnlineTestEnabled = true
\ No newline at end of file