fog-azure-rm 0.1.2 → 0.2.0

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

@@ -0,0 +1,31 @@
+module Fog
+  module Network
+    class AzureRM
+      # Real class for Network Request
+      class Real
+        def list_load_balancers_in_subscription
+          msg = 'Getting list of Load-Balancers in subscription'
+          Fog::Logger.debug msg
+          begin
+            load_balancers = @network_client.load_balancers.list_all
+          rescue MsRestAzure::AzureOperationError => e
+            raise_azure_exception(e, msg)
+          end
+          Fog::Logger.debug 'Successfully retrieved load balancers in subscription'
+          load_balancers
+        end
+      end
+
+      # Mock class for Network Request
+      class Mock
+        def list_load_balancers_in_subscription
+          lb = Azure::ARM::Network::Models::LoadBalancer.new
+          lb.name = 'fogtestloadbalancer'
+          lb.location = 'West US'
+          lb.properties = Azure::ARM::Network::Models::LoadBalancerPropertiesFormat.new
+          [lb]
+        end
+      end
+    end
+  end
+end

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

@@ -0,0 +1,74 @@
+module Fog
+  module Network
+    class AzureRM
+      # This class provides the actual implementation for service calls.
+      class Real
+        def list_virtual_networks_in_subscription
+          msg = 'Listing Virtual Networks in a subscription'
+          Fog::Logger.debug msg
+          begin
+            virtual_networks = @network_client.virtual_networks.list_all
+          rescue MsRestAzure::AzureOperationError => e
+            raise_azure_exception(e, msg)
+          end
+          virtual_networks
+        end
+      end
+
+      # This class provides the mock implementation for unit tests.
+      class Mock
+        def list_virtual_networks_in_subscription
+          virtual_network = {
+            'value' => [
+              {
+                'id' => '/subscriptions/########-####-####-####-############/resourceGroups/TestRG/providers/Microsoft.Network/virtualNetworks/testVnet',
+                'name' => 'testVnet',
+                'type' => 'Microsoft.Network/virtualNetworks',
+                'location' => 'westus',
+                'properties' =>
+                  {
+                    'addressSpace' =>
+                      {
+                        'addressPrefixes' =>
+                          %w(10.1.0.0/16 10.2.0.0/16)
+                      },
+                    'subnets' =>
+                      [
+                        {
+                          'id' => '/subscriptions/########-####-####-####-############/resourceGroups/TestRG/providers/Microsoft.Network/virtualNetworks/testVnet/subnets/subnet_0_testVnet',
+                          'properties' =>
+                            {
+                              'addressPrefix' => '10.1.0.0/24',
+                              'provisioningState' => 'Succeeded'
+                            },
+                          'name' => 'subnet_0_testVnet'
+                        },
+                        {
+                          'id' => '/subscriptions/########-####-####-####-############/resourceGroups/TestRG/providers/Microsoft.Network/virtualNetworks/testVnet/subnets/fog-test-subnet',
+                          'properties' =>
+                            {
+                              'addressPrefix' => '10.2.0.0/16',
+                              'ipConfigurations' =>
+                                [
+                                  {
+                                    'id' => '/subscriptions/########-####-####-####-############/resourceGroups/TestRG/providers/Microsoft.Network/networkInterfaces/test-NIC/ipConfigurations/ipconfig1'
+                                  }
+                                ],
+                              'provisioningState' => 'Succeeded'
+                            },
+                          'name' => 'fog-test-subnet'
+                        }
+                      ],
+                    'resourceGuid' => 'c573f8e2-d916-493f-8b25-a681c31269ef',
+                    'provisioningState' => 'Succeeded'
+                  }
+              }
+            ]
+          }
+          vnet_mapper = Azure::ARM::Network::Models::VirtualNetworkListResult.mapper
+          @network_client.deserialize(vnet_mapper, virtual_network, 'result.body')
+        end
+      end
+    end
+  end
+end

data/lib/fog/azurerm/requests/resources/check_azure_resource_exists.rb

