fog-azure-rm 0.1.0 → 0.1.1

data/test/requests/storage/test_release_blob_lease.rb

@@ -1,16 +1,34 @@
 require File.expand_path '../../test_helper', __dir__
 
 # Storage Blob Class
-class TestListBlobs < Minitest::Test
+class TestReleaseBlobLease < Minitest::Test
   # This class posesses the test cases for the requests of releasing blob lease.
   def setup
+    Fog.mock!
+    @mock_service = Fog::Storage::AzureRM.new(storage_account_credentials)
+    Fog.unmock!
+    @mocked_response = mocked_storage_http_error
+
     @service = Fog::Storage::AzureRM.new(storage_account_credentials)
     @blob_client = @service.instance_variable_get(:@blob_client)
   end
 
   def test_release_blob_lease_success
     @blob_client.stub :release_blob_lease, true do
-      assert @service.release_blob_lease('test_container', 'test_blob', '78659cee-b7cb-4f18-b9f8-0e9e8bf849f6'), true
+      assert @service.release_blob_lease('test_container', 'test_blob', 'lease_id')
+    end
+  end
+
+  def test_release_blob_lease_http_exception
+    http_exception = ->(*) { raise Azure::Core::Http::HTTPError.new(@mocked_response) }
+    @blob_client.stub :release_blob_lease, http_exception do
+      assert_raises(RuntimeError) do
+        @service.release_blob_lease('test_container', 'test_blob', 'lease_id')
+      end
     end
   end
+
+  def test_release_blob_lease_mock
+    assert @mock_service.release_blob_lease('test_container', 'test_blob', 'lease_id')
+  end
 end

data/test/requests/storage/test_release_container_lease.rb

@@ -1,16 +1,34 @@
 require File.expand_path '../../test_helper', __dir__
 
-# Storage Blob Class
-class TestListBlobs < Minitest::Test
+# Storage Container Class
+class TestReleaseContainerLease < Minitest::Test
   # This class posesses the test cases for the requests of releasing container lease.
   def setup
+    Fog.mock!
+    @mock_service = Fog::Storage::AzureRM.new(storage_account_credentials)
+    Fog.unmock!
+    @mocked_response = mocked_storage_http_error
+
     @service = Fog::Storage::AzureRM.new(storage_account_credentials)
     @blob_client = @service.instance_variable_get(:@blob_client)
   end
 
   def test_release_container_lease_success
     @blob_client.stub :release_container_lease, true do
-      assert @service.release_container_lease('test_container', '78659cee-b7cb-4f18-b9f8-0e9e8bf849f6'), true
+      assert @service.release_container_lease('test_container', 'lease_id')
+    end
+  end
+
+  def test_release_container_lease_http_exception
+    http_exception = ->(*) { raise Azure::Core::Http::HTTPError.new(@mocked_response) }
+    @blob_client.stub :release_container_lease, http_exception do
+      assert_raises(RuntimeError) do
+        @service.release_container_lease('test_container', 'lease_id')
+      end
     end
   end
+
+  def test_release_container_lease_mock
+    assert @mock_service.release_container_lease('test_container', 'lease_id')
+  end
 end

data/test/requests/storage/test_save_page_blob.rb

