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

@@ -2,43 +2,34 @@
 
 # Test class for Create Subnet Request
 class TestCreateSubnet < 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_create_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 :create_or_update, @promise do
-        assert_equal @service.create_subnet('fog-test-rg', 'fog-test-subnet', 'fog-test-virtual-network', '10.1.0.0/24', 'nsg-id', 'table-id'), expected_response
-      end
+    mocked_response = ApiStub::Requests::Network::Subnet.create_subnet_response(@network_client)
+    @subnets.stub :create_or_update, mocked_response do
+      assert_equal @service.create_subnet('fog-test-rg', 'fog-test-subnet', 'fog-test-virtual-network', '10.1.0.0/24', 'nsg-id', 'table-id'), mocked_response
     end
   end
 
   def test_create_subnet_argument_error_failure
-    response = ApiStub::Requests::Network::Subnet.create_subnet_response
-    @promise.stub :value!, response do
-      @subnets.stub :create_or_update, @promise do
-        assert_raises ArgumentError do
-          @service.create_subnet('fog-test-rg', 'fog-test-subnet', 'fog-test-virtual-network')
-        end
+    response = ApiStub::Requests::Network::Subnet.create_subnet_response(@network_client)
+    @subnets.stub :create_or_update, response do
+      assert_raises ArgumentError do
+        @service.create_subnet('fog-test-rg', 'fog-test-subnet', 'fog-test-virtual-network')
       end
     end
   end
 
   def test_create_subnet_exception_failure
-    response = -> { raise MsRestAzure::AzureOperationError.new(nil, nil, 'error' => { 'message' => 'mocked exception' }) }
-    @promise.stub :value!, response do
-      @subnets.stub :create_or_update, @promise do
-        assert_raises RuntimeError do
-          @service.create_subnet('fog-test-rg', 'fog-test-subnet', 'fog-test-virtual-network', '10.1.0.0/24', 'nsg-id', 'table-id')
-        end
+    response = proc { raise MsRestAzure::AzureOperationError.new(nil, nil, 'error' => { 'message' => 'mocked exception' }) }
+    @subnets.stub :create_or_update, response do
+      assert_raises RuntimeError do
+        @service.create_subnet('fog-test-rg', 'fog-test-subnet', 'fog-test-virtual-network', '10.1.0.0/24', 'nsg-id', 'table-id')
       end
     end
   end
 end