fog-azure-rm 0.0.6 → 0.0.8

data/lib/fog/azurerm/requests/network/create_or_update_express_route_circuit_peering.rb

@@ -19,25 +19,35 @@ module Fog
         private
 
         def get_circuit_peering_object(circuit_peering_params)
-          circuit_peering = Azure::ARM::Network::Models::ExpressRouteCircuitPeering.new
-          circuit_peering.name = circuit_peering_params[:peering_name]
-          circuit_peering.peering_type = circuit_peering_params[:peering_type]
-          circuit_peering.peer_asn = circuit_peering_params[:peer_asn]
-          circuit_peering.primary_peer_address_prefix = circuit_peering_params[:primary_peer_address_prefix]
-          circuit_peering.secondary_peer_address_prefix = circuit_peering_params[:secondary_peer_address_prefix]
-          circuit_peering.vlan_id = circuit_peering_params[:vlan_id]
+          circuit_peering = create_express_route_circuit_peering(circuit_peering_params[:peering_type], circuit_peering_params[:peer_asn], circuit_peering_params[:primary_peer_address_prefix], circuit_peering_params[:secondary_peer_address_prefix], circuit_peering_params[:vlan_id], circuit_peering_params[:name])
 
-          if circuit_peering_params[:peering_type].casecmp(MICROSOFT_PEERING) == 0
-            peering_config = Azure::ARM::Network::Models::ExpressRouteCircuitPeeringConfig.new
-            peering_config.advertised_public_prefixes = circuit_peering_params[:advertised_public_prefixes]
-            peering_config.advertised_public_prefixes_state = circuit_peering_params[:advertised_public_prefix_state]
-            peering_config.customer_asn = circuit_peering_params[:customer_asn]
-            peering_config.routing_registry_name = circuit_peering_params[:routing_registry_name]
+          if circuit_peering_params[:peering_type].casecmp(MICROSOFT_PEERING).zero?
+            peering_config = create_express_route_circuit_peering_config(circuit_peering_params[:advertised_public_prefixes], circuit_peering_params[:advertised_public_prefix_state], circuit_peering_params[:customer_asn], circuit_peering_params[:routing_registry_name])
             circuit_peering.microsoft_peering_config = peering_config
           end
 
           circuit_peering
         end
+
+        def create_express_route_circuit_peering(peering_type, peer_asn, primary_peer_address_prefix, secondary_peer_address_prefix, vlan_id, name)
+          circuit_peering = Azure::ARM::Network::Models::ExpressRouteCircuitPeering.new
+          circuit_peering.peering_type = peering_type
+          circuit_peering.peer_asn = peer_asn
+          circuit_peering.primary_peer_address_prefix = primary_peer_address_prefix
+          circuit_peering.secondary_peer_address_prefix = secondary_peer_address_prefix
+          circuit_peering.vlan_id = vlan_id
+          circuit_peering.name = name
+          circuit_peering
+        end
+
+        def create_express_route_circuit_peering_config(advertised_public_prefixes, advertised_public_prefix_state, customer_asn, routing_registry_name)
+          peering_config = Azure::ARM::Network::Models::ExpressRouteCircuitPeeringConfig.new
+          peering_config.advertised_public_prefixes = advertised_public_prefixes
+          peering_config.advertised_public_prefixes_state = advertised_public_prefix_state
+          peering_config.customer_asn = customer_asn
+          peering_config.routing_registry_name = routing_registry_name
+          peering_config
+        end
       end
 
       # Mock class for Network Request

data/lib/fog/azurerm/requests/network/create_or_update_network_security_rule.rb

