fog-azure-rm 0.0.4 → 0.0.5

data/lib/fog/azurerm/storage.rb

@@ -54,7 +54,7 @@ module Fog
         end
       end
 
-      # This class provides the actual implemention for service calls.
+      # This class provides the actual implementation for service calls.
       class Real
         def initialize(options)
           begin

data/lib/fog/azurerm/traffic_manager.rb

@@ -0,0 +1,56 @@
+require 'fog/azurerm/core'
+module Fog
+  module TrafficManager
+    # This class registers models, requests and collections
+    class AzureRM < Fog::Service
+      requires :tenant_id
+      requires :client_id
+      requires :client_secret
+      requires :subscription_id
+
+      request_path 'fog/azurerm/requests/traffic_manager'
+      request :create_traffic_manager_profile
+      request :delete_traffic_manager_profile
+      request :get_traffic_manager_profile
+      request :list_traffic_manager_profiles
+
+      request :create_traffic_manager_endpoint
+      request :delete_traffic_manager_endpoint
+      request :get_traffic_manager_endpoint
+
+      model_path 'fog/azurerm/models/traffic_manager'
+      model :traffic_manager_profile
+      collection :traffic_manager_profiles
+
+      model :traffic_manager_end_point
+      collection :traffic_manager_end_points
+
+      # This class provides the actual implementation for service calls.
+      class Real
+        def initialize(options)
+          begin
+            require 'azure_mgmt_traffic_manager'
+          rescue LoadError => e
+            retry if require('rubygems')
+            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)
+          @traffic_mgmt_client.subscription_id = options[:subscription_id]
+        end
+      end
+      # This class provides the mock implementation for unit tests.
+      class Mock
+        def initialize(_options = {})
+          begin
+            require 'azure_mgmt_traffic_manager'
+          rescue LoadError => e
+            retry if require('rubygems')
+            raise e.message
+          end
+        end
+      end
+    end
+  end
+end

data/lib/fog/azurerm/utilities/general.rb

@@ -8,6 +8,26 @@ def get_virtual_network_from_id(subnet_id)
   subnet_id.split('/')[8]
 end
 
+# Extract Endpoint type from (String)
+def get_end_point_type(endpoint_type)
+  endpoint_type.split('/')[2]
+end
+
+# Extract Traffic Manager Profile Name from Endpoint id(String)
+def get_traffic_manager_profile_name_from_endpoint_id(endpoint_id)
+  endpoint_id.split('/')[8]
+end
+
+def get_record_type(type)
+  type.split('/').last
+end
+
+def raise_azure_exception(exception, msg)
+  message = "Exception in #{msg} #{exception.body['error']['message'] unless exception.body['error']['message'].nil?} Type: #{exception.class} \n "
+  Fog::Logger.debug exception.backtrace
+  raise message
+end
+
 # Make sure if input_params(Hash) contains all keys present in required_params(Array)
 def validate_params(required_params, input_params)
   missing_params = required_params.select { |param| param unless input_params.key?(param) }
@@ -16,3 +36,11 @@ def validate_params(required_params, input_params)
     raise(ArgumentError, "Missing Parameters: #{missing_params.join(', ')} required for this operation")
   end
 end
+
+def get_resource_from_resource_id(resource_id, position)
+  data = resource_id.split('/') unless resource_id.nil?
+
+  raise 'Invalid Resource ID' if data.count < 9 && data.count != 5
+
+  data[position]
+end

data/lib/fog/azurerm/version.rb

@@ -1,5 +1,5 @@
 module Fog
   module AzureRM
-    VERSION = '0.0.4'.freeze
+    VERSION = '0.0.5'.freeze
   end
 end

data/test/api_stub.rb

@@ -4,7 +4,6 @@ require File.expand_path 'api_stub/models/resources/resource_group', __dir__
 require File.expand_path 'api_stub/models/resources/deployment', __dir__
 require File.expand_path 'api_stub/models/resources/resource', __dir__
 require File.expand_path 'api_stub/models/storage/storageaccount', __dir__
