fog-azure-rm 0.1.2 → 0.2.0

data/lib/fog/azurerm/resources.rb

@@ -6,19 +6,23 @@ module Fog
       requires :client_id
       requires :client_secret
       requires :subscription_id
+      recognizes :environment
 
       request_path 'fog/azurerm/requests/resources'
       request :create_resource_group
       request :list_resource_groups
       request :delete_resource_group
       request :get_resource_group
+      request :check_resource_group_exists
       request :create_deployment
       request :delete_deployment
       request :list_deployments
       request :get_deployment
+      request :check_deployment_exists
       request :delete_resource_tag
       request :list_tagged_resources
       request :tag_resource
+      request :check_azure_resource_exists
 
       model_path 'fog/azurerm/models/resources'
       model :resource_group
@@ -53,9 +57,13 @@ module Fog
             raise e.message
           end
 
-          credentials = Fog::Credentials::AzureRM.get_credentials(options[:tenant_id], options[:client_id], options[:client_secret])
-          @rmc = ::Azure::ARM::Resources::ResourceManagementClient.new(credentials, resource_manager_endpoint_url)
+          options[:environment] = 'AzureCloud' if options[:environment].nil?
+
+          credentials = Fog::Credentials::AzureRM.get_credentials(options[:tenant_id], options[:client_id], options[:client_secret], options[:environment])
+          telemetry = "fog-azure-rm/#{Fog::AzureRM::VERSION}"
+          @rmc = ::Azure::ARM::Resources::ResourceManagementClient.new(credentials, resource_manager_endpoint_url(options[:environment]))
           @rmc.subscription_id = options[:subscription_id]
+          @rmc.add_user_agent_information(telemetry)
         end
       end
     end

data/lib/fog/azurerm/sql.rb

@@ -6,22 +6,26 @@ module Fog
       requires :client_id
       requires :client_secret
       requires :subscription_id
+      recognizes :environment
 
       request_path 'fog/azurerm/requests/sql'
       request :create_or_update_sql_server
       request :delete_sql_server
       request :get_sql_server
       request :list_sql_servers
+      request :check_sql_server_exists
 
       request :create_or_update_database
       request :delete_database
       request :get_database
       request :list_databases
+      request :check_database_exists
 
       request :create_or_update_firewall_rule
       request :delete_firewall_rule
       request :get_firewall_rule
       request :list_firewall_rules
+      request :check_firewall_rule_exists
 
       model_path 'fog/azurerm/models/sql'
       model :sql_server
@@ -36,16 +40,20 @@ module Fog
       # This class provides the actual implementation for service calls.
       class Real
         def initialize(options)
-          @tenant_id = options[:tenant_id]
-          @client_id = options[:client_id]
-          @client_secret = options[:client_secret]
-          @subscription_id = options[:subscription_id]
-          @resources = Fog::Resources::AzureRM.new(
-            tenant_id: options[:tenant_id],
-            client_id: options[:client_id],
-            client_secret: options[:client_secret],
-            subscription_id: options[:subscription_id]
-          )
+          begin
+            require 'azure_mgmt_sql'
+          rescue LoadError => e
+            retry if require('rubygems')
+            raise e.message
+          end
+
+          options[:environment] = 'AzureCloud' if options[:environment].nil?
+
+          credentials = Fog::Credentials::AzureRM.get_credentials(options[:tenant_id], options[:client_id], options[:client_secret], options[:environment])
+          @sql_mgmt_client = ::Azure::ARM::SQL::SqlManagementClient.new(credentials, resource_manager_endpoint_url(options[:environment]))
+          @sql_mgmt_client.subscription_id = options[:subscription_id]
+          telemetry = "fog-azure-rm/#{Fog::AzureRM::VERSION}"
+          @sql_mgmt_client.add_user_agent_information(telemetry)
         end
       end
       # This class provides the mock implementation for unit tests.

data/lib/fog/azurerm/storage.rb

@@ -24,6 +24,7 @@ module Fog
       request :list_storage_account_for_rg
       request :check_storage_account_name_availability
       request :get_storage_access_keys
+      request :check_storage_account_exists
       # Azure Storage Disk requests
       request :delete_disk
       request :create_disk
@@ -38,6 +39,7 @@ module Fog
       request :get_container_acl
       request :put_container_acl
       request :get_container_url
+      request :check_container_exists
       # Azure Storage Blob requests
       request :list_blobs
       request :put_blob_metadata
@@ -125,9 +127,11 @@ module Fog
           @environment = options[:environment]
 
           credentials = Fog::Credentials::AzureRM.get_credentials(@tenant_id, @client_id, @client_secret, @environment)
