fog-azure-rm-temp 0.0.4 → 0.0.5

data/test/requests/compute/test_attach_data_disk_to_vm.rb

@@ -14,7 +14,7 @@ class TestAttachDataDiskToVM < Minitest::Test
     @disks = @compute_client.disks
     @get_managed_disk_response = ApiStub::Requests::Compute::VirtualMachine.get_managed_disk_response(@compute_client)
     @get_vm_managed_disk_response = ApiStub::Requests::Compute::VirtualMachine.get_vm_with_managed_disk_response(@compute_client)
-    @input_params = {vm_name: "fog-test-vm", vm_resource_group: "fog-test-rg", disk_name: "disk1", disk_size_gb: 1, storage_account_name: "mystorage1"}
+    @input_params = { vm_name: 'fog-test-vm', vm_resource_group: 'fog-test-rg', disk_name: 'disk1', disk_size_gb: 1, storage_account_name: 'mystorage1' }
   end
 
   def test_attach_data_disk_to_vm_success
@@ -37,7 +37,7 @@ class TestAttachDataDiskToVM < Minitest::Test
     @virtual_machines.stub :get, @get_vm_response do
       @disks.stub :get, @get_managed_disk_response do
         @virtual_machines.stub :create_or_update, @get_vm_managed_disk_response do
-          input_params = {vm_name: "ManagedVM", vm_resource_group: "ManagedRG", disk_name: "ManagedDataDisk1", disk_size_gb: 100, disk_resource_group: "ManagedRG"}
+          input_params = { vm_name: 'ManagedVM', vm_resource_group: 'ManagedRG', disk_name: 'ManagedDataDisk1', disk_size_gb: 100, disk_resource_group: 'ManagedRG' }
           assert_equal @service.attach_data_disk_to_vm(input_params, false), @get_vm_managed_disk_response
         end
       end

data/test/requests/compute/test_check_availability_set_exists.rb

@@ -16,14 +16,21 @@ class TestCheckAvailabilitySetExists < Minitest::Test
   end
 
   def test_check_availability_set_exists_failure
-    response = proc { raise MsRestAzure::AzureOperationError.new(nil, nil, 'error' => { 'message' => 'mocked exception', 'code' => 'ResourceNotFound' }) }
+    response = proc { raise MsRestAzure::AzureOperationError.new(nil, create_mock_response, 'error' => { 'message' => 'mocked exception', 'code' => 'ResourceNotFound' }) }
+    @availability_sets.stub :get, response do
+      assert !@service.check_availability_set_exists('myrg1', 'myavset1')
+    end
+  end
+
+  def test_check_availability_set_resource_group_exists_failure
+    response = proc { raise MsRestAzure::AzureOperationError.new(nil, create_mock_response, 'error' => { 'message' => 'mocked exception', 'code' => 'ResourceGroupNotFound' }) }
     @availability_sets.stub :get, response do
       assert !@service.check_availability_set_exists('myrg1', 'myavset1')
     end
   end
 
   def test_check_availability_set_exists_exception
-    response = proc { raise MsRestAzure::AzureOperationError.new(nil, nil, 'error' => { 'message' => 'mocked exception', 'code' => 'ResourceGroupNotFound' }) }
+    response = proc { raise MsRestAzure::AzureOperationError.new(nil, create_mock_response, 'error' => { 'message' => 'mocked exception', 'code' => 'Exception' }) }
     @availability_sets.stub :get, response do
       assert_raises(RuntimeError) { @service.check_availability_set_exists('myrg1', 'myavset1') }
     end

data/test/requests/compute/test_check_managed_disk_exists.rb

@@ -16,14 +16,21 @@ class TestCheckManagedDiskExists < Minitest::Test
   end
 
   def test_check_managed_disk_exists_failure
-    response = proc { raise MsRestAzure::AzureOperationError.new(nil, nil, 'error' => { 'message' => 'mocked exception', 'code' => 'ResourceNotFound' }) }
+    response = proc { raise MsRestAzure::AzureOperationError.new(nil, create_mock_response, 'error' => { 'message' => 'mocked exception', 'code' => 'ResourceNotFound' }) }
+    @managed_disks.stub :get, response do
+      assert !@service.check_managed_disk_exists('myrg1', 'mydisk1')
+    end
+  end
+
+  def test_check_managed_disk_resource_group_exists_failure
+    response = proc { raise MsRestAzure::AzureOperationError.new(nil, create_mock_response, 'error' => { 'message' => 'mocked exception', 'code' => 'ResourceGroupNotFound' }) }
     @managed_disks.stub :get, response do
       assert !@service.check_managed_disk_exists('myrg1', 'mydisk1')
     end
   end
 
   def test_check_managed_disk_exists_exception
