fog-azure-rm 0.0.3 → 0.0.4

data/test/models/storage/test_containers.rb

@@ -0,0 +1,56 @@
+require File.expand_path '../../test_helper', __dir__
+
+# Test class for Container Collection
+class TestContainers < Minitest::Test
+  def setup
+    @service = Fog::Storage::AzureRM.new(storage_account_credentials)
+    @containers = Fog::Storage::AzureRM::Containers.new(service: @service)
+    @response = ApiStub::Models::Storage::Container.test_get_container_metadata
+    @list_results = ApiStub::Models::Storage::Container.list_containers
+    @acl_results = ApiStub::Models::Storage::Container.get_container_access_control_list
+  end
+
+  def test_collection_methods
+    methods = [
+      :set_container_metadata,
+      :get_container_metadata,
+      :all,
+      :get
+    ]
+    methods.each do |method|
+      assert @containers.respond_to? method, true
+    end
+  end
+
+  def test_get_container_metadata
+    @service.stub :get_container_metadata, @response do
+      assert_equal @response, @containers.get_container_metadata('Test-container')
+    end
+  end
+
+  def test_set_container_metadata
+    @service.stub :set_container_metadata, true do
+      assert @containers.set_container_metadata('Test-container', @response)
+    end
+  end
+
+  def test_all_method
+    containers = Fog::Storage::AzureRM::Containers.new(service: @service)
+    @service.stub :list_containers, @list_results do
+      assert_instance_of Fog::Storage::AzureRM::Containers, containers.all
+      assert containers.all.size >= 1
+      containers.all.each do |container|
+        assert_instance_of Fog::Storage::AzureRM::Container, container
+      end
+    end
+  end
+
+  def test_get_method
+    @service.stub :list_containers, @list_results do
+      @service.stub :get_container_access_control_list, @acl_results do
+        assert_instance_of Fog::Storage::AzureRM::Container, @containers.get('testcontainer1')
+        assert @containers.get('wrong-name').nil?, true
+      end
+    end
+  end
+end

data/test/models/storage/test_data_disk.rb

@@ -0,0 +1,29 @@
+require File.expand_path '../../test_helper', __dir__
+
+# Test class for Data Disk Model
+class TestDataDisk < Minitest::Test
+  def setup
+    @service = Fog::Storage::AzureRM.new(credentials)
+    @data_disk = Fog::Storage::AzureRM::DataDisk.new
+  end
+
+  def test_model_attributes
+    attributes = [
+      :name,
+      :disk_size_gb,
+      :lun,
+      :vhd_uri,
+      :caching,
+      :create_option
+    ]
+    attributes.each do |attribute|
+      assert @data_disk.respond_to? attribute, true
+    end
+  end
+
+  def test_parse_method_response
+    mocked_response = ApiStub::Models::Storage::DataDisk.create_data_disk_response
+    expected_response = ApiStub::Models::Storage::DataDisk.expected_create_data_disk_response
+    assert_equal Fog::Storage::AzureRM::DataDisk.parse(mocked_response), expected_response
+  end
+end

data/test/models/storage/test_storage_account.rb

@@ -0,0 +1,62 @@
+require File.expand_path '../../test_helper', __dir__
+# Test class for Storage Account Model
+class TestStorageAccount < Minitest::Test
+  def setup
+    @service = Fog::Storage::AzureRM.new(credentials)
+    @storage_account = storage_account(@service)
+    @response = ApiStub::Models::Storage::StorageAccount.create_storage_account
+  end
+
+  def test_model_methods
+    methods = [
+      :save,
+      :destroy,
+      :get_access_keys
+    ]
+    methods.each do |method|
+      assert @storage_account.respond_to? method, true
+    end
+  end
+
+  def test_model_attributes
+    attributes = [
+      :name,
+      :location,
+      :resource_group,
+      :account_type
+    ]
+    @service.stub :create_storage_account, @response do
+      attributes.each do |attribute|
+        assert_respond_to @storage_account, attribute
+      end
+    end
+  end
+
+  def test_save_method_response
+    @service.stub :create_storage_account, @response do
+      assert_instance_of Fog::Storage::AzureRM::StorageAccount, @storage_account.save
+    end
+  end
+
+  def test_get_access_keys_method_response
+    response = {
+      'key1' => 'key1 value',
+      'key2' => 'key2 value'
+    }
+    @service.stub :get_storage_access_keys, response do
+      assert_equal @storage_account.get_access_keys, response
+    end
+  end
+
+  def test_destroy_method_true_response
+    @service.stub :delete_storage_account, true do
+      assert @storage_account.destroy
+    end
+  end
+
+  def test_destroy_method_false_response
+    @service.stub :delete_storage_account, false do
+      assert !@storage_account.destroy
+    end
+  end
+end

