diff options
Diffstat (limited to 'cmd')
| -rw-r--r-- | cmd/gotosocial/action/testrig/testrig.go | 12 | ||||
| -rw-r--r-- | cmd/gotosocial/testrig.go | 2 |
2 files changed, 12 insertions, 2 deletions
diff --git a/cmd/gotosocial/action/testrig/testrig.go b/cmd/gotosocial/action/testrig/testrig.go index 6010d4047..2b70a3447 100644 --- a/cmd/gotosocial/action/testrig/testrig.go +++ b/cmd/gotosocial/action/testrig/testrig.go @@ -92,9 +92,14 @@ var Start action.GTSAction = func(ctx context.Context) error { } if state.DB != nil { + // Clean up database by + // dropping tables if required. + if !config.GetTestrigSkipDBTeardown() { + testrig.StandardDBTeardown(state.DB) + } + // Lastly, if database service was started, // ensure it gets closed now all else stopped. - testrig.StandardDBTeardown(state.DB) if err := state.DB.Close(); err != nil { log.Errorf(ctx, "error stopping database: %v", err) } @@ -125,7 +130,10 @@ var Start action.GTSAction = func(ctx context.Context) error { // that twice, we can just start the initialized caches. state.Caches.Start() - testrig.StandardDBSetup(state.DB, nil) + // Populate database tables + data if required. + if !config.GetTestrigSkipDBSetup() { + testrig.StandardDBSetup(state.DB, nil) + } // Get the instance account (we'll need this later). instanceAccount, err := state.DB.GetInstanceAccount(ctx, "") diff --git a/cmd/gotosocial/testrig.go b/cmd/gotosocial/testrig.go index ee21b7312..55498243b 100644 --- a/cmd/gotosocial/testrig.go +++ b/cmd/gotosocial/testrig.go @@ -19,6 +19,7 @@ package main import ( "code.superseriousbusiness.org/gotosocial/cmd/gotosocial/action/testrig" + "code.superseriousbusiness.org/gotosocial/internal/config" "github.com/spf13/cobra" ) @@ -38,6 +39,7 @@ func testrigCommands() *cobra.Command { } testrigCmd.AddCommand(testrigStartCmd) + config.AddTestrig(testrigCmd) return testrigCmd } return nil |