-    response = proc { raise MsRestAzure::AzureOperationError.new(nil, nil, 'error' => { 'message' => 'mocked exception', 'code' => 'ResourceGroupNotFound' }) }
+    response = proc { raise MsRestAzure::AzureOperationError.new(nil, create_mock_response, 'error' => { 'message' => 'mocked exception', 'code' => 'Exception' }) }
     @managed_disks.stub :get, response do
       assert_raises(RuntimeError) { @service.check_managed_disk_exists('myrg1', 'mydisk1') }
     end

data/test/requests/compute/test_check_vm_exists.rb

@@ -16,14 +16,21 @@ class TestCheckVirtualMachineExists < Minitest::Test
   end
 
   def test_check_vm_exists_failure
-    response = proc { raise MsRestAzure::AzureOperationError.new(nil, nil, 'error' => { 'message' => 'mocked exception', 'code' => 'ResourceNotFound' }) }
+    response = proc { raise MsRestAzure::AzureOperationError.new(nil, create_mock_response, 'error' => { 'message' => 'mocked exception', 'code' => 'ResourceNotFound' }) }
+    @virtual_machines.stub :get, response do
+      assert !@service.check_vm_exists('fog-test-rg', 'fog-test-server', false)
+    end
+  end
+
+  def test_check_vm_resource_group_exists_failure
+    response = proc { raise MsRestAzure::AzureOperationError.new(nil, create_mock_response, 'error' => { 'message' => 'mocked exception', 'code' => 'ResourceGroupNotFound' }) }
     @virtual_machines.stub :get, response do
       assert !@service.check_vm_exists('fog-test-rg', 'fog-test-server', false)
     end
   end
 
   def test_check_vm_exists_exception
-    response = proc { raise MsRestAzure::AzureOperationError.new(nil, nil, 'error' => { 'message' => 'mocked exception', 'code' => 'ResourceGroupNotFound' }) }
+    response = proc { raise MsRestAzure::AzureOperationError.new(nil, create_mock_response, 'error' => { 'message' => 'mocked exception', 'code' => 'Exception' }) }
     @virtual_machines.stub :get, response do
       assert_raises(RuntimeError) { @service.check_vm_exists('fog-test-rg', 'fog-test-server', false) }
     end

data/test/requests/compute/test_check_vm_extension_exists.rb

@@ -16,14 +16,21 @@ class TestCheckVMExtensionExists < Minitest::Test
   end
 
   def test_check_vm_extension_exists_failure
-    response = proc { raise MsRestAzure::AzureOperationError.new(nil, nil, 'error' => { 'message' => 'mocked exception', 'code' => 'ResourceNotFound' }) }
+    response = proc { raise MsRestAzure::AzureOperationError.new(nil, create_mock_response, 'error' => { 'message' => 'mocked exception', 'code' => 'ResourceNotFound' }) }
+    @vm_extension.stub :get, response do
+      assert !@service.check_vm_extension_exists('fog-test-rg', 'fog-test-vm', 'fog-test-extension')
+    end
+  end
+
+  def test_check_vm_extension_resource_group_exists_failure
+    response = proc { raise MsRestAzure::AzureOperationError.new(nil, create_mock_response, 'error' => { 'message' => 'mocked exception', 'code' => 'ResourceGroupNotFound' }) }
     @vm_extension.stub :get, response do
       assert !@service.check_vm_extension_exists('fog-test-rg', 'fog-test-vm', 'fog-test-extension')
     end
   end
 
   def test_check_vm_extension_exists_exception
-    response = proc { raise MsRestAzure::AzureOperationError.new(nil, nil, 'error' => { 'message' => 'mocked exception', 'code' => 'ResourceGroupNotFound' }) }
+    response = proc { raise MsRestAzure::AzureOperationError.new(nil, create_mock_response, 'error' => { 'message' => 'mocked exception', 'code' => 'Exception' }) }
     @vm_extension.stub :get, response do
       assert_raises(RuntimeError) { @service.check_vm_extension_exists('fog-test-rg', 'fog-test-vm', 'fog-test-extension') }
     end

data/test/requests/compute/test_create_image.rb