@@ -0,0 +1,115 @@
+require File.expand_path '../../test_helper', __dir__
+
+# Storage Blob Class
+class TestSavePageBlob < Minitest::Test
+  # This class posesses the test cases for the requests of saving storage page blob.
+  def setup
+    Fog.mock!
+    @mock_service = Fog::Storage::AzureRM.new(storage_account_credentials)
+    Fog.unmock!
+    @mocked_response = mocked_storage_http_error
+
+    @service = Fog::Storage::AzureRM.new(storage_account_credentials)
+  end
+
+  def test_save_page_blob_success
+    body = 'd' * 5 * 1024 * 1024 # MAXIMUM_CHUNK_SIZE is 4 * 1024 * 1024
+
+    @service.stub :create_page_blob, true do
+      @service.stub :put_blob_pages, true do
+        assert @service.save_page_blob('test_container', 'test_blob', body, worker_thread_num: 6, content_md5: 'oafL1+HS78x65+e39PGIIg==')
+      end
+    end
+  end
+
+  def test_save_page_blob_with_file_handle_success
+    data_length = 5 * 1024 * 1024 # MAXIMUM_CHUNK_SIZE is 4 * 1024 * 1024
+    i = 0
+    multiple_values = lambda do |*|
+      i += 1
+      return 'd' * 4 * 1024 * 1024 if i == 1
+      return 'd' * 1 * 1024 * 1024 if i == 2
+      return nil
+    end
+    temp_file = '/dev/null'
+    File.open(temp_file, 'r') do |file_handle|
+      file_handle.stub :read, multiple_values do
+        file_handle.stub :size, data_length do
+          file_handle.stub :rewind, nil do
+            @service.stub :create_page_blob, true do
+              @service.stub :put_blob_pages, true do
+                assert @service.save_page_blob('test_container', 'test_blob', file_handle, worker_thread_num: 0)
+              end
+            end
+          end
+        end
+      end
+    end
+  end
+
+  def test_save_page_blob_with_file_handle_do_not_support_rewind_success
+    data_length = 5 * 1024 * 1024 # MAXIMUM_CHUNK_SIZE is 4 * 1024 * 1024
+    i = 0
+    multiple_values = lambda do |*|
+      i += 1
+      return 'd' * 4 * 1024 * 1024 if i == 1
+      return 'd' * 1 * 1024 * 1024 if i == 2
+      return nil
+    end
+    temp_file = '/dev/null'
+    exception = ->(*) { raise 'do not support rewind' }
+    File.open(temp_file, 'r') do |file_handle|
+      file_handle.stub :read, multiple_values do
+        file_handle.stub :size, data_length do
+          file_handle.stub :rewind, exception do
+            @service.stub :create_page_blob, true do
+              @service.stub :put_blob_pages, true do
+                assert @service.save_page_blob('test_container', 'test_blob', file_handle, worker_thread_num: 'invalid')
+              end
+            end
+          end
+        end
+      end
+    end
+  end
+
+  def test_save_page_blob_http_exception
+    body = 'd' * 5 * 1024 * 1024 # MAXIMUM_CHUNK_SIZE is 4 * 1024 * 1024
+
+    http_exception = ->(*) { raise Azure::Core::Http::HTTPError.new(@mocked_response) }
+    @service.stub :create_page_blob, http_exception do
+      @service.stub :delete_blob, true do
+        assert_raises(Azure::Core::Http::HTTPError) do
+          @service.save_page_blob('test_container', 'test_blob', body, {})
+        end
+      end
+    end
+  end
+
+  def test_save_page_blob_invalid_size_exception
+    body = 'd' * 1025 # The page blob size must be aligned to a 512-byte boundary.
+
+    @service.stub :delete_blob, true do
+      assert_raises(RuntimeError) do
+        @service.save_page_blob('test_container', 'test_blob', body, {})
+      end
+    end
+  end
+
+  def test_save_page_blob_fail_when_delete_blob_http_exception
+    body = 'd' * 5 * 1024 * 1024 # MAXIMUM_CHUNK_SIZE is 4 * 1024 * 1024
+
+    http_exception = ->(*) { raise Azure::Core::Http::HTTPError.new(@mocked_response) }
+    @service.stub :create_page_blob, http_exception do
+      @service.stub :delete_blob, http_exception do
+        assert_raises(Azure::Core::Http::HTTPError) do
+          @service.save_page_blob('test_container', 'test_blob', body, {})
+        end
+      end
+    end
+  end
+
+  def test_save_page_blob_mock
+    assert @mock_service.save_page_blob('test_container', 'test_blob', 'content', {})
+  end
+end

data/test/requests/storage/test_wait_blob_copy_operation_to_finish.rb

