fog-azure-rm-temp 0.0.1

data/test/models/traffic_manager/test_traffic_manager_end_point.rb

@@ -0,0 +1,54 @@
+require File.expand_path '../../test_helper', __dir__
+
+# Test class for Traffic Manager End Point Model
+class TestTrafficManagerEndPoint < Minitest::Test
+  def setup
+    @service = Fog::TrafficManager::AzureRM.new(credentials)
+    @traffic_manager_end_point = traffic_manager_end_point(@service)
+    @traffic_manager_client = @service.instance_variable_get(:@traffic_mgmt_client)
+  end
+
+  def test_model_methods
+    methods = [
+      :save,
+      :destroy
+    ]
+    methods.each do |method|
+      assert_respond_to @traffic_manager_end_point, method
+    end
+  end
+
+  def test_model_attributes
+    attributes = [
+      :name,
+      :traffic_manager_profile_name,
+      :id,
+      :resource_group,
+      :type,
+      :target_resource_id,
+      :target,
+      :endpoint_status,
+      :endpoint_monitor_status,
+      :weight,
+      :priority,
+      :endpoint_location,
+      :min_child_endpoints
+    ]
+    attributes.each do |attribute|
+      assert_respond_to @traffic_manager_end_point, attribute
+    end
+  end
+
+  def test_save_method_response
+    response = ApiStub::Models::TrafficManager::TrafficManagerEndPoint.create_traffic_manager_end_point_response(@traffic_manager_client)
+    @service.stub :create_or_update_traffic_manager_endpoint, response do
+      assert_instance_of Fog::TrafficManager::AzureRM::TrafficManagerEndPoint, @traffic_manager_end_point.save
+    end
+  end
+
+  def test_destroy_method_response
+    @service.stub :delete_traffic_manager_endpoint, true do
+      assert @traffic_manager_end_point.destroy
+    end
+  end
+end

data/test/models/traffic_manager/test_traffic_manager_end_points.rb

@@ -0,0 +1,56 @@
+require File.expand_path '../../test_helper', __dir__
+
+# Test class for Traffic Manager End Point Collection
+class TestTrafficManagerEndPoints < Minitest::Test
+  def setup
+    @service = Fog::TrafficManager::AzureRM.new(credentials)
+    @traffic_manager_end_points = Fog::TrafficManager::AzureRM::TrafficManagerEndPoints.new(resource_group: 'fog-test-rg', traffic_manager_profile_name: 'fog-test-profile', service: @service)
+    traffic_manager_client = @service.instance_variable_get(:@traffic_mgmt_client)
+    @response_endpoint = ApiStub::Models::TrafficManager::TrafficManagerEndPoint.create_traffic_manager_end_point_response(traffic_manager_client)
+    @response_endpoints = ApiStub::Models::TrafficManager::TrafficManagerProfile.traffic_manager_profile_response(traffic_manager_client)
+  end
+
+  def test_collection_methods
+    methods = [
+      :all,
+      :get,
+      :check_traffic_manager_endpoint_exists
+    ]
+    methods.each do |method|
+      assert_respond_to @traffic_manager_end_points, method
+    end
+  end
+
+  def test_collection_attributes
+    assert_respond_to @traffic_manager_end_points, :resource_group
+    assert_respond_to @traffic_manager_end_points, :traffic_manager_profile_name
+  end
+
+  def test_all_method_response
+    @service.stub :get_traffic_manager_profile, @response_endpoints do
+      assert_instance_of Fog::TrafficManager::AzureRM::TrafficManagerEndPoints, @traffic_manager_end_points.all
+      assert @traffic_manager_end_points.all.size >= 1
+      @traffic_manager_end_points.all.each do |endpoint|
+        assert_instance_of Fog::TrafficManager::AzureRM::TrafficManagerEndPoint, endpoint
+      end
+    end
+  end
+
+  def test_get_method_response
+    @service.stub :get_traffic_manager_end_point, @response_endpoint do
+      assert_instance_of Fog::TrafficManager::AzureRM::TrafficManagerEndPoint, @traffic_manager_end_points.get('resource-group-name', 'profile-name', 'endpoint-name1', 'endpoint-type')
+    end
+  end
+
+  def test_check_traffic_manager_endpoint_exists_true_response
+    @service.stub :check_traffic_manager_endpoint_exists, true do
+      assert @traffic_manager_end_points.check_traffic_manager_endpoint_exists('resource-group-name', 'profile-name', 'endpoint-name1', 'endpoint-type')
+    end
+  end
+
+  def test_check_traffic_manager_endpoint_exists_false_response
+    @service.stub :check_traffic_manager_endpoint_exists, false do
+      assert !@traffic_manager_end_points.check_traffic_manager_endpoint_exists('resource-group-name', 'profile-name', 'endpoint-name1', 'endpoint-type')
+    end
+  end
+end

