fog-azure-rm 0.0.3 → 0.0.4

data/test/requests/network/test_list_subnets.rb

@@ -0,0 +1,31 @@
+require File.expand_path '../../test_helper', __dir__
+
+# Test class for List Subnets Request
+class TestListSubnets < Minitest::Test
+  def setup
+    @service = Fog::Network::AzureRM.new(credentials)
+    client = @service.instance_variable_get(:@network_client)
+    @subnets = client.subnets
+    @promise = Concurrent::Promise.execute do
+    end
+  end
+
+  def test_list_subnets_success
+    mocked_response = ApiStub::Requests::Network::Subnet.list_subnets_response
+    expected_response = Azure::ARM::Network::Models::SubnetListResult.serialize_object(mocked_response.body)
+    @promise.stub :value!, mocked_response do
+      @subnets.stub :list, @promise do
+        assert_equal @service.list_subnets('fog-test-rg', 'fog-test-virtual-network'), expected_response['value']
+      end
+    end
+  end
+
+  def test_list_subnets_failure
+    response = -> { raise MsRestAzure::AzureOperationError.new(nil, nil, 'error' => { 'message' => 'mocked exception' }) }
+    @promise.stub :value!, response do
+      @subnets.stub :list, @promise do
+        assert_raises(RuntimeError) { @service.list_subnets('fog-test-rg', 'fog-test-virtual-network') }
+      end
+    end
+  end
+end

data/test/requests/network/test_list_traffic_manager_profiles.rb

@@ -0,0 +1,29 @@
+require File.expand_path '../../test_helper', __dir__
+
+# Test class for List Traffic Manager Profiles
+class TestListTrafficManagerProfiles < Minitest::Test
+  def setup
+    @service = Fog::Network::AzureRM.new(credentials)
+    @token_provider = Fog::Credentials::AzureRM.instance_variable_get(:@token_provider)
+  end
+
+  def test_list_traffic_manager_profiles_success
+    response = ApiStub::Requests::Network::TrafficManagerProfile.list_traffic_manager_profiles_response
+    @token_provider.stub :get_authentication_header, 'Bearer <some-token>' do
+      RestClient.stub :get, response do
+        assert @service.list_traffic_manager_profiles('fog-test-rg')
+      end
+    end
+  end
+
+  def test_list_traffic_manager_profiles_failure
+    exception = RestClient::Exception.new
+    exception.instance_variable_set(:@response, '{"code": "ResourceNotFound", "message": "mocked exception message"}')
+    response = -> { raise exception }
+    @token_provider.stub :get_authentication_header, response do
+      assert_raises RuntimeError do
+        @service.list_traffic_manager_profiles('fog-test-rg')
+      end
+    end
+  end
+end

data/test/requests/network/test_list_virtual_networks.rb

@@ -0,0 +1,31 @@
+require File.expand_path '../../test_helper', __dir__
+
+# Test class for List Virtual Networks Request
+class TestListVirtualNetworks < Minitest::Test
+  def setup
+    @service = Fog::Network::AzureRM.new(credentials)
+    client = @service.instance_variable_get(:@network_client)
+    @virtual_networks = client.virtual_networks
+    @promise = Concurrent::Promise.execute do
+    end
+  end
+
+  def test_list_virtual_networks_success
+    mocked_response = ApiStub::Requests::Network::VirtualNetwork.list_virtual_networks_response
+    expected_response = Azure::ARM::Network::Models::VirtualNetworkListResult.serialize_object(mocked_response.body)
+    @promise.stub :value!, mocked_response do
+      @virtual_networks.stub :list, @promise do
+        assert_equal @service.list_virtual_networks('fog-test-rg'), expected_response['value']
+      end
+    end
+  end
+
+  def test_list_virtual_networks_failure
+    response = -> { raise MsRestAzure::AzureOperationError.new(nil, nil, 'error' => { 'message' => 'mocked exception' }) }
+    @promise.stub :value!, response do
+      @virtual_networks.stub :list, @promise do
+        assert_raises(RuntimeError) { @service.list_virtual_networks('fog-test-rg') }
+      end
+    end
+  end
+end

data/test/requests/network/test_remove_address_prefixes_from_virtual_network.rb