data/test/models/storage/test_storage_accounts.rb

@@ -0,0 +1,65 @@
+require File.expand_path '../../test_helper', __dir__
+
+# Test class for Availability Set Collection
+class TestStorageAccounts < Minitest::Test
+  def setup
+    @service = Fog::Storage::AzureRM.new(credentials)
+    @storage_accounts = Fog::Storage::AzureRM::StorageAccounts.new(resource_group: 'fog-test-rg', service: @service)
+    @response = [ApiStub::Models::Storage::StorageAccount.list_storage_account]
+  end
+
+  def test_collection_methods
+    methods = [
+      :all,
+      :get
+    ]
+    methods.each do |method|
+      assert @storage_accounts.respond_to? method, true
+    end
+  end
+
+  def test_collection_attributes
+    assert @storage_accounts.respond_to? :resource_group, true
+  end
+
+  def test_all_method_response_for_rg
+    @service.stub :list_storage_account_for_rg, @response do
+      assert_instance_of Fog::Storage::AzureRM::StorageAccounts, @storage_accounts.all
+      assert @storage_accounts.all.size >= 1
+      @storage_accounts.all.each do |s|
+        assert_instance_of Fog::Storage::AzureRM::StorageAccount, s
+      end
+    end
+  end
+
+  def test_all_method_response
+    storage_accounts = Fog::Storage::AzureRM::StorageAccounts.new(service: @service)
+    @service.stub :list_storage_accounts, @response do
+      assert_instance_of Fog::Storage::AzureRM::StorageAccounts, storage_accounts.all
+      assert storage_accounts.all.size >= 1
+      storage_accounts.all.each do |s|
+        assert_instance_of Fog::Storage::AzureRM::StorageAccount, s
+      end
+    end
+  end
+
+  def test_get_method_response
+    response = [ApiStub::Models::Storage::StorageAccount.list_storage_account]
+    @service.stub :list_storage_account_for_rg, response do
+      assert_instance_of Fog::Storage::AzureRM::StorageAccount, @storage_accounts.get('fog-test-storage-account')
+      assert @storage_accounts.get('wrong-name').nil?, true
+    end
+  end
+
+  def test_check_name_availability_true_case
+    @service.stub :check_storage_account_name_availability, true do
+      assert @storage_accounts.check_name_availability('fog-test-storage-account')
+    end
+  end
+
+  def test_check_name_availability_false_case
+    @service.stub :check_storage_account_name_availability, false do
+      assert !@storage_accounts.check_name_availability('fog-test-storage-account')
+    end
+  end
+end

data/test/requests/compute/test_attach_data_disk_to_vm.rb