data/test/models/traffic_manager/test_traffic_manager_profile.rb

@@ -0,0 +1,62 @@
+require File.expand_path '../../test_helper', __dir__
+
+# Test class for Traffic Manager Profile Model
+class TestTrafficManagerProfile < Minitest::Test
+  def setup
+    @service = Fog::TrafficManager::AzureRM.new(credentials)
+    @traffic_manager_profile = traffic_manager_profile(@service)
+    @traffic_manager_client = @service.instance_variable_get(:@traffic_mgmt_client)
+    @traffic_manager_profile_response = ApiStub::Models::TrafficManager::TrafficManagerProfile.traffic_manager_profile_response(@traffic_manager_client)
+  end
+
+  def test_model_methods
+    methods = [
+      :save,
+      :destroy,
+      :update
+    ]
+    methods.each do |method|
+      assert_respond_to @traffic_manager_profile, method
+    end
+  end
+
+  def test_model_attributes
+    attributes = [
+      :name,
+      :id,
+      :resource_group,
+      :location,
+      :profile_status,
+      :traffic_routing_method,
+      :relative_name,
+      :fqdn,
+      :ttl,
+      :profile_monitor_status,
+      :protocol,
+      :port,
+      :path,
+      :endpoints
+    ]
+    attributes.each do |attribute|
+      assert_respond_to @traffic_manager_profile, attribute
+    end
+  end
+
+  def test_save_method_response
+    @service.stub :create_or_update_traffic_manager_profile, @traffic_manager_profile_response do
+      assert_instance_of Fog::TrafficManager::AzureRM::TrafficManagerProfile, @traffic_manager_profile.save
+    end
+  end
+
+  def test_update_method_response
+    @service.stub :create_or_update_traffic_manager_profile, @traffic_manager_profile_response do
+      assert_instance_of Fog::TrafficManager::AzureRM::TrafficManagerProfile, @traffic_manager_profile.update({})
+    end
+  end
+
+  def test_destroy_method_response
+    @service.stub :delete_traffic_manager_profile, true do
+      assert @traffic_manager_profile.destroy
+    end
+  end
+end

data/test/models/traffic_manager/test_traffic_manager_profiles.rb

@@ -0,0 +1,55 @@
+require File.expand_path '../../test_helper', __dir__
+
+# Test class for Traffic Manager Profile Collection
+class TestTrafficManagerProfiles < Minitest::Test
+  def setup
+    @service = Fog::TrafficManager::AzureRM.new(credentials)
+    @traffic_manager_profiles = Fog::TrafficManager::AzureRM::TrafficManagerProfiles.new(resource_group: 'fog-test-rg', service: @service)
+    traffic_manager_client = @service.instance_variable_get(:@traffic_mgmt_client)
+    @profiles_list = [ApiStub::Models::TrafficManager::TrafficManagerProfile.traffic_manager_profile_response(traffic_manager_client)]
+    @profile = ApiStub::Models::TrafficManager::TrafficManagerProfile.traffic_manager_profile_response(traffic_manager_client)
+  end
+
+  def test_collection_methods
+    methods = [
+      :all,
+      :get,
+      :check_traffic_manager_profile_exists
+    ]
+    methods.each do |method|
+      assert_respond_to @traffic_manager_profiles, method
+    end
+  end
+
+  def test_collection_attributes
+    assert_respond_to @traffic_manager_profiles, :resource_group
+  end
+
+  def test_all_method_response
+    @service.stub :list_traffic_manager_profiles, @profiles_list do
+      assert_instance_of Fog::TrafficManager::AzureRM::TrafficManagerProfiles, @traffic_manager_profiles.all
+      assert @traffic_manager_profiles.all.size >= 1
+      @traffic_manager_profiles.all.each do |endpoint|
+        assert_instance_of Fog::TrafficManager::AzureRM::TrafficManagerProfile, endpoint
+      end
+    end
+  end
+
+  def test_get_method_response
+    @service.stub :get_traffic_manager_profile, @profile do
+      assert_instance_of Fog::TrafficManager::AzureRM::TrafficManagerProfile, @traffic_manager_profiles.get('resource-group-name', 'fog-test-profile-name')
+    end
+  end
+
+  def test_check_traffic_manager_profile_exists_true_response
+    @service.stub :check_traffic_manager_profile_exists, true do
+      assert @traffic_manager_profiles.check_traffic_manager_profile_exists('resource-group-name', 'fog-test-profile-name')
+    end
+  end
+
+  def test_check_traffic_manager_profile_exists_false_response
+    @service.stub :check_traffic_manager_profile_exists, false do
+      assert !@traffic_manager_profiles.check_traffic_manager_profile_exists('resource-group-name', 'fog-test-profile-name')
+    end
+  end
+end