@@ -0,0 +1,40 @@
+module Fog
+  module Resources
+    class AzureRM
+      # This class provides the actual implementation for service calls.
+      class Real
+        def check_azure_resource_exists(resource_id, api_version)
+          split_resource = resource_id.split('/') unless resource_id.nil?
+          raise 'Invalid Resource Id' if split_resource.count != 9
+
+          resource_group_name = get_resource_from_resource_id(resource_id, RESOURCE_GROUP_NAME)
+          resource_provider_namespace = get_resource_from_resource_id(resource_id, RESOURCE_PROVIDER_NAMESPACE)
+          resource_type = get_resource_from_resource_id(resource_id, RESOURCE_TYPE)
+          resource_name = get_resource_from_resource_id(resource_id, RESOURCE_NAME)
+          parent_resource_id = ''
+
+          msg = "Checking Resource #{resource_name}"
+          Fog::Logger.debug msg
+          begin
+            @rmc.resources.check_existence(resource_group_name, resource_provider_namespace, parent_resource_id, resource_type, resource_name, api_version)
+            Fog::Logger.debug "Resource #{resource_name} exists."
+            true
+          rescue MsRestAzure::AzureOperationError => e
+            if e.response.status == 405
+              Fog::Logger.debug "Resource #{resource_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_azure_resource_exists(*)
+          true
+        end
+      end
+    end
+  end
+end

data/lib/fog/azurerm/requests/resources/check_deployment_exists.rb

@@ -0,0 +1,30 @@
+module Fog
+  module Resources
+    class AzureRM
+      # This class provides the actual implementation for service calls.
+      class Real
+        def check_deployment_exists(resource_group_name, deployment_name)
+          msg = "Checking Deployment #{deployment_name}"
+          Fog::Logger.debug msg
+          begin
+            flag = @rmc.deployments.check_existence(resource_group_name, deployment_name)
+            if flag
+              Fog::Logger.debug "Deployment #{deployment_name} exists."
+            else
+              Fog::Logger.debug "Deployment #{deployment_name} doesn't exist."
+            end
+            flag
+          rescue MsRestAzure::AzureOperationError => e
+            raise_azure_exception(e, msg)
+          end
+        end
+      end
+      # This class provides the mock implementation for unit tests.
+      class Mock
+        def check_deployment_exists(*)
+          true
+        end
+      end
+    end
+  end
+end

data/lib/fog/azurerm/requests/resources/check_resource_group_exists.rb

@@ -0,0 +1,30 @@
+module Fog
+  module Resources
+    class AzureRM
+      # This class provides the actual implementation for service calls.
+      class Real
+        def check_resource_group_exists(resource_group_name)
+          msg = "Checking Resource Group #{resource_group_name}"
+          Fog::Logger.debug msg
+          begin
+            flag = @rmc.resource_groups.check_existence(resource_group_name)
+            if flag
+              Fog::Logger.debug "Resource Group #{resource_group_name} exists."
+            else
+              Fog::Logger.debug "Resource Group #{resource_group_name} doesn't exist."
+            end
+            flag
+          rescue MsRestAzure::AzureOperationError => e
+            raise_azure_exception(e, msg)
+          end
+        end
+      end
+      # This class provides the mock implementation for unit tests.
+      class Mock
+        def check_resource_group_exists(*)
+          true
+        end
+      end
+    end
+  end
+end

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

@@ -0,0 +1,21 @@
+module Fog
+  module Sql
+    class AzureRM
+      # Mock class for Sql Request
+      class Real
+        def check_database_exists(resource_group, server_name, name)
+          msg = "Checking SQL Database #{name}"
+          Fog::Logger.debug msg
+          # This module needs to be updated to azure sdk
+        end
+      end
+
+      # Mock class for Sql Request
+      class Mock
+        def check_database_exists(*)
+          true
+        end
+      end
+    end
+  end
+end

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

@@ -0,0 +1,21 @@
+module Fog
+  module Sql
+    class AzureRM
+      # Mock class for Sql Request
+      class Real
+        def check_firewall_rule_exists(resource_group, server_name, rule_name)
+          msg = "Checking Firewall Rule #{rule_name}"
+          Fog::Logger.debug msg
+          # This module needs to be updated to azure sdk
+        end
+      end
+
+      # Mock class for Sql Request
+      class Mock
+        def check_firewall_rule_exists(*)
+          true
+        end
+      end
+    end
+  end
+end

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

@@ -0,0 +1,21 @@
+module Fog
+  module Sql
+    class AzureRM
+      # Mock class for Sql Request
+      class Real
+        def check_sql_server_exists(resource_group, server_name)
+          msg = "Checking SQL Server #{server_name}"
+          Fog::Logger.debug msg
+          # This module needs to be updated to azure sdk
+        end
+      end
+
+      # Mock class for Sql Request
+      class Mock
+        def check_sql_server_exists(*)
+          true
+        end
+      end
+    end
+  end
+end

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