-require File.expand_path 'api_stub/models/storage/data_disk', __dir__
 require File.expand_path 'api_stub/models/storage/blob', __dir__
 require File.expand_path 'api_stub/models/storage/container', __dir__
 require File.expand_path 'api_stub/models/network/public_ip', __dir__
@@ -13,9 +12,13 @@ require File.expand_path 'api_stub/models/network/virtual_network', __dir__
 require File.expand_path 'api_stub/models/network/network_interface', __dir__
 require File.expand_path 'api_stub/models/network/load_balancer', __dir__
 require File.expand_path 'api_stub/models/network/network_security_group', __dir__
-require File.expand_path 'api_stub/models/network/application_gateway', __dir__
-require File.expand_path 'api_stub/models/network/traffic_manager_end_point', __dir__
-require File.expand_path 'api_stub/models/network/traffic_manager_profile', __dir__
+require File.expand_path 'api_stub/models/application_gateway/gateway', __dir__
+require File.expand_path 'api_stub/models/network/virtual_network_gateway', __dir__
+require File.expand_path 'api_stub/models/network/express_route_circuit', __dir__
+require File.expand_path 'api_stub/models/network/express_route_circuit_peering', __dir__
+require File.expand_path 'api_stub/models/network/express_route_service_provider', __dir__
+require File.expand_path 'api_stub/models/traffic_manager/traffic_manager_end_point', __dir__
+require File.expand_path 'api_stub/models/traffic_manager/traffic_manager_profile', __dir__
 require File.expand_path 'api_stub/models/dns/zone', __dir__
 require File.expand_path 'api_stub/models/dns/record_set', __dir__
 
@@ -33,9 +36,13 @@ require File.expand_path 'api_stub/requests/network/virtual_network', __dir__
 require File.expand_path 'api_stub/requests/network/network_interface', __dir__
 require File.expand_path 'api_stub/requests/network/load_balancer', __dir__
 require File.expand_path 'api_stub/requests/network/network_security_group', __dir__
-require File.expand_path 'api_stub/requests/network/application_gateway', __dir__
-require File.expand_path 'api_stub/requests/network/traffic_manager_endpoint', __dir__
-require File.expand_path 'api_stub/requests/network/traffic_manager_profile', __dir__
+require File.expand_path 'api_stub/requests/application_gateway/gateway', __dir__
+require File.expand_path 'api_stub/requests/network/virtual_network_gateway', __dir__
+require File.expand_path 'api_stub/requests/network/express_route_circuit', __dir__
+require File.expand_path 'api_stub/requests/network/express_route_circuit_peering', __dir__
+require File.expand_path 'api_stub/requests/network/express_route_service_provider', __dir__
+require File.expand_path 'api_stub/requests/traffic_manager/traffic_manager_endpoint', __dir__
+require File.expand_path 'api_stub/requests/traffic_manager/traffic_manager_profile', __dir__
 require File.expand_path 'api_stub/requests/dns/zone', __dir__
 require File.expand_path 'api_stub/requests/dns/record_set', __dir__
 

data/test/api_stub/models/{network/application_gateway.rb → application_gateway/gateway.rb}

@@ -1,8 +1,8 @@
 module ApiStub
   module Models
-    module Network
-      class ApplicationGateway
-        def self.create_application_gateway_response
+    module ApplicationGateway
+      class Gateway
+        def self.create_application_gateway_response(gateway_client)
           gateway = '{
               "id": "/subscriptions/{guid}/resourceGroups/{resourceGroupName}/providers/Microsoft.Network/applicationGateways/ag-demoagplat",
               "name": "gateway",
@@ -59,9 +59,9 @@ module ApiStub
                     "protocol": "<probeProtocol>",
                     "host": "<probeHost>",
                     "path": "<probePath>",
