fog-azure-rm 0.0.4 → 0.0.5

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

@@ -1,23 +1,24 @@
 module Fog
   module Storage
     class AzureRM
-      # This class provides the actual implemention for service calls.
+      # This class provides the actual implementation for service calls.
       class Real
         def set_container_metadata(name, metadata)
-          Fog::Logger.debug "Set Container #{name} metadata."
+          msg = "Setting Container #{name} metadata."
+          Fog::Logger.debug msg
           begin
             @blob_client.set_container_metadata(name, metadata)
-            Fog::Logger.debug "Setting metadata of container #{name} successfully."
-            true
           rescue Azure::Core::Http::HTTPError => ex
-            raise "Exception in setting metadata of Container #{name}: #{ex.inspect}"
+            raise_azure_exception(ex, msg)
           end
+          Fog::Logger.debug "Setting metadata of container #{name} successfully."
+          true
         end
       end
       # This class provides the mock implementation for unit tests.
       class Mock
-        def set_container_metadata(name, metadata)
-          Fog::Logger.debug "Set Container #{name} metadata #{metadata} successfully."
+        def set_container_metadata(*)
+          Fog::Logger.debug "Set Container testcontainer1 metadata successfully."
           true
         end
       end

data/lib/fog/azurerm/requests/traffic_manager/create_traffic_manager_endpoint.rb

@@ -0,0 +1,54 @@
+module Fog
+  module TrafficManager
+    class AzureRM
+      # Real class for Traffic Manager Request
+      class Real
+        def create_traffic_manager_endpoint(endpoint_hash)
+          msg = "Creating Traffic Manager Endpoint: #{endpoint_hash[:name]}."
+          Fog::Logger.debug msg
+          endpoint_parameters = get_endpoint_object(endpoint_hash[:target_resource_id], endpoint_hash[:target], endpoint_hash[:weight], endpoint_hash[:priority], endpoint_hash[:endpoint_location], endpoint_hash[:min_child_endpoints])
+          begin
+            traffic_manager_endpoint = @traffic_mgmt_client.endpoints.create_or_update(endpoint_hash[:resource_group], endpoint_hash[:traffic_manager_profile_name],
+                                                                                       endpoint_hash[:type], endpoint_hash[:name], endpoint_parameters)
+          rescue MsRestAzure::AzureOperationError => e
+            raise_azure_exception(e, msg)
+          end
+          Fog::Logger.debug "Traffic Manager Endpoint: #{endpoint_hash[:name]} created successfully."
+          traffic_manager_endpoint
+        end
+
+        private
+
+        def get_endpoint_object(target_resource_id, target, weight, priority, endpoint_location, min_child_endpoints)
+          endpoint = Azure::ARM::TrafficManager::Models::Endpoint.new
+          endpoint.min_child_endpoints = min_child_endpoints
+          endpoint.target_resource_id = target_resource_id
+          endpoint.endpoint_location = endpoint_location
+          endpoint.priority = priority
+          endpoint.target = target
+          endpoint.weight = weight
+          endpoint
+        end
+      end
+
+      # Mock class for TrafficManager Request
+      class Mock
+        def create_traffic_manager_endpoint(*)
+          endpoint = {
+            'name' => '{endpoint-name}',
+            'type' => 'Microsoft.Network/trafficManagerProfiles/externalEndpoints',
+            'properties' => {
+              'target' => 'myendpoint.contoso.com',
+              'endpointStatus' => 'Enabled',
+              'weight' => 10,
+              'priority' => 5,
+              'endpointLocation' => 'northeurope'
+            }
+          }
+          endpoint_mapper = Azure::ARM::TrafficManager::Models::Endpoint.mapper
+          @traffic_mgmt_client.deserialize(endpoint_mapper, endpoint, 'result.body')
+        end
+      end
+    end
+  end
+end

data/lib/fog/azurerm/requests/traffic_manager/create_traffic_manager_profile.rb

