diff --git a/integration_tests/create_test.go b/integration_tests/create_test.go index 275e537..0de44ac 100644 --- a/integration_tests/create_test.go +++ b/integration_tests/create_test.go @@ -36,6 +36,15 @@ func (suite *CreateUserTestSuite) TearDownSuite() { } +func (suite *CreateUserTestSuite) TearDownTest() { + err := clearTable(suite.Service, "users") + if err != nil { + fmt.Printf("Tear down test error %v\n", err.Error()) + return + } + fmt.Println("Tear down test success") +} + func (suite *CreateUserTestSuite) TestCreate() { e := echo.New() e.HTTPErrorHandler = responses.HTTPErrorHandler diff --git a/integration_tests/util.go b/integration_tests/util.go index 38f5cd8..8513471 100644 --- a/integration_tests/util.go +++ b/integration_tests/util.go @@ -74,6 +74,16 @@ func LndHubTestServiceInit() (svc *service.LndhubService, err error) { return svc, nil } +func clearTable(svc *service.LndhubService, tableName string) error { + dbConn, err := db.Open(svc.Config.DatabaseUri) + if err != nil { + return fmt.Errorf("failed to connect to database: %w", err) + } + + _, err = dbConn.Exec(fmt.Sprintf("DELETE FROM %s", tableName)) + return err +} + func createUsers(svc *service.LndhubService, usersToCreate int) (logins []controllers.CreateUserResponseBody, tokens []string, err error) { logins = []controllers.CreateUserResponseBody{} tokens = []string{}