data/test/requests/application_gateway/test_check_ag_exists.rb

@@ -0,0 +1,31 @@
+require File.expand_path '../../test_helper', __dir__
+
+# Test class for Check Application Gateway Exists Request
+class TestCheckAGExists < Minitest::Test
+  def setup
+    @service = Fog::ApplicationGateway::AzureRM.new(credentials)
+    @gateway_client = @service.instance_variable_get(:@network_client)
+    @gateways = @gateway_client.application_gateways
+  end
+
+  def test_check_ag_exists_success
+    mocked_response = ApiStub::Requests::ApplicationGateway::Gateway.create_application_gateway_response(@gateway_client)
+    @gateways.stub :get, mocked_response do
+      assert @service.check_ag_exists('fog-test-rg', 'fogRM-rg')
+    end
+  end
+
+  def test_check_ag_exists_failure
+    response = proc { raise MsRestAzure::AzureOperationError.new(nil, nil, 'error' => { 'message' => 'mocked exception', 'code' => 'ResourceNotFound' }) }
+    @gateways.stub :get, response do
+      assert !@service.check_ag_exists('fog-test-rg', 'fogRM-rg')
+    end
+  end
+
+  def test_check_ag_exists_exception
+    response = proc { raise MsRestAzure::AzureOperationError.new(nil, nil, 'error' => { 'message' => 'mocked exception', 'code' => 'ResourceGroupNotFound' }) }
+    @gateways.stub :get, response do
+      assert_raises(RuntimeError) { @service.check_ag_exists('fog-test-rg', 'fogRM-rg') }
+    end
+  end
+end

data/test/requests/application_gateway/test_create_or_update_application_gateway.rb

@@ -0,0 +1,35 @@
+require File.expand_path '../../test_helper', __dir__
+
+# Test class for Create Application Gateway Request
+class TestCreateOrUpdateApplicationGateway < Minitest::Test
+  def setup
+    @service = Fog::ApplicationGateway::AzureRM.new(credentials)
+    gateway_client = @service.instance_variable_get(:@network_client)
+    @gateways = gateway_client.application_gateways
+    @response = ApiStub::Requests::ApplicationGateway::Gateway.create_application_gateway_response(gateway_client)
+    @gateway_params = ApiStub::Requests::ApplicationGateway::Gateway.gateway_params
+  end
+
+  def test_create_or_update_application_gateway_success
+    @gateways.stub :create_or_update, @response do
+      assert_equal @service.create_or_update_application_gateway(@gateway_params), @response
+    end
+  end
+
+  def test_create_or_update_application_gateway_argument_error_failure
+    @gateways.stub :create_or_update, @response do
+      assert_raises ArgumentError do
+        @service.create_or_update_application_gateway('gateway', 'East US', 'fogRM-rg', 'Standard_Medium', 'Standard', 2)
+      end
+    end
+  end
+
+  def test_create_or_update_application_gateway_exception_failure
+    response = proc { raise MsRestAzure::AzureOperationError.new(nil, nil, 'error' => { 'message' => 'mocked exception' }) }
+    @gateways.stub :create_or_update, response do
+      assert_raises RuntimeError do
+        @service.create_or_update_application_gateway(@gateway_params)
+      end
+    end
+  end
+end

data/test/requests/application_gateway/test_delete_application_gateway.rb

@@ -0,0 +1,23 @@
+require File.expand_path '../../test_helper', __dir__
+
+# Test class for Delete Application Gateway Request
+class TestDeleteApplicationGateway < Minitest::Test
+  def setup
+    @service = Fog::ApplicationGateway::AzureRM.new(credentials)
+    gateway_client = @service.instance_variable_get(:@network_client)
+    @gateways = gateway_client.application_gateways
+  end
+
+  def test_delete_application_gateway_success
+    @gateways.stub :delete, true do
+      assert @service.delete_application_gateway('fogRM-rg', 'gateway'), true
+    end
+  end
+
+  def test_delete_application_gateway_failure
+    response = proc { raise MsRestAzure::AzureOperationError.new(nil, nil, 'error' => { 'message' => 'mocked exception' }) }
+    @gateways.stub :delete, response do
+      assert_raises(RuntimeError) { @service.delete_application_gateway('fogRM-rg', 'gateway') }
+    end
+  end
+end

