fog-azure-rm 0.0.3 → 0.0.4

data/test/requests/dns/test_check_for_zone.rb

@@ -0,0 +1,52 @@
+require File.expand_path '../../test_helper', __dir__
+
+# Test class for Check for Zone Request
+class TestCheckForZone < Minitest::Test
+  def setup
+    @service = Fog::DNS::AzureRM.new(credentials)
+    @zones = @service.zones
+    @token_provider = Fog::Credentials::AzureRM.instance_variable_get(:@token_provider)
+  end
+
+  def test_check_for_zone_success
+    response = ApiStub::Requests::DNS::Zone.zone_response
+    @token_provider.stub :get_authentication_header, 'Bearer <some-token>' do
+      RestClient.stub :get, response do
+        assert @service.check_for_zone('fog-test-rg', 'fog-test-zone')
+      end
+    end
+  end
+
+  def test_check_for_zone_success_if_not_found
+    response = -> { fail RestClient::Exception.new('{"error": {"code": "ResourceNotFound"}}') }
+    @token_provider.stub :get_authentication_header, response do
+      assert !@service.check_for_zone('fog-test-rg', 'fog-test-zone')
+    end
+  end
+
+  def test_check_for_zone_success_if_not_found_failure
+    response = -> { fail RestClient::Exception.new('{"error": {"code": "InvalidCode"}}') }
+    @token_provider.stub :get_authentication_header, response do
+      assert_raises Exception do
+        @service.check_for_zone('fog-test-rg', 'fog-test-zone')
+      end
+    end
+  end
+
+  def test_check_for_zone_failure
+    @token_provider.stub :get_authentication_header, 'Bearer <some-token>' do
+      assert_raises ArgumentError do
+        @service.check_for_zone('fog-test-rg')
+      end
+    end
+  end
+
+  def test_check_for_zone_exception
+    response = -> { fail Exception.new('mocked exception') }
+    @token_provider.stub :get_authentication_header, response do
+      assert_raises Exception do
+        @service.check_for_zone('fog-test-rg', 'fog-test-zone')
+      end
+    end
+  end
+end

data/test/requests/dns/test_create_record_set.rb

@@ -0,0 +1,45 @@
+require File.expand_path '../../test_helper', __dir__
+
+# Test class for Create Record Set Request
+class TestCreateRecordSet < Minitest::Test
+  def setup
+    @service = Fog::DNS::AzureRM.new(credentials)
+    @record_sets = @service.record_sets
+    @token_provider = Fog::Credentials::AzureRM.instance_variable_get(:@token_provider)
+  end
+
+  def test_create_record_set_a_type
+    response = ApiStub::Requests::DNS::RecordSet.rest_client_put_method_for_record_set_a_type_response
+    @token_provider.stub :get_authentication_header, 'Bearer <some-token>' do
+      RestClient.stub :put, response do
+        assert_equal @service.create_record_set('fog-test-rg', 'fog-test-record-set', 'fog-test-zone', ['1.2.3.4', '1.2.3.3'], 'A', 60), JSON.parse(response)
+      end
+    end
+  end
+
+  def test_create_record_set_cname_type
+    response = ApiStub::Requests::DNS::RecordSet.rest_client_put_method_for_record_set_cname_type_response
+    @token_provider.stub :get_authentication_header, 'Bearer <some-token>' do
+      RestClient.stub :put, response do
+        assert_equal @service.create_record_set('fog-test-rg', 'fog-test-record-set', 'fog-test-zone', ['test.fog.com'], 'CNAME', 60), JSON.parse(response)
+      end
+    end
+  end
+
+  def test_create_record_set_failure
+    @token_provider.stub :get_authentication_header, 'Bearer <some-token>' do
+      assert_raises ArgumentError do
+        @service.create_record_set('fog-test-rg', 'fog-test-record-set', 'fog-test-zone', ['test.fog.com'], 'CNAME')
+      end
+    end
+  end
+
+  def test_create_record_set_exception
+    response = -> { fail Exception.new('mocked exception') }
+    @token_provider.stub :get_authentication_header, response do
+      assert_raises Exception do
+        @service.create_record_set('fog-test-rg', 'fog-test-record-set', 'fog-test-zone', ['test.fog.com'], 'CNAME', 60)
+      end
+    end
+  end
+end