@@ -0,0 +1,27 @@
+require File.expand_path('../../test_helper', __dir__)
+
+# Test class for Create Image Request
+class TestCreateImage < Minitest::Test
+  def setup
+    @service = Fog::Compute::AzureRM.new(credentials)
+    compute_client = @service.instance_variable_get(:@compute_mgmt_client)
+    @image = compute_client.images
+    @response = ApiStub::Requests::Compute::Image.create_image(compute_client)
+    @input_params = ApiStub::Requests::Compute::Image.image_params
+  end
+
+  def test_create_image_success
+    @image.stub :create_or_update, @response do
+      assert_equal @service.create_image(@input_params), @response
+    end
+  end
+
+  def test_create_image_failure
+    response = proc { raise MsRestAzure::AzureOperationError.new(nil, nil, 'error' => { 'message' => 'mocked exception' }) }
+    @image.stub :create_or_update, response do
+      assert_raises RuntimeError do
+        @service.create_image(@input_params)
+      end
+    end
+  end
+end

data/test/requests/compute/test_create_virtual_machine.rb

@@ -10,80 +10,104 @@ class TestCreateVirtualMachine < Minitest::Test
     @custom_data_response = ApiStub::Requests::Compute::VirtualMachine.create_virtual_machine_with_custom_data_response(compute_client)
     @linux_virtual_machine_hash = ApiStub::Requests::Compute::VirtualMachine.linux_virtual_machine_params
     @windows_virtual_machine_hash = ApiStub::Requests::Compute::VirtualMachine.windows_virtual_machine_params
+    @error_response = proc { raise MsRestAzure::AzureOperationError.new(nil, nil, 'error' => { 'message' => 'mocked exception' }) }
   end
 
   def test_create_linux_virtual_machine_success
     @virtual_machines.stub :create_or_update, @response do
-      assert_equal @service.create_virtual_machine(@linux_virtual_machine_hash), @response
+      @virtual_machines.stub :get, @error_response do
+        assert_equal @service.create_virtual_machine(@linux_virtual_machine_hash), @response
+      end
     end
 
     # Async
     @virtual_machines.stub :create_or_update_async, @response do
-      assert_equal @service.create_virtual_machine(@linux_virtual_machine_hash, true), @response
+      @virtual_machines.stub :get, @error_response do
+        assert_equal @service.create_virtual_machine(@linux_virtual_machine_hash, true), @response
+      end
     end
   end
 
   def test_create_windows_virtual_machine_success
     @virtual_machines.stub :create_or_update, @response do
-      assert_equal @service.create_virtual_machine(@windows_virtual_machine_hash), @response
+      @virtual_machines.stub :get, @error_response do
+        assert_equal @service.create_virtual_machine(@windows_virtual_machine_hash), @response
+      end
     end
 
     # Async
     @virtual_machines.stub :create_or_update_async, @response do
-      assert_equal @service.create_virtual_machine(@windows_virtual_machine_hash, true), @response
+      @virtual_machines.stub :get, @error_response do
+        assert_equal @service.create_virtual_machine(@windows_virtual_machine_hash, true), @response
+      end
     end
   end
 
   def test_create_linux_virtual_machine_from_custom_image_success
     linux_virtual_machine_with_custom_image_hash = ApiStub::Requests::Compute::VirtualMachine.linux_virtual_machine_with_custom_image_params
     @virtual_machines.stub :create_or_update, @response do
-      assert_equal @service.create_virtual_machine(linux_virtual_machine_with_custom_image_hash), @response
+      @virtual_machines.stub :get, @error_response do
+        assert_equal @service.create_virtual_machine(linux_virtual_machine_with_custom_image_hash), @response
+      end
     end
 
     # Async
     @virtual_machines.stub :create_or_update_async, @response do
-      assert_equal @service.create_virtual_machine(linux_virtual_machine_with_custom_image_hash, true), @response
+      @virtual_machines.stub :get, @error_response do
+        assert_equal @service.create_virtual_machine(linux_virtual_machine_with_custom_image_hash, true), @response
+      end
     end
   end
 
   def test_create_windows_virtual_machine_from_custom_image_success
     windows_virtual_machine_with_custom_image_hash = ApiStub::Requests::Compute::VirtualMachine.windows_virtual_machine_with_custom_image_params
     @virtual_machines.stub :create_or_update, @response do
-      assert_equal @service.create_virtual_machine(windows_virtual_machine_with_custom_image_hash), @response
+      @virtual_machines.stub :get, @error_response do
+        assert_equal @service.create_virtual_machine(windows_virtual_machine_with_custom_image_hash), @response
+      end
     end
 
     # Async
     @virtual_machines.stub :create_or_update_async, @response do
