Function TestMigrateGet

cmd/river/rivercli/river_cli_test.go:450–551

(t *testing.T)

}

func TestMigrateGet(t *testing.T) {
	t.Parallel()

	ctx := context.Background()

	type testBundle struct {
		migratorStub *MigratorStub
		out          *bytes.Buffer
	}

	setup := func(t *testing.T) (*migrateGet, *testBundle) {
		t.Helper()

		cmd, out := withCommandBase(t, &migrateGet{})

		migratorStub := &MigratorStub{}
		migratorStub.allVersionsStub = func() []rivermigrate.Migration { return testMigrationAll }
		migratorStub.getVersionStub = func(version int) (rivermigrate.Migration, error) {
			switch version {
			case 1:
				return testMigration01, nil
			case 2:
				return testMigration02, nil
			case 3:
				return testMigration03, nil
			}
			return rivermigrate.Migration{}, fmt.Errorf("unknown version: %d", version)
		}
		migratorStub.existingVersionsStub = func(ctx context.Context) ([]rivermigrate.Migration, error) { return nil, nil }

		cmd.GetCommandBase().DriverProcurer = &DriverProcurerStub{
			getMigratorStub: func(config *rivermigrate.Config) (MigratorInterface, error) { return migratorStub, nil },
		}

		return cmd, &testBundle{
			out:          out,
			migratorStub: migratorStub,
		}
	}

	t.Run("DownMigration", func(t *testing.T) {
		t.Parallel()

		cmd, bundle := setup(t)

		_, err := runCommand(ctx, t, cmd, &migrateGetOpts{Down: true, Version: []int{1}})
		require.NoError(t, err)

		require.Equal(t, strings.TrimSpace(`
-- River main migration 001 [down]
SELECT 'down 1' FROM river_table
		`), strings.TrimSpace(bundle.out.String()))
	})

	t.Run("UpMigration", func(t *testing.T) {
		t.Parallel()

		cmd, bundle := setup(t)