fog-azure-rm 0.1.2 → 0.2.0

data/lib/fog/azurerm/requests/sql/delete_database.rb

@@ -6,16 +6,9 @@ module Fog
         def delete_database(resource_group, server_name, name)
           msg = "Deleting SQL Database: #{name}."
           Fog::Logger.debug msg
-          resource_url = "#{resource_manager_endpoint_url}/subscriptions/#{@subscription_id}/resourceGroups/#{resource_group}/providers/Microsoft.Sql/servers/#{server_name}/databases/#{name}?api-version=2014-04-01-preview"
           begin
-            token = Fog::Credentials::AzureRM.get_token(@tenant_id, @client_id, @client_secret)
-            RestClient.delete(
-              resource_url,
-              accept: :json,
-              content_type: :json,
-              authorization: token
-            )
-          rescue RestClient::Exception => e
+            @sql_mgmt_client.databases.delete(resource_group, server_name, name)
+          rescue MsRestAzure::AzureOperationError => e
             raise_azure_exception(e, msg)
           end
           Fog::Logger.debug "SQL Database: #{name} deleted successfully."

data/lib/fog/azurerm/requests/sql/delete_firewall_rule.rb

@@ -7,16 +7,9 @@ module Fog
           msg = "Deleting SQL Firewall Rule: #{rule_name}."
           Fog::Logger.debug msg
 
-          resource_url = "#{resource_manager_endpoint_url}/subscriptions/#{@subscription_id}/resourceGroups/#{resource_group}/providers/Microsoft.Sql/servers/#{server_name}/firewallRules/#{rule_name}?api-version=2014-04-01-preview"
           begin
-            token = Fog::Credentials::AzureRM.get_token(@tenant_id, @client_id, @client_secret)
-            RestClient.delete(
-              resource_url,
-              accept: :json,
-              content_type: :json,
-              authorization: token
-            )
-          rescue RestClient::Exception => e
+            @sql_mgmt_client.servers.delete_firewall_rule(resource_group, server_name, rule_name)
+          rescue MsRestAzure::AzureOperationError => e
             raise_azure_exception(e, msg)
           end
           Fog::Logger.debug "SQL Firewall Rule: #{rule_name} deleted successfully."

data/lib/fog/azurerm/requests/sql/delete_sql_server.rb

@@ -6,16 +6,10 @@ module Fog
         def delete_sql_server(resource_group, name)
           msg = "Deleting SQL Server: #{name}."
           Fog::Logger.debug msg
-          resource_url = "#{resource_manager_endpoint_url}/subscriptions/#{@subscription_id}/resourceGroups/#{resource_group}/providers/Microsoft.Sql/servers/#{name}?api-version=2014-04-01-preview"
+
           begin
-            token = Fog::Credentials::AzureRM.get_token(@tenant_id, @client_id, @client_secret)
-            RestClient.delete(
-              resource_url,
-              accept: :json,
-              content_type: :json,
-              authorization: token
-            )
-          rescue RestClient::Exception => e
+            @sql_mgmt_client.servers.delete(resource_group, name)
+          rescue MsRestAzure::AzureOperationError => e
             raise_azure_exception(e, msg)
           end
           Fog::Logger.debug "SQL Server: #{name} deleted successfully."

data/lib/fog/azurerm/requests/sql/get_database.rb

@@ -6,20 +6,14 @@ module Fog
         def get_database(resource_group, server_name, name)
           msg = "Getting Sql Database: #{name} in Resource Group: #{resource_group}."
           Fog::Logger.debug msg
-          resource_url = "#{resource_manager_endpoint_url}/subscriptions/#{@subscription_id}/resourceGroups/#{resource_group}/providers/Microsoft.Sql/servers/#{server_name}/databases/#{name}?api-version=2014-04-01-preview"
+
           begin
-            token = Fog::Credentials::AzureRM.get_token(@tenant_id, @client_id, @client_secret)
-            response = RestClient.get(
-              resource_url,
-              accept: :json,
-              content_type: :json,
-              authorization: token
-            )
-          rescue RestClient::Exception => e
+            database = @sql_mgmt_client.databases.get(resource_group, server_name, name)
+          rescue MsRestAzure::AzureOperationError => e
             raise_azure_exception(e, msg)
           end
           Fog::Logger.debug "Sql Database fetched successfully in Resource Group: #{resource_group}"
-          Fog::JSON.decode(response)
+          database
         end
       end
 

data/lib/fog/azurerm/requests/sql/get_firewall_rule.rb

