Fix EdgeHub CI Tests (#276)

Azure · Sep 12, 2018 · 4f3ccd6 · 4f3ccd6
1 parent edb15fb
commit 4f3ccd6
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 2 deletions.
diff --git a/...-hub/test/Microsoft.Azure.Devices.Edge.Hub.CloudProxy.Test/CloudConnectionProviderTest.cs b/...-hub/test/Microsoft.Azure.Devices.Edge.Hub.CloudProxy.Test/CloudConnectionProviderTest.cs
@@ -27,7 +27,9 @@ public class CloudConnectionProviderTest
         [Integration]
         public async Task ConnectTest()
         {
+            var edgeHub = Mock.Of<IEdgeHub>();
             ICloudConnectionProvider cloudConnectionProvider = new CloudConnectionProvider(MessageConverterProvider, ConnectionPoolSize, new ClientProvider(), Option.None<UpstreamProtocol>(), TokenProvider, DeviceScopeIdentitiesCache, TimeSpan.FromMinutes(60));
+            cloudConnectionProvider.BindEdgeHub(edgeHub);
             string deviceConnectionString = await SecretsHelper.GetSecretFromConfigKey("device1ConnStrKey");
             var deviceIdentity = Mock.Of<IDeviceIdentity>(m => m.Id == ConnectionStringHelper.GetDeviceId(deviceConnectionString));
             var clientCredentials = new SharedKeyCredentials(deviceIdentity, deviceConnectionString, null);
@@ -41,8 +43,9 @@ public async Task ConnectTest()
         [Integration]
         public async Task ConnectWithInvalidConnectionStringTest()
         {
+            var edgeHub = Mock.Of<IEdgeHub>();
             ICloudConnectionProvider cloudConnectionProvider = new CloudConnectionProvider(MessageConverterProvider, ConnectionPoolSize, new ClientProvider(), Option.None<UpstreamProtocol>(), TokenProvider, DeviceScopeIdentitiesCache, TimeSpan.FromMinutes(60));
-
+            cloudConnectionProvider.BindEdgeHub(edgeHub);
             var deviceIdentity1 = Mock.Of<IIdentity>(m => m.Id == "device1");
             var clientCredentials1 = new SharedKeyCredentials(deviceIdentity1, "dummyConnStr", null);
             Try<ICloudConnection> result = await cloudConnectionProvider.Connect(clientCredentials1, null);

diff --git a/edge-hub/test/Microsoft.Azure.Devices.Edge.Hub.E2E.Test/ProtocolHeadFixture.cs b/edge-hub/test/Microsoft.Azure.Devices.Edge.Hub.E2E.Test/ProtocolHeadFixture.cs
@@ -174,7 +174,7 @@ async Task StartProtocolHead()
 
                 // CloudConnectionProvider and RoutingEdgeHub have a circular dependency. So set the
                 // EdgeHub on the CloudConnectionProvider before any other operation
-                var cloudConnectionProvider = this.container.Resolve<ICloudConnectionProvider>();
+                ICloudConnectionProvider cloudConnectionProvider = await this.container.Resolve<Task<ICloudConnectionProvider>>();
                 IEdgeHub edgeHub = await this.container.Resolve<Task<IEdgeHub>>();
                 cloudConnectionProvider.BindEdgeHub(edgeHub);