-      assert_equal @service.create_virtual_machine(windows_virtual_machine_with_custom_image_hash, true), @response
+      @virtual_machines.stub :get, @error_response do
+        assert_equal @service.create_virtual_machine(windows_virtual_machine_with_custom_image_hash, true), @response
+      end
     end
   end
 
   def test_create_linux_virtual_machine_with_custom_data_success
     linux_virtual_machine_with_custom_data_hash = ApiStub::Requests::Compute::VirtualMachine.linux_virtual_machine_with_custom_data_params
     @virtual_machines.stub :create_or_update, @custom_data_response do
-      assert_equal @service.create_virtual_machine(linux_virtual_machine_with_custom_data_hash), @custom_data_response
+      @virtual_machines.stub :get, @error_response do
+        assert_equal @service.create_virtual_machine(linux_virtual_machine_with_custom_data_hash), @custom_data_response
+      end
     end
   end
 
   def test_create_windows_virtual_machine_with_custom_data_success
     windows_virtual_machine_with_custom_data_hash = ApiStub::Requests::Compute::VirtualMachine.windows_virtual_machine_with_custom_data_params
     @virtual_machines.stub :create_or_update, @custom_data_response do
-      assert_equal @service.create_virtual_machine(windows_virtual_machine_with_custom_data_hash), @custom_data_response
+      @virtual_machines.stub :get, @error_response do
+        assert_equal @service.create_virtual_machine(windows_virtual_machine_with_custom_data_hash), @custom_data_response
+      end
     end
   end
 
   def test_create_virtual_machine_failure
-    response = proc { raise MsRestAzure::AzureOperationError.new(nil, nil, 'error' => { 'message' => 'mocked exception' }) }
-    @virtual_machines.stub :create_or_update, response do
-      assert_raises RuntimeError do
-        @service.create_virtual_machine(@linux_virtual_machine_hash)
+    @virtual_machines.stub :create_or_update, @error_response do
+      @virtual_machines.stub :get, @error_response do
+        assert_raises RuntimeError do
+          @service.create_virtual_machine(@linux_virtual_machine_hash)
+        end
       end
     end
 
     # Async
-    @virtual_machines.stub :create_or_update_async, response do
-      assert_raises RuntimeError do
-        @service.create_virtual_machine(@linux_virtual_machine_hash, true)
+    @virtual_machines.stub :create_or_update_async, @error_response do
+      @virtual_machines.stub :get, @error_response do
+        assert_raises RuntimeError do
+          @service.create_virtual_machine(@linux_virtual_machine_hash, true)
+        end
       end
     end
   end

data/test/requests/compute/test_delete_image.rb

@@ -0,0 +1,23 @@
+require File.expand_path('../../test_helper', __dir__)
+
+# Test class for Delete Image Request
+class TestDeleteImage < Minitest::Test
+  def setup
+    @service = Fog::Compute::AzureRM.new(credentials)
+    compute_client = @service.instance_variable_get(:@compute_mgmt_client)
+    @image = compute_client.images
+  end
+
+  def test_delete_image_success
+    @image.stub :delete, true do
+      assert @service.delete_image('fog-test-rg', 'fog-test-server-osImage')
+    end
+  end
+
+  def test_delete_image_failure
+    response = proc { raise MsRestAzure::AzureOperationError.new(nil, nil, 'error' => { 'message' => 'mocked exception' }) }
+    @image.stub :delete, response do
+      assert_raises(RuntimeError) { @service.delete_image('fog-test-rg', 'fog-test-server-osImage') }
+    end
+  end
+end

data/test/requests/compute/test_get_image.rb

@@ -0,0 +1,23 @@
+require File.expand_path('../../test_helper', __dir__)
+
+# Test class for Get Image Request
+class TestGetImage < Minitest::Test
+  def setup
+    @service = Fog::Compute::AzureRM.new(credentials)
+    compute_client = @service.instance_variable_get(:@compute_mgmt_client)
+    @image = compute_client.images
+  end
+
+  def test_get_image_success
+    @image.stub :get, true do
+      assert @service.get_image('fog-test-rg', 'TestImage')
+    end
+  end
+
+  def test_get_image_failure
+    response = proc { raise MsRestAzure::AzureOperationError.new(nil, nil, 'error' => { 'message' => 'mocked exception' }) }
+    @image.stub :get, response do
+      assert_raises(RuntimeError) { @service.get_image('fog-test-rg', 'TestImage') }
+    end
+  end
+end