@@ -6,20 +6,14 @@ module Fog
         def get_firewall_rule(resource_group, server_name, rule_name)
           msg = "Getting Sql Server Firewall Rule: #{rule_name} from SQL Server: #{server_name} in Resource Group: #{resource_group}..."
           Fog::Logger.debug msg
-          resource_url = "#{resource_manager_endpoint_url}/subscriptions/#{@subscription_id}/resourceGroups/#{resource_group}/providers/Microsoft.Sql/servers/#{server_name}/firewallRules/#{rule_name}?api-version=2014-04-01-preview"
+
           begin
-            token = Fog::Credentials::AzureRM.get_token(@tenant_id, @client_id, @client_secret)
-            response = RestClient.get(
-              resource_url,
-              accept: :json,
-              content_type: :json,
-              authorization: token
-            )
-          rescue RestClient::Exception => e
+            firewall_rule = @sql_mgmt_client.servers.get_firewall_rule(resource_group, server_name, rule_name)
+          rescue MsRestAzure::AzureOperationError => e
             raise_azure_exception(e, msg)
           end
           Fog::Logger.debug "Sql Server Firewall Rule fetched successfully from Server: #{server_name}, Resource Group: #{resource_group}"
-          Fog::JSON.decode(response)
+          firewall_rule
         end
       end
 

data/lib/fog/azurerm/requests/sql/get_sql_server.rb

@@ -6,20 +6,13 @@ module Fog
         def get_sql_server(resource_group, server_name)
           msg = "Getting Sql Server: #{server_name} in Resource Group: #{resource_group}..."
           Fog::Logger.debug msg
-          resource_url = "#{resource_manager_endpoint_url}/subscriptions/#{@subscription_id}/resourceGroups/#{resource_group}/providers/Microsoft.Sql/servers/#{server_name}?api-version=2014-04-01-preview"
           begin
-            token = Fog::Credentials::AzureRM.get_token(@tenant_id, @client_id, @client_secret)
-            response = RestClient.get(
-              resource_url,
-              accept: :json,
-              content_type: :json,
-              authorization: token
-            )
-          rescue RestClient::Exception => e
+            sql_server = @sql_mgmt_client.servers.get_by_resource_group(resource_group, server_name)
+          rescue MsRestAzure::AzureOperationError => e
             raise_azure_exception(e, msg)
           end
           Fog::Logger.debug "Sql Server fetched successfully in Resource Group: #{resource_group}"
-          Fog::JSON.decode(response)
+          sql_server
         end
       end
 

data/lib/fog/azurerm/requests/sql/list_databases.rb

@@ -6,20 +6,13 @@ module Fog
         def list_databases(resource_group, server_name)
           msg = "Listing Sql Databases in Resource Group: #{resource_group}."
           Fog::Logger.debug msg
-          resource_url = "#{resource_manager_endpoint_url}/subscriptions/#{@subscription_id}/resourceGroups/#{resource_group}/providers/Microsoft.Sql/servers/#{server_name}/databases?api-version=2014-04-01-preview"
           begin
-            token = Fog::Credentials::AzureRM.get_token(@tenant_id, @client_id, @client_secret)
-            response = RestClient.get(
-              resource_url,
-              accept: :json,
-              content_type: :json,
-              authorization: token
-            )
-          rescue RestClient::Exception => e
+            databases = @sql_mgmt_client.databases.list_by_server(resource_group, server_name)
+          rescue MsRestAzure::AzureOperationError => e
             raise_azure_exception(e, msg)
           end
           Fog::Logger.debug "Sql Databases listed successfully in Resource Group: #{resource_group}"
-          Fog::JSON.decode(response)['value']
+          databases
         end
       end
 

data/lib/fog/azurerm/requests/sql/list_firewall_rules.rb

@@ -6,20 +6,14 @@ module Fog
         def list_firewall_rules(resource_group, server_name)
           msg = "Listing Sql Server Firewall rules on server: #{server_name} in Resource Group: #{resource_group}."
           Fog::Logger.debug msg
-          resource_url = "#{resource_manager_endpoint_url}/subscriptions/#{@subscription_id}/resourceGroups/#{resource_group}/providers/Microsoft.Sql/servers/#{server_name}/firewallRules/?api-version=2014-04-01-preview"
+
           begin