@@ -0,0 +1,45 @@
+require File.expand_path '../../test_helper', __dir__
+
+# Test class for Remove Address Prefixes in Virtual Network Request
+class TestRemoveddressPrefixesInVirtualNetwork < Minitest::Test
+  def setup
+    @service = Fog::Network::AzureRM.new(credentials)
+    client = @service.instance_variable_get(:@network_client)
+    @virtual_networks = client.virtual_networks
+  end
+
+  def test_remove_address_prefixes_in_virtual_network_success
+    promise_get = Concurrent::Promise.execute do
+    end
+    promise_create = Concurrent::Promise.execute do
+    end
+
+    mocked_response = ApiStub::Requests::Network::VirtualNetwork.create_virtual_network_response
+    expected_response = ApiStub::Requests::Network::VirtualNetwork.create_virtual_network_response
+    expected_response = Azure::ARM::Network::Models::VirtualNetwork.serialize_object(expected_response.body)
+    expected_response['properties']['addressSpace']['addressPrefixes'].delete('10.1.0.0/16')
+
+    promise_get.stub :value!, mocked_response do
+      promise_create.stub :value!, mocked_response do
+        @virtual_networks.stub :get, promise_get do
+          @virtual_networks.stub :create_or_update, promise_create do
+            assert_equal @service.remove_address_prefixes_from_virtual_network('fog-test-rg', 'fog-test-virtual-network', ['10.1.0.0/16']), expected_response
+          end
+        end
+      end
+    end
+  end
+
+  def test_remove_address_prefixes_in_virtual_network_failure
+    promise = Concurrent::Promise.execute do
+    end
+    response = -> { raise MsRestAzure::AzureOperationError.new(nil, nil, 'error' => { 'message' => 'mocked exception' }) }
+    promise.stub :value!, response do
+      @virtual_networks.stub :get, promise do
+        assert_raises RuntimeError do
+          @service.remove_address_prefixes_from_virtual_network('fog-test-rg', 'fog-test-virtual-network', ['10.1.0.0/16'])
+        end
+      end
+    end
+  end
+end

data/test/requests/network/test_remove_dns_servers_from_virtual_network.rb

@@ -0,0 +1,45 @@
+require File.expand_path '../../test_helper', __dir__
+
+# Test class for Remove Dns Servers in Virtual Network Request
+class TestRemoveDnsServersInVirtualNetwork < Minitest::Test
+  def setup
+    @service = Fog::Network::AzureRM.new(credentials)
+    client = @service.instance_variable_get(:@network_client)
+    @virtual_networks = client.virtual_networks
+  end
+
+  def test_remove_dns_servers_in_virtual_network_success
+    promise_get = Concurrent::Promise.execute do
+    end
+    promise_create = Concurrent::Promise.execute do
+    end
+
+    mocked_response = ApiStub::Requests::Network::VirtualNetwork.create_virtual_network_response
+    expected_response = ApiStub::Requests::Network::VirtualNetwork.create_virtual_network_response
+    expected_response = Azure::ARM::Network::Models::VirtualNetwork.serialize_object(expected_response.body)
+    expected_response['properties']['dhcpOptions']['dnsServers'].delete('10.1.0.5')
+
+    promise_get.stub :value!, mocked_response do
+      promise_create.stub :value!, mocked_response do
+        @virtual_networks.stub :get, promise_get do
+          @virtual_networks.stub :create_or_update, promise_create do
+            assert_equal @service.remove_dns_servers_from_virtual_network('fog-test-rg', 'fog-test-virtual-network', ['10.1.0.5']), expected_response
+          end
+        end
+      end
+    end
+  end
+
+  def test_remove_dns_servers_in_virtual_network_failure
+    promise = Concurrent::Promise.execute do
+    end
+    response = -> { raise MsRestAzure::AzureOperationError.new(nil, nil, 'error' => { 'message' => 'mocked exception' }) }
+    promise.stub :value!, response do
+      @virtual_networks.stub :get, promise do
+        assert_raises RuntimeError do
+          @service.remove_dns_servers_from_virtual_network('fog-test-rg', 'fog-test-virtual-network', ['10.1.0.5'])
+        end
+      end
+    end
+  end
+end