data/test/requests/dns/test_check_record_set_exists.rb

@@ -16,14 +16,21 @@ class TestCheckRecordSetExists < Minitest::Test
   end
 
   def test_check_record_set_exists_failure
-    response = proc { raise MsRestAzure::AzureOperationError.new(nil, nil, 'code' => 'NotFound' ) }
+    response = proc { raise MsRestAzure::AzureOperationError.new(nil, create_mock_response, 'code' => 'NotFound') }
+    @record_sets.stub :get, response do
+      assert !@service.check_record_set_exists('fog-test-rg', 'fog-test-result', 'fog-test-zone', 'CNAME')
+    end
+  end
+
+  def test_check_record_set_resource_group_exists_failure
+    response = proc { raise MsRestAzure::AzureOperationError.new(nil, create_mock_response, 'error' => { 'message' => 'mocked exception', 'code' => 'ResourceGroupNotFound' }) }
     @record_sets.stub :get, response do
       assert !@service.check_record_set_exists('fog-test-rg', 'fog-test-result', 'fog-test-zone', 'CNAME')
     end
   end
 
   def test_check_record_set_exists_exception
-    response = proc { raise MsRestAzure::AzureOperationError.new(nil, nil, 'error' => { 'message' => 'mocked exception', 'code' => 'ResourceGroupNotFound' }) }
+    response = proc { raise MsRestAzure::AzureOperationError.new(nil, create_mock_response, 'error' => { 'message' => 'mocked exception', 'code' => 'Exception' }) }
     @record_sets.stub :get, response do
       assert_raises(RuntimeError) { @service.check_record_set_exists('fog-test-rg', 'fog-test-result', 'fog-test-zone', 'CNAME') }
     end

data/test/requests/dns/test_check_zone_exists.rb

@@ -15,7 +15,7 @@ class TestCheckZoneExists < Minitest::Test
     end
   end
 
-  def test_check_zone_exists_failure
+  def test_check_zone_arguement_error
     response = ApiStub::Requests::DNS::RecordSet.list_record_sets_response(@dns_client)
     @zones.stub :get, response do
       assert_raises ArgumentError do
@@ -24,8 +24,22 @@ class TestCheckZoneExists < Minitest::Test
     end
   end
 
+  def test_check_zone_exists_failure
+    response = proc { raise MsRestAzure::AzureOperationError.new(nil, create_mock_response, 'error' => { 'message' => 'mocked exception', 'code' => 'ResourceNotFound' }) }
+    @zones.stub :get, response do
+      assert !@service.check_zone_exists('fog-test-rg', 'zone_name')
+    end
+  end
+
+  def test_check_zone_resource_group_exists_failure
+    response = proc { raise MsRestAzure::AzureOperationError.new(nil, create_mock_response, 'error' => { 'message' => 'mocked exception', 'code' => 'ResourceGroupNotFound' }) }
+    @zones.stub :get, response do
+      assert !@service.check_zone_exists('fog-test-rg', 'zone_name')
+    end
+  end
+
   def test_check_zone_exists_exception
-    response = proc { raise MsRestAzure::AzureOperationError.new(nil, nil, 'error' => { 'message' => 'mocked exception' }) }
+    response = proc { raise MsRestAzure::AzureOperationError.new(nil, create_mock_response, 'error' => { 'message' => 'mocked exception', 'code' => 'Exception' }) }
     @zones.stub :get, response do
       assert_raises RuntimeError do
         @service.check_zone_exists('fog-test-rg', 'zone_name')

data/test/requests/key_vault/test_check_vault_exists.rb

@@ -16,14 +16,21 @@ class TestCheckVaultExists < Minitest::Test
   end
 
   def test_check_vault_exists_failure
-    response = proc { raise MsRestAzure::AzureOperationError.new(nil, nil, 'error' => { 'message' => 'mocked exception', 'code' => 'ResourceNotFound' }) }
+    response = proc { raise MsRestAzure::AzureOperationError.new(nil, create_mock_response, 'error' => { 'message' => 'mocked exception', 'code' => 'ResourceNotFound' }) }
+    @vaults.stub :get, response do
+      assert !@service.check_vault_exists('fog-test-rg', 'fog-test-kv')
+    end
+  end
+
+  def test_check_vault_resource_group_exists_failure
+    response = proc { raise MsRestAzure::AzureOperationError.new(nil, create_mock_response, 'error' => { 'message' => 'mocked exception', 'code' => 'ResourceGroupNotFound' }) }
     @vaults.stub :get, response do
       assert !@service.check_vault_exists('fog-test-rg', 'fog-test-kv')
     end
   end
 
   def test_check_vault_exists_exception