@@ -0,0 +1,148 @@
+require File.expand_path '../../test_helper', __dir__
+
+# Storage Blob Class
+class TestWaitBlobCopyOperationToFinish < Minitest::Test
+  # This class posesses the test cases for the requests of waiting storage blob copy operation to finish.
+  def setup
+    Fog.mock!
+    @mock_service = Fog::Storage::AzureRM.new(storage_account_credentials)
+    Fog.unmock!
+    @mocked_response = mocked_storage_http_error
+    @blob = storage_blob
+
+    @service = Fog::Storage::AzureRM.new(storage_account_credentials)
+    @blob_client = @service.instance_variable_get(:@blob_client)
+  end
+
+  def test_wait_blob_copy_operation_to_finish_with_success
+    copy_id = @blob.properties[:copy_id]
+    copy_status = 'pending'
+
+    i = 0
+    multiple_values = lambda do |*|
+      i += 1
+      if i == 1
+        @blob.properties[:copy_status] = copy_status
+        @blob.properties[:copy_progress] = '0/1024'
+        @blob.properties[:copy_status_description] = 'in progress'
+        return @blob
+      end
+
+      if i == 2
+        @blob.properties[:copy_status] = copy_status
+        @blob.properties[:copy_progress] = '2/1024'
+        @blob.properties[:copy_status_description] = 'in progress'
+        return @blob
+      end
+
+      if i == 3
+        @blob.properties[:copy_status] = copy_status
+        @blob.properties[:copy_progress] = '1023/1024'
+        @blob.properties[:copy_status_description] = 'in progress'
+        return @blob
+      end
+
+      @blob.properties[:copy_status] = 'success'
+      @blob.properties[:copy_progress] = '1024/1024'
+      @blob.properties[:copy_status_description] = 'finish'
+      @blob
+    end
+
+    @service.stub :get_blob_properties, multiple_values do
+      assert @service.wait_blob_copy_operation_to_finish('test_container', 'test_blob', copy_id, copy_status)
+    end
+  end
+
+  def test_wait_blob_copy_operation_to_finish_with_copy_finished_success
+    copy_id = nil
+    copy_status = 'success'
+
+    @service.stub :get_blob_properties, @blob do
+      assert @service.wait_blob_copy_operation_to_finish('test_container', 'test_blob', copy_id, copy_status)
+    end
+  end
+
+  def test_wait_blob_copy_operation_to_finish_with_copy_id_is_nil
+    copy_id = nil
+    copy_status = 'pending'
+
+    @service.stub :get_blob_properties, @blob do
+      assert @service.wait_blob_copy_operation_to_finish('test_container', 'test_blob', copy_id, copy_status)
+    end
+  end
+
+  def test_wait_blob_copy_operation_to_finish_with_copy_id_not_match_exception
+    copy_id = 'copy_id'
+    copy_status = 'pending'
+
+    @service.stub :get_blob_properties, @blob do
+      @service.stub :delete_blob, true do
+        assert_raises(RuntimeError) do
+          @service.wait_blob_copy_operation_to_finish('test_container', 'test_blob', copy_id, copy_status)
+        end
+      end
+    end
+  end
+
+  def test_wait_blob_copy_operation_to_finish_with_timeout_exception
+    copy_id = @blob.properties[:copy_id]
+    copy_status = 'pending'
+
+    i = 0
+    multiple_values = lambda do |*|
+      i += 1
+      if i == 1
+        @blob.properties[:copy_status] = copy_status
+        @blob.properties[:copy_progress] = '0/1024'
+        @blob.properties[:copy_status_description] = 'in progress'
+        return @blob
+      end
+
+      @blob.properties[:copy_status] = copy_status
+      @blob.properties[:copy_progress] = '0/1024'
+      @blob.properties[:copy_status_description] = 'in progress'
+      @blob
+    end
+
+    @service.stub :get_blob_properties, multiple_values do
+      @service.stub :delete_blob, true do
+        assert_raises(TimeoutError) do
+          @service.wait_blob_copy_operation_to_finish('test_container', 'test_blob', copy_id, copy_status, 2)
+        end
+      end
+    end
+  end
+
+  def test_wait_blob_copy_operation_to_finish_with_fail_exception
+    copy_id = @blob.properties[:copy_id]
+    copy_status = 'pending'
+
+    @blob.properties[:copy_status] = 'failed'
+    @service.stub :get_blob_properties, @blob do
+      @service.stub :delete_blob, true do
+        assert_raises(RuntimeError) do
+          @service.wait_blob_copy_operation_to_finish('test_container', 'test_blob', copy_id, copy_status)
+        end
+      end
+    end
+  end
+
+  def test_wait_blob_copy_operation_to_finish_with_fail_when_delete_blob_http_exception
+    copy_id = @blob.properties[:copy_id]
+    copy_status = 'pending'
+
+    @blob.properties[:copy_status] = 'failed'
+    http_exception = ->(*) { raise Azure::Core::Http::HTTPError.new(@mocked_response) }
+    @service.stub :get_blob_properties, @blob do
+      @service.stub :delete_blob, http_exception do
+        assert_raises(RuntimeError) do
+          @service.wait_blob_copy_operation_to_finish('test_container', 'test_blob', copy_id, copy_status)
+        end
+      end
+    end
+  end
+
+  def test_wait_blob_copy_operation_to_finish_mock
+    assert @mock_service.wait_blob_copy_operation_to_finish('test_container', 'test_blob', 'copy_id', 'pending')
+  end
+end

data/test/test_helper.rb

@@ -7,11 +7,16 @@ if ENV['COVERAGE']
 end
 
 if ENV['CODECLIMATE_REPO_TOKEN']
-  require 'codeclimate-test-reporter'
-  CodeClimate::TestReporter.start
+  require 'simplecov'
+  SimpleCov.start do
+    add_filter 'test'
+    command_name 'Minitest'
+  end
 end
 
 require 'minitest/autorun'
+require 'azure/core/http/http_error'
+require 'azure/core/http/http_response'
 $LOAD_PATH.unshift(File.expand_path '../lib', __dir__)
 require File.expand_path '../lib/fog/azurerm', __dir__
 require File.expand_path './api_stub', __dir__