@@ -0,0 +1,175 @@
+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)
+    vm_client = @service.instance_variable_get(:@compute_mgmt_client)
+    storage_client = @service.instance_variable_get(:@storage_mgmt_client)
+    @virtual_machines = vm_client.virtual_machines
+    @storage_accounts = storage_client.storage_accounts
+  end
+
+  def test_attach_data_disk_to_vm_success
+    promise_get_vm = Concurrent::Promise.execute do
+    end
+
+    promise_update_vm = Concurrent::Promise.execute do
+    end
+
+    promise_storage_access_keys = Concurrent::Promise.execute do
+    end
+
+    get_vm_response = ApiStub::Requests::Compute::VirtualMachine.create_virtual_machine_response
+    update_vm_response = ApiStub::Requests::Compute::VirtualMachine.update_virtual_machine_response
+    expected_update_vm_response = Azure::ARM::Compute::Models::VirtualMachine.serialize_object(update_vm_response.body)
+    storage_access_keys_response = ApiStub::Requests::Storage::StorageAccount.list_keys_response
+    promise_get_vm.stub :value!, get_vm_response do
+      @virtual_machines.stub :get, promise_get_vm do
+        promise_storage_access_keys.stub :value!, storage_access_keys_response do
+          @storage_accounts.stub :list_keys, promise_storage_access_keys do
+            promise_update_vm.stub :value!, update_vm_response do
+              @virtual_machines.stub :create_or_update, promise_update_vm do
+                @service.stub :check_blob_exist, true do
+                  assert_equal @service.attach_data_disk_to_vm('fog-test-rg', 'fog-test-vm', 'disk1', 1, 'mystorage1'), expected_update_vm_response
+                end
+              end
+            end
+          end
+        end
+      end
+    end
+  end
+
+  def test_attach_data_disk_to_vm_failure
+    promise_get_vm = Concurrent::Promise.execute do
+    end
+
+    promise_update_vm = Concurrent::Promise.execute do
+    end
+
+    promise_storage_access_keys = Concurrent::Promise.execute do
+    end
+
+    get_vm_response = ApiStub::Requests::Compute::VirtualMachine.virtual_machine_15_data_disks_response
+    update_vm_response = ApiStub::Requests::Compute::VirtualMachine.update_virtual_machine_response
+    storage_access_keys_response = ApiStub::Requests::Storage::StorageAccount.list_keys_response
+    promise_get_vm.stub :value!, get_vm_response do
+      @virtual_machines.stub :get, promise_get_vm do
+        promise_storage_access_keys.stub :value!, storage_access_keys_response do
+          @storage_accounts.stub :list_keys, promise_storage_access_keys do
+            promise_update_vm.stub :value!, update_vm_response do
+              @virtual_machines.stub :create_or_update, promise_update_vm do
+                @service.stub :check_blob_exist, true 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
+        end
+      end
+    end
+  end
+
+  def test_update_vm_failure
+    promise_get_vm = Concurrent::Promise.execute do
+    end
+
+    promise_update_vm = Concurrent::Promise.execute do
+    end
+
+    promise_storage_access_keys = Concurrent::Promise.execute do
+    end
+
+    get_vm_response = ApiStub::Requests::Compute::VirtualMachine.create_virtual_machine_response
+    update_vm_response = -> { fail MsRestAzure::AzureOperationError.new(nil, nil, 'error' => { 'message' => 'mocked exception' }) }
+    storage_access_keys_response = ApiStub::Requests::Storage::StorageAccount.list_keys_response
+    promise_get_vm.stub :value!, get_vm_response do
+      @virtual_machines.stub :get, promise_get_vm do
+        promise_storage_access_keys.stub :value!, storage_access_keys_response do
+          @storage_accounts.stub :list_keys, promise_storage_access_keys do
+            promise_update_vm.stub :value!, update_vm_response do
+              @virtual_machines.stub :create_or_update, promise_update_vm do
+                @service.stub :check_blob_exist, true 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
+        end
+      end
+    end
+  end
+
+  def test_update_vm_blob_not_exist_failure
+    promise_get_vm = Concurrent::Promise.execute do
+    end
+
+    promise_update_vm = Concurrent::Promise.execute do
+    end
+
+    promise_storage_access_keys = Concurrent::Promise.execute do
+    end
+
+    get_vm_response = ApiStub::Requests::Compute::VirtualMachine.create_virtual_machine_response
+    update_vm_response = -> { fail MsRestAzure::AzureOperationError.new(nil, nil, 'error' => { 'message' => 'mocked exception' }) }
+    storage_access_keys_response = ApiStub::Requests::Storage::StorageAccount.list_keys_response
+    promise_get_vm.stub :value!, get_vm_response do
+      @virtual_machines.stub :get, promise_get_vm do
+        promise_storage_access_keys.stub :value!, storage_access_keys_response do
+          @storage_accounts.stub :list_keys, promise_storage_access_keys do
+            promise_update_vm.stub :value!, update_vm_response do
+              @virtual_machines.stub :create_or_update, promise_update_vm do
+                @service.stub :check_blob_exist, false 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
+        end
+      end
+    end
+  end
+
+  def test_get_vm_failure
+    promise_get_vm = Concurrent::Promise.execute do
+    end
+
+    get_vm_response = -> { fail MsRestAzure::AzureOperationError.new(nil, nil, 'error' => { 'message' => 'mocked exception' }) }
+    promise_get_vm.stub :value!, get_vm_response do
+      @virtual_machines.stub :get, promise_get_vm do
+        assert_raises RuntimeError do
+          @service.attach_data_disk_to_vm('fog-test-rg', 'fog-test-vm', 'disk1', 1, 'mystorage1')
+        end
+      end
+    end
+  end
+
+  def test_get_storage_key_failure
+    promise_get_vm = Concurrent::Promise.execute do
+    end
+
+    promise_storage_access_keys = Concurrent::Promise.execute do
+    end
+
+    get_vm_response = ApiStub::Requests::Compute::VirtualMachine.create_virtual_machine_response
+    storage_access_keys_response = -> { fail MsRestAzure::AzureOperationError.new(nil, nil, 'error' => { 'message' => 'mocked exception' }) }
+    promise_get_vm.stub :value!, get_vm_response do
+      @virtual_machines.stub :get, promise_get_vm do
+        promise_storage_access_keys.stub :value!, storage_access_keys_response do
+          @storage_accounts.stub :list_keys, promise_storage_access_keys 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
+  end
+end