data/test/requests/network/test_remove_security_rule.rb

@@ -0,0 +1,49 @@
+require File.expand_path '../../test_helper', __dir__
+
+# Test class for Remove Security Rule
+class TestRemoveSecurityRule < Minitest::Test
+  def setup
+    @service = Fog::Network::AzureRM.new(credentials)
+    client = @service.instance_variable_get(:@network_client)
+    @network_security_groups = client.network_security_groups
+  end
+
+  def test_remove_security_rule_success
+    mocked_response = ApiStub::Requests::Network::NetworkSecurityGroup.add_security_rules_response
+    expected_response_json = ApiStub::Requests::Network::NetworkSecurityGroup.create_network_security_group_response
+    expected_response = Azure::ARM::Network::Models::NetworkSecurityGroup.serialize_object(expected_response_json.body)
+    get_promise = Concurrent::Promise.execute do
+    end
+    create_promise = Concurrent::Promise.execute do
+    end
+
+    get_promise .stub :value!, mocked_response do
+      @network_security_groups.stub :get, get_promise do
+        create_promise .stub :value!, expected_response_json do
+          @network_security_groups.stub :begin_create_or_update, create_promise do
+            assert_equal @service.remove_security_rule('fog-test-rg', 'fog-test-nsg', 'testRule'), expected_response
+          end
+        end
+      end
+    end
+  end
+
+  def test_remove_security_rule_failure
+    response = -> { raise MsRestAzure::AzureOperationError.new(nil, nil, 'error' => { 'message' => 'mocked exception' }) }
+    get_promise = Concurrent::Promise.execute do
+    end
+    create_promise = Concurrent::Promise.execute do
+    end
+    get_promise .stub :value!, response do
+      @network_security_groups.stub :get, get_promise do
+        create_promise.stub :value!, response do
+          @network_security_groups.stub :begin_create_or_update, create_promise do
+            assert_raises MsRestAzure::AzureOperationError do
+              @service.remove_security_rule('fog-test-rg', 'fog-test-nsg', 'testRule')
+            end
+          end
+        end
+      end
+    end
+  end
+end

data/test/requests/network/test_remove_subnets_from_virtual_network.rb

@@ -0,0 +1,45 @@
+require File.expand_path '../../test_helper', __dir__
+
+# Test class for Remove Subnets in Virtual Network Request
+class TestRemoveSubnetsInVirtualNetwork < Minitest::Test
+  def setup
+    @service = Fog::Network::AzureRM.new(credentials)
+    client = @service.instance_variable_get(:@network_client)
+    @virtual_networks = client.virtual_networks
+  end
+
+  def test_remove_subnets_in_virtual_network_success
+    promise_get = Concurrent::Promise.execute do
+    end
+    promise_create = Concurrent::Promise.execute do
+    end
+
+    mocked_response = ApiStub::Requests::Network::VirtualNetwork.create_virtual_network_response
+    expected_response = ApiStub::Requests::Network::VirtualNetwork.create_virtual_network_response
+    expected_response = Azure::ARM::Network::Models::VirtualNetwork.serialize_object(expected_response.body)
+    expected_response['properties']['subnets'].delete_at(0)
+
+    promise_get.stub :value!, mocked_response do
+      promise_create.stub :value!, mocked_response do
+        @virtual_networks.stub :get, promise_get do
+          @virtual_networks.stub :create_or_update, promise_create do
+            assert_equal @service.remove_subnets_from_virtual_network('fog-test-rg', 'fog-test-virtual-network', ['mysubnet1']), expected_response
+          end
+        end
+      end
+    end
+  end
+
+  def test_remove_subnets_in_virtual_network_failure
+    promise = Concurrent::Promise.execute do
+    end
+    response = -> { raise MsRestAzure::AzureOperationError.new(nil, nil, 'error' => { 'message' => 'mocked exception' }) }
+    promise.stub :value!, response do
+      @virtual_networks.stub :get, promise do
+        assert_raises RuntimeError do
+          @service.remove_subnets_from_virtual_network('fog-test-rg', 'fog-test-virtual-network', ['mysubnet1'])
+        end
+      end
+    end
+  end
+end

