fog-azure-rm 0.0.5 → 0.0.6

data/test/integration/server.rb

@@ -94,7 +94,8 @@ compute.servers.create(
   offer: 'UbuntuServer',
   sku: '14.04.2-LTS',
   version: 'latest',
-  platform: 'linux'
+  platform: 'linux',
+  custom_data: 'echo customData'
 )
 
 ########################################################################################################################
@@ -134,8 +135,8 @@ nic.destroy
 vnet = network.virtual_networks.get('TestRG-VM', 'testVnet')
 vnet.destroy
 
-storage = storage.storage_accounts(resource_group: 'TestRG-VM').get('fogstorageac')
+storage = storage.storage_accounts.get('TestRG-VM', 'fogstorageac')
 storage.destroy
 
-rg = rs.resource_groups.get('TestRG-VM')
-rg.destroy
+resource_group = rs.resource_groups.get('TestRG-VM')
+resource_group.destroy

data/test/integration/storage_account.rb

@@ -75,16 +75,16 @@ storage.storage_accounts.create(
 ######################                         Get and Delete Storage Account                     ######################
 ########################################################################################################################
 
-standard_storage_account = storage.storage_accounts(resource_group: 'TestRG-SA').get('fogstandardsalrs')
+standard_storage_account = storage.storage_accounts.get('TestRG-SA', 'fogstandardsalrs')
 standard_storage_account.destroy
-standard_storage_account = storage.storage_accounts(resource_group: 'TestRG-SA').get('fogstandardsagrs')
+standard_storage_account = storage.storage_accounts.get('TestRG-SA', 'fogstandardsagrs')
 standard_storage_account.destroy
-premium_storage_account = storage.storage_accounts(resource_group: 'TestRG-SA').get('fogpremiumsa')
+premium_storage_account = storage.storage_accounts.get('TestRG-SA', 'fogpremiumsa')
 premium_storage_account.destroy
 
 ########################################################################################################################
 ######################                                   CleanUp                                  ######################
 ########################################################################################################################
 
-rg = rs.resource_groups.get('TestRG-SA')
-rg.destroy
+resource_group = rs.resource_groups.get('TestRG-SA')
+resource_group.destroy

data/test/integration/traffic_manager.rb

@@ -58,6 +58,16 @@ traffic_manager.traffic_manager_end_points.create(
   target: 'test-app.com',
   endpoint_location: 'eastus'
 )
+########################################################################################################################
+######################                   Get and Update Traffic Manager Endpoint                ######################
+########################################################################################################################
+
+end_point = traffic_manager.traffic_manager_end_points.get('TestRG-TM', 'test-tmp', 'myendpoint', 'externalEndpoints')
+end_point.update(
+  type: 'externalEndpoints',
+  target: 'test-app1.com',
+  endpoint_location: 'centralus'
+)
 
 ########################################################################################################################
 ######################                   Get and Destroy Traffic Manager Endpoint                ######################
@@ -66,6 +76,17 @@ traffic_manager.traffic_manager_end_points.create(
 end_point = traffic_manager.traffic_manager_end_points.get('TestRG-TM', 'test-tmp', 'myendpoint', 'externalEndpoints')
 end_point.destroy
 
+########################################################################################################################
+######################                    Get and Update Traffic Manager Profile                 ######################
+########################################################################################################################
+
+traffic_manager_profile = traffic_manager.traffic_manager_profiles.get('TestRG-TM', 'test-tmp')
+traffic_manager_profile.update(traffic_routing_method: 'Weighted',
+                               ttl: '35',
+                               protocol: 'https',
+                               port: '90',
+                               path: '/monitorpage1.aspx')
+
 ########################################################################################################################
 ######################                    Get and Destroy Traffic Manager Profile                 ######################
 ########################################################################################################################

data/test/integration/virtual_network_gateway.rb

@@ -31,6 +31,42 @@ resource.resource_groups.create(
   location: 'eastus'
 )
 
+network.virtual_networks.create(
+  name: 'testVnet',
+  location: 'eastus',
+  resource_group: 'TestRG-VNG',
+  network_address_list: '10.1.0.0/16,10.2.0.0/16'
+)
+
+network.subnets.create(
+  name: 'GatewaySubnet',
+  resource_group: 'TestRG-VNG',
+  virtual_network_name: 'testVnet',
+  address_prefix: '10.2.0.0/24'
+)
+
+network.public_ips.create(
+  name: 'mypubip',
+  resource_group: 'TestRG-VNG',
+  location: 'eastus',
+  public_ip_allocation_method: 'Dynamic'
+)
+
+network.local_network_gateways.create(
+  name: 'testlocalnetworkgateway',
+  location: 'eastus',
+  tags: {
+    key1: 'value1',
+    key2: 'value2'
+  },
+  resource_group: 'TestRG-VNG',
+  gateway_ip_address: '192.168.1.1',
+  local_network_address_space_prefixes: [],
+  asn: 100,
+  bgp_peering_address: '192.168.1.2',
+  peer_weight: 3
+)
+
 ########################################################################################################################
 ######################                           Create Virtual Network Gateway                   ######################
 ########################################################################################################################
@@ -46,16 +82,16 @@ network.virtual_network_gateways.create(
     {
       name: 'default',
       private_ipallocation_method: 'Dynamic',
-      public_ipaddress_id: '/subscriptions/{subscription_id}/resourceGroups/{resource_group_name}/providers/Microsoft.Network/publicIPAddresses/{public_ip_name}',
-      subnet_id: '/subscriptions/{subscription_id}/resourceGroups/{resource_group_name}/providers/Microsoft.Network/virtualNetworks/{virtual_network_name}/subnets/{subnet_name}',
+      public_ipaddress_id: "/subscriptions/#{azure_credentials['subscription_id']}/resourceGroups/TestRG-VNG/providers/Microsoft.Network/publicIPAddresses/mypubip",
+      subnet_id: "/subscriptions/#{azure_credentials['subscription_id']}/resourceGroups/TestRG-VNG/providers/Microsoft.Network/virtualNetworks/testVnet/subnets/GatewaySubnet",
       private_ipaddress: nil
     }
   ],
-  resource_group: 'learn_fog',
+  resource_group: 'TestRG-VNG',
   sku_name: 'Standard',
   sku_tier: 'Standard',
   sku_capacity: 2,
-  gateway_type: 'ExpressRoute',
+  gateway_type: 'vpn',
   enable_bgp: true,
   gateway_size: nil,
   asn: 100,
@@ -63,23 +99,7 @@ network.virtual_network_gateways.create(
   peer_weight: 3,
   vpn_type: 'RouteBased',
   vpn_client_address_pool: [],
-  gateway_default_site: '/subscriptions/{subscription_id}/resourceGroups/{resource_group_name}/providers/Microsoft.Network/localNetworkGateways/{local_network_gateway_name}',
-  default_sites: [],
-  vpn_client_configuration: {
-    address_pool: ['192.168.0.4', '192.168.0.5'],
-    root_certificates: [
-      {
-        name: 'root',
-        public_cert_data: 'certificate data'
-      }
-    ],
-    revoked_certificates: [
-      {
-        name: 'revoked',
-        thumbprint: 'thumb print detail'
-      }
-    ]
-  }
+  gateway_default_site: "/subscriptions/#{azure_credentials['subscription_id']}/resourceGroups/TestRG-VNG/providers/Microsoft.Network/localNetworkGateways/testlocalnetworkgateway"
 )
 
 ########################################################################################################################
@@ -95,7 +115,16 @@ end
 ######################                  Get Virtual Network Gateway and CleanUp                   ######################
 ########################################################################################################################
 
-network_gateway = network.virtual_network_gateways.get('learn_fog', 'testVNG')
+network_gateway = network.virtual_network_gateways.get('TestRG-VNG', 'testnetworkgateway')
 puts network_gateway.name.to_s
 
 network_gateway.destroy
+
+pubip = network.public_ips.get('TestRG-VNG', 'mypubip')
+pubip.destroy
+
+local_network_gateway = network.local_network_gateways.get('TestRG-VNG', 'testlocalnetworkgateway')
+local_network_gateway.destroy
+
+rg = resource.resource_groups.get('TestRG-VNG')
+rg.destroy

data/test/models/application_gateway/test_gateway.rb

@@ -7,6 +7,13 @@ class TestGateway < Minitest::Test
     @gateway = gateway(@service)
     @gateway_client = @service.instance_variable_get(:@network_client)
     @response = ApiStub::Models::ApplicationGateway::Gateway.create_application_gateway_response(@gateway_client)
+    gateways = Fog::ApplicationGateway::AzureRM::Gateways.new(resource_group: 'fog-test-rg', service: @service)
+    @service.stub :get_application_gateway, @response do
+      @gateway_obj = gateways.get('fog-test-rg', 'gateway')
+    end
+    @ssl_certifcate = ApiStub::Models::ApplicationGateway::Gateway.ssl_certifcate
+    @frontend_port = ApiStub::Models::ApplicationGateway::Gateway.frontend_port
+    @probe = ApiStub::Models::ApplicationGateway::Gateway.probe
   end
 
   def test_model_methods
@@ -44,11 +51,59 @@ class TestGateway < Minitest::Test
   end
 
   def test_save_method_response
-    @service.stub :create_application_gateway, @response do
+    @service.stub :create_or_update_application_gateway, @response do
       assert_instance_of Fog::ApplicationGateway::AzureRM::Gateway, @gateway.save
     end
   end
 
+  def test_update_sku
+    @service.stub :update_sku_attributes, @response do
+      assert_instance_of Fog::ApplicationGateway::AzureRM::Gateway, @gateway_obj.update_sku('Standard_Medium', '2')
+    end
+  end
+
+  def test_update_gateway_ip_configuration
+    @service.stub :update_subnet_id_in_gateway_ip_configuration, @response do
+      assert_instance_of Fog::ApplicationGateway::AzureRM::Gateway, @gateway_obj.update_gateway_ip_configuration('/subscriptions/{guid}/resourceGroups/{resourceGroupName}/providers/Microsoft.Network/virtualNetworks/vnet/subnets/GatewaySubnet')
+    end
+  end
+
+  def test_add_ssl_certificate
+    @service.stub :create_or_update_application_gateway, @response do
+      assert_instance_of Fog::ApplicationGateway::AzureRM::Gateway, @gateway_obj.add_ssl_certificate(@ssl_certifcate)
+    end
+  end
+
+  def test_remove_ssl_certificate
+    @service.stub :create_or_update_application_gateway, @response do
+      assert_instance_of Fog::ApplicationGateway::AzureRM::Gateway, @gateway_obj.remove_ssl_certificate(@ssl_certifcate)
+    end
+  end
+
+  def test_add_frontend_port
+    @service.stub :create_or_update_application_gateway, @response do
+      assert_instance_of Fog::ApplicationGateway::AzureRM::Gateway, @gateway_obj.add_frontend_port(@frontend_port)
+    end
+  end
+
+  def test_remove_frontend_port
+    @service.stub :create_or_update_application_gateway, @response do
+      assert_instance_of Fog::ApplicationGateway::AzureRM::Gateway, @gateway_obj.remove_frontend_port(@frontend_port)
+    end
+  end
+
+  def test_add_probe
+    @service.stub :create_or_update_application_gateway, @response do
+      assert_instance_of Fog::ApplicationGateway::AzureRM::Gateway, @gateway_obj.add_probe(@probe)
+    end
+  end
+
+  def test_remove_probe
+    @service.stub :create_or_update_application_gateway, @response do
+      assert_instance_of Fog::ApplicationGateway::AzureRM::Gateway, @gateway_obj.remove_probe(@probe)
+    end
+  end
+
   def test_destroy_method_response
     @service.stub :delete_application_gateway, true do
       assert @gateway.destroy

data/test/models/application_gateway/test_gateways.rb

@@ -6,7 +6,7 @@ class TestGateways < Minitest::Test
     @service = Fog::ApplicationGateway::AzureRM.new(credentials)
     @gateways = Fog::ApplicationGateway::AzureRM::Gateways.new(resource_group: 'fog-test-rg', service: @service)
     @gateway_client = @service.instance_variable_get(:@network_client)
-    @response = [ApiStub::Models::ApplicationGateway::Gateway.create_application_gateway_response(@gateway_client)]
+    @response = ApiStub::Models::ApplicationGateway::Gateway.create_application_gateway_response(@gateway_client)
   end
 
   def test_collection_methods
@@ -24,7 +24,8 @@ class TestGateways < Minitest::Test
   end
 
   def test_all_method_response
-    @service.stub :list_application_gateways, @response do
+    response = [@response]
+    @service.stub :list_application_gateways, response do
       assert_instance_of Fog::ApplicationGateway::AzureRM::Gateways, @gateways.all
       assert @gateways.all.size >= 1
       @gateways.all.each do |application_gateway|
@@ -34,9 +35,8 @@ class TestGateways < Minitest::Test
   end
 
   def test_get_method_response
-    @service.stub :list_application_gateways, @response do
-      assert_instance_of Fog::ApplicationGateway::AzureRM::Gateway, @gateways.get('gateway')
-      assert @gateways.get('wrong-name').nil?, true
+    @service.stub :get_application_gateway, @response do
+      assert_instance_of Fog::ApplicationGateway::AzureRM::Gateway, @gateways.get('fog-test-rg', 'gateway')
     end
   end
 end

data/test/models/compute/test_servers.rb

@@ -6,14 +6,13 @@ class TestServers < Minitest::Test
     @service = Fog::Compute::AzureRM.new(credentials)
     @servers = Fog::Compute::AzureRM::Servers.new(resource_group: 'fog-test-rg', service: @service)
     @compute_client = @service.instance_variable_get(:@compute_mgmt_client)
-    @response = [ApiStub::Models::Compute::Server.create_windows_virtual_machine_response(@compute_client)]
+    @response = ApiStub::Models::Compute::Server.create_windows_virtual_machine_response(@compute_client)
   end
 
   def test_collection_methods
     methods = [
       :all,
-      :get,
-      :get_from_remote
+      :get
     ]
     methods.each do |method|
       assert @servers.respond_to? method, true
@@ -25,7 +24,8 @@ class TestServers < Minitest::Test
   end
 
   def test_all_method_response
-    @service.stub :list_virtual_machines, @response do
+    response = [@response]
+    @service.stub :list_virtual_machines, response do
       assert_instance_of Fog::Compute::AzureRM::Servers, @servers.all
       assert @servers.all.size >= 1
       @servers.all.each do |s|
@@ -35,16 +35,8 @@ class TestServers < Minitest::Test
   end
 
   def test_get_method_response
-    @service.stub :list_virtual_machines, @response do
-      assert_instance_of Fog::Compute::AzureRM::Server, @servers.get('fog-test-server')
-      assert @servers.get('wrong-name').nil?, true
-    end
-  end
-
-  def test_get_from_remote_method_response
-    response = ApiStub::Models::Compute::Server.create_windows_virtual_machine_response(@compute_client)
-    @service.stub :get_virtual_machine, response do
-      assert_instance_of Fog::Compute::AzureRM::Server, @servers.get_from_remote('fog-test-rg', 'fog-test-server')
+    @service.stub :get_virtual_machine, @response do
+      assert_instance_of Fog::Compute::AzureRM::Server, @servers.get('fog-test-rg', 'fog-test-server')
     end
   end
 end

data/test/models/network/test_express_route_circuit_authorization.rb

@@ -0,0 +1,50 @@
+require File.expand_path '../../test_helper', __dir__
+
+# Test class for ExpressRouteCircuitAuthorization Model
+class TestExpressRouteCircuitAuthorization < Minitest::Test
+  def setup
+    @service = Fog::Network::AzureRM.new(credentials)
+    @circuit_authorization = express_route_circuit_authorization(@service)
+    @network_client = @service.instance_variable_get(:@network_client)
+  end
+
+  def test_model_methods
+    methods = [
+      :save,
+      :destroy
+    ]
+    methods.each do |method|
+      assert @circuit_authorization.respond_to? method
+    end
+  end
+
+  def test_model_attributes
+    attributes = [
+      :name,
+      :id,
+      :resource_group,
+      :authorization_name,
+      :authorization_key,
+      :authorization_status,
+      :provisioning_state,
+      :etag,
+      :circuit_name
+    ]
+    attributes.each do |attribute|
+      assert @circuit_authorization.respond_to? attribute
+    end
+  end
+
+  def test_save_method_response
+    response = ApiStub::Models::Network::ExpressRouteCircuitAuthorization.create_express_route_circuit_authorization_response(@network_client)
+    @service.stub :create_or_update_express_route_circuit_authorization, response do
+      assert_instance_of Fog::Network::AzureRM::ExpressRouteCircuitAuthorization, @circuit_authorization.save
+    end
+  end
+
+  def test_destroy_method_response
+    @service.stub :delete_express_route_circuit_authorization, true do
+      assert @circuit_authorization.destroy
+    end
+  end
+end

data/test/models/network/test_express_route_circuit_authorizations.rb

@@ -0,0 +1,43 @@
+require File.expand_path '../../test_helper', __dir__
+
+# Test class for ExpressRouteCircuitAuthorizations Collection
+class TestExpressRouteCircuitAuthorizations < Minitest::Test
+  def setup
+    @service = Fog::Network::AzureRM.new(credentials)
+    @circuit_authorizations = Fog::Network::AzureRM::ExpressRouteCircuitAuthorizations.new(resource_group: 'fog-test-rg', circuit_name: 'testCircuit', service: @service)
+    @network_client = @service.instance_variable_get(:@network_client)
+  end
+
+  def test_collection_methods
+    methods = [
+      :all,
+      :get
+    ]
+    methods.each do |method|
+      assert @circuit_authorizations.respond_to? method
+    end
+  end
+
+  def test_collection_attributes
+    assert @circuit_authorizations.respond_to? :resource_group
+    assert @circuit_authorizations.respond_to? :circuit_name
+  end
+
+  def test_all_method_response
+    response = [ApiStub::Models::Network::ExpressRouteCircuitAuthorization.create_express_route_circuit_authorization_response(@network_client)]
+    @service.stub :list_express_route_circuit_authorizations, response do
+      assert_instance_of Fog::Network::AzureRM::ExpressRouteCircuitAuthorizations, @circuit_authorizations.all
+      assert @circuit_authorizations.all.size >= 1
+      @circuit_authorizations.all.each do |circuit_authorization|
+        assert_instance_of Fog::Network::AzureRM::ExpressRouteCircuitAuthorization, circuit_authorization
+      end
+    end
+  end
+
+  def test_get_method_response
+    response = ApiStub::Models::Network::ExpressRouteCircuitAuthorization.create_express_route_circuit_authorization_response(@network_client)
+    @service.stub :get_express_route_circuit_authorization, response do
+      assert_instance_of Fog::Network::AzureRM::ExpressRouteCircuitAuthorization, @circuit_authorizations.get('HaiderRG', 'testCircuit', 'auth-name')
+    end
+  end
+end

data/test/models/network/test_load_balancers.rb

@@ -5,8 +5,8 @@ class TestLoadBalancers < Minitest::Test
   def setup
     @service = Fog::Network::AzureRM.new(credentials)
     @load_balancers = Fog::Network::AzureRM::LoadBalancers.new(resource_group: 'fog-test-rg', service: @service)
-    network_client = @service.instance_variable_get(:@network_client)
-    @response = [ApiStub::Models::Network::LoadBalancer.create_load_balancer_response(network_client)]
+    @network_client = @service.instance_variable_get(:@network_client)
+    @response = ApiStub::Models::Network::LoadBalancer.create_load_balancer_response(@network_client)
   end
 
   def test_collection_methods
@@ -24,7 +24,8 @@ class TestLoadBalancers < Minitest::Test
   end
 
   def test_all_method_response
-    @service.stub :list_load_balancers, @response do
+    response = [@response]
+    @service.stub :list_load_balancers, response do
       assert_instance_of Fog::Network::AzureRM::LoadBalancers, @load_balancers.all
       assert @load_balancers.all.size >= 1
       @load_balancers.all.each do |lb|
@@ -34,9 +35,8 @@ class TestLoadBalancers < Minitest::Test
   end
 
   def test_get_method_response
-    @service.stub :list_load_balancers, @response do
-      assert_instance_of Fog::Network::AzureRM::LoadBalancer, @load_balancers.get('mylb1')
-      assert @load_balancers.get('wrong-name').nil?, true
+    @service.stub :get_load_balancer, @response do
+      assert_instance_of Fog::Network::AzureRM::LoadBalancer, @load_balancers.get('fog-test-rg', 'mylb1')
     end
   end
 end

data/test/models/network/test_local_network_gateway.rb

@@ -0,0 +1,53 @@
+require File.expand_path '../../test_helper', __dir__
+
+# Test class for LocalNetworkGateway Model
+class TestLocalNetworkGateway < Minitest::Test
+  def setup
+    @service = Fog::Network::AzureRM.new(credentials)
+    @local_network_gateway = local_network_gateway(@service)
+    network_client = @service.instance_variable_get(:@network_client)
+    @response = ApiStub::Models::Network::LocalNetworkGateway.create_local_network_gateway_response(network_client)
+  end
+
+  def test_model_methods
+    methods = [
+      :save,
+      :destroy
+    ]
+    methods.each do |method|
+      assert @local_network_gateway.respond_to? method
+    end
+  end
+
+  def test_model_attributes
+    attributes = [
+      :name,
+      :id,
+      :location,
+      :type,
+      :resource_group,
+      :tags,
+      :local_network_address_space_prefixes,
+      :gateway_ip_address,
+      :asn,
+      :bgp_peering_address,
+      :peer_weight,
+      :provisioning_state
+    ]
+    attributes.each do |attribute|
+      assert @local_network_gateway.respond_to? attribute
+    end
+  end
+
+  def test_save_method_response
+    @service.stub :create_or_update_local_network_gateway, @response do
+      assert_instance_of Fog::Network::AzureRM::LocalNetworkGateway, @local_network_gateway.save
+    end
+  end
+
+  def test_destroy_method_response
+    @service.stub :delete_local_network_gateway, true do
+      assert @local_network_gateway.destroy
+    end
+  end
+end

data/test/models/network/test_local_network_gateways.rb

@@ -0,0 +1,42 @@
+require File.expand_path '../../test_helper', __dir__
+
+# Test class for LocalNetworkGateway Collection
+class TestLocalNetworkGateways < Minitest::Test
+  def setup
+    @service = Fog::Network::AzureRM.new(credentials)
+    @local_network_gateways = Fog::Network::AzureRM::LocalNetworkGateways.new(resource_group: 'fog-rg', service: @service)
+    @network_client = @service.instance_variable_get(:@network_client)
+    @response = ApiStub::Models::Network::LocalNetworkGateway.create_local_network_gateway_response(@network_client)
+  end
+
+  def test_collection_methods
+    methods = [
+      :all,
+      :get
+    ]
+    methods.each do |method|
+      assert @local_network_gateways.respond_to? method
+    end
+  end
+
+  def test_collection_attributes
+    assert @local_network_gateways.respond_to? :resource_group
+  end
+
+  def test_all_method_response
+    response = [@response]
+    @service.stub :list_local_network_gateways, response do
+      assert_instance_of Fog::Network::AzureRM::LocalNetworkGateways, @local_network_gateways.all
+      assert @local_network_gateways.all.size >= 1
+      @local_network_gateways.all.each do |local_network_gateway|
+        assert_instance_of Fog::Network::AzureRM::LocalNetworkGateway, local_network_gateway
+      end
+    end
+  end
+
+  def test_get_method_response
+    @service.stub :get_local_network_gateway, @response do
+      assert_instance_of Fog::Network::AzureRM::LocalNetworkGateway, @local_network_gateways.get('fog-rg', 'mylocalgateway1')
+    end
+  end
+end

data/test/models/network/test_public_ip.rb

@@ -4,20 +4,18 @@ require File.expand_path '../../test_helper', __dir__
 class TestPublicIp < Minitest::Test
   def setup
     @service = Fog::Network::AzureRM.new(credentials)
-    network_client = @service.instance_variable_get(:@network_client)
+    @network_client = @service.instance_variable_get(:@network_client)
     @public_ip = public_ip(@service)
-    @response = ApiStub::Models::Network::PublicIp.create_public_ip_response(network_client)
   end
 
   def test_model_methods
     methods = [
       :save,
-      :destroy
+      :destroy,
+      :update
     ]
-    @service.stub :create_public_ip, @response do
-      methods.each do |method|
-        assert @public_ip.respond_to? method
-      end
+    methods.each do |method|
+      assert @public_ip.respond_to? method
     end
   end
 
@@ -35,19 +33,25 @@ class TestPublicIp < Minitest::Test
       :fqdn,
       :reverse_fqdn
     ]
-    @service.stub :create_public_ip, @response do
-      attributes.each do |attribute|
-        assert @public_ip.respond_to? attribute
-      end
+    attributes.each do |attribute|
+      assert @public_ip.respond_to? attribute
     end
   end
 
   def test_save_method_response
-    @service.stub :create_public_ip, @response do
+    response = ApiStub::Models::Network::PublicIp.create_public_ip_response(@network_client)
+    @service.stub :create_or_update_public_ip, response do
       assert_instance_of Fog::Network::AzureRM::PublicIp, @public_ip.save
     end
   end
 
+  def test_update_method_response
+    response = ApiStub::Models::Network::PublicIp.create_public_ip_response(@network_client)
+    @service.stub :create_or_update_public_ip, response do
+      assert_instance_of Fog::Network::AzureRM::PublicIp, @public_ip.update({})
+    end
+  end
+
   def test_destroy_method_response
     response = MsRestAzure::AzureOperationResponse.new(MsRest::HttpOperationRequest.new('', '', ''), Faraday::Response.new)
     @service.stub :delete_public_ip, response do