-            token = Fog::Credentials::AzureRM.get_token(@tenant_id, @client_id, @client_secret)
-            response = RestClient.get(
-              resource_url,
-              accept: :json,
-              content_type: :json,
-              authorization: token
-            )
-          rescue RestClient::Exception => e
+            firewall_rules = @sql_mgmt_client.servers.list_firewall_rules(resource_group, server_name)
+          rescue MsRestAzure::AzureOperationError => e
             raise_azure_exception(e, msg)
           end
           Fog::Logger.debug "Sql Server Firewall Rules listed successfully on server: #{server_name} in Resource Group: #{resource_group}"
-          Fog::JSON.decode(response)['value']
+          firewall_rules
         end
       end
 

data/lib/fog/azurerm/requests/sql/list_sql_servers.rb

@@ -6,20 +6,13 @@ module Fog
         def list_sql_servers(resource_group)
           msg = "Listing Sql Servers in Resource Group: #{resource_group}."
           Fog::Logger.debug msg
-          resource_url = "#{resource_manager_endpoint_url}/subscriptions/#{@subscription_id}/resourceGroups/#{resource_group}/providers/Microsoft.Sql/servers?api-version=2014-04-01-preview"
           begin
-            token = Fog::Credentials::AzureRM.get_token(@tenant_id, @client_id, @client_secret)
-            response = RestClient.get(
-              resource_url,
-              accept: :json,
-              content_type: :json,
-              authorization: token
-            )
-          rescue RestClient::Exception => e
+            servers = @sql_mgmt_client.servers.list_by_resource_group(resource_group)
+          rescue MsRestAzure::AzureOperationError => e
             raise_azure_exception(e, msg)
           end
           Fog::Logger.debug "Sql Servers listed successfully in Resource Group: #{resource_group}"
-          Fog::JSON.decode(response)['value']
+          servers
         end
       end
 

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

@@ -0,0 +1,30 @@
+module Fog
+  module Storage
+    class AzureRM
+      # This class provides the actual implementation for service calls.
+      class Real
+        def check_container_exists(name)
+          msg = "Checking container #{name}."
+          Fog::Logger.debug msg
+          begin
+            get_container_properties(name)
+            Fog::Logger.debug "Container #{name} exists."
+            true
+          rescue Exception => e
+            if e.message.include? 'NotFound'
+              Fog::Logger.debug "The specified container #{name} does not exist."
+              false
+            end
+          end
+        end
+      end
+
+      # This class provides the mock implementation for unit tests.
+      class Mock
+        def check_container_exists(*)
+          true
+        end
+      end
+    end
+  end
+end

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

@@ -0,0 +1,31 @@
+module Fog
+  module Storage
+    class AzureRM
+      # This class provides the actual implementation for service calls.
+      class Real
+        def check_storage_account_exists(resource_group_name, storage_account_name)
+          msg = "Checking Storage Account: #{storage_account_name}."
+          Fog::Logger.debug msg
+          begin
+            @storage_mgmt_client.storage_accounts.get_properties(resource_group_name, storage_account_name)
+            Fog::Logger.debug "Storage Account #{storage_account_name} exists."
+            true
+          rescue MsRestAzure::AzureOperationError => e
+            if e.body['error']['code'] == 'ResourceNotFound'
+              Fog::Logger.debug "Storage Account #{storage_account_name} doesn't exist."
+              false
+            else
+              raise_azure_exception(e, msg)
+            end
+          end
+        end
+      end
+      # This class provides the mock implementation.
+      class Mock
+        def check_storage_account_exists(*)
+          true
+        end
+      end
+    end
+  end
+end

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

@@ -7,7 +7,7 @@ module Fog
           msg = "Creating/Updating Recovery Vault #{name} in Resource Group #{resource_group}"
           Fog::Logger.debug msg
 
