fog-azure-rm 0.0.4 → 0.0.5

data/test/requests/compute/test_deallocate_virtual_machine.rb

@@ -4,26 +4,20 @@ require File.expand_path '../../test_helper', __dir__
 class TestDeallocateVirtualMachine < 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
+    compute_client = @service.instance_variable_get(:@compute_mgmt_client)
+    @virtual_machines = compute_client.virtual_machines
   end
 
   def test_deallocate_virtual_machine_success
-    @promise.stub :value!, true do
-      @virtual_machines.stub :deallocate, @promise do
-        assert @service.deallocate_virtual_machine('fog-test-rg', 'fog-test-server')
-      end
+    @virtual_machines.stub :deallocate, true do
+      assert @service.deallocate_virtual_machine('fog-test-rg', 'fog-test-server')
     end
   end
 
   def test_deallocate_virtual_machine_failure
-    response = -> { fail MsRestAzure::AzureOperationError.new(nil, nil, 'error' => { 'message' => 'mocked exception' }) }
-    @promise.stub :value!, response do
-      @virtual_machines.stub :deallocate, @promise do
-        assert_raises(RuntimeError) { @service.deallocate_virtual_machine('fog-test-rg', 'fog-test-server') }
-      end
+    response = proc { fail MsRestAzure::AzureOperationError.new(nil, nil, 'error' => { 'message' => 'mocked exception' }) }
+    @virtual_machines.stub :deallocate, response do
+      assert_raises(RuntimeError) { @service.deallocate_virtual_machine('fog-test-rg', 'fog-test-server') }
     end
   end
 end

data/test/requests/compute/test_delete_availability_set.rb

@@ -6,24 +6,18 @@ class TestDeleteAvailabilitySet < Minitest::Test
     @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_delete_availability_set_success
-    @promise.stub :value!, true do
-      @availability_sets.stub :delete, @promise do
-        assert @service.delete_availability_set('fog-test-rg', 'fog-test-as')
-      end
+    @availability_sets.stub :delete, true do
+      assert @service.delete_availability_set('fog-test-rg', 'fog-test-as')
     end
   end
 
   def test_delete_availability_set_failure
-    response = -> { fail MsRestAzure::AzureOperationError.new(nil, nil, 'error' => { 'message' => 'mocked exception' }) }
-    @promise.stub :value!, response do
-      @availability_sets.stub :delete, @promise do
-        assert_raises(RuntimeError) { @service.delete_availability_set('fog-test-rg', 'fog-test-as') }
-      end
+    response = proc { raise MsRestAzure::AzureOperationError.new(nil, nil, 'error' => { 'message' => 'mocked exception' }) }
+    @availability_sets.stub :delete, response do
+      assert_raises(RuntimeError) { @service.delete_availability_set('fog-test-rg', 'fog-test-as') }
     end
   end
 end

data/test/requests/compute/test_delete_virtual_machine.rb

@@ -4,26 +4,20 @@ require File.expand_path '../../test_helper', __dir__
 class TestDeleteVirtualMachine < 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
+    compute_client = @service.instance_variable_get(:@compute_mgmt_client)
+    @virtual_machines = compute_client.virtual_machines
   end
 
   def test_delete_virtual_machine_success
-    @promise.stub :value!, true do
-      @virtual_machines.stub :delete, @promise do
-        assert @service.delete_virtual_machine('fog-test-rg', 'fog-test-server')
-      end
+    @virtual_machines.stub :delete, true do
+      assert @service.delete_virtual_machine('fog-test-rg', 'fog-test-server')
     end
   end
 
   def test_delete_virtual_machine_failure
-    response = -> { fail MsRestAzure::AzureOperationError.new(nil, nil, 'error' => { 'message' => 'mocked exception' }) }
-    @promise.stub :value!, response do
-      @virtual_machines.stub :delete, @promise do
-        assert_raises(RuntimeError) { @service.delete_virtual_machine('fog-test-rg', 'fog-test-server') }
-      end
+    response = proc { fail MsRestAzure::AzureOperationError.new(nil, nil, 'error' => { 'message' => 'mocked exception' }) }
+    @virtual_machines.stub :delete, response do
+      assert_raises(RuntimeError) { @service.delete_virtual_machine('fog-test-rg', 'fog-test-server') }
     end
   end
 end

data/test/requests/compute/test_detach_data_disk_from_vm.rb

