test/requests/network/test_get_subnet.rb in fog-azure-rm-0.0.4 vs test/requests/network/test_get_subnet.rb in fog-azure-rm-0.0.5

@@ -2,32 +2,25 @@
 
 # Test class for Get Subnet Request
 class TestGetSubnet < Minitest::Test
   def setup
     @service = Fog::Network::AzureRM.new(credentials)
-    client = @service.instance_variable_get(:@network_client)
-    @subnets = client.subnets
-    @promise = Concurrent::Promise.execute do
-    end
+    @network_client = @service.instance_variable_get(:@network_client)
+    @subnets = @network_client.subnets
   end
 
   def test_get_subnet_success
-    mocked_response = ApiStub::Requests::Network::Subnet.create_subnet_response
-    expected_response = Azure::ARM::Network::Models::Subnet.serialize_object(mocked_response.body)
-    @promise.stub :value!, mocked_response do
-      @subnets.stub :get, @promise do
-        assert_equal @service.get_subnet('fog-test-rg', 'fog-test-virtual-network', 'fog-test-subnet'), expected_response
-      end
+    mocked_response = ApiStub::Requests::Network::Subnet.create_subnet_response(@network_client)
+    @subnets.stub :get, mocked_response do
+      assert_equal @service.get_subnet('fog-test-rg', 'fog-test-virtual-network', 'fog-test-subnet'), mocked_response
     end
   end
 
   def test_get_subnet_exception_failure
-    response = -> { raise MsRestAzure::AzureOperationError.new(nil, nil, 'error' => { 'message' => 'mocked exception' }) }
-    @promise.stub :value!, response do
-      @subnets.stub :get, @promise do
-        assert_raises RuntimeError do
-          @service.get_subnet('fog-test-rg', 'fog-test-virtual-network', 'fog-test-subnet')
-        end
+    response = proc { raise MsRestAzure::AzureOperationError.new(nil, nil, 'error' => { 'message' => 'mocked exception' }) }
+    @subnets.stub :get, response do
+      assert_raises RuntimeError do
+        @service.get_subnet('fog-test-rg', 'fog-test-virtual-network', 'fog-test-subnet')
       end
     end
   end
 end