Update to latest mssqldriver (#7613)

* New driver does not tolerate USE - handle this by closing db and reopening db in the new dbname

1 files changed, 9 insertions, 1 deletions

diff --git a/integrations/migration-test/migration_test.go b/integrations/migration-test/migration_test.go
index 15b086c6e6..fd0d0b7c72 100644
--- a/integrations/migration-test/migration_test.go
+++ b/integrations/migration-test/migration_test.go
@@ -181,11 +181,19 @@ func restoreOldDB(t *testing.T, version string) bool {
 		assert.NoError(t, err)
 		defer db.Close()
 
-		_, err = db.Exec("DROP DATABASE IF EXISTS gitea")
+		_, err = db.Exec("DROP DATABASE IF EXISTS [gitea]")
 		assert.NoError(t, err)
 
 		statements := strings.Split(data, "\nGO\n")
 		for _, statement := range statements {
+			if len(statement) > 5 && statement[:5] == "USE [" {
+				dbname := statement[5 : len(statement)-1]
+				db.Close()
+				db, err = sql.Open("mssql", fmt.Sprintf("server=%s; port=%s; database=%s; user id=%s; password=%s;",
+					host, port, dbname, models.DbCfg.User, models.DbCfg.Passwd))
+				assert.NoError(t, err)
+				defer db.Close()
+			}
 			_, err = db.Exec(statement)
 			assert.NoError(t, err, "Failure whilst running: %s\nError: %v", statement, err)
 		}