feat: new testing incoming

Author: qbart

spec/action_migrate_up_test.go

@@ -5,6 +5,29 @@ import (
 )
 
 func TestActionMigrateUp(t *testing.T) {
+    cli := NewTestCLI(t)
+    defer cli.Teardown()
+
+    cli.DefaultDatabase()
+    cli.Set(
+        "animals.fyml",
+        `
+apiVersion: migrations/v1
+kind: MigrationSet
+metadata:
+  name: public
+spec:
+  namespace:
+    name: public
+  migrations:
+    - create_animals
+        `,
+    )
+    cli.AssertRun("migrate", "up", "--set", "public", "-d", "test")
+    cli.AssertOutputContains()
+    cli.AssertSchemaMigrate
+
+
 	c := mockCli(mockConfig(`
 migration "do_nothing" {
   version = "v1"
@@ -18,7 +41,7 @@ migration_set "public" {
 }
 `))
 	defer c.Teardown()
-	c.AssertSuccessfulRun(t, []string{"migrate", "up", "public"})
+	c.AssertSuccessfulRun(t, []string{"migrate", "up", "--set", "public", "-d", "test"})
 	c.AssertOutputContains(t, "\x1b[0;32mOK  \x1b[0mv1 do_nothing")
 	c.AssertSchemaMigrationTable(t, "public", "v1")
 }

spec/helpers_test.go

@@ -8,11 +8,10 @@ import (
 	"strings"
 	"testing"
 
+	"github.com/google/uuid"
 	_ "github.com/jackc/pgx/v5"
 	_ "github.com/jackc/pgx/v5/stdlib"
-	"github.com/ohkrab/krab/cli"
 	"github.com/ohkrab/krab/krab"
-	"github.com/ohkrab/krab/krabcli"
 	"github.com/ohkrab/krab/krabdb"
 	"github.com/ohkrab/krab/web"
 	"github.com/spf13/afero"
@@ -21,16 +20,55 @@ import (
 )
 
 type cliMock struct {
-	connection    *mockDBConnection
-	config        *krab.Config
-	app           *krabcli.App
+	connection *mockDBConnection
+	// config        *krab.Config
+	// app           *krabcli.App
 	exitCode      int
 	err           error
 	uiWriter      bytes.Buffer
 	uiErrorWriter bytes.Buffer
 	helpWriter    bytes.Buffer
 	errorWriter   bytes.Buffer
 	fs            afero.Afero
+	id            string
+}
+
+func NewTestCLI(t *testing.T) *cliMock {
+	memfs := afero.NewMemMapFs()
+
+	return &cliMock{
+		id: uuid.Must(uuid.NewV7()).String(),
+		fs: afero.Afero{Fs: memfs},
+	}
+}
+
+func (c *cliMock) DefaultDatabase() {
+	c.Files(
+		"config.fyml",
+		`
+apiVersion: drivers/v1
+kind: Driver
+metadata:
+  name: test
+spec:
+  driver: postgresql
+  config:
+    dsn: postgres://test:test@localhost:5433/test
+        `,
+	)
+}
+
+func (c *cliMock) Files(pathContentPair ...string) {
+	for i := 1; i < len(pathContentPair); i += 2 {
+		path := pathContentPair[i-1]
+		content := pathContentPair[i]
+		afero.WriteFile(
+			c.fs,
+			path,
+			[]byte(content),
+			0644,
+		)
+	}
 }
 
 func (m *cliMock) setup(args []string) {