data/test/requests/resources/test_create_deployment.rb

@@ -0,0 +1,49 @@
+require File.expand_path '../../test_helper', __dir__
+
+# Test class for Create Deployment Request
+class TestCreateDeployment < Minitest::Test
+  def setup
+    @service = Fog::Resources::AzureRM.new(credentials)
+    client = @service.instance_variable_get(:@rmc)
+    @deployments = client.deployments
+    @resource_group = 'fog-test-rg'
+    @deployment_name = 'fog-test-deployment'
+    @template_link = 'https://test.com/template.json'
+    @parameters_link = 'https://test.com/parameters.json'
+  end
+
+  def test_create_deployment_success
+    validate_deployment_promise = Concurrent::Promise.execute do
+    end
+    create_deployment_promise = Concurrent::Promise.execute do
+    end
+    mocked_response = ApiStub::Requests::Resources::Deployment.create_deployment_response
+    expected_response = Azure::ARM::Resources::Models::DeploymentExtended.serialize_object(mocked_response.body)
+    validate_deployment_promise.stub :value!, true do
+      @deployments.stub :validate, validate_deployment_promise do
+        create_deployment_promise.stub :value!, mocked_response do
+          @deployments.stub :create_or_update, create_deployment_promise do
+            assert_equal @service.create_deployment(@resource_group, @deployment_name, @template_link, @parameters_link), expected_response
+          end
+        end
+      end
+    end
+  end
+
+  def test_create_deployment_failure
+    validate_deployment_promise = Concurrent::Promise.execute do
+    end
+    create_deployment_promise = Concurrent::Promise.execute do
+    end
+    response = -> { raise MsRestAzure::AzureOperationError.new(nil, nil, 'error' => { 'message' => 'mocked exception' }) }
+    validate_deployment_promise.stub :value!, true do
+      @deployments.stub :validate, validate_deployment_promise do
+        create_deployment_promise.stub :value!, response do
+          @deployments.stub :create_or_update, create_deployment_promise do
+            assert_raises(RuntimeError) { @service.create_deployment(@resource_group, @deployment_name, @template_link, @parameters_link) }
+          end
+        end
+      end
+    end
+  end
+end

data/test/requests/resources/test_create_resource_group.rb

@@ -0,0 +1,31 @@
+require File.expand_path '../../test_helper', __dir__
+
+# Test class for Create Resource Group Request
+class TestCreateResourceGroup < Minitest::Test
+  def setup
+    @service = Fog::Resources::AzureRM.new(credentials)
+    client = @service.instance_variable_get(:@rmc)
+    @resource_groups = client.resource_groups
+    @promise = Concurrent::Promise.execute do
+    end
+  end
+
+  def test_create_resource_group_success
+    mocked_response = ApiStub::Requests::Resources::ResourceGroup.create_resource_group_response
+    expected_response = Azure::ARM::Resources::Models::ResourceGroup.serialize_object(mocked_response.body)
+    @promise.stub :value!, mocked_response do
+      @resource_groups.stub :create_or_update, @promise do
+        assert_equal @service.create_resource_group('fog-test-rg', 'west us'), expected_response
+      end
+    end
+  end
+
+  def test_create_resource_group_failure
+    response = -> { fail MsRestAzure::AzureOperationError.new(nil, nil, 'error' => { 'message' => 'mocked exception' }) }
+    @promise.stub :value!, response do
+      @resource_groups.stub :create_or_update, @promise do
+        assert_raises(RuntimeError) { @service.create_resource_group('fog-test-rg', 'west us') }
+      end
+    end
+  end
+end

data/test/requests/resources/test_delete_deployment.rb