-    response = proc { raise MsRestAzure::AzureOperationError.new(nil, nil, 'error' => { 'message' => 'mocked exception', 'code' => 'ResourceGroupNotFound' }) }
+    response = proc { raise MsRestAzure::AzureOperationError.new(nil, create_mock_response, 'error' => { 'message' => 'mocked exception', 'code' => 'Exception' }) }
     @vaults.stub :get, response do
       assert_raises(RuntimeError) { @service.check_vault_exists('fog-test-rg', 'fog-test-kv') }
     end

data/test/requests/network/test_check_express_route_cir_auth_exists.rb

@@ -16,14 +16,21 @@ class TestCheckExpressRouteCirAuthExists < Minitest::Test
   end
 
   def test_check_express_route_cir_auth_exists_failure
-    response = proc { raise MsRestAzure::AzureOperationError.new(nil, nil, 'error' => { 'message' => 'mocked exception', 'code' => 'ResourceNotFound' }) }
+    response = proc { raise MsRestAzure::AzureOperationError.new(nil, create_mock_response, 'error' => { 'message' => 'mocked exception', 'code' => 'NotFound' }) }
+    @circuit_authorization.stub :get, response do
+      assert !@service.check_express_route_cir_auth_exists('Fog-rg', 'testCircuit', 'auth-name')
+    end
+  end
+
+  def test_check_express_route_cir_auth_resource_group_exists_failure
+    response = proc { raise MsRestAzure::AzureOperationError.new(nil, create_mock_response, 'error' => { 'message' => 'mocked exception', 'code' => 'ResourceGroupNotFound' }) }
     @circuit_authorization.stub :get, response do
       assert !@service.check_express_route_cir_auth_exists('Fog-rg', 'testCircuit', 'auth-name')
     end
   end
 
   def test_check_express_route_cir_auth_exists_exception
-    response = proc { raise MsRestAzure::AzureOperationError.new(nil, nil, 'error' => { 'message' => 'mocked exception', 'code' => 'ResourceGroupNotFound' }) }
+    response = proc { raise MsRestAzure::AzureOperationError.new(nil, create_mock_response, 'error' => { 'message' => 'mocked exception', 'code' => 'Exception' }) }
     @circuit_authorization.stub :get, response do
       assert_raises(RuntimeError) { @service.check_express_route_cir_auth_exists('Fog-rg', 'testCircuit', 'auth-name') }
     end

data/test/requests/network/test_check_express_route_circuit_exists.rb

@@ -16,14 +16,21 @@ class TestCheckExpressRouteCircuitExists < Minitest::Test
   end
 
   def test_check_express_route_circuit_exists_failure
-    response = proc { raise MsRestAzure::AzureOperationError.new(nil, nil, 'error' => { 'message' => 'mocked exception', 'code' => 'ResourceNotFound' }) }
+    response = proc { raise MsRestAzure::AzureOperationError.new(nil, create_mock_response, 'error' => { 'message' => 'mocked exception', 'code' => 'ResourceNotFound' }) }
+    @circuit.stub :get, response do
+      assert !@service.check_express_route_circuit_exists('fog-test-rg', 'testCircuit')
+    end
+  end
+
+  def test_check_express_route_circuit_resource_group_exists_failure
+    response = proc { raise MsRestAzure::AzureOperationError.new(nil, create_mock_response, 'error' => { 'message' => 'mocked exception', 'code' => 'ResourceGroupNotFound' }) }
     @circuit.stub :get, response do
       assert !@service.check_express_route_circuit_exists('fog-test-rg', 'testCircuit')
     end
   end
 
   def test_check_express_route_circuit_exists_exception
-    response = proc { raise MsRestAzure::AzureOperationError.new(nil, nil, 'error' => { 'message' => 'mocked exception', 'code' => 'ResourceGroupNotFound' }) }
+    response = proc { raise MsRestAzure::AzureOperationError.new(nil, create_mock_response, 'error' => { 'message' => 'mocked exception', 'code' => 'Exception' }) }
     @circuit.stub :get, response do
       assert_raises(RuntimeError) { @service.check_express_route_circuit_exists('fog-test-rg', 'testCircuit') }
     end