test: fix failing tests

Author: jahvon

go.mod

@@ -95,3 +95,5 @@ require (
 	golang.org/x/text v0.26.0 // indirect
 	golang.org/x/tools v0.33.0 // indirect
 )
+
+replace github.com/jahvon/tuikit => ../tuikit

internal/io/vault/view.go

@@ -82,7 +82,7 @@ func (vc *vaultCollection) Plural() string {
 }
 
 func (vc *vaultCollection) Items() []*types.EntityInfo {
-	items := make([]*types.EntityInfo, 0, len(vc.Vaults))
+	items := make([]*types.EntityInfo, len(vc.Vaults))
 	for i, v := range vc.Vaults {
 		items[i] = &types.EntityInfo{
 			Header:    v.Name,

internal/services/run/run_test.go

@@ -52,7 +52,8 @@ var _ = Describe("Run", func() {
 				logger.EXPECT().LogMode().DoAndReturn(func() tuikitIO.LogMode {
 					return tuikitIO.Text
 				}).AnyTimes()
-				logger.EXPECT().Println("foo").Times(1)
+				logger.EXPECT().Print("foo").Times(1)
+				logger.EXPECT().Print("\n").Times(1)
 				err := run.RunCmd("echo \"foo\"", "", nil, tuikitIO.Text, logger, os.Stdin, nil)
 				Expect(err).NotTo(HaveOccurred())
 			})
@@ -126,7 +127,8 @@ var _ = Describe("Run", func() {
 			logger.EXPECT().LogMode().DoAndReturn(func() tuikitIO.LogMode {
 				return tuikitIO.Text
 			}).AnyTimes()
-			logger.EXPECT().Println("foo").Times(1)
+			logger.EXPECT().Print("foo").Times(1)
+			logger.EXPECT().Print("\n").Times(1)
 			filename := filepath.Base(testfile.Name())
 			filedir := filepath.Dir(testfile.Name())
 			err := run.RunFile(filename, filedir, nil, tuikitIO.Logfmt, logger, os.Stdin, nil)

internal/utils/utils.go

@@ -164,3 +164,11 @@ func IsZeroValue(v interface{}) bool {
 	}
 	return reflect.DeepEqual(v, reflect.Zero(reflect.TypeOf(v)).Interface())
 }
+
+const ansi = "[\u001B\u009B][[\\]()#;?]*(?:(?:(?:[a-zA-Z\\d]*(?:;[a-zA-Z\\d]*)*)?\u0007)|(?:(?:\\d{1,4}(?:;\\d{0,4})*)?[\\dA-PRZcf-ntqry=><~]))"
+
+var re = regexp.MustCompile(ansi)
+
+func StripAnsi(str string) string {
+	return re.ReplaceAllString(str, "")
+}

tests/secret_cmds_e2e_test.go

@@ -10,6 +10,7 @@ import (
 	. "github.com/onsi/ginkgo/v2"
 	. "github.com/onsi/gomega"
 
+	internalUtils "github.com/jahvon/flow/internal/utils"
 	"github.com/jahvon/flow/tests/utils"
 )
 
@@ -44,7 +45,7 @@ var _ = Describe("vault/secrets e2e", Ordered, func() {
 			Expect(lines).ToNot(BeEmpty())
 			parts := strings.Split(strings.TrimSpace(lines[0]), ":")
 			Expect(parts).To(HaveLen(2))
-			encryptionKey := strings.TrimSpace(parts[1])
+			encryptionKey := strings.TrimSpace(internalUtils.StripAnsi(parts[1]))
 			Expect(os.Setenv(keyEnv, encryptionKey)).To(Succeed())
 		})
 

tests/utils/context.go

@@ -2,6 +2,7 @@ package utils
 
 import (
 	stdCtx "context"
+	"fmt"
 	"os"
 	"path/filepath"
 	"runtime"
@@ -49,12 +50,13 @@ func NewContext(ctx stdCtx.Context, t ginkgo.FullGinkgoTInterface) *Context {
 		tuikitIO.WithOutput(stdOut),
 		tuikitIO.WithTheme(io.Theme("")),
 		tuikitIO.WithMode(tuikitIO.Text),
-		tuikitIO.WithExitFunc(func() {
+		tuikitIO.WithExitFunc(func(msg string, args ...any) {
+			msg = fmt.Sprintf(msg, args...)
 			_, file, line, ok := runtime.Caller(3)
 			if ok {
-				t.Fatalf("logger exit called from %s:%d", file, line)
+				t.Fatalf("logger exit called from %s:%d - %s", file, line, msg)
 			} else {
-				t.Fatalf("logger exit called")
+				t.Fatalf("logger exit called - %s", msg)
 			}
 		}),
 	)
@@ -123,12 +125,13 @@ func ResetTestContext(ctx *Context, t ginkgo.FullGinkgoTInterface) {
 		tuikitIO.WithOutput(stdOut),
 		tuikitIO.WithTheme(io.Theme("")),
 		tuikitIO.WithMode(tuikitIO.Text),
-		tuikitIO.WithExitFunc(func() {
+		tuikitIO.WithExitFunc(func(msg string, args ...any) {
+			msg = fmt.Sprintf(msg, args...)
 			_, file, line, ok := runtime.Caller(3)
 			if ok {
-				t.Fatalf("logger exit called from %s:%d", file, line)
+				t.Fatalf("logger exit called from %s:%d - %s", file, line, msg)
 			} else {
-				t.Fatalf("logger exit called")
+				t.Fatalf("logger exit called - %s", msg)
 			}
 		}),
 	)