From 9712d54096fe6654ff42b4079b00c10b6eea478b Mon Sep 17 00:00:00 2001
From: Raymond Truong <ratruong@microsoft.com>
Date: Thu, 13 Oct 2022 20:18:33 -0700
Subject: [PATCH] [SQL Migration] Update server name query (#20848)

* Update server name query

* Escape quotes
---
 extensions/sql-migration/src/models/stateMachine.ts | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/extensions/sql-migration/src/models/stateMachine.ts b/extensions/sql-migration/src/models/stateMachine.ts
index 654c2fa1d6..9fefe28543 100644
--- a/extensions/sql-migration/src/models/stateMachine.ts
+++ b/extensions/sql-migration/src/models/stateMachine.ts
@@ -380,7 +380,7 @@ export class MigrationStateModel implements Model, vscode.Disposable {
 			const connectionProfile = await this.getSourceConnectionProfile();
 			const connectionUri = await azdata.connection.getUriForConnection(this._sourceConnectionId);
 			const queryProvider = azdata.dataprotocol.getProvider<azdata.QueryProvider>(connectionProfile.providerId, azdata.DataProviderType.QueryProvider);
-			const queryString = 'select @@servername;';
+			const queryString = 'SELECT SERVERPROPERTY(\'ServerName\');';
 			const queryResult = await queryProvider.runQueryAndReturn(connectionUri, queryString);
 
 			if (queryResult.rowCount > 0) {