@@ -0,0 +1,59 @@
+module Fog
+  module Network
+    class AzureRM
+      # Real class for Network Request
+      class Real
+        def create_or_update_network_security_rule(security_rule)
+          msg = "Creating/Updating Network Security Rule #{security_rule[:name]} in Resource Group #{security_rule[:resource_group]}."
+          Fog::Logger.debug msg
+          security_rule_params = get_security_rule_params(security_rule)
+          begin
+            security_rule_obj = @network_client.security_rules.create_or_update(security_rule[:resource_group], security_rule[:network_security_group_name], security_rule[:name], security_rule_params)
+          rescue MsRestAzure::AzureOperationError => e
+            raise_azure_exception(e, msg)
+          end
+          Fog::Logger.debug "Network Security Rule #{security_rule[:name]} Created/Updated Successfully!"
+          security_rule_obj
+        end
+
+        def get_security_rule_params(security_rule_hash)
+          security_rule = Azure::ARM::Network::Models::SecurityRule.new
+          security_rule.protocol = security_rule_hash[:protocol]
+          security_rule.source_port_range = security_rule_hash[:source_port_range]
+          security_rule.destination_port_range = security_rule_hash[:destination_port_range]
+          security_rule.source_address_prefix = security_rule_hash[:source_address_prefix]
+          security_rule.destination_address_prefix = security_rule_hash[:destination_address_prefix]
+          security_rule.access = security_rule_hash[:access]
+          security_rule.priority = security_rule_hash[:priority]
+          security_rule.direction = security_rule_hash[:direction]
+          security_rule
+        end
+      end
+
+      # Mock class for Network Security Rule Request
+      class Mock
+        def create_or_update_network_security_rule(*)
+          network_security_rule = '{
+            "name":"myNsRule",
+            "id":"/subscriptions/{guid}/resourceGroups/{resourceGroupName}/providers/Microsoft.Network/networkSecurityGroups/myNsg/securityRules/myNsRule",
+            "etag":"W/\"00000000-0000-0000-0000-000000000000\"",
+            "properties":{
+              "provisioningState":"Succeeded",
+              "description":"description-of-this-rule",
+              "protocol": "*",
+              "sourcePortRange":"source-port-range",
+              "destinationPortRange":"destination-port-range",
+              "sourceAddressPrefix":"*",
+              "destinationAddressPrefix":"*",
+              "access":"Allow",
+              "priority":6500,
+              "direction":"Inbound"
+            }
+          }'
+          network_security_rule_mapper = Azure::ARM::Network::Models::SecurityRule.mapper
+          @network_client.deserialize(network_security_rule_mapper, JSON.load(network_security_rule), 'result.body')
+        end
+      end
+    end
+  end
+end

data/lib/fog/azurerm/requests/network/create_or_update_virtual_network_gateway.rb

@@ -22,31 +22,23 @@ module Fog
         def get_network_gateway_object(virtual_network_params)
           network_gateway = Azure::ARM::Network::Models::VirtualNetworkGateway.new
 
-          default_site = MsRestAzure::SubResource.new
-          default_site.id = virtual_network_params[:gateway_default_site]
-
-          sku = Azure::ARM::Network::Models::VirtualNetworkGatewaySku.new
-          sku.name = virtual_network_params[:sku_name]
-          sku.capacity = virtual_network_params[:sku_capacity]
-          sku.tier = virtual_network_params[:sku_tier]
-
           network_gateway.enable_bgp = virtual_network_params[:enable_bgp]
           network_gateway.gateway_type = virtual_network_params[:gateway_type]
           network_gateway.provisioning_state = virtual_network_params[:provisioning_state]
           network_gateway.vpn_type = virtual_network_params[:vpn_type]
-          network_gateway.sku = sku
-          network_gateway.gateway_default_site = default_site
+          network_gateway.sku = get_virtual_network_gateway_sku(virtual_network_params)
+          if virtual_network_params[:gateway_default_site]
+            default_site = MsRestAzure::SubResource.new
+            default_site.id = virtual_network_params[:gateway_default_site]
+            network_gateway.gateway_default_site = default_site
+          end
           if virtual_network_params[:ip_configurations]
             ip_configurations = get_ip_configurations(virtual_network_params[:ip_configurations])
             network_gateway.ip_configurations = ip_configurations
           end
 
           if virtual_network_params[:enable_bgp]
-            bgp_settings = Azure::ARM::Network::Models::BgpSettings.new
-            bgp_settings.asn = virtual_network_params[:asn]
-            bgp_settings.bgp_peering_address = virtual_network_params[:bgp_peering_address]
-            bgp_settings.peer_weight = virtual_network_params[:peer_weight]
-            network_gateway.bgp_settings = bgp_settings
+            network_gateway.bgp_settings = get_bgp_settings(virtual_network_params)
           end
 
           if virtual_network_params[:vpn_client_configuration]