@@ -6,46 +6,40 @@ module Fog
         def create_or_update_database(database_hash)
           msg = "Creating SQL Database: #{database_hash[:name]}."
           Fog::Logger.debug msg
-          resource_url = "#{resource_manager_endpoint_url}/subscriptions/#{@subscription_id}/resourceGroups/#{database_hash[:resource_group]}/providers/Microsoft.Sql/servers/#{database_hash[:server_name]}/databases/#{database_hash[:name]}?api-version=2014-04-01-preview"
-          request_parameters = format_database_parameters(database_hash[:location], database_hash[:create_mode], database_hash[:edition], database_hash[:source_database_id], database_hash[:collation], database_hash[:max_size_bytes], database_hash[:requested_service_objective_name], database_hash[:restore_point_in_time], database_hash[:source_database_deletion_date], database_hash[:elastic_pool_name], database_hash[:requested_service_objective_id])
           begin
-            token = Fog::Credentials::AzureRM.get_token(@tenant_id, @client_id, @client_secret)
-            response = RestClient.put(
-              resource_url,
-              request_parameters.to_json,
-              accept: :json,
-              content_type: :json,
-              authorization: token
-            )
-          rescue RestClient::Exception => e
+            sql_database = @sql_mgmt_client.databases.create_or_update(database_hash[:resource_group],
+                                                                       database_hash[:server_name],
+                                                                       database_hash[:name],
+                                                                       format_database_parameters(database_hash[:location],
+                                                                                                  database_hash[:create_mode],
+                                                                                                  database_hash[:edition],
+                                                                                                  database_hash[:source_database_id],
+                                                                                                  database_hash[:collation],
+                                                                                                  database_hash[:max_size_bytes],
+                                                                                                  database_hash[:requested_service_objective_name],
+                                                                                                  database_hash[:elastic_pool_name],
+                                                                                                  database_hash[:requested_service_objective_id]))
+          rescue MsRestAzure::AzureOperationError => e
             raise_azure_exception(e, msg)
           end
           Fog::Logger.debug "SQL Database: #{database_hash[:name]} created successfully."
-          Fog::JSON.decode(response)
+          sql_database
         end
 
         private
 
-        def format_database_parameters(location, create_mode, edition, source_database_id, collation, max_size_bytes, requested_service_objective_name, restore_point_in_time, source_database_deletion_date, elastic_pool_name, requested_service_objective_id)
-          parameters = {}
-          properties = {}
-
-          properties['edition'] = edition unless edition.nil?
-          properties['collation'] = collation unless collation.nil?
-          properties['createMode'] = create_mode unless create_mode.nil?
-          properties['maxSizeBytes'] = max_size_bytes unless max_size_bytes.nil?
-          properties['elasticPoolName'] = elastic_pool_name unless elastic_pool_name.nil?
-          properties['sourceDatabaseId'] = source_database_id unless source_database_id.nil?
-          properties['restorePointInTime'] = restore_point_in_time unless restore_point_in_time.nil?
-          properties['sourceDatabaseDeletionDate'] = source_database_deletion_date unless source_database_deletion_date.nil?
-          properties['requestedServiceObjectiveId'] = requested_service_objective_id unless requested_service_objective_id.nil?
-          properties['requestedServiceObjectiveName'] = requested_service_objective_name unless requested_service_objective_name.nil?
-
-          parameters['tags'] = {}
-          parameters['location'] = location
-          parameters['properties'] = properties
-
-          parameters
+        def format_database_parameters(location, create_mode, edition, source_database_id, collation, max_size_bytes, requested_service_objective_name, elastic_pool_name, requested_service_objective_id)
+          database = Azure::ARM::SQL::Models::Database.new
+          database.location = location
+          database.edition = edition unless edition.nil?
+          database.collation = collation unless collation.nil?
+          database.create_mode = create_mode unless create_mode.nil?
+          database.max_size_bytes = max_size_bytes unless max_size_bytes.nil?
+          database.elastic_pool_name = elastic_pool_name unless elastic_pool_name.nil?
+          database.source_database_id = source_database_id unless source_database_id.nil?
+          database.requested_service_objective_id = requested_service_objective_id unless requested_service_objective_id.nil?
+          database.requested_service_objective_name = requested_service_objective_name unless requested_service_objective_name.nil?
+          database
         end
       end
 

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

