Fix Unit Tests

#46 non-breaking
sharpbrick · Aug 1, 2020 · a7a0028 · a7a0028
1 parent 3a4f709
commit a7a0028
Showing 1 changed file with 6 additions and 4 deletions.
diff --git a/test/SharpBrick.PoweredUp.Test/Devices/VoltageTest.cs b/test/SharpBrick.PoweredUp.Test/Devices/VoltageTest.cs
@@ -73,14 +73,16 @@ await mock.WriteUpstreamAsync(new PortInputFormatSingleMessage()
         {
             var serviceProvider = new ServiceCollection()
                 .AddLogging()
-                .AddSingleton<PoweredUpBluetoothAdapterMock>()
+                .AddSingleton<IPoweredUpBluetoothAdapter, PoweredUpBluetoothAdapterMock>()
+                .AddSingleton<BluetoothKernel>()
+                .AddSingleton<IPoweredUpProtocol, PoweredUpProtocol>()
                 .AddSingleton<IDeviceFactory, DeviceFactory>() // for protocol knowledge init
+
                 .BuildServiceProvider();
 
-            var poweredUpBluetoothAdapterMock = serviceProvider.GetService<PoweredUpBluetoothAdapterMock>();
+            var poweredUpBluetoothAdapterMock = serviceProvider.GetService<IPoweredUpBluetoothAdapter>() as PoweredUpBluetoothAdapterMock;
 
-            var kernel = ActivatorUtilities.CreateInstance<BluetoothKernel>(serviceProvider, (IPoweredUpBluetoothAdapter)poweredUpBluetoothAdapterMock, (ulong)0);
-            var protocol = ActivatorUtilities.CreateInstance<PoweredUpProtocol>(serviceProvider, kernel);
+            var protocol = serviceProvider.GetService<IPoweredUpProtocol>();
 
             protocol.ConnectAsync().Wait();