@@ -4,48 +4,27 @@ require File.expand_path '../../test_helper', __dir__
 class TestDetachDataDiskFromVM < Minitest::Test
   def setup
     @service = Fog::Compute::AzureRM.new(credentials)
-    vm_client = @service.instance_variable_get(:@compute_mgmt_client)
-    @virtual_machines = vm_client.virtual_machines
+    @compute_client = @service.instance_variable_get(:@compute_mgmt_client)
+    @virtual_machines = @compute_client.virtual_machines
   end
 
   def test_detach_data_disk_from_vm_success
-    promise_get_vm = Concurrent::Promise.execute do
-    end
-
-    promise_update_vm = Concurrent::Promise.execute do
-    end
-
-    get_vm_response = ApiStub::Requests::Compute::VirtualMachine.create_virtual_machine_response
-    update_vm_response = ApiStub::Requests::Compute::VirtualMachine.detach_data_disk_from_vm_response
-    expected_update_vm_response = Azure::ARM::Compute::Models::VirtualMachine.serialize_object(update_vm_response.body)
-    promise_get_vm.stub :value!, get_vm_response do
-      @virtual_machines.stub :get, promise_get_vm do
-        promise_update_vm.stub :value!, update_vm_response do
-          @virtual_machines.stub :create_or_update, promise_update_vm do
-            assert_equal @service.detach_data_disk_from_vm('fog-test-rg', 'fog-test-vm', 'mydatadisk1'), expected_update_vm_response
-          end
-        end
+    get_vm_response = ApiStub::Requests::Compute::VirtualMachine.create_virtual_machine_response(@compute_client)
+    update_vm_response = ApiStub::Requests::Compute::VirtualMachine.detach_data_disk_from_vm_response(@compute_client)
+    @virtual_machines.stub :get, get_vm_response do
+      @virtual_machines.stub :create_or_update, update_vm_response do
+        assert_equal @service.detach_data_disk_from_vm('fog-test-rg', 'fog-test-vm', 'mydatadisk1'), update_vm_response
       end
     end
   end
 
   def test_detach_data_disk_from_vm_failure
-    promise_get_vm = Concurrent::Promise.execute do
-    end
-
-    promise_update_vm = 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' }) }
-    promise_get_vm.stub :value!, get_vm_response do
-      @virtual_machines.stub :get, promise_get_vm do
-        promise_update_vm.stub :value!, update_vm_response do
-          @virtual_machines.stub :create_or_update, promise_update_vm do
-            assert_raises RuntimeError do
-              @service.detach_data_disk_from_vm('fog-test-rg', 'fog-test-vm', 'mydatadisk1')
-            end
-          end
+    get_vm_response = ApiStub::Requests::Compute::VirtualMachine.create_virtual_machine_response(@compute_client)
+    update_vm_response = proc { fail MsRestAzure::AzureOperationError.new(nil, nil, 'error' => { 'message' => 'mocked exception' }) }
+    @virtual_machines.stub :get, get_vm_response do
+      @virtual_machines.stub :create_or_update, update_vm_response do
+        assert_raises RuntimeError do
+          @service.detach_data_disk_from_vm('fog-test-rg', 'fog-test-vm', 'mydatadisk1')
         end
       end
     end

data/test/requests/compute/test_generalize_virtual_machine.rb

@@ -4,26 +4,20 @@ require File.expand_path '../../test_helper', __dir__
 class TestGeneralizeVirtualMachine < 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
+    compute_client = @service.instance_variable_get(:@compute_mgmt_client)
+    @virtual_machines = compute_client.virtual_machines
   end
 
   def test_generalize_virtual_machine_success
-    @promise.stub :value!, true do
-      @virtual_machines.stub :generalize, @promise do
-        assert @service.generalize_virtual_machine('fog-test-rg', 'fog-test-server')
-      end
+    @virtual_machines.stub :generalize, true do
+      assert @service.generalize_virtual_machine('fog-test-rg', 'fog-test-server')
     end
   end
 
   def test_generalize_virtual_machine_failure
-    response = -> { fail MsRestAzure::AzureOperationError.new(nil, nil, 'error' => { 'message' => 'mocked exception' }) }
-    @promise.stub :value!, response do
-      @virtual_machines.stub :generalize, @promise do
-        assert_raises(RuntimeError) { @service.generalize_virtual_machine('fog-test-rg', 'fog-test-server') }
-      end
+    response = proc { fail MsRestAzure::AzureOperationError.new(nil, nil, 'error' => { 'message' => 'mocked exception' }) }
+    @virtual_machines.stub :generalize, response do
+      assert_raises(RuntimeError) { @service.generalize_virtual_machine('fog-test-rg', 'fog-test-server') }
     end
   end
 end