data/test/requests/application_gateway/test_get_application_gateway.rb

@@ -0,0 +1,24 @@
+require File.expand_path '../../test_helper', __dir__
+
+# Test class for Get Application Gateway Request
+class TestGetApplicationGateway < Minitest::Test
+  def setup
+    @service = Fog::ApplicationGateway::AzureRM.new(credentials)
+    @gateway_client = @service.instance_variable_get(:@network_client)
+    @gateways = @gateway_client.application_gateways
+  end
+
+  def test_get_application_gateway_success
+    mocked_response = ApiStub::Requests::ApplicationGateway::Gateway.create_application_gateway_response(@gateway_client)
+    @gateways.stub :get, mocked_response do
+      assert_equal @service.get_application_gateway('fog-test-rg', 'fogRM-rg'), mocked_response
+    end
+  end
+
+  def test_get_application_gateway_failure
+    response = proc { raise MsRestAzure::AzureOperationError.new(nil, nil, 'error' => { 'message' => 'mocked exception' }) }
+    @gateways.stub :get, response do
+      assert_raises(RuntimeError) { @service.get_application_gateway('fog-test-rg', 'fogRM-rg') }
+    end
+  end
+end

data/test/requests/application_gateway/test_list_application_gateways.rb

@@ -0,0 +1,24 @@
+require File.expand_path '../../test_helper', __dir__
+
+# Test class for List Application Gateway Request
+class TestListApplicationGateways < Minitest::Test
+  def setup
+    @service = Fog::ApplicationGateway::AzureRM.new(credentials)
+    @gateway_client = @service.instance_variable_get(:@network_client)
+    @gateways = @gateway_client.application_gateways
+  end
+
+  def test_list_application_gateways_success
+    mocked_response = ApiStub::Requests::ApplicationGateway::Gateway.list_application_gateway_response(@gateway_client)
+    @gateways.stub :list_as_lazy, mocked_response do
+      assert_equal @service.list_application_gateways('fogRM-rg'), mocked_response.value
+    end
+  end
+
+  def test_list_application_gateways_failure
+    response = proc { raise MsRestAzure::AzureOperationError.new(nil, nil, 'error' => { 'message' => 'mocked exception' }) }
+    @gateways.stub :list_as_lazy, response do
+      assert_raises(RuntimeError) { @service.list_application_gateways('fogRM-rg') }
+    end
+  end
+end

data/test/requests/application_gateway/test_start_application_gateway.rb

@@ -0,0 +1,25 @@
+require File.expand_path '../../test_helper', __dir__
+
+# Test class for Start Application Gateway Request
+class TestStartApplicationGateway < Minitest::Test
+  def setup
+    @service = Fog::ApplicationGateway::AzureRM.new(credentials)
+    gateway_client = @service.instance_variable_get(:@network_client)
+    @gateways = gateway_client.application_gateways
+  end
+
+  def test_start_application_gateway_success
+    @gateways.stub :start, nil do
+      assert_equal @service.start_application_gateway('test-rg', 'test-ag'), true
+    end
+  end
+
+  def test_start_application_gateway_exception_failure
+    response = proc { raise MsRestAzure::AzureOperationError.new(nil, nil, 'error' => { 'message' => 'mocked exception' }) }
+    @gateways.stub :start, response do
+      assert_raises RuntimeError do
+        @service.start_application_gateway('test-rg', 'test-ag')
+      end
+    end
+  end
+end

data/test/requests/application_gateway/test_stop_application_gateway.rb

@@ -0,0 +1,25 @@
+require File.expand_path '../../test_helper', __dir__
+
+# Test class for Stop Application Gateway Request
+class TestStopApplicationGateway < Minitest::Test
+  def setup
+    @service = Fog::ApplicationGateway::AzureRM.new(credentials)
+    gateway_client = @service.instance_variable_get(:@network_client)
+    @gateways = gateway_client.application_gateways
+  end
+
+  def test_stop_application_gateway_success
+    @gateways.stub :stop, nil do
+      assert_equal @service.stop_application_gateway('test-rg', 'test-ag'), true
+    end
+  end
+
+  def test_stop_application_gateway_exception_failure
+    response = proc { raise MsRestAzure::AzureOperationError.new(nil, nil, 'error' => { 'message' => 'mocked exception' }) }
+    @gateways.stub :stop, response do
+      assert_raises RuntimeError do
+        @service.stop_application_gateway('test-rg', 'test-ag')
+      end
+    end
+  end
+end

