(Feat): Initial Commit

internal/util/command/command_test.go

@@ -0,0 +1,34 @@
+package command_test
+
+import (
+	"context"
+	"errors"
+	"testing"
+
+	"allaboutapps.dev/aw/go-starter/internal/api"
+	"allaboutapps.dev/aw/go-starter/internal/test"
+	"allaboutapps.dev/aw/go-starter/internal/util/command"
+	"github.com/stretchr/testify/assert"
+	"github.com/stretchr/testify/require"
+)
+
+func TestWithServer(t *testing.T) {
+	test.WithTestServer(t, func(s *api.Server) {
+		ctx := t.Context()
+
+		var testError = errors.New("test error")
+
+		s.Config.Logger.PrettyPrintConsole = false
+		resultErr := command.WithServer(ctx, s.Config, func(ctx context.Context, s *api.Server) error {
+			var database string
+			err := s.DB.QueryRowContext(ctx, "SELECT current_database();").Scan(&database)
+			require.NoError(t, err)
+
+			assert.NotEmpty(t, database)
+
+			return testError
+		})
+
+		assert.Equal(t, testError, resultErr)
+	})
+}