data/test/requests/compute/test_get_availability_set.rb

@@ -0,0 +1,24 @@
+require File.expand_path '../../test_helper', __dir__
+
+# Test class for List Availability Sets Request
+class TestGetAvailabilitySet < Minitest::Test
+  def setup
+    @service = Fog::Compute::AzureRM.new(credentials)
+    @client = @service.instance_variable_get(:@compute_mgmt_client)
+    @availability_sets = @client.availability_sets
+  end
+
+  def test_list_availability_set_success
+    mocked_response = ApiStub::Requests::Compute::AvailabilitySet.get_availability_set_response(@client)
+    @availability_sets.stub :get, mocked_response do
+      assert_equal @service.get_availability_set('myrg1', 'myavset1'), mocked_response
+    end
+  end
+
+  def test_list_availability_set_failure
+    response = proc { raise MsRestAzure::AzureOperationError.new(nil, nil, 'error' => { 'message' => 'mocked exception' }) }
+    @availability_sets.stub :get, response do
+      assert_raises(RuntimeError) { @service.get_availability_set('myrg1', 'myavset1') }
+    end
+  end
+end

data/test/requests/compute/test_get_virtual_machine.rb

@@ -4,28 +4,21 @@ require File.expand_path '../../test_helper', __dir__
 class TestGetVirtualMachine < 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
+    @compute_client = @service.instance_variable_get(:@compute_mgmt_client)
+    @virtual_machines = @compute_client.virtual_machines
   end
 
   def test_get_virtual_machine_success
-    response = ApiStub::Requests::Compute::VirtualMachine.create_virtual_machine_response
-    compare_result = ApiStub::Requests::Compute::VirtualMachine.virtual_machine_response
-    @promise.stub :value!, response do
-      @virtual_machines.stub :get, @promise do
-        assert_equal compare_result, @service.get_virtual_machine('fog-test-rg', 'fog-test-server')
-      end
+    response = ApiStub::Requests::Compute::VirtualMachine.create_virtual_machine_response(@compute_client)
+    @virtual_machines.stub :get, response do
+      assert_equal @service.get_virtual_machine('fog-test-rg', 'fog-test-server'), response
     end
   end
 
   def test_get_virtual_machine_failure
-    response = -> { fail MsRestAzure::AzureOperationError.new(nil, nil, 'error' => { 'message' => 'mocked exception' }) }
-    @promise.stub :value!, response do
-      @virtual_machines.stub :get, @promise do
-        assert_raises(RuntimeError) { @service.get_virtual_machine('fog-test-rg', 'fog-test-server') }
-      end
+    response = proc { fail MsRestAzure::AzureOperationError.new(nil, nil, 'error' => { 'message' => 'mocked exception' }) }
+    @virtual_machines.stub :get, response do
+      assert_raises(RuntimeError) { @service.get_virtual_machine('fog-test-rg', 'fog-test-server') }
     end
   end
 end

data/test/requests/compute/test_get_vm_status.rb

@@ -4,28 +4,22 @@ require File.expand_path '../../test_helper', __dir__
 class TestGetVirtualMachineStatus < 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
+    @compute_client = @service.instance_variable_get(:@compute_mgmt_client)
+    @virtual_machines = @compute_client.virtual_machines
   end
 
   def test_vm_status_success
-    response = ApiStub::Requests::Compute::VirtualMachine.virtual_machine_instance_view_response
+    response = ApiStub::Requests::Compute::VirtualMachine.virtual_machine_instance_view_response(@compute_client)
     compare_result = ApiStub::Requests::Compute::VirtualMachine.vm_status_response
-    @promise.stub :value!, response do
-      @virtual_machines.stub :get, @promise do
-        assert_equal compare_result, @service.check_vm_status('fog-test-rg', 'fog-test-server')
-      end
+    @virtual_machines.stub :get, response do
+      assert_equal @service.check_vm_status('fog-test-rg', 'fog-test-server'), compare_result
     end
   end
 
   def test_vm_status_failure