@@ -61,6 +53,22 @@ module Fog
           network_gateway
         end
 
+        def get_bgp_settings(virtual_network_params)
+          bgp_settings = Azure::ARM::Network::Models::BgpSettings.new
+          bgp_settings.asn = virtual_network_params[:asn]
+          bgp_settings.bgp_peering_address = virtual_network_params[:bgp_peering_address]
+          bgp_settings.peer_weight = virtual_network_params[:peer_weight]
+          bgp_settings
+        end
+
+        def get_virtual_network_gateway_sku(virtual_network_params)
+          sku = Azure::ARM::Network::Models::VirtualNetworkGatewaySku.new
+          sku.name = virtual_network_params[:sku_name]
+          sku.capacity = virtual_network_params[:sku_capacity]
+          sku.tier = virtual_network_params[:sku_tier]
+          sku
+        end
+
         def get_vpn_client_config(vpn_client_config)
           client_config = Azure::ARM::Network::Models::VpnClientConfiguration.new
 

data/lib/fog/azurerm/requests/network/delete_network_security_rule.rb

@@ -0,0 +1,30 @@
+module Fog
+  module Network
+    class AzureRM
+      # Real class for Network Request
+      class Real
+        def delete_network_security_rule(resource_group, network_security_group_name, security_rule_name)
+          msg = "Deleting Network Security Rule #{security_rule_name} in Security Group #{network_security_group_name}"
+          Fog::Logger.debug msg
+
+          begin
+            @network_client.security_rules.delete(resource_group, network_security_group_name, security_rule_name)
+          rescue MsRestAzure::AzureOperationError => e
+            raise_azure_exception(e, msg)
+          end
+
+          Fog::Logger.debug "Network Security Rule #{security_rule_name} deleted successfully."
+          true
+        end
+      end
+
+      # Mock class for Network Request
+      class Mock
+        def delete_network_security_rule(*)
+          Fog::Logger.debug 'Network Security Rule test-security-rule from Resource group test-rg deleted successfully.'
+          true
+        end
+      end
+    end
+  end
+end

data/lib/fog/azurerm/requests/network/get_available_ipaddresses_count.rb

@@ -0,0 +1,23 @@
+module Fog
+  module Network
+    class AzureRM
+      # Real class for Network Request
+      class Real
+        def get_available_ipaddresses_count(subnet_name, address_prefix, ip_configurations_ids, express_route_enabled)
+          Fog::Logger.debug "Getting free IP Address count of Subnet #{subnet_name}"
+          total_ipaddresses = (2**(32 - address_prefix.split('/').last.to_i)) - 2
+          total_ipaddresses -= 3 if express_route_enabled
+          used_ip_address = ip_configurations_ids.nil? ? 0 : ip_configurations_ids.count
+          total_ipaddresses - used_ip_address
+        end
+      end
+
+      # Mock class for Network Request
+      class Mock
+        def get_available_ipaddress_count(*)
+          65_531
+        end
+      end
+    end
+  end
+end

data/lib/fog/azurerm/requests/network/get_network_security_rule.rb

@@ -0,0 +1,47 @@
+module Fog
+  module Network
+    class AzureRM
+      # Real class for Network Security rule Request
+      class Real
+        def get_network_security_rule(resource_group_name, network_security_group_name, security_rule_name)
+          msg = "Getting Network Security Rule #{security_rule_name} from Resource Group #{resource_group_name}."
+          Fog::Logger.debug msg
+
+          begin
+            security_rule = @network_client.security_rules.get(resource_group_name, network_security_group_name, security_rule_name)
+          rescue MsRestAzure::AzureOperationError => e
+            raise_azure_exception(e, msg)
+          end
+
+          Fog::Logger.debug "Network Security Rule #{security_rule_name} retrieved successfully."
+          security_rule
+        end
+      end
+
+      # Mock class for Network Security Rule Request
+      class Mock
+        def get_network_security_rule(*)
+          nsr = '{
+              "name":"myNsRule",
+              "id":"/subscriptions/{guid}/resourceGroups/{resourceGroupName}/providers/Microsoft.Network/networkSecurityGroups/myNsg/securityRules/myNsRule",
+              "etag":"W/\"00000000-0000-0000-0000-000000000000\"",
+              "properties":{
+                "provisioningState":"Succeeded",
+                "description":"description-of-this-rule",
+                "protocol": "*",
+                "sourcePortRange":"source-port-range",
+                "destinationPortRange":"destination-port-range",
+                "sourceAddressPrefix":"*",
+                "destinationAddressPrefix":"*",
+                "access":"Allow",
+                "priority":6500,
+                "direction":"Inbound"
+              }
+          }'
+          nsr_mapper = Azure::ARM::Network::Models::SecurityRule.mapper
+          @network_client.deserialize(nsr_mapper, JSON.load(nsr), 'result.body')
+        end
+      end
+    end
+  end
+end