@@ -7,36 +7,22 @@ module Fog
           msg = "Creating SQL Firewall Rule : #{firewall_hash[:name]}."
           Fog::Logger.debug msg
 
-          resource_url = "#{resource_manager_endpoint_url}/subscriptions/#{@subscription_id}/resourceGroups/#{firewall_hash[:resource_group]}/providers/Microsoft.Sql/servers/#{firewall_hash[:server_name]}/firewallRules/#{firewall_hash[:name]}?api-version=2014-04-01-preview"
-          request_parameters = get_server_firewall_parameters(firewall_hash[:start_ip], firewall_hash[:end_ip])
           begin
-            token = Fog::Credentials::AzureRM.get_token(@tenant_id, @client_id, @client_secret)
-            response = RestClient.put(
-              resource_url,
-              request_parameters.to_json,
-              accept: :json,
-              content_type: :json,
-              authorization: token
-            )
-          rescue RestClient::Exception => e
+            server_firewall_rule = @sql_mgmt_client.servers.create_or_update_firewall_rule(firewall_hash[:resource_group], firewall_hash[:server_name], firewall_hash[:name], format_server_firewall_parameters(firewall_hash[:start_ip], firewall_hash[:end_ip]))
+          rescue MsRestAzure::AzureOperationError => e
             raise_azure_exception(e, msg)
           end
           Fog::Logger.debug "SQL Firewall Rule : #{firewall_hash[:name]} created successfully."
-          Fog::JSON.decode(response)
+          server_firewall_rule
         end
 
         private
 
-        def get_server_firewall_parameters(start_ip, end_ip)
-          parameters = {}
-          properties = {}
-
-          properties['startIpAddress'] = start_ip
-          properties['endIpAddress'] = end_ip
-
-          parameters['properties'] = properties
-
-          parameters
+        def format_server_firewall_parameters(start_ip, end_ip)
+          firewall_rule = Azure::ARM::SQL::Models::ServerFirewallRule.new
+          firewall_rule.start_ip_address = start_ip
+          firewall_rule.end_ip_address = end_ip
+          firewall_rule
         end
       end
 

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

@@ -6,39 +6,29 @@ module Fog
         def create_or_update_sql_server(server_hash)
           msg = "Creating SQL Server: #{server_hash[:name]}."
           Fog::Logger.debug msg
-          resource_url = "#{resource_manager_endpoint_url}/subscriptions/#{@subscription_id}/resourceGroups/#{server_hash[:resource_group]}/providers/Microsoft.Sql/servers/#{server_hash[:name]}?api-version=2014-04-01-preview"
-          request_parameters = format_server_parameters(server_hash[:location], server_hash[:version], server_hash[:administrator_login], server_hash[:administrator_login_password])
           begin
-            token = Fog::Credentials::AzureRM.get_token(@tenant_id, @client_id, @client_secret)
-            response = RestClient.put(
-              resource_url,
-              request_parameters.to_json,
-              accept: :json,
-              content_type: :json,
-              authorization: token
-            )
-          rescue RestClient::Exception => e
+            sql_server = @sql_mgmt_client.servers.create_or_update(server_hash[:resource_group],
+                                                                   server_hash[:name],
+                                                                   format_server_parameters(server_hash[:location],
+                                                                                            server_hash[:version],
+                                                                                            server_hash[:administrator_login],
+                                                                                            server_hash[:administrator_login_password]))
+          rescue MsRestAzure::AzureOperationError => e
             raise_azure_exception(e, msg)
           end
           Fog::Logger.debug "SQL Server: #{server_hash[:name]} created successfully."
-          Fog::JSON.decode(response)
+          sql_server
         end
 
         private
 
         def format_server_parameters(location, version, admin_login, admin_password)
-          parameters = {}
-          properties = {}
-
-          properties['version'] = version
-          properties['administratorLogin'] = admin_login
-          properties['administratorLoginPassword'] = admin_password
-
-          parameters['properties'] = properties
-          parameters['location'] = location
-          parameters['tags'] = {}
-
-          parameters
+          server = Azure::ARM::SQL::Models::Server.new
+          server.version = version
+          server.location = location
+          server.administrator_login = admin_login
+          server.administrator_login_password = admin_password
+          server
         end
       end