diff --git a/go/libraries/doltcore/dtestutils/sql_server_driver/cmd.go b/go/libraries/doltcore/dtestutils/sql_server_driver/cmd.go index c37b36c46a..25fea37c89 100644 --- a/go/libraries/doltcore/dtestutils/sql_server_driver/cmd.go +++ b/go/libraries/doltcore/dtestutils/sql_server_driver/cmd.go @@ -107,6 +107,10 @@ func (u DoltUser) MakeRepoStore() (RepoStore, error) { return RepoStore{u, tmpdir}, nil } +func (u DoltUser) Cleanup() error { + return os.RemoveAll(u.tmpdir) +} + type RepoStore struct { user DoltUser Dir string diff --git a/integration-tests/go-sql-server-driver/testdef.go b/integration-tests/go-sql-server-driver/testdef.go index ea545fa5e6..19314303df 100644 --- a/integration-tests/go-sql-server-driver/testdef.go +++ b/integration-tests/go-sql-server-driver/testdef.go @@ -112,6 +112,9 @@ func (test Test) Run(t *testing.T) { u, err := driver.NewDoltUser() require.NoError(t, err) + t.Cleanup(func() { + u.Cleanup() + }) rs, err := u.MakeRepoStore() require.NoError(t, err) @@ -130,6 +133,9 @@ func (test Test) Run(t *testing.T) { // Each MultiRepo gets its own dolt config --global. u, err := driver.NewDoltUser() require.NoError(t, err) + t.Cleanup(func() { + u.Cleanup() + }) rs, err = u.MakeRepoStore() require.NoError(t, err) for _, r := range mr.Repos {