-    response = -> { fail MsRestAzure::AzureOperationError.new(nil, nil, 'error' => { 'message' => 'mocked exception' }) }
-    @promise.stub :value!, response do
-      @virtual_machines.stub :get, @promise do
-        assert_raises(RuntimeError) { @service.check_vm_status('fog-test-rg', 'fog-test-server') }
-      end
+    response = proc { fail MsRestAzure::AzureOperationError.new(nil, nil, 'error' => { 'message' => 'mocked exception' }) }
+    @virtual_machines.stub :get, response do
+      assert_raises(RuntimeError) { @service.check_vm_status('fog-test-rg', 'fog-test-server') }
     end
   end
 end

data/test/requests/compute/test_list_availability_sets.rb

@@ -4,27 +4,21 @@ require File.expand_path '../../test_helper', __dir__
 class TestListAvailabilitySet < 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
+    @client = @service.instance_variable_get(:@compute_mgmt_client)
+    @availability_sets = @client.availability_sets
   end
 
   def test_list_availability_set_success
-    response = ApiStub::Requests::Compute::AvailabilitySet.list_availability_set_response
-    @promise.stub :value!, response do
-      @availability_sets.stub :list, @promise do
-        assert @service.list_availability_sets('fog-test-rg'), response
-      end
+    mocked_response = ApiStub::Requests::Compute::AvailabilitySet.list_availability_set_response(@client)
+    @availability_sets.stub :list, mocked_response do
+      assert_equal @service.list_availability_sets('fog-test-rg'), mocked_response.value
     end
   end
 
   def test_list_availability_set_failure
-    response = -> { fail MsRestAzure::AzureOperationError.new(nil, nil, 'error' => { 'message' => 'mocked exception' }) }
-    @promise.stub :value!, response do
-      @availability_sets.stub :list, @promise do
-        assert_raises(RuntimeError) { @service.list_availability_sets('fog-test-rg') }
-      end
+    response = proc { raise MsRestAzure::AzureOperationError.new(nil, nil, 'error' => { 'message' => 'mocked exception' }) }
+    @availability_sets.stub :list, response do
+      assert_raises(RuntimeError) { @service.list_availability_sets('fog-test-rg') }
     end
   end
 end

data/test/requests/compute/test_list_available_sizes_for_virtual_machine.rb

@@ -4,28 +4,21 @@ require File.expand_path '../../test_helper', __dir__
 class TestListAvailableSizesForVirtualMachine < 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
+    @compute_client = @service.instance_variable_get(:@compute_mgmt_client)
+    @virtual_machines = @compute_client.virtual_machines
   end
 
   def test_list_available_sizes_for_virtual_machine_success
-    response = ApiStub::Requests::Compute::VirtualMachine.list_available_sizes_for_virtual_machine_response
-    compare_value = Azure::ARM::Compute::Models::VirtualMachineSizeListResult.serialize_object(response.body)['value']
-    @promise.stub :value!, response do
-      @virtual_machines.stub :list_available_sizes, @promise do
-        assert_equal compare_value, @service.list_available_sizes_for_virtual_machine('fog-test-rg', 'fog-test-server')
-      end
+    response = ApiStub::Requests::Compute::VirtualMachine.list_available_sizes_for_virtual_machine_response(@compute_client)
+    @virtual_machines.stub :list_available_sizes, response do
+      assert_equal @service.list_available_sizes_for_virtual_machine('fog-test-rg', 'fog-test-server'), response.value
     end
   end
 
   def test_list_available_sizes_for_virtual_machine_failure
-    response = -> { fail MsRestAzure::AzureOperationError.new(nil, nil, 'error' => { 'message' => 'mocked exception' }) }
-    @promise.stub :value!, response do
-      @virtual_machines.stub :list_available_sizes, @promise do
-        assert_raises(RuntimeError) { @service.list_available_sizes_for_virtual_machine('fog-test-rg', 'fog-test-server') }
-      end
+    response = proc { fail MsRestAzure::AzureOperationError.new(nil, nil, 'error' => { 'message' => 'mocked exception' }) }
+    @virtual_machines.stub :list_available_sizes, response do
+      assert_raises(RuntimeError) { @service.list_available_sizes_for_virtual_machine('fog-test-rg', 'fog-test-server') }
     end
   end
 end

data/test/requests/compute/test_list_virtual_machines.rb

@@ -4,27 +4,21 @@ require File.expand_path '../../test_helper', __dir__
 class TestListVirtualMachines < 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
+    @compute_client = @service.instance_variable_get(:@compute_mgmt_client)
+    @virtual_machines = @compute_client.virtual_machines
   end
 
   def test_list_virtual_machines_success