@@ -0,0 +1,107 @@
+module Fog
+  module TrafficManager
+    class AzureRM
+      # This class provides the actual implementation for service call.
+      class Real
+        def create_traffic_manager_profile(resource_group, name, traffic_routing_method, relative_name, ttl, protocol, port, path)
+          msg = "Creating Traffic Manager Profile: #{name}."
+          Fog::Logger.debug msg
+          profile_parameters = get_profile_object(traffic_routing_method, relative_name, ttl, protocol, port, path)
+          begin
+            traffic_manager_profile = @traffic_mgmt_client.profiles.create_or_update(resource_group, name, profile_parameters)
+          rescue MsRestAzure::AzureOperationError => e
+            raise_azure_exception(e, msg)
+          end
+          Fog::Logger.debug "Traffic Manager Profile: #{name} created successfully."
+          traffic_manager_profile
+        end
+
+        private
+
+        def get_profile_object(traffic_routing_method, relative_name, ttl, protocol, port, path)
+          traffic_manager_profile = Azure::ARM::TrafficManager::Models::Profile.new
+          traffic_manager_profile.traffic_routing_method = traffic_routing_method
+          traffic_manager_profile.location = GLOBAL
+
+          traffic_manager_profile.dns_config = get_traffic_manager_dns_config(relative_name, ttl)
+          traffic_manager_profile.monitor_config = get_traffic_manager_monitor_config(protocol, port, path)
+          traffic_manager_profile
+        end
+
+        def get_traffic_manager_dns_config(relative_name, ttl)
+          traffic_manager_dns_config = Azure::ARM::TrafficManager::Models::DnsConfig.new
+          traffic_manager_dns_config.relative_name = relative_name
+          traffic_manager_dns_config.ttl = ttl
+          traffic_manager_dns_config
+        end
+
+        def get_traffic_manager_monitor_config(protocol, port, path)
+          traffic_manager_monitor_config = Azure::ARM::TrafficManager::Models::MonitorConfig.new
+          traffic_manager_monitor_config.path = path
+          traffic_manager_monitor_config.protocol = protocol
+          traffic_manager_monitor_config.port = port
+          traffic_manager_monitor_config
+        end
+      end
+      # This class provides the mock implementation for unit tests.
+      class Mock
+        def create_traffic_manager_profile(*)
+          profile = {
+            'location' => 'global',
+            'tags' => {},
+            'properties' => {
+              'profileStatus' => 'Enabled',
+              'trafficRoutingMethod' => 'Performance',
+              'dnsConfig' => {
+                'relativeName' => 'myapp',
+                'ttl' => 30
+              },
+              'monitorConfig' => {
+                'protocol' => 'http',
+                'port' => 80,
+                'path' => '/monitorpage.aspx'
+              },
+              'endpoints' => [
+                {
+                  'name' => '{endpoint-name}',
+                  'type' => 'Microsoft.Network/trafficManagerProfiles/azureEndpoints',
+                  'properties' => {
+                    'targetResourceId' => '{resource ID of target resource in Azure}',
+                    'endpointStatus' => 'Enabled',
+                    'weight' => 10,
+                    'priority' => 3
+                  }
+                },
+                {
+                  'name' => '{endpoint-name}',
+                  'type' => 'Microsoft.Network/trafficManagerProfiles/externalEndpoints',
+                  'properties' => {
+                    'target' => 'myendpoint.contoso.com',
+                    'endpointStatus' => 'Enabled',
+                    'weight' => 10,
+                    'priority' => 5,
+                    'endpointLocation' => 'northeurope'
+                  }
+                },
+                {
+                  'name' => '{endpoint-name}',
+                  'type' => 'Microsoft.Network/trafficManagerProfiles/nestedEndpoints',
+                  'properties' => {
+                    'targetResourceId' => '{resource ID of child Traffic Manager profile}',
+                    'endpointStatus' => 'Enabled',
+                    'weight' => 10,
+                    'priority' => 1,
+                    'endpointLocation' => 'westeurope',
+                    'minChildEndpoints' => 1
+                  }
+                }
+              ]
+            }
+          }
+          profile_mapper = Azure::ARM::TrafficManager::Models::Profile.mapper
+          @traffic_mgmt_client.deserialize(profile_mapper, profile, 'result.body')
+        end
+      end
+    end
+  end
+end

data/lib/fog/azurerm/requests/traffic_manager/delete_traffic_manager_endpoint.rb

@@ -0,0 +1,28 @@
+module Fog
+  module TrafficManager
+    class AzureRM
+      # Real class for TrafficManager Request
+      class Real
+        def delete_traffic_manager_endpoint(resource_group, name, traffic_manager_profile_name, type)
+          msg = "Deleting Traffic Manager Endpoint: #{name}."
+          Fog::Logger.debug msg
+          begin
+            @traffic_mgmt_client.endpoints.delete(resource_group, traffic_manager_profile_name, type, name)
+          rescue MsRestAzure::AzureOperationError => e
+            raise_azure_exception(e, msg)
+          end
+          Fog::Logger.debug "Traffic Manager Endpoint: #{name} deleted successfully."
+          true
+        end
+      end
+
+      # Mock class for TrafficManager Request
+      class Mock
+        def delete_traffic_manager_endpoint(resource_group, name, _traffic_manager_profile_name, _type)
+          Fog::Logger.debug "Traffic Manager End Point #{name} from Resource group #{resource_group} deleted successfully."
+          true
+        end
+      end
+    end
+  end
+end