data/test/requests/dns/test_create_zone.rb

@@ -0,0 +1,36 @@
+require File.expand_path '../../test_helper', __dir__
+
+# Test class for Create Zone Request
+class TestCreateZone < Minitest::Test
+  def setup
+    @service = Fog::DNS::AzureRM.new(credentials)
+    @zones = @service.zones
+    @token_provider = Fog::Credentials::AzureRM.instance_variable_get(:@token_provider)
+  end
+
+  def test_create_zone_success
+    response = ApiStub::Requests::DNS::Zone.rest_client_put_method_for_zone_resonse
+    @token_provider.stub :get_authentication_header, 'Bearer <some-token>' do
+      RestClient.stub :put, response do
+        assert_equal @service.create_zone('fog-test-rg', 'fog-test-zone'), JSON.parse(response)
+      end
+    end
+  end
+
+  def test_create_zone_failure
+    @token_provider.stub :get_authentication_header, 'Bearer <some-token>' do
+      assert_raises ArgumentError do
+        @service.create_zone('fog-test-zone')
+      end
+    end
+  end
+
+  def test_create_zone_exception
+    response = -> { fail RestClient::Exception.new("'body': {'error': {'code': 'ResourceNotFound', 'message': 'mocked exception message'}}") }
+    @token_provider.stub :get_authentication_header, response do
+      assert_raises Exception do
+        @service.create_zone('fog-test-rg', 'fog-test-zone')
+      end
+    end
+  end
+end

data/test/requests/dns/test_delete_record_set.rb

@@ -0,0 +1,35 @@
+require File.expand_path '../../test_helper', __dir__
+
+# Test class for Delete Record Set Request
+class TestDeleteRecordSet < Minitest::Test
+  def setup
+    @service = Fog::DNS::AzureRM.new(credentials)
+    @record_sets = @service.record_sets
+    @token_provider = Fog::Credentials::AzureRM.instance_variable_get(:@token_provider)
+  end
+
+  def test_delete_record_set_success
+    @token_provider.stub :get_authentication_header, 'Bearer <some-token>' do
+      RestClient.stub :delete, true do
+        assert @service.delete_record_set('fog-test-rg', 'fog-test-record-set', 'fog-test-zone', 'A')
+      end
+    end
+  end
+
+  def test_delte_record_set_failure
+    @token_provider.stub :get_authentication_header, 'Bearer <some-token>' do
+      assert_raises ArgumentError do
+        @service.delete_record_set('fog-test-rg', 'fog-test-record-set', 'fog-test-zone')
+      end
+    end
+  end
+
+  def test_create_record_set_exception
+    response = -> { fail Exception.new('mocked exception') }
+    @token_provider.stub :get_authentication_header, response do
+      assert_raises Exception do
+        assert @service.delete_record_set('fog-test-rg', 'fog-test-record-set', 'fog-test-zone', 'A')
+      end
+    end
+  end
+end

data/test/requests/dns/test_delete_zone.rb

@@ -0,0 +1,35 @@
+require File.expand_path '../../test_helper', __dir__
+
+# Test class for Dellete Zone Request
+class TestDeleteZone < Minitest::Test
+  def setup
+    @service = Fog::DNS::AzureRM.new(credentials)
+    @zones = @service.zones
+    @token_provider = Fog::Credentials::AzureRM.instance_variable_get(:@token_provider)
+  end
+
+  def test_delete_zone_success
+    @token_provider.stub :get_authentication_header, 'Bearer <some-token>' do
+      RestClient.stub :delete, true do
+        assert @service.delete_zone('fog-test-rg', 'fog-test-zone')
+      end
+    end
+  end
+
+  def test_delte_zone_failure
+    @token_provider.stub :get_authentication_header, 'Bearer <some-token>' do
+      assert_raises ArgumentError do
+        @service.delete_zone('fog-test-zone')
+      end
+    end
+  end
+
+  def test_create_zone_exception
+    response = -> { fail Exception.new('mocked exception') }
+    @token_provider.stub :get_authentication_header, response do
+      assert_raises Exception do
+        assert @service.delete_zone('fog-test-rg', 'fog-test-zone')
+      end
+    end
+  end
+end