-    response = ApiStub::Requests::Compute::VirtualMachine.list_virtual_machines_response
-    @promise.stub :value!, response do
-      @virtual_machines.stub :list, @promise do
-        assert_equal @service.list_virtual_machines('fog-test-rg'), Azure::ARM::Compute::Models::VirtualMachineListResult.serialize_object(response.body)['value']
-      end
+    response = ApiStub::Requests::Compute::VirtualMachine.list_virtual_machines_response(@compute_client)
+    @virtual_machines.stub :list_as_lazy, response do
+      assert_equal @service.list_virtual_machines('fog-test-rg'), response.value
     end
   end
 
   def test_list_virtual_machines_failure
-    response = -> { fail MsRestAzure::AzureOperationError.new(nil, nil, 'error' => { 'message' => 'mocked exception' }) }
-    @promise.stub :value!, response do
-      @virtual_machines.stub :list, @promise do
-        assert_raises(RuntimeError) { @service.list_virtual_machines('fog-test-rg') }
-      end
+    response = proc { fail MsRestAzure::AzureOperationError.new(nil, nil, 'error' => { 'message' => 'mocked exception' }) }
+    @virtual_machines.stub :list_as_lazy, response do
+      assert_raises(RuntimeError) { @service.list_virtual_machines('fog-test-rg') }
     end
   end
 end

data/test/requests/compute/test_power_off_virtual_machine.rb

@@ -4,26 +4,20 @@ require File.expand_path '../../test_helper', __dir__
 class TestPowerOffVirtualMachine < 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
+    compute_client = @service.instance_variable_get(:@compute_mgmt_client)
+    @virtual_machines = compute_client.virtual_machines
   end
 
   def test_power_off_virtual_machine_success
-    @promise.stub :value!, true do
-      @virtual_machines.stub :power_off, @promise do
-        assert @service.power_off_virtual_machine('fog-test-rg', 'fog-test-server')
-      end
+    @virtual_machines.stub :power_off, true do
+      assert @service.power_off_virtual_machine('fog-test-rg', 'fog-test-server')
     end
   end
 
   def test_power_off_virtual_machine_failure
-    response = -> { fail MsRestAzure::AzureOperationError.new(nil, nil, 'error' => { 'message' => 'mocked exception' }) }
-    @promise.stub :value!, response do
-      @virtual_machines.stub :power_off, @promise do
-        assert_raises(RuntimeError) { @service.power_off_virtual_machine('fog-test-rg', 'fog-test-server') }
-      end
+    response = proc { fail MsRestAzure::AzureOperationError.new(nil, nil, 'error' => { 'message' => 'mocked exception' }) }
+    @virtual_machines.stub :power_off, response do
+      assert_raises(RuntimeError) { @service.power_off_virtual_machine('fog-test-rg', 'fog-test-server') }
     end
   end
 end

data/test/requests/compute/test_redeploy_virtual_machine.rb

@@ -4,26 +4,20 @@ require File.expand_path '../../test_helper', __dir__
 class TestRedeployVirtualMachine < 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
+    compute_client = @service.instance_variable_get(:@compute_mgmt_client)
+    @virtual_machines = compute_client.virtual_machines
   end
 
   def test_redeploy_virtual_machine_success
-    @promise.stub :value!, true do
-      @virtual_machines.stub :redeploy, @promise do
-        assert @service.redeploy_virtual_machine('fog-test-rg', 'fog-test-server')
-      end
+    @virtual_machines.stub :redeploy, true do
+      assert @service.redeploy_virtual_machine('fog-test-rg', 'fog-test-server')
     end
   end
 
   def test_redeploy_virtual_machine_failure
-    response = -> { fail MsRestAzure::AzureOperationError.new(nil, nil, 'error' => { 'message' => 'mocked exception' }) }
-    @promise.stub :value!, response do
-      @virtual_machines.stub :redeploy, @promise do
-        assert_raises(RuntimeError) { @service.redeploy_virtual_machine('fog-test-rg', 'fog-test-server') }
-      end
+    response = proc { fail MsRestAzure::AzureOperationError.new(nil, nil, 'error' => { 'message' => 'mocked exception' }) }
+    @virtual_machines.stub :redeploy, response do
+      assert_raises(RuntimeError) { @service.redeploy_virtual_machine('fog-test-rg', 'fog-test-server') }
     end
   end
 end