data/lib/fog/azurerm/requests/traffic_manager/delete_traffic_manager_profile.rb

@@ -0,0 +1,26 @@
+module Fog
+  module TrafficManager
+    class AzureRM
+      # Real class for TrafficManager Request
+      class Real
+        def delete_traffic_manager_profile(resource_group, name)
+          msg = "Deleting Traffic Manager Profile: #{name}."
+          Fog::Logger.debug msg
+          begin
+            @traffic_mgmt_client.profiles.delete(resource_group, name)
+            true
+          rescue MsRestAzure::AzureOperationError => e
+            raise_azure_exception(e, msg)
+          end
+        end
+      end
+      # Mock class for TrafficManager Request
+      class Mock
+        def delete_subnet(resource_group, name)
+          Fog::Logger.debug "Traffic Manager Profile #{name} from Resource group #{resource_group} deleted successfully."
+          true
+        end
+      end
+    end
+  end
+end

data/lib/fog/azurerm/requests/traffic_manager/get_traffic_manager_endpoint.rb

@@ -0,0 +1,39 @@
+module Fog
+  module TrafficManager
+    class AzureRM
+      # Real class for TrafficManager Request
+      class Real
+        def get_traffic_manager_end_point(resource_group, traffic_manager_profile_name, traffic_manager_end_point, type)
+          msg = "Getting Traffic Manager Endpoint: #{traffic_manager_end_point} in Profile: #{traffic_manager_profile_name}."
+          Fog::Logger.debug msg
+          begin
+            endpoint = @traffic_mgmt_client.endpoints.get(resource_group, traffic_manager_profile_name, type, traffic_manager_end_point)
+          rescue MsRestAzure::AzureOperationError => e
+            raise_azure_exception(e, msg)
+          end
+          Fog::Logger.debug "Traffic Manager Endpoint fetched successfully in Resource Group: #{resource_group}"
+          endpoint
+        end
+      end
+
+      # Mock class for Network Request
+      class Mock
+        def get_traffic_manager_end_point(*)
+          endpoint = {
+            'name' => '{endpoint-name}',
+            'type' => 'Microsoft.Network/trafficManagerProfiles/externalEndpoints',
+            'properties' => {
+              'target' => 'myendpoint.contoso.com',
+              'endpointStatus' => 'Enabled',
+              'weight' => 10,
+              'priority' => 5,
+              'endpointLocation' => 'northeurope'
+            }
+          }
+          endpoint_mapper = Azure::ARM::TrafficManager::Models::Endpoint.mapper
+          @traffic_mgmt_client.deserialize(endpoint_mapper, endpoint, 'result.body')
+        end
+      end
+    end
+  end
+end

data/lib/fog/azurerm/requests/traffic_manager/get_traffic_manager_profile.rb