data/test/requests/dns/test_get_records_from_record_set.rb

@@ -0,0 +1,55 @@
+require File.expand_path '../../test_helper', __dir__
+
+# Test class for Get Records From Record Set Request
+class TestGetRecordsFromRecordSet < Minitest::Test
+  def setup
+    @service = Fog::DNS::AzureRM.new(credentials)
+    @record_sets = @service.record_sets
+    @token_provider = Fog::Credentials::AzureRM.instance_variable_get(:@token_provider)
+  end
+
+  def test_get_records_from_record_set_of_a_type_success
+    response = ApiStub::Requests::DNS::RecordSet.record_set_response_for_a_type_response
+    @token_provider.stub :get_authentication_header, 'Bearer <some-token>' do
+      RestClient.stub :get, response do
+        assert_equal @service.get_records_from_record_set('fog-test-rg', 'fog-test-record-set', 'fog-test-zone', 'A'), ['4.3.2.1', '5.3.2.1']
+      end
+    end
+  end
+
+  def test_get_records_from_record_set_of_cname_type_success
+    response = ApiStub::Requests::DNS::RecordSet.record_set_response_for_cname_type
+    @token_provider.stub :get_authentication_header, 'Bearer <some-token>' do
+      RestClient.stub :get, response do
+        assert_equal @service.get_records_from_record_set('fog-test-rg', 'fog-test-record-set', 'fog-test-zone', 'CNAME'), ['test.fog.com']
+      end
+    end
+  end
+
+  def test_get_records_from_record_set_failure
+    @token_provider.stub :get_authentication_header, 'Bearer <some-token>' do
+      assert_raises ArgumentError do
+        @service.get_records_from_record_set('fog-test-record-set')
+      end
+    end
+  end
+
+  def test_get_records_from_record_set_exception
+    response = -> { fail Exception.new('mocked exception') }
+    @token_provider.stub :get_authentication_header, response do
+      assert_raises Exception do
+        @service.get_records_from_record_set('fog-test-rg', 'fog-test-record-set', 'fog-test-zone', 'A')
+      end
+    end
+  end
+
+  def test_get_records_from_record_set_parsing_exception
+    @token_provider.stub :get_authentication_header, 'Bearer <some-token>' do
+      RestClient.stub :get, '{invalid json}' do
+        assert_raises Exception do
+          @service.get_records_from_record_set('fog-test-rg', 'fog-test-record-set', 'fog-test-zone', 'A')
+        end
+      end
+    end
+  end
+end

data/test/requests/dns/test_list_record_sets.rb

@@ -0,0 +1,36 @@
+require File.expand_path '../../test_helper', __dir__
+
+# Test class for List Record Sets Request
+class TestListRecordSets < Minitest::Test
+  def setup
+    @service = Fog::DNS::AzureRM.new(credentials)
+    @record_sets = @service.record_sets
+    @token_provider = Fog::Credentials::AzureRM.instance_variable_get(:@token_provider)
+  end
+
+  def test_list_record_sets_success
+    response = ApiStub::Requests::DNS::RecordSet.list_record_sets_response
+    @token_provider.stub :get_authentication_header, 'Bearer <some-token>' do
+      RestClient.stub :get, response do
+        assert_equal @service.list_record_sets('fog-test-rg', 'fog-test-zone'), JSON.parse(response)['value']
+      end
+    end
+  end
+
+  def test_list_record_sets_failure
+    @token_provider.stub :get_authentication_header, 'Bearer <some-token>' do
+      assert_raises ArgumentError do
+        @service.list_record_sets('fog-test-rg')
+      end
+    end
+  end
+
+  def test_list_record_sets_exception
+    response = -> { fail Exception.new('mocked exception') }
+    @token_provider.stub :get_authentication_header, response do
+      assert_raises Exception do
+        @service.list_record_sets('fog-test-rg', 'fog-test-zone')
+      end
+    end
+  end
+end

data/test/requests/dns/test_list_zones.rb