-                    "interval": "<probeIntervalInSec>",
-                    "timeout": "<probeTimeoutInSec>",
-                    "unhealthyThreshold": "<probeUnhealthyThreshold>"
+                    "interval": "30",
+                    "timeout": "20",
+                    "unhealthyThreshold": "20"
                   }
                 }],
                 "backendAddressPools": [{
@@ -127,7 +127,8 @@ module ApiStub
                 "provisioningState": "Succeeded"
               }
             }'
-          JSON.parse(gateway)
+          gateway_mapper = Azure::ARM::Network::Models::ApplicationGateway.mapper
+          gateway_client.deserialize(gateway_mapper, JSON.load(gateway), 'result.body')
         end
       end
     end

data/test/api_stub/models/compute/availability_set.rb

@@ -3,17 +3,46 @@ module ApiStub
     module Compute
       # Mock class for Availability Set Model
       class AvailabilitySet
-        def self.create_availability_set_response
-          {
+        def self.create_availability_set_response(sdk_compute_client)
+          avail_set = {
             'id' => '/subscriptions/########-####-####-####-############/resourceGroups/fog-test-rg/providers/Microsoft.Compute/availabilitySets/fog-test-availability-set',
             'name' => 'fog-test-availability-set',
             'type' => 'Microsoft.Compute/availabilitySets',
             'location' => 'westus',
-            'properties' => {
-              'platformUpdateDomainCount' => 5,
-              'platformFaultDomainCount' => 3
-            }
+            'platformUpdateDomainCount' => UPDATE_DOMAIN_COUNT,
+            'platformFaultDomainCount' => FAULT_DOMAIN_COUNT
+
           }
+          result_mapper = Azure::ARM::Compute::Models::AvailabilitySet.mapper
+          sdk_compute_client.deserialize(result_mapper, avail_set, 'result.body')
+        end
+
+        def self.list_availability_set_response(sdk_compute_client)
+          avail_set = {
+              'id' => '/subscriptions/########-####-####-####-############/resourceGroups/fog-test-rg/providers/Microsoft.Compute/availabilitySets/fog-test-availability-set',
+              'name' => 'fog-test-availability-set',
+              'type' => 'Microsoft.Compute/availabilitySets',
+              'location' => 'westus',
+              'platformUpdateDomainCount' => UPDATE_DOMAIN_COUNT,
+              'platformFaultDomainCount' => FAULT_DOMAIN_COUNT
+
+          }
+          result_mapper = Azure::ARM::Compute::Models::AvailabilitySet.mapper
+          sdk_compute_client.deserialize(result_mapper, avail_set, 'result.body')
+        end
+
+        def self.get_availability_set_response(sdk_compute_client)
+          avail_set = {
+              'id' => '/subscriptions/########-####-####-####-############/resourceGroups/fog-test-rg/providers/Microsoft.Compute/availabilitySets/fog-test-availability-set',
+              'name' => 'fog-test-availability-set',
+              'type' => 'Microsoft.Compute/availabilitySets',
+              'location' => 'westus',
+              'platformUpdateDomainCount' => UPDATE_DOMAIN_COUNT,
+              'platformFaultDomainCount' => FAULT_DOMAIN_COUNT
+
+          }
+          result_mapper = Azure::ARM::Compute::Models::AvailabilitySet.mapper
+          sdk_compute_client.deserialize(result_mapper, avail_set, 'result.body')
         end
       end
     end

data/test/api_stub/models/compute/server.rb

@@ -3,8 +3,8 @@ module ApiStub
     module Compute
       # Mock class for Server Model
       class Server
-        def self.create_linux_virtual_machine_response
-          {
+        def self.create_linux_virtual_machine_response(compute_client)
+          body = {
             'id' => '/subscriptions/########-####-####-####-############/resourceGroups/fog-test-rg/providers/Microsoft.Compute/virtualMachines/fog-test-server',
             'name' => 'fog-test-server',
             'location' => 'West US',
@@ -52,10 +52,12 @@ module ApiStub
               }
             }
           }