+          telemetry = "fog-azure-rm/#{Fog::AzureRM::VERSION}"
           unless credentials.nil?
             @storage_mgmt_client = ::Azure::ARM::Storage::StorageManagementClient.new(credentials, resource_manager_endpoint_url(@environment))
             @storage_mgmt_client.subscription_id = @subscription_id
+            @storage_mgmt_client.add_user_agent_information(telemetry)
           end
 
           return unless @azure_storage_account_name != options[:azure_storage_account_name] ||

data/lib/fog/azurerm/traffic_manager.rb

@@ -6,16 +6,19 @@ module Fog
       requires :client_id
       requires :client_secret
       requires :subscription_id
+      recognizes :environment
 
       request_path 'fog/azurerm/requests/traffic_manager'
       request :create_or_update_traffic_manager_profile
       request :delete_traffic_manager_profile
       request :get_traffic_manager_profile
       request :list_traffic_manager_profiles
+      request :check_traffic_manager_profile_exists
 
       request :create_or_update_traffic_manager_endpoint
       request :delete_traffic_manager_endpoint
       request :get_traffic_manager_endpoint
+      request :check_traffic_manager_endpoint_exists
 
       model_path 'fog/azurerm/models/traffic_manager'
       model :traffic_manager_profile
@@ -34,9 +37,13 @@ module Fog
             raise e.message
           end
 
-          credentials = Fog::Credentials::AzureRM.get_credentials(options[:tenant_id], options[:client_id], options[:client_secret])
-          @traffic_mgmt_client = ::Azure::ARM::TrafficManager::TrafficManagerManagementClient.new(credentials, resource_manager_endpoint_url)
+          options[:environment] = 'AzureCloud' if options[:environment].nil?
+
+          credentials = Fog::Credentials::AzureRM.get_credentials(options[:tenant_id], options[:client_id], options[:client_secret], options[:environment])
+          telemetry = "fog-azure-rm/#{Fog::AzureRM::VERSION}"
+          @traffic_mgmt_client = ::Azure::ARM::TrafficManager::TrafficManagerManagementClient.new(credentials, resource_manager_endpoint_url(options[:environment]))
           @traffic_mgmt_client.subscription_id = options[:subscription_id]
+          @traffic_mgmt_client.add_user_agent_information(telemetry)
         end
       end
       # This class provides the mock implementation for unit tests.

data/lib/fog/azurerm/version.rb

@@ -1,5 +1,5 @@
 module Fog
   module AzureRM
-    VERSION = '0.1.2'.freeze
+    VERSION = '0.2.0'.freeze
   end
 end

data/rake-script.sh

@@ -1,11 +1,4 @@
 #!/bin/bash
-if [ $TRAVIS_PULL_REQUEST != "true" ]
-then
-  rake cc_coverage
-  export CODECLIMATE_REPO_TOKEN="b1401494baa004d90402414cb33a7fc6420fd3693e60c677a120ddefd7d84cfd"
-  codeclimate-test-reporter --directory /home/travis/build/fog/fog-azure-rm/coverage
-else
-  rake cc_coverage
-  export CODECLIMATE_REPO_TOKEN="b1401494baa004d90402414cb33a7fc6420fd3693e60c677a120ddefd7d84cfd"
-  codeclimate-test-reporter --directory /home/travis/build/fog/fog-azure-rm/coverage
-fi
+rake cc_coverage
+export CODECLIMATE_REPO_TOKEN="b1401494baa004d90402414cb33a7fc6420fd3693e60c677a120ddefd7d84cfd"
+codeclimate-test-reporter

data/test/api_stub.rb

@@ -56,6 +56,10 @@ module ApiStub
       autoload :SqlDatabase, File.expand_path('api_stub/models/sql/sql_database', __dir__)
       autoload :SqlFirewallRule, File.expand_path('api_stub/models/sql/sql_firewall_rule', __dir__)
     end
+
+    module KeyVault
+      autoload :Vault, File.expand_path('api_stub/models/key_vault/vault', __dir__)
+    end
   end
 
   module Requests
@@ -114,5 +118,9 @@ module ApiStub
       autoload :SqlDatabase, File.expand_path('api_stub/requests/sql/database', __dir__)
       autoload :FirewallRule, File.expand_path('api_stub/requests/sql/firewall_rule', __dir__)
     end
+
+    module KeyVault
+      autoload :Vault, File.expand_path('api_stub/requests/key_vault/vault', __dir__)
+    end
   end
 end

data/test/api_stub/models/key_vault/vault.rb