@@ -0,0 +1,40 @@
+require File.expand_path '../../test_helper', __dir__
+
+# Test class for List Zones Request
+class TestListZones < Minitest::Test
+  def setup
+    @service = Fog::DNS::AzureRM.new(credentials)
+    @zones = @service.zones
+    @token_provider = Fog::Credentials::AzureRM.instance_variable_get(:@token_provider)
+  end
+
+  def test_list_zones_success
+    client = @service.instance_variable_get(:@resources)
+
+    list_resources_response = ApiStub::Requests::Resources::ResourceGroup.list_resource_groups_for_zones
+    list_zones_response = ApiStub::Requests::DNS::Zone.list_zones_response
+
+    client.stub :resource_groups, list_resources_response do
+      @token_provider.stub :get_authentication_header, 'Bearer <some-token>' do
+        RestClient.stub :get, list_zones_response do
+          assert_equal @service.list_zones, JSON.parse(list_zones_response)['value']
+        end
+      end
+    end
+  end
+
+  def test_list_zones_exception
+    client = @service.instance_variable_get(:@resources)
+
+    list_resources_response = ApiStub::Requests::Resources::ResourceGroup.list_resource_groups_for_zones
+    exception_response = -> { fail Exception.new('mocked exception') }
+
+    client.stub :resource_groups, list_resources_response do
+      @token_provider.stub :get_authentication_header, exception_response do
+        assert_raises Exception do
+          @service.list_zones
+        end
+      end
+    end
+  end
+end

data/test/requests/network/test_add_address_prefixes_in_virtual_network.rb

@@ -0,0 +1,45 @@
+require File.expand_path '../../test_helper', __dir__
+
+# Test class for Add Address Prefixes in Virtual Network Request
+class TestAddAddressPrefixesInVirtualNetwork < 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_add_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'].push('10.3.0.0/16', '10.4.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.add_address_prefixes_in_virtual_network('fog-test-rg', 'fog-test-virtual-network', ['10.3.0.0/16', '10.4.0.0/16']), expected_response
+          end
+        end
+      end
+    end
+  end
+
+  def test_add_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.add_address_prefixes_in_virtual_network('fog-test-rg', 'fog-test-virtual-network', ['10.1.0.0/16', '10.2.0.0/16'])
+        end
+      end
+    end
+  end
+end

data/test/requests/network/test_add_dns_servers_in_virtual_network.rb

@@ -0,0 +1,45 @@
+require File.expand_path '../../test_helper', __dir__
+
+# Test class for Add Dns Servers in Virtual Network Request
+class TestAddDnsServersInVirtualNetwork < 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_add_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'].push('10.1.0.7', '10.1.0.8')
+
+    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.add_dns_servers_in_virtual_network('fog-test-rg', 'fog-test-virtual-network', ['10.1.0.7', '10.1.0.8']), expected_response
+          end
+        end
+      end
+    end
+  end
+
+  def test_add_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.add_dns_servers_in_virtual_network('fog-test-rg', 'fog-test-virtual-network', ['10.1.0.7', '10.1.0.8'])
+        end
+      end
+    end
+  end
+end

data/test/requests/network/test_add_security_rules.rb

@@ -0,0 +1,63 @@
+require File.expand_path '../../test_helper', __dir__
+
+# Test class for Add Security Rules
+class TestAddSecurityRules < 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_add_security_rules_success
+    mocked_response = ApiStub::Requests::Network::NetworkSecurityGroup.create_network_security_group_response
+    expected_response_json = ApiStub::Requests::Network::NetworkSecurityGroup.add_security_rules_response
+    expected_response = Azure::ARM::Network::Models::NetworkSecurityGroup.serialize_object(expected_response_json.body)
+    security_rule =
+      {
+        name: 'testRule',
+        protocol: 'tcp',
+        source_port_range: '22',
+        destination_port_range: '22',
+        source_address_prefix: '0.0.0.0/0',
+        destination_address_prefix: '0.0.0.0/0',
+        access: 'Allow',
+        priority: '100',
+        direction: 'Inbound',
+        description: 'This is a test rule'
+      }
+
+    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.add_security_rules('fog-test-rg', 'fog-test-nsg', [security_rule]), expected_response
+          end
+        end
+      end
+    end
+  end
+
+  def test_add_security_rules_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.add_security_rules('fog-test-rg', 'fog-test-nsg', [])
+            end
+          end
+        end
+      end
+    end
+  end
+end