+          vm_mapper = Azure::ARM::Compute::Models::VirtualMachine.mapper
+          compute_client.deserialize(vm_mapper, body, 'result.body')
         end
 
-        def self.create_windows_virtual_machine_response
-          {
+        def self.create_windows_virtual_machine_response(compute_client)
+          body = {
             'id' => '/subscriptions/########-####-####-####-############/resourceGroups/fog-test-rg/providers/Microsoft.Compute/virtualMachines/fog-test-server',
             'name' => 'fog-test-server',
             'location' => 'West US',
@@ -88,12 +90,6 @@ module ApiStub
                       'certificateUrl' => 'certificateUrl'
                     }]
                   },
-                  'additionalUnattendContent' => {
-                    'pass' => 'oobesystem',
-                    'component' => 'Microsoft-Windows-Shell-Setup',
-                    'settingName' => 'FirstLogonCommands|AutoLogon',
-                    'content' => '<XML unattend content>'
-                  },
                   'enableAutomaticUpdates' => true
                 },
                 'secrets' => []
@@ -107,9 +103,11 @@ module ApiStub
               }
             }
           }
+          vm_mapper = Azure::ARM::Compute::Models::VirtualMachine.mapper
+          compute_client.deserialize(vm_mapper, body, 'result.body')
         end
 
-        def self.list_available_sizes_for_virtual_machine_response
+        def self.list_available_sizes_for_virtual_machine_response(compute_client)
           body = '{
             "value": [
               {
@@ -122,13 +120,12 @@ module ApiStub
               }
             ]
           }'
-          result = MsRestAzure::AzureOperationResponse.new(MsRest::HttpOperationRequest.new('', '', ''), Faraday::Response.new)
-          result.body = Azure::ARM::Compute::Models::VirtualMachineSizeListResult.deserialize_object(JSON.load(body))
-          result.body.value
+          vm_mapper = Azure::ARM::Compute::Models::VirtualMachineSizeListResult.mapper
+          compute_client.deserialize(vm_mapper, JSON.load(body), 'result.body').value
         end
 
-        def self.attach_data_disk_response
-          {
+        def self.attach_data_disk_response(compute_client)
+          body = {
             'id' => '/subscriptions/########-####-####-####-############/resourceGroups/fog-test-rg/providers/Microsoft.Compute/virtualMachines/fog-test-server',
             'name' => 'fog-test-server',
             'location' => 'West US',
@@ -184,6 +181,8 @@ module ApiStub
               }
             }
           }
+          vm_mapper = Azure::ARM::Compute::Models::VirtualMachine.mapper
+          compute_client.deserialize(vm_mapper, body, 'result.body')
         end
       end
     end

data/test/api_stub/models/dns/record_set.rb

@@ -53,6 +53,26 @@ module ApiStub
               }
           }
         end
+
+        def self.response_for_cname
+          {
+            'id' => '/subscriptions/########-####-####-####-############/resourceGroups/EdgeMonitoring2/providers/Microsoft.Network/dnszones/edgemonitoring2.com./CNAME/www',
+            'location' => 'global',
+            'name' => 'www',
+            'tags' => {},
+            'type' => 'Microsoft.Network/dnszones/CNAME',
+            'etag' => '5b83020b-b59c-44be-8f19-a052ebe80fe7',
+            'properties' => {
+              'metadata' => 'nil',
+              'fqdn' => 'fog-test-record_set.fog-test-zone.com.',
+              'TTL' => '60',
+              'CNAMERecord' =>
+                {
+                  'cname' => 'test.fog.com'
+                }
+            }
+          }
+        end
       end
     end
   end

data/test/api_stub/models/network/express_route_circuit.rb