data/lib/fog/azurerm/requests/network/list_network_security_groups.rb

@@ -8,7 +8,7 @@ module Fog
           Fog::Logger.debug msg
 
           begin
-            nsg_list_result = @network_client.network_security_groups.list(resource_group)
+            nsg_list_result = @network_client.network_security_groups.list_as_lazy(resource_group)
           rescue MsRestAzure::AzureOperationError => e
             raise_azure_exception(e, msg)
           end

data/lib/fog/azurerm/requests/network/list_network_security_rules.rb

@@ -0,0 +1,51 @@
+module Fog
+  module Network
+    class AzureRM
+      # Real class for Network Request
+      class Real
+        def list_network_security_rules(resource_group, network_security_group_name)
+          msg = "Getting list of Network Security Rule in Security Group #{network_security_group_name} from Resource Group #{resource_group}."
+          Fog::Logger.debug msg
+
+          begin
+            nsr_list_result = @network_client.security_rules.list_as_lazy(resource_group, network_security_group_name)
+          rescue MsRestAzure::AzureOperationError => e
+            raise_azure_exception(e, msg)
+          end
+
+          Fog::Logger.debug "Network Security Rules list retrieved successfully from Resource Group #{resource_group}."
+          nsr_list_result.value
+        end
+      end
+
+      # Mock class for Network Security Rule Request
+      class Mock
+        def list_network_security_rules(*)
+          nsr_list = '{
+            "value":[
+              {
+                "name":"myNsRule",
+                "id":"/subscriptions/{guid}/resourceGroups/{resourceGroupName}/providers/Microsoft.Network/networkSecurityGroups/myNsg/securityRules/myNsRule",
+                "etag":"W/\"00000000-0000-0000-0000-000000000000\"",
+                "properties":{
+                  "provisioningState":"Succeeded",
+                  "description":"description-of-this-rule",
+                  "protocol": "*",
+                  "sourcePortRange":"source-port-range",
+                  "destinationPortRange":"destination-port-range",
+                  "sourceAddressPrefix":"*",
+                  "destinationAddressPrefix":"*",
+                  "access":"Allow",
+                  "priority":100,
+                  "direction":"Inbound"
+                }
+              }
+          ]
+          }'
+          nsr_mapper = Azure::ARM::Network::Models::SecurityRuleListResult.mapper
+          @network_client.deserialize(nsr_mapper, JSON.load(nsr_list), 'result.body')
+        end
+      end
+    end
+  end
+end

data/lib/fog/azurerm/requests/storage/acquire_blob_lease.rb

@@ -0,0 +1,29 @@
+module Fog
+  module Storage
+    class AzureRM
+      # This class provides the actual implementation for service calls.
+      class Real
+        def acquire_blob_lease(container_name, name, options={})
+          msg = "Leasing blob: #{name} of container #{container_name}"
+          Fog::Logger.debug msg
+          begin
+            lease_id = @blob_client.acquire_blob_lease(container_name, name, options)
+          rescue Azure::Core::Http::HTTPError => ex
+            raise_azure_exception(ex, msg)
+          end
+          Fog::Logger.debug "Blob #{name} leased successfully."
+          lease_id
+        end
+      end
+
+      # This class provides the mock implementation for unit tests.
+      class Mock
+        def acquire_blob_lease(*)
+          {
+            'leaseId' => 'abc123'
+          }
+        end
+      end
+    end
+  end
+end