@@ -0,0 +1,39 @@
+module ApiStub
+  module Models
+    module KeyVault
+      # Mock class for Vault Model
+      class Vault
+        def self.create_vault_response(key_vault_client)
+          vault = {
+            'id' => '/subscriptions/<AZURE_SUBSCRIPTION_ID>/resourceGroups/fog-test-rg/providers/Microsoft.KeyVault/vaults/fog-test-kv',
+            'name' => 'fog-test-kv',
+            'type' => 'Microsoft.KeyVault/vaults',
+            'location' => 'westus',
+            'tags' => {},
+            'properties' => {
+              'sku' => {
+                'family' => 'A',
+                'name' => 'standard'
+              },
+              'tenantId' => '<AZURE_TENANT_ID>',
+              'accessPolicies' => [
+                {
+                  'tenantId' => '<AZURE_TENANT_ID>',
+                  'objectId' => '<AZURE_OBJECT_ID>',
+                  'permissions' => {
+                    'keys' => ['all'],
+                    'secrets' => ['all']
+                  }
+                }
+              ],
+              'enabledForDeployment' => false,
+              'vaultUri' => 'https://fog-test-kv.vault.azure.net/'
+            }
+          }
+          vault_mapper = Azure::ARM::KeyVault::Models::Vault.mapper
+          key_vault_client.deserialize(vault_mapper, vault, 'result.body')
+        end
+      end
+    end
+  end
+end

data/test/api_stub/models/sql/sql_database.rb

@@ -4,23 +4,25 @@ module ApiStub
       # Mock class for Database
       class SqlDatabase
         # This class contain two mocks, for collection and for model
-        def self.create_database
-          {
-            'id' => '/subscriptions/########-####-####-####-############/resourceGroups/{resource-group-name}/providers/Microsoft.Sql/servers/{test-sql-server-name}/databases/{database-name}',
-            'location' => '{database-location}',
-            'properties' => {
-              'createMode' => '{creation-mode}',
-              'sourceDatabaseId' => '{source-database-id}',
-              'edition' => '{database-edition}',
-              'collation' => '{collation-name}',
-              'maxSizeBytes' => '{max-database-size}',
-              'requestedServiceObjectiveId' => '{requested-service-id}',
-              'requestedServiceObjectiveName' => '{requested-service-id}',
-              'restorePointInTime' => '{restore-time}',
-              'sourceDatabaseDeletionDate' => '{source-deletion-date}',
-              'elasticPoolName' => '{elastic-pool-name}'
+        def self.create_database(sql_manager_client)
+          database = '{
+            "id" : "/subscriptions/########-####-####-####-############/resourceGroups/{resource-group-name}/providers/Microsoft.Sql/servers/{test-sql-server-name}/databases/{database-name}",
+            "location" : "{database-location}",
+            "properties" : {
+              "createMode" : "{creation-mode}",
+              "sourceDatabaseId" : "{source-database-id}",
+              "edition" : "{database-edition}",
+              "collation" : "{collation-name}",
+              "maxSizeBytes" : "{max-database-size}",
+              "requestedServiceObjectiveId" : "{requested-service-id}",
+              "requestedServiceObjectiveName" : "{requested-service-id}",
+              "restorePointInTime" : "{restore-time}",
+              "sourceDatabaseDeletionDate" : "{source-deletion-date}",
+              "elasticPoolName" : "{elastic-pool-name}"
             }
-          }
+          }'
+          database_mapper = Azure::ARM::SQL::Models::Database.mapper
+          sql_manager_client.deserialize(database_mapper, Fog::JSON.decode(database), 'result.body')
         end
       end
     end

data/test/api_stub/models/sql/sql_firewall_rule.rb

@@ -4,17 +4,19 @@ module ApiStub
       # Mock class for Firewall Rule
       class SqlFirewallRule
         # This class contain mock
-        def self.create_firewall_rule
-          {
-            'id' => '/subscriptions/########-####-####-####-############/resourceGroups/{resource-group-name}/providers/Microsoft.Sql/servers/{server-name}/firewallRules/{rule-name}',
-            'name' => '{rule-name}',
-            'type' => '{rule-type}',
-            'location' => '{server-location}',
-            'properties' => {
-              'startIpAddress' => '{start-ip-address}',
-              'endIpAddress' => '{end-ip-address}'
+        def self.create_firewall_rule(firewall_manager_client)
+          firewall_rule = '{
+            "id" : "/subscriptions/########-####-####-####-############/resourceGroups/{resource-group-name}/providers/Microsoft.Sql/servers/{server-name}/firewallRules/{rule-name}",
+            "name" : "{rule-name}",
+            "type" : "{rule-type}",
+            "location" : "{server-location}",
+            "properties" : {
+              "startIpAddress" : "{start-ip-address}",
+              "endIpAddress" : "{end-ip-address}"
             }
-          }
+          }'
+          firewall_rule_mapper = Azure::ARM::SQL::Models::ServerFirewallRule.mapper
+          firewall_manager_client.deserialize(firewall_rule_mapper, Fog::JSON.decode(firewall_rule), 'result.body')
         end
       end
     end