data/test/requests/compute/test_attach_data_disk_to_vm.rb

@@ -0,0 +1,90 @@
+require File.expand_path '../../test_helper', __dir__
+
+# Test class for Attach Data Disk to VM Request
+class TestAttachDataDiskToVM < Minitest::Test
+  def setup
+    @service = Fog::Compute::AzureRM.new(credentials)
+    @compute_client = @service.instance_variable_get(:@compute_mgmt_client)
+    storage_client = @service.instance_variable_get(:@storage_mgmt_client)
+    @virtual_machines = @compute_client.virtual_machines
+    @storage_accounts = storage_client.storage_accounts
+    @get_vm_response = ApiStub::Requests::Compute::VirtualMachine.create_virtual_machine_response(@compute_client)
+    @update_vm_response = ApiStub::Requests::Compute::VirtualMachine.update_virtual_machine_response(@compute_client)
+    @storage_access_keys_response = ApiStub::Requests::Storage::StorageAccount.list_keys_response
+  end
+
+  def test_attach_data_disk_to_vm_success
+    blob_service = Minitest::Mock.new
+    blob_service.expect :get_blob_properties, 'blob_props' do
+      Azure::Storage::Blob::BlobService.stub :new, blob_service do
+        @virtual_machines.stub :get, @get_vm_response do
+          @storage_accounts.stub :list_keys, @storage_access_keys_response do
+            @virtual_machines.stub :create_or_update, @update_vm_response do
+              assert_equal @service.attach_data_disk_to_vm('fog-test-rg', 'fog-test-vm', 'disk1', 1, 'mystorage1'), @update_vm_response
+              blob_service.verify
+            end
+          end
+        end
+      end
+    end
+  end
+
+  def test_attach_data_disk_to_vm_failure
+    get_vm_response = ApiStub::Requests::Compute::VirtualMachine.virtual_machine_15_data_disks_response(@compute_client)
+    @virtual_machines.stub :get, get_vm_response do
+      @storage_accounts.stub :list_keys, @storage_access_keys_response do
+        @virtual_machines.stub :create_or_update, @update_vm_response do
+          assert_raises RuntimeError do
+            @service.attach_data_disk_to_vm('fog-test-rg', 'fog-test-vm', 'disk1', 1, 'mystorage1')
+          end
+        end
+      end
+    end
+  end
+
+  def test_update_vm_failure
+    update_vm_response = proc { fail MsRestAzure::AzureOperationError.new(nil, nil, 'error' => { 'message' => 'mocked exception' }) }
+    @virtual_machines.stub :get, @get_vm_response do
+      @storage_accounts.stub :list_keys, @storage_access_keys_response do
+        @virtual_machines.stub :create_or_update, update_vm_response do
+          assert_raises RuntimeError do
+            @service.attach_data_disk_to_vm('fog-test-rg', 'fog-test-vm', 'disk1', 1, 'mystorage1')
+          end
+        end
+      end
+    end
+  end
+
+  def test_update_vm_blob_not_exist_failure
+    update_vm_response = proc { fail MsRestAzure::AzureOperationError.new(nil, nil, 'error' => { 'message' => 'mocked exception' }) }
+    @virtual_machines.stub :get, @get_vm_response do
+      @storage_accounts.stub :list_keys, @storage_access_keys_response do
+        @virtual_machines.stub :create_or_update, update_vm_response do
+          assert_raises RuntimeError do
+            @service.attach_data_disk_to_vm('fog-test-rg', 'fog-test-vm', 'disk1', 1, 'mystorage1')
+          end
+        end
+      end
+    end
+  end
+
+  def test_get_vm_failure
+    get_vm_response = proc { fail MsRestAzure::AzureOperationError.new(nil, nil, 'error' => { 'message' => 'mocked exception' }) }
+    @virtual_machines.stub :get, get_vm_response do
+      assert_raises RuntimeError do
+        @service.attach_data_disk_to_vm('fog-test-rg', 'fog-test-vm', 'disk1', 1, 'mystorage1')
+      end
+    end
+  end
+
+  def test_get_storage_key_failure
+    storage_access_keys_response = proc { fail MsRestAzure::AzureOperationError.new(nil, nil, 'error' => { 'message' => 'mocked exception' }) }
+    @virtual_machines.stub :get, @get_vm_response do
+      @storage_accounts.stub :list_keys, storage_access_keys_response do
+        assert_raises RuntimeError do
+          @service.attach_data_disk_to_vm('fog-test-rg', 'fog-test-vm', 'disk1', 1, 'mystorage1')
+        end
+      end
+    end
+  end
+end