@@ -65,6 +70,11 @@ def mocked_storage_http_error
   Azure::Core::Http::HttpResponse.new mocked_net_response, 'mocked_uri'
 end
 
+def mocked_storage_http_not_found_error
+  mocked_net_response = MockResponse.new '404', 'mocked_body', a: 'a', b: 'b'
+  Azure::Core::Http::HttpResponse.new mocked_net_response, 'mocked_uri'
+end
+
 def server(service)
   Fog::Compute::AzureRM::Server.new(
     name: 'fog-test-server',
@@ -170,28 +180,29 @@ def premium_check_for_invalid_replications(service)
   )
 end
 
-def storage_container(service)
+def directory(service)
   Fog::Storage::AzureRM::Directory.new(
-    key: 'storage-test-container',
-    last_modified: 'Tue, 04 Aug 2015 06:01:08 GMT',
+    key: 'test_container',
+    acl: 'container',
     etag: '0x8D29C92176C8352',
-    lease_status: 'unlocked',
-    lease_state: 'available',
+    last_modified: Time.parse('Tue, 04 Aug 2015 06:01:08 GMT'),
     lease_duration: nil,
+    lease_state: 'available',
+    lease_status: 'unlocked',
     metadata: {
       'key1' => 'value1',
       'key2' => 'value2'
     },
-    public_access_level: nil,
-    service: service
+    service: service,
+    collection: Fog::Storage::AzureRM::Directories.new(service: @service)
   )
 end
 
-def storage_blob(service)
+def file(service)
   Fog::Storage::AzureRM::File.new(
-    key: 'storage-test-blob',
-    directory: 'storage-test-container',
-    last_modified: 'Tue, 04 Aug 2015 06:01:08 GMT',
+    key: 'test_blob',
+    directory: directory(service),
+    last_modified: Time.parse('Tue, 04 Aug 2015 06:01:08 GMT'),
     etag: '0x8D29C92176C8352',
     metadata: {
       'key1' => 'value1',
@@ -208,7 +219,7 @@ def storage_blob(service)
     content_md5: 'tXAohIyxuu/t94Lp/ujeRw==',
     cache_control: nil,
     sequence_number: 0,
-    blob_type: 'PageBlob',
+    blob_type: 'BlockBlob',
     copy_id: '095adc3b-e277-4c3d-97e0-0abca881f60c',
     copy_status: 'success',
     copy_source: 'https://testaccount.blob.core.windows.net/testblob/4m?snapshot=2016-02-04T08%3A35%3A50.3157696Z',
@@ -216,14 +227,14 @@ def storage_blob(service)
     copy_completion_time: 'Thu, 04 Feb 2016 08:35:52 GMT',
     copy_status_description: nil,
     accept_ranges: 0,
-    service: service
+    service: service,
+    collection: Fog::Storage::AzureRM::Files.new(service: @service, directory: directory(service))
   )
 end
 
-def storage_cloud_blob
+def storage_blob
   mock_blob = MockBlob.new
-  mock_blob.name = 'storage-test-blob'
-  mock_blob.snapshot = nil
+  mock_blob.name = 'test_blob'
   mock_blob.properties = {
     lease_status: 'unlocked',
     lease_state: 'available',
@@ -239,11 +250,13 @@ def storage_cloud_blob
     blob_type: 'PageBlob',
     copy_id: '095adc3b-e277-4c3d-97e0-0abca881f60c',
     copy_status: 'success',
-    copy_source: 'https://testaccount.blob.core.windows.net/testblob/4m?snapshot=2016-02-04T08%3A35%3A50.3157696Z',
+    copy_source: 'https://sa.blob.core.windows.net/test_container/test_blob?snapshot=2016-02-04T08%3A35%3A50.3157696Z',
     copy_progress: '4194304/4194304',
     copy_completion_time: 'Thu, 04 Feb 2016 08:35:52 GMT',
     copy_status_description: nil,
-    accept_ranges: 0
+    accept_ranges: 0,
+    last_modified: 'Tue, 04 Aug 2015 06:01:08 GMT',
+    etag: '"0x8D29C92176C8352"'
   }
   mock_blob.metadata = {
     'key1' => 'value1',
@@ -541,7 +554,7 @@ def gateway(service)
               backend_address_pool_id: '/subscriptions/########-####-####-####-############/resourceGroups/fogRM-rg/providers/Microsoft.Network/applicationGateways/gateway/backendAddressPools/AG-BackEndAddressPool',
               backend_http_settings_id: '/subscriptions/########-####-####-####-############/resourceGroups/fogRM-rg/providers/Microsoft.Network/applicationGateways/gateway/backendHttpSettingsCollection/gateway_settings',
               paths: [
-                %w('/usr', '/etc')
+                %w(/usr /etc)
               ]
             }
           ]