data/test/api_stub/models/sql/sql_server.rb

@@ -4,17 +4,19 @@ module ApiStub
       # Mock class for Sql Server
       class SqlServer
         # This class contain two mocks, for collection and for model
-        def self.create_sql_server
-          {
-            'id' => '/subscriptions/########-####-####-####-############/resourceGroups/{resource-group-name}/providers/Microsoft.Sql/servers/{server-name}',
-            'name' => 'server-name',
-            'location' => '{server-location}',
-            'properties' => {
-              'version' => '{server-version}',
-              'administratorLogin' => '{admin-name}',
-              'administratorLoginPassword' => '{admin-password}'
+        def self.create_sql_server(server_manager_client)
+          sql_server = '{
+            "id" : "/subscriptions/########-####-####-####-############/resourceGroups/{resource-group-name}/providers/Microsoft.Sql/servers/{server-name}",
+            "name" : "server-name",
+            "location" : "{server-location}",
+            "properties" : {
+              "version" : "{server-version}",
+              "administratorLogin" : "{admin-name}",
+              "administratorLoginPassword" : "{admin-password}"
             }
-          }
+          }'
+          server_mapper = Azure::ARM::SQL::Models::Server.mapper
+          server_manager_client.deserialize(server_mapper, Fog::JSON.decode(sql_server), 'result.body')
         end
       end
     end

data/test/api_stub/requests/key_vault/vault.rb

@@ -0,0 +1,95 @@
+module ApiStub
+  module Requests
+    module KeyVault
+      # Mock class for Vault Requests
+      class Vault
+        def self.create_vault_response(key_vault_client)
+          response = '{
+            "id" : "/subscriptions/{AZURE_SUBSCRIPTION_ID}/resourceGroups/fog-test-rg/providers/Microsoft.KeyVault/vaults/fog-test-kv",
+            "name" : "fog-test-kv",
+            "type" : "Microsoft.KeyVault/vaults",
+            "location" : "westus",
+            "tags" : {},
+            "properties" : {
+              "sku" : {
+                "family" : "A",
+                "name" : "standard"
+              },
+              "tenantId" : "{AZURE_TENANT_ID}",
+              "accessPolicies" : [
+                {
+                   "tenantId" : "{AZURE_TENANT_ID}",
+                   "objectId" : "{AZURE_OBJECT_ID}",
+                   "permissions" : {
+                     "keys" : ["all"],
+                     "secrets" : ["all"]
+                   }
+                }
+              ],
+              "enabledForDeployment" : false,
+              "vaultUri" : "https://fog-test-kv.vault.azure.net/"
+            }
+          }'
+          vault_mapper = Azure::ARM::KeyVault::Models::Vault.mapper
+          key_vault_client.deserialize(vault_mapper, Fog::JSON.decode(response), 'result.body')
+        end
+
+        def self.vault_params
+          access_policies_arr = [
+            {
+              tenant_id: '{AZURE_TENANT_ID}',
+              object_id: '{AZURE_TENANT_ID}',
+              permissions: {
+                keys: ['all'],
+                secrets: ['all']
+              }
+            }
+          ]
+
+          {
+            name: 'test-tmp',
+            resource_group: 'TestRG-KV',
+            location: 'eastus',
+            tenant_id: '{AZURE_TENANT_ID}',
+            sku_family: 'A',
+            sku_name: 'standard',
+            access_policies: access_policies_arr
+          }
+        end
+
+        def self.list_vault_response(key_vault_client)
+          response = '{
+            "value": [{
+              "id" : "/subscriptions/{AZURE_SUBSCRIPTION_ID}/resourceGroups/fog-test-rg/providers/Microsoft.KeyVault/vaults/fog-test-kv",
+              "name" : "fog-test-kv",
+              "type" : "Microsoft.KeyVault/vaults",
+              "location" : "westus",
+              "tags" : {},
+              "properties" : {
+                "sku" : {
+                  "family" : "A",
+                  "name" : "standard"
+                },
+                "tenantId" : "{AZURE_TENANT_ID}",
+                "accessPolicies" : [
+                  {
+                     "tenantId" : "{AZURE_TENANT_ID}",
+                     "objectId" : "{AZURE_OBJECT_ID}",
+                     "permissions" : {
+                       "keys" : ["all"],
+                       "secrets" : ["all"]
+                     }
+                  }
+                ],
+                "enabledForDeployment" : false,
+                "vaultUri" : "https://fog-test-kv.vault.azure.net/"
+              }
+            }]
+          }'
+          vaults_mapper = Azure::ARM::KeyVault::Models::VaultListResult.mapper
+          key_vault_client.deserialize(vaults_mapper, Fog::JSON.decode(response), 'result.body')
+        end
+      end
+    end
+  end
+end