@@ -0,0 +1,80 @@
+module Fog
+  module TrafficManager
+    class AzureRM
+      # Real class for TrafficManager Request
+      class Real
+        def get_traffic_manager_profile(resource_group, traffic_manager_profile_name)
+          msg = "Getting Traffic Manager Profile: #{traffic_manager_profile_name} in Resource Group: #{resource_group}..."
+          Fog::Logger.debug msg
+          begin
+            profile = @traffic_mgmt_client.profiles.get(resource_group, traffic_manager_profile_name)
+          rescue MsRestAzure::AzureOperationError => e
+            raise_azure_exception(e, msg)
+          end
+          Fog::Logger.debug "Traffic Manager Profile fetched successfully in Resource Group: #{resource_group}"
+          profile
+        end
+      end
+
+      # Mock class for Network Request
+      class Mock
+        def get_traffic_manager_profile(*)
+          profile = {
+            'location' => 'global',
+            'tags' => {},
+            'properties' => {
+              'profileStatus' => 'Enabled',
+              'trafficRoutingMethod' => 'Performance',
+              'dnsConfig' => {
+                'relativeName' => 'myapp',
+                'ttl' => 30
+              },
+              'monitorConfig' => {
+                'protocol' => 'http',
+                'port' => 80,
+                'path' => '/monitorpage.aspx'
+              },
+              'endpoints' => [
+                {
+                  'name' => '{endpoint-name}',
+                  'type' => 'Microsoft.Network/trafficManagerProfiles/azureEndpoints',
+                  'properties' => {
+                    'targetResourceId' => '{resource ID of target resource in Azure}',
+                    'endpointStatus' => 'Enabled',
+                    'weight' => 10,
+                    'priority' => 3
+                  }
+                },
+                {
+                  'name' => '{endpoint-name}',
+                  'type' => 'Microsoft.Network/trafficManagerProfiles/externalEndpoints',
+                  'properties' => {
+                    'target' => 'myendpoint.contoso.com',
+                    'endpointStatus' => 'Enabled',
+                    'weight' => 10,
+                    'priority' => 5,
+                    'endpointLocation' => 'northeurope'
+                  }
+                },
+                {
+                  'name' => '{endpoint-name}',
+                  'type' => 'Microsoft.Network/trafficManagerProfiles/nestedEndpoints',
+                  'properties' => {
+                    'targetResourceId' => '{resource ID of child Traffic Manager profile}',
+                    'endpointStatus' => 'Enabled',
+                    'weight' => 10,
+                    'priority' => 1,
+                    'endpointLocation' => 'westeurope',
+                    'minChildEndpoints' => 1
+                  }
+                }
+              ]
+            }
+          }
+          profile_mapper = Azure::ARM::TrafficManager::Models::Profile.mapper
+          @traffic_mgmt_client.deserialize(profile_mapper, profile, 'result.body')
+        end
+      end
+    end
+  end
+end

data/lib/fog/azurerm/requests/traffic_manager/list_traffic_manager_profiles.rb

@@ -0,0 +1,82 @@
+module Fog
+  module TrafficManager
+    class AzureRM
+      # Real class for TrafficManager Request
+      class Real
+        def list_traffic_manager_profiles(resource_group)
+          msg = "Listing Traffic Manager Profiles in Resource Group: #{resource_group}."
+          Fog::Logger.debug msg
+          begin
+            profiles = @traffic_mgmt_client.profiles.list_all_in_resource_group(resource_group)
+          rescue MsRestAzure::AzureOperationError => e
+            raise_azure_exception(e, msg)
+          end
+          Fog::Logger.debug "Traffic Manager Profiles listed successfully in Resource Group: #{resource_group}"
+          profiles.value
+        end
+      end
+
+      # Mock class for TrafficManager Request
+      class Mock
+        def list_traffic_manager_profiles(*)
+          profiles = [
+            {
+              'location' => 'global',
+              'tags' => {},
+              'properties' => {
+                'profileStatus' => 'Enabled',
+                'trafficRoutingMethod' => 'Performance',
+                'dnsConfig' => {
+                  'relativeName' => 'myapp',
+                  'ttl' => 30
+                },
+                'monitorConfig' => {
+                  'protocol' => 'http',
+                  'port' => 80,
+                  'path' => '/monitorpage.aspx'
+                },
+                'endpoints' => [
+                  {
+                    'name' => '{endpoint-name}',
+                    'type' => 'Microsoft.Network/trafficManagerProfiles/azureEndpoints',
+                    'properties' => {
+                      'targetResourceId' => '{resource ID of target resource in Azure}',
+                      'endpointStatus' => 'Enabled',
+                      'weight' => 10,
+                      'priority' => 3
+                    }
+                  },
+                  {
+                    'name' => '{endpoint-name}',
+                    'type' => 'Microsoft.Network/trafficManagerProfiles/externalEndpoints',
+                    'properties' => {
+                      'target' => 'myendpoint.contoso.com',
+                      'endpointStatus' => 'Enabled',
+                      'weight' => 10,
+                      'priority' => 5,
+                      'endpointLocation' => 'northeurope'
+                    }
+                  },
+                  {
+                    'name' => '{endpoint-name}',
+                    'type' => 'Microsoft.Network/trafficManagerProfiles/nestedEndpoints',
+                    'properties' => {
+                      'targetResourceId' => '{resource ID of child Traffic Manager profile}',
+                      'endpointStatus' => 'Enabled',
+                      'weight' => 10,
+                      'priority' => 1,
+                      'endpointLocation' => 'westeurope',
+                      'minChildEndpoints' => 1
+                    }
+                  }
+                ]
+              }
+            }
+          ]
+          profiles_mapper = Azure::ARM::TrafficManager::Models::Profile.mapper
+          @traffic_mgmt_client.deserialize(profiles_mapper, profiles, 'result.body')
+        end
+      end
+    end
+  end
+end