-          resource_url = "#{AZURE_RESOURCE}/subscriptions/#{@subscription_id}/resourceGroups/#{resource_group}/providers/Microsoft.RecoveryServices/vaults/#{name}?api-version=2016-05-01"
+          resource_url = "#{AZURE_RESOURCE}/subscriptions/#{@subscription_id}/resourceGroups/#{resource_group}/providers/Microsoft.RecoveryServices/vaults/#{name}?api-version=#{REST_CLIENT_API_VERSION[1]}"
           body = {
             location: location,
             tags: {},

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

@@ -6,7 +6,7 @@ module Fog
         def delete_recovery_vault(resource_group, name)
           msg = "Deleting Recovery Vault #{name} from Resource Group #{resource_group}"
           Fog::Logger.debug msg
-          resource_url = "#{AZURE_RESOURCE}/subscriptions/#{@subscription_id}/resourceGroups/#{resource_group}/providers/Microsoft.RecoveryServices/vaults/#{name}?api-version=2016-05-01"
+          resource_url = "#{AZURE_RESOURCE}/subscriptions/#{@subscription_id}/resourceGroups/#{resource_group}/providers/Microsoft.RecoveryServices/vaults/#{name}?api-version=#{REST_CLIENT_API_VERSION[1]}"
           begin
             token = Fog::Credentials::AzureRM.get_token(@tenant_id, @client_id, @client_secret)
             RestClient.delete(

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

@@ -12,7 +12,7 @@ module Fog
           set_recovery_vault_context(rv_resource_group, rv_name)
           vm_id = get_virtual_machine_id(vm_resource_group, vm_name)
 
-          resource_url = "#{AZURE_RESOURCE}/subscriptions/#{@subscription_id}/resourceGroups/#{rv_resource_group}/providers/Microsoft.RecoveryServices/vaults/#{rv_name}/backupFabrics/Azure/protectionContainers/iaasvmcontainer;iaasvmcontainerv2;#{vm_resource_group.downcase};#{vm_name.downcase}/protectedItems/vm;iaasvmcontainerv2;#{vm_resource_group.downcase};#{vm_name.downcase}?api-version=2016-05-01"
+          resource_url = "#{AZURE_RESOURCE}/subscriptions/#{@subscription_id}/resourceGroups/#{rv_resource_group}/providers/Microsoft.RecoveryServices/vaults/#{rv_name}/backupFabrics/Azure/protectionContainers/iaasvmcontainer;iaasvmcontainerv2;#{vm_resource_group.downcase};#{vm_name.downcase}/protectedItems/vm;iaasvmcontainerv2;#{vm_resource_group.downcase};#{vm_name.downcase}?api-version=#{REST_CLIENT_API_VERSION[1]}"
           body = {
             properties: {
               protectedItemType: 'Microsoft.Compute/virtualMachines',

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

@@ -14,7 +14,7 @@ module Fog
           policy = backup_protection_policy.select { |item| item['name'].eql? 'DefaultPolicy' }[0]
           vm_id = get_virtual_machine_id(vm_resource_group, vm_name)
 
-          resource_url = "#{AZURE_RESOURCE}/subscriptions/#{@subscription_id}/resourceGroups/#{rv_resource_group}/providers/Microsoft.RecoveryServices/vaults/#{rv_name}/backupFabrics/Azure/protectionContainers/iaasvmcontainer;iaasvmcontainerv2;#{vm_resource_group.downcase};#{vm_name.downcase}/protectedItems/vm;iaasvmcontainerv2;#{vm_resource_group.downcase};#{vm_name.downcase}?api-version=2016-05-01"
+          resource_url = "#{AZURE_RESOURCE}/subscriptions/#{@subscription_id}/resourceGroups/#{rv_resource_group}/providers/Microsoft.RecoveryServices/vaults/#{rv_name}/backupFabrics/Azure/protectionContainers/iaasvmcontainer;iaasvmcontainerv2;#{vm_resource_group.downcase};#{vm_name.downcase}/protectedItems/vm;iaasvmcontainerv2;#{vm_resource_group.downcase};#{vm_name.downcase}?api-version=#{REST_CLIENT_API_VERSION[1]}"
           body = {
             properties: {
               protectedItemType: 'Microsoft.Compute/virtualMachines',

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

@@ -7,7 +7,7 @@ module Fog
           msg = "Getting all backup jobs for Recovery Vault #{rv_name}"
           Fog::Logger.debug msg
 
-          resource_url = "#{AZURE_RESOURCE}/subscriptions/#{@subscription_id}/resourceGroups/#{rv_resource_group}/providers/Microsoft.RecoveryServices/vaults/#{rv_name}/backupJobs?api-version=2016-05-01&$filter=status eq 'InProgress'"
+          resource_url = "#{AZURE_RESOURCE}/subscriptions/#{@subscription_id}/resourceGroups/#{rv_resource_group}/providers/Microsoft.RecoveryServices/vaults/#{rv_name}/backupJobs?api-version=#{REST_CLIENT_API_VERSION[1]}&$filter=status eq 'InProgress'"
           begin
             token = Fog::Credentials::AzureRM.get_token(@tenant_id, @client_id, @client_secret)
             response = RestClient.get(

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

@@ -7,7 +7,7 @@ module Fog
           msg = "Getting backup container from Recovery Vault #{rv_name}"
           Fog::Logger.debug msg
 
-          resource_url = "#{AZURE_RESOURCE}/subscriptions/#{@subscription_id}/resourceGroups/#{resource_group}/providers/Microsoft.RecoveryServices/vaults/#{rv_name}/backupProtectionContainers?api-version=2016-05-01&$filter=backupManagementType eq 'AzureIaasVM' and status eq 'Registered' and friendlyName eq '#{vm_name}'"
+          resource_url = "#{AZURE_RESOURCE}/subscriptions/#{@subscription_id}/resourceGroups/#{resource_group}/providers/Microsoft.RecoveryServices/vaults/#{rv_name}/backupProtectionContainers?api-version=#{REST_CLIENT_API_VERSION[1]}&$filter=backupManagementType eq 'AzureIaasVM' and status eq 'Registered' and friendlyName eq '#{vm_name}'"
           begin
             token = Fog::Credentials::AzureRM.get_token(@tenant_id, @client_id, @client_secret)
             response = RestClient.get(

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

@@ -7,7 +7,7 @@ module Fog
           msg = "Getting backup item from Recovery Vault #{rv_name}"
           Fog::Logger.debug msg
 
-          resource_url = "#{AZURE_RESOURCE}/subscriptions/#{@subscription_id}/resourceGroups/#{resource_group}/providers/Microsoft.RecoveryServices/vaults/#{rv_name}/backupProtectedItems?api-version=2016-05-01&$filter=backupManagementType eq 'AzureIaasVM' and itemType eq 'VM'"
+          resource_url = "#{AZURE_RESOURCE}/subscriptions/#{@subscription_id}/resourceGroups/#{resource_group}/providers/Microsoft.RecoveryServices/vaults/#{rv_name}/backupProtectedItems?api-version=#{REST_CLIENT_API_VERSION[1]}&$filter=backupManagementType eq 'AzureIaasVM' and itemType eq 'VM'"
           begin
             token = Fog::Credentials::AzureRM.get_token(@tenant_id, @client_id, @client_secret)
             response = RestClient.get(

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

@@ -7,7 +7,7 @@ module Fog
           msg = "Get backup protection policy from Resource Group #{resource_group}"
           Fog::Logger.debug msg
 
-          resource_url = "#{AZURE_RESOURCE}/subscriptions/#{@subscription_id}/resourceGroups/#{resource_group}/providers/Microsoft.RecoveryServices/vaults/#{name}/backupPolicies?api-version=2016-05-01&$filter=backupManagementType eq 'AzureIaasVM'"
+          resource_url = "#{AZURE_RESOURCE}/subscriptions/#{@subscription_id}/resourceGroups/#{resource_group}/providers/Microsoft.RecoveryServices/vaults/#{name}/backupPolicies?api-version=#{REST_CLIENT_API_VERSION[1]}&$filter=backupManagementType eq 'AzureIaasVM'"
           begin
             token = Fog::Credentials::AzureRM.get_token(@tenant_id, @client_id, @client_secret)
             response = RestClient.get(

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

@@ -6,7 +6,7 @@ module Fog
         def get_recovery_vault(resource_group, name)
           msg = "Retrieving Recovery Vault #{name} from Resource Group #{resource_group}"
           Fog::Logger.debug msg
-          resource_url = "#{AZURE_RESOURCE}/subscriptions/#{@subscription_id}/resourceGroups/#{resource_group}/providers/Microsoft.RecoveryServices/vaults?api-version=2016-05-01"
+          resource_url = "#{AZURE_RESOURCE}/subscriptions/#{@subscription_id}/resourceGroups/#{resource_group}/providers/Microsoft.RecoveryServices/vaults?api-version=#{REST_CLIENT_API_VERSION[1]}"
           begin
             token = Fog::Credentials::AzureRM.get_token(@tenant_id, @client_id, @client_secret)
             recovery_vault_response = RestClient.get(

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

@@ -6,7 +6,7 @@ module Fog
         def list_recovery_vaults(resource_group)
           msg = "Listing Recovery Vaults in Resource Group #{resource_group}"
           Fog::Logger.debug msg
-          resource_url = "#{AZURE_RESOURCE}/subscriptions/#{@subscription_id}/resourceGroups/#{resource_group}/providers/Microsoft.RecoveryServices/vaults?api-version=2016-05-01"
+          resource_url = "#{AZURE_RESOURCE}/subscriptions/#{@subscription_id}/resourceGroups/#{resource_group}/providers/Microsoft.RecoveryServices/vaults?api-version=#{REST_CLIENT_API_VERSION[1]}"
           begin
             token = Fog::Credentials::AzureRM.get_token(@tenant_id, @client_id, @client_secret)
             recovery_vaults_response = RestClient.get(

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

@@ -7,7 +7,7 @@ module Fog
           msg = "Set context for Recovery Vault #{name}"
           Fog::Logger.debug msg
 
-          resource_url = "#{AZURE_RESOURCE}/subscriptions/#{@subscription_id}/resourceGroups/#{resource_group}/providers/Microsoft.RecoveryServices/vaults/#{name}?api-version=2016-05-01"
+          resource_url = "#{AZURE_RESOURCE}/subscriptions/#{@subscription_id}/resourceGroups/#{resource_group}/providers/Microsoft.RecoveryServices/vaults/#{name}?api-version=#{REST_CLIENT_API_VERSION[1]}"
           begin
             token = Fog::Credentials::AzureRM.get_token(@tenant_id, @client_id, @client_secret)
             RestClient.get(

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

@@ -19,7 +19,7 @@ module Fog
           backup_items = get_backup_item(resource_group, name)
           backup_item = backup_items.select { |item| (item['properties']['friendlyName'].eql? vm_name.downcase) && (vm_resource_group.eql? get_resource_group_from_id(item['properties']['virtualMachineId'])) }[0]
 
-          resource_url = "#{AZURE_RESOURCE}/subscriptions/#{@subscription_id}/resourceGroups/#{resource_group}/providers/Microsoft.RecoveryServices/vaults/#{name}/backupFabrics/Azure/protectionContainers/iaasvmcontainer;#{backup_item['name']}/protectedItems/vm;#{backup_item['name']}/backup?api-version=2016-05-01"
+          resource_url = "#{AZURE_RESOURCE}/subscriptions/#{@subscription_id}/resourceGroups/#{resource_group}/providers/Microsoft.RecoveryServices/vaults/#{name}/backupFabrics/Azure/protectionContainers/iaasvmcontainer;#{backup_item['name']}/protectedItems/vm;#{backup_item['name']}/backup?api-version=#{REST_CLIENT_API_VERSION[1]}"
           body = {
             properties: {
               objectType: IAAS_VM_BACKUP_REQUEST

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

@@ -0,0 +1,31 @@
+module Fog
+  module TrafficManager
+    class AzureRM
+      # This class provides the actual implementation for service calls.
+      class Real
+        def check_traffic_manager_endpoint_exists(resource_group, traffic_manager_profile_name, traffic_manager_end_point, type)
+          msg = "Checking Traffic Manager Endpoint #{traffic_manager_end_point}"
+          Fog::Logger.debug msg
+          begin
+            @traffic_mgmt_client.endpoints.get(resource_group, traffic_manager_profile_name, type, traffic_manager_end_point)
+            Fog::Logger.debug "Traffic Manager Endpoint #{traffic_manager_end_point} exists."
+            true
+          rescue MsRestAzure::AzureOperationError => e
+            if e.body['code'] == 'NotFound'
+              Fog::Logger.debug "Traffic Manager Endpoint #{traffic_manager_end_point} doesn't exist."
+              false
+            else
+              raise_azure_exception(e, msg)
+            end
+          end
+        end
+      end
+      # This class provides the mock implementation for unit tests.
+      class Mock
+        def check_traffic_manager_endpoint_exists(*)
+          true
+        end
+      end
+    end
+  end
+end

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

@@ -0,0 +1,31 @@
+module Fog
+  module TrafficManager
+    class AzureRM
+      # This class provides the actual implementation for service calls.
+      class Real
+        def check_traffic_manager_profile_exists(resource_group, traffic_manager_profile_name)
+          msg = "Checking Traffic Manager Profile #{traffic_manager_profile_name}"
+          Fog::Logger.debug msg
+          begin
+            @traffic_mgmt_client.profiles.get(resource_group, traffic_manager_profile_name)
+            Fog::Logger.debug "Traffic Manager Profile #{traffic_manager_profile_name} exists."
+            true
+          rescue MsRestAzure::AzureOperationError => e
+            if e.body['error']['code'] == 'ResourceNotFound'
+              Fog::Logger.debug "Traffic Manager Profile #{traffic_manager_profile_name} doesn't exist."
+              false
+            else
+              raise_azure_exception(e, msg)
+            end
+          end
+        end
+      end
+      # This class provides the mock implementation for unit tests.
+      class Mock
+        def check_traffic_manager_profile_exists(*)
+          true
+        end
+      end
+    end
+  end
+end