@@ -0,0 +1,31 @@
+require File.expand_path '../../test_helper', __dir__
+
+# Test class for Delete Deployment Request
+class TestDeleteDeployment < Minitest::Test
+  def setup
+    @service = Fog::Resources::AzureRM.new(credentials)
+    client = @service.instance_variable_get(:@rmc)
+    @deployments = client.deployments
+    @promise = Concurrent::Promise.execute do
+    end
+    @resource_group = 'fog-test-rg'
+    @deployment_name = 'fog-test-deployment'
+  end
+
+  def test_delete_deployment_success
+    @promise.stub :value!, true do
+      @deployments.stub :delete, @promise do
+        assert @service.delete_deployment(@resource_group, @deployment_name)
+      end
+    end
+  end
+
+  def test_list_deployment_failure
+    response = -> { raise MsRestAzure::AzureOperationError.new(nil, nil, 'error' => { 'message' => 'mocked exception' }) }
+    @promise.stub :value!, response do
+      @deployments.stub :delete, @promise do
+        assert_raises(RuntimeError) { @service.delete_deployment(@resource_group, @deployment_name) }
+      end
+    end
+  end
+end

data/test/requests/resources/test_delete_resource_group.rb

@@ -0,0 +1,30 @@
+require File.expand_path '../../test_helper', __dir__
+
+# Test class for Delete Resource Group Request
+class TestDeleteResourceGroup < Minitest::Test
+  def setup
+    @service = Fog::Resources::AzureRM.new(credentials)
+    client = @service.instance_variable_get(:@rmc)
+    @resource_groups = client.resource_groups
+    @promise = Concurrent::Promise.execute do
+    end
+  end
+
+  def test_delete_resource_group_success
+    response = ApiStub::Requests::Resources::ResourceGroup.create_resource_group_response
+    @promise.stub :value!, response do
+      @resource_groups.stub :delete, @promise do
+        assert @service.delete_resource_group('fog-test-rg')
+      end
+    end
+  end
+
+  def test_delete_resource_group_failure
+    response = -> { fail MsRestAzure::AzureOperationError.new(nil, nil, 'error' => { 'message' => 'mocked exception' }) }
+    @promise.stub :value!, response do
+      @resource_groups.stub :delete, @promise do
+        assert_raises(RuntimeError) { @service.delete_resource_group('fog-test-rg') }
+      end
+    end
+  end
+end

data/test/requests/resources/test_delete_resource_tag.rb

@@ -0,0 +1,51 @@
+require File.expand_path '../../test_helper', __dir__
+
+# Test class for Delete Resource Tag Request
+class TestDeleteResourceTag < Minitest::Test
+  def setup
+    @service = Fog::Resources::AzureRM.new(credentials)
+    client = @service.instance_variable_get(:@rmc)
+    @resources = client.resources
+  end
+
+  def test_delete_resource_tag_success
+    response = ApiStub::Requests::Resources::AzureResource.azure_resource_response
+    promise_get = Concurrent::Promise.execute do
+    end
+    promise_update = Concurrent::Promise.execute do
+    end
+    promise_get.stub :value!, response do
+      @resources.stub :get, promise_get do
+        promise_update.stub :value!, response do
+          @resources.stub :create_or_update, promise_update do
+            resource_id = '/subscriptions/########-####-####-####-############/resourceGroups/{RESOURCE-GROUP}/providers/Microsoft.Network/{PROVIDER-NAME}/{RESOURCE-NAME}'
+            assert @service.delete_resource_tag(resource_id, 'tag_name', 'tag_value')
+          end
+        end
+      end
+    end
+  end
+
+  def test_delete_resource_tag_failure
+    response = -> { raise MsRestAzure::AzureOperationError.new(nil, nil, 'error' => { 'message' => 'mocked exception' }) }
+    promise_get = Concurrent::Promise.execute do
+    end
+    promise_update = Concurrent::Promise.execute do
+    end
+    promise_get.stub :value!, response do
+      @resources.stub :get, promise_get do
+        promise_update.stub :value!, response do
+          @resources.stub :create_or_update, promise_update do
+            resource_id = '/subscriptions/########-####-####-####-############/resourceGroups/{RESOURCE-GROUP}/providers/Microsoft.Network/{PROVIDER-NAME}/{RESOURCE-NAME}'
+            assert_raises(RuntimeError) { @service.delete_resource_tag(resource_id, 'tag_name', 'tag_value') }
+          end
+        end
+      end
+    end
+  end
+
+  def test_invalid_resource_id_exeception
+    resource_id = 'Invalid-Resource-ID'
+    assert_raises(RuntimeError) { @service.delete_resource_tag(resource_id, 'tag_name', 'tag_value') }
+  end
+end