data/test/requests/compute/test_create_availability_set.rb

@@ -0,0 +1,30 @@
+require File.expand_path '../../test_helper', __dir__
+
+# Test class for Create Availability Set Request
+class TestCreateAvailabilitySet < Minitest::Test
+  def setup
+    @service = Fog::Compute::AzureRM.new(credentials)
+    client = @service.instance_variable_get(:@compute_mgmt_client)
+    @availability_sets = client.availability_sets
+    @promise = Concurrent::Promise.execute do
+    end
+  end
+
+  def test_create_availability_set_success
+    response = ApiStub::Requests::Compute::AvailabilitySet.create_availability_set_response
+    @promise.stub :value!, response do
+      @availability_sets.stub :create_or_update, @promise do
+        assert_equal @service.create_availability_set('fog-test-rg', 'fog-test-as', 'west us'), Azure::ARM::Compute::Models::AvailabilitySet.serialize_object(response.body)
+      end
+    end
+  end
+
+  def test_create_availability_set_failure
+    response = -> { fail MsRestAzure::AzureOperationError.new(nil, nil, 'error' => { 'message' => 'mocked exception' }) }
+    @promise.stub :value!, response do
+      @availability_sets.stub :create_or_update, @promise do
+        assert_raises(RuntimeError) { @service.create_availability_set('fog-test-rg', 'fog-test-as', 'west us') }
+      end
+    end
+  end
+end

data/test/requests/compute/test_create_virtual_machine.rb

@@ -0,0 +1,50 @@
+require File.expand_path '../../test_helper', __dir__
+
+# Test class for Create Virtual Machine Request
+class TestCreateVirtualMachine < Minitest::Test
+  def setup
+    @service = Fog::Compute::AzureRM.new(credentials)
+    client = @service.instance_variable_get(:@compute_mgmt_client)
+    @virtual_machines = client.virtual_machines
+    @promise = Concurrent::Promise.execute do
+    end
+  end
+
+  def test_create_linux_virtual_machine_success
+    response = ApiStub::Requests::Compute::VirtualMachine.create_virtual_machine_response
+    @promise.stub :value!, response do
+      @virtual_machines.stub :create_or_update, @promise do
+        assert_equal @service.create_virtual_machine('fog-test-rg', 'fog-test-server', 'westus', 'Basic_A0',
+                                                     'fogstrg', 'fog', 'fog', false, '/home', 'key', 'nic-id',
+                                                     'as-id', 'Canonical', 'UbuntuServer', '14.04.2-LTS', 'latest', 'Linux', nil, nil),
+                     Azure::ARM::Compute::Models::VirtualMachine.serialize_object(response.body)
+      end
+    end
+  end
+
+  def test_create_windows_virtual_machine_success
+    response = ApiStub::Requests::Compute::VirtualMachine.create_virtual_machine_response
+    @promise.stub :value!, response do
+      @virtual_machines.stub :create_or_update, @promise do
+        assert_equal @service.create_virtual_machine('fog-test-rg', 'fog-test-server', 'westus', 'Basic_A0',
+                                                     'fogstrg', 'fog', 'fog', nil, '/home', 'key', 'nic-id',
+                                                     'as-id', 'MicrosoftWindowsServerEssentials', 'WindowsServerEssentials',
+                                                     'WindowsServerEssentials', 'latest', 'Windows', true, true),
+                     Azure::ARM::Compute::Models::VirtualMachine.serialize_object(response.body)
+      end
+    end
+  end
+
+  def test_create_virtual_machine_failure
+    response = -> { raise MsRestAzure::AzureOperationError.new(nil, nil, 'error' => { 'message' => 'mocked exception' }) }
+    @promise.stub :value!, response do
+      @virtual_machines.stub :create_or_update, @promise do
+        assert_raises RuntimeError do
+          @service.create_virtual_machine('fog-test-rg', 'fog-test-server', 'westus', 'Basic_A0', 'fogstrg', 'fog',
+                                          'fog', false, '/home', 'key', 'nic-id', 'as-id', 'Canonical',
+                                          'UbuntuServer', '14.04.2-LTS', 'latest', 'Linux', nil, nil)
+        end
+      end
+    end
+  end
+end