Add fixes

ajanikow · ajanikow · commit d41a79e60f4c · 2025-02-06T10:56:04.000Z
diff --git a/cmd/init_container_version_check_test.go b/cmd/init_container_version_check_test.go
@@ -23,6 +23,8 @@ package cmd
 import (
 	"encoding/json"
 	"fmt"
+	"github.com/arangodb/kube-arangodb/pkg/util/cli"
+	"github.com/arangodb/kube-arangodb/pkg/util/errors"
 	"os"
 	"testing"
 
@@ -57,7 +59,9 @@ func Test_extractVersionFromData(t *testing.T) {
 			if valid {
 				require.NoError(t, err)
 			} else {
-				ensureExitCode(t, err, cmdVersionCheckInitContainersInvalidVersionExitCode)
+				var v cli.CommandExitCode
+				require.True(t, errors.As(err, &v))
+				require.EqualValues(t, v.ExitCode, cmdVersionCheckInitContainersInvalidVersionExitCode)
 			}
 		})
 	}
diff --git a/pkg/util/pretty/marsh_table_test.go b/pkg/util/pretty/marsh_table_test.go
@@ -22,17 +22,14 @@ package pretty
 
 import (
 	"testing"
-
-	"github.com/stretchr/testify/require"
 )
 
 func Test_MarshTable(t *testing.T) {
 	type q struct {
 		B string `table:"Table Name" table_align:"center"`
 	}
 
-	z, err := NewTable[q]()
-	require.NoError(t, err)
+	z := NewTable[q]()
 
 	println(z.Add(q{
 		B: "TEST",