@@ -0,0 +1,47 @@
+module ApiStub
+  module Models
+    module Network
+      # Mock class for Express Route Circuit Model
+      class ExpressRouteCircuit
+        def self.create_express_route_circuit_response(network_client)
+          circuit = '{
+            "name": "<circuit name>",
+            "id": "/subscriptions/{guid}/resourceGroups/{resourceGroupName}/providers/Microsoft.Network/expressRouteCircuits/testCircuit",
+            "location": "<location>",
+            "tags": {
+              "key1": "value1",
+              "key2": "value2"
+            },
+            "sku": {
+              "name": "Standard_MeteredData",
+              "tier": "Standard",
+              "family": "MeteredData"
+            },
+            "properties": {
+              "serviceProviderProperties": {
+                  "serviceProviderName": "serviceProviderName",
+                  "peeringLocation": "<peering location>",
+                  "bandwidthInMbps": 100
+              },
+              "peerings": [
+                {
+                  "name": "AzurePublicPeering",
+                  "id": "/subscriptions/{guid}/resourceGroups/{resourceGroupName}/providers/Microsoft.Network/expressRouteCircuits/{circuitName}/peerings/{peeringName}",
+                  "properties": {
+                    "peeringType": "AzurePublicPeering",
+                    "peerASN": 100,
+                    "PrimaryPeerAddressPrefix": "192.168.1.0/30",
+                    "SecondaryPeerAddressPrefix": "192.168.2.0/30",
+                    "vlanId": 200
+                  }
+                }
+              ]
+            }
+          }'
+          express_route_circuit_mapper = Azure::ARM::Network::Models::ExpressRouteCircuit.mapper
+          network_client.deserialize(express_route_circuit_mapper, JSON.load(circuit), 'result.body')
+        end
+      end
+    end
+  end
+end

data/test/api_stub/models/network/express_route_circuit_peering.rb

@@ -0,0 +1,33 @@
+module ApiStub
+  module Models
+    module Network
+      # Mock class for Express Route Circuit Peering Model
+      class ExpressRouteCircuitPeering
+        def self.create_express_route_circuit_peering_response(network_client)
+          peering = '{
+                        "name": "MicrosoftPeering",
+                        "id": "/subscriptions/{guid}/resourceGroups/{resourceGroupName}/providers/Microsoft.Network/expressRouteCircuits/{circuitName}/peerings/{peeringName}",
+                        "properties": {
+                            "peeringType": "MicrosoftPeering",
+                            "peerASN": 100,
+                            "primaryPeerAddressPrefix": "192.168.1.0/30",
+                            "secondaryPeerAddressPrefix": "192.168.2.0/30",
+                            "vlanId": 200,
+                            "microsoftPeeringConfig": {
+                              "advertisedpublicprefixes": [
+                                "11.2.3.4/30",
+                                "12.2.3.4/30"
+                              ],
+                              "advertisedPublicPrefixState": "NotConfigured ",
+                              "customerAsn": 200,
+                              "routingRegistryName": "<name>"
+                            }
+                          }
+                  }'
+          express_route_circuit_peering_mapper = Azure::ARM::Network::Models::ExpressRouteCircuitPeering.mapper
+          network_client.deserialize(express_route_circuit_peering_mapper, JSON.load(peering), 'result.body')
+        end
+      end
+    end
+  end
+end

data/test/api_stub/models/network/express_route_service_provider.rb

@@ -0,0 +1,29 @@
+module ApiStub
+  module Models
+    module Network
+      # Mock class for Express Route Service Provider Model
+      class ExpressRouteServiceProvider
+        def self.list_express_route_service_provider_response(network_client)
+          service_provider = '{
+              "value": [
+              {
+                "name": "providername",
+                "peeringLocations": [
+                  "location1",
+                  "location2"
+                ],
+                "bandwidthsOffered": [
+                  {
+                    "offerName": "100Mbps",
+                    "valueInMbps": 100
+                  }
+                ]
+              }
+            ]}'
+          express_route_servcie_provider_mapper = Azure::ARM::Network::Models::ExpressRouteServiceProviderListResult.mapper
+          network_client.deserialize(express_route_servcie_provider_mapper, JSON.load(service_provider), 'result.body').value
+        end
+      end
+    end
+  end
+end