data/lib/fog/azurerm/requests/storage/acquire_container_lease.rb

@@ -0,0 +1,29 @@
+module Fog
+  module Storage
+    class AzureRM
+      # This class provides the actual implementation for service calls.
+      class Real
+        def acquire_container_lease(name, options={})
+          msg = "Leasing container: #{name}"
+          Fog::Logger.debug msg
+          begin
+            lease_id = @blob_client.acquire_container_lease(name, options)
+          rescue Azure::Core::Http::HTTPError => ex
+            raise_azure_exception(ex, msg)
+          end
+          Fog::Logger.debug "Container #{name} leased successfully."
+          lease_id
+        end
+      end
+
+      # This class provides the mock implementation for unit tests.
+      class Mock
+        def acquire_container_lease(*)
+          {
+            'leaseId' => 'abc123'
+          }
+        end
+      end
+    end
+  end
+end

data/lib/fog/azurerm/requests/storage/check_blob_exist.rb

@@ -0,0 +1,25 @@
+module Fog
+  module Storage
+    class AzureRM
+      # This class provides the actual implemention for service calls.
+      class Real
+        def check_blob_exist(container_name, name, options = {})
+          Fog::Logger.debug "Check Blob #{name} exist in container #{container_name}."
+          begin
+            @blob_client.get_blob_properties(container_name, name, options)
+          rescue Azure::Core::Http::HTTPError
+            return false
+          end
+          Fog::Logger.debug "Blob #{name} Exists."
+          true
+        end
+      end
+      # This class provides the mock implementation for unit tests.
+      class Mock
+        def check_blob_exist(*)
+          true
+        end
+      end
+    end
+  end
+end

data/lib/fog/azurerm/requests/storage/compare_blob.rb

@@ -0,0 +1,73 @@
+module Fog
+  module Storage
+    class AzureRM
+      # This class provides the actual implementation for service calls.
+      class Real
+        def compare_blob(container1, container2)
+          msg = "Comparing blobs from container #{container1} to container #{container2}"
+          Fog::Logger.debug msg
+          begin
+            identical_blobs = get_identical_blobs_from_containers(container1, container2)
+          rescue Azure::Core::Http::HTTPError => ex
+            raise_azure_exception(ex, msg)
+          end
+          identical_blobs
+        end
+
+        private
+
+        def get_identical_blobs_from_containers(container1, container2)
+          container1_blobs = list_blobs(container1)
+          container2_blobs = list_blobs(container2)
+
+          identical_blobs = []
+          container1_blobs.each do |container1_blob|
+            container2_blobs.each do |container2_blob|
+              container1_prop = get_blob_properties(container1, container1_blob.name)
+              container2_prop = get_blob_properties(container2, container2_blob.name)
+              if container1_blob.name == container2_blob.name && container1_prop.properties[:content_md5] == container2_prop.properties[:content_md5]
+                identical_blobs.push(container1_blob)
+              end
+            end
+          end
+          identical_blobs
+        end
+      end
+
+      # This class provides the mock implementation for unit tests.
+      class Mock
+        def compare_blob(*)
+          [
+            {
+              'name' => 'blob_name',
+              'metadata' => {},
+              'properties' =>
+                {
+                  'last_modified' => 'Thu, 28 Jul 2016 06:53:05 GMT',
+                  'etag' => '0x8D3B6B3D353FFCA',
+                  'lease_status' => 'unlocked',
+                  'lease_state' => 'available',
+                  'lease_duration' => nil,
+                  'content_length' => 4_194_304,
+                  'content_type' => 'application/atom+xml; charset=utf-8',
+                  'content_encoding' => 'ASCII-8BIT',
+                  'content_language' => nil,
+                  'content_disposition' => nil,
+                  'content_md5' => 'tXAohIyxuu/t94Lp/ujeRw==',
+                  'cache_control' => nil,
+                  'blob_type' => 'BlockBlob',
+                  'copy_id' => nil,
+                  'copy_status' => nil,
+                  'copy_source' => nil,
+                  'copy_progress' => nil,
+                  'copy_completion_time' => nil,
+                  'copy_status_description' => nil,
+                  'accept_ranges' => 0
+                }
+            }
+          ]
+        end
+      end
+    end
+  end
+end