Fix Connection tests break

This commit is contained in:
Karl Burtram
2016-09-19 12:35:59 -07:00
parent 03315475cb
commit 30a6fdd26c

View File

@@ -138,8 +138,9 @@ namespace Microsoft.SqlTools.ServiceLayer.Test.Connection
// register disconnect callback // register disconnect callback
connectionService.RegisterOnDisconnectTask( connectionService.RegisterOnDisconnectTask(
(result) => { (result, uri) => {
callbackInvoked = true; callbackInvoked = true;
Assert.True(uri.Equals(ownerUri));
return Task.FromResult(true); return Task.FromResult(true);
} }
); );
@@ -433,8 +434,9 @@ namespace Microsoft.SqlTools.ServiceLayer.Test.Connection
// register disconnect callback // register disconnect callback
connectionService.RegisterOnDisconnectTask( connectionService.RegisterOnDisconnectTask(
(result) => { (result, uri) => {
callbackInvoked = true; callbackInvoked = true;
Assert.True(uri.Equals(ownerUri));
return Task.FromResult(true); return Task.FromResult(true);
} }
); );