fog-azure-rm 0.1.0 → 0.1.1

data/test/models/storage/test_files.rb

@@ -3,65 +3,232 @@ require File.expand_path '../../test_helper', __dir__
 # Test class for Blob Collection
 class TestFiles < Minitest::Test
   def setup
-    Fog.mock!
-    @mock_service = Fog::Storage::AzureRM.new(storage_account_credentials)
-    @mock_files = Fog::Storage::AzureRM::Files.new(directory: 'test-container', service: @mock_service)
-    Fog.unmock!
     @service = Fog::Storage::AzureRM.new(storage_account_credentials)
-    @blob_client = @service.instance_variable_get(:@blob_client)
-    @files = Fog::Storage::AzureRM::Files.new(directory: 'test-container', service: @service)
+    @directory = directory(@service)
+    @files = @directory.files
+
     @mocked_response = mocked_storage_http_error
-    @list_results = ApiStub::Models::Storage::File.list_blobs
+    @container = ApiStub::Models::Storage::File.container
+    @blob = ApiStub::Models::Storage::File.blob
+    @blob_list = ApiStub::Models::Storage::File.blob_list
+    @blob_https_url = ApiStub::Models::Storage::File.blob_https_url
   end
 
   def test_collection_methods
     methods = [
       :all,
-      :get
+      :each,
+      :each_file_this_page,
+      :get,
+      :get_url,
+      :get_http_url,
+      :get_https_url,
+      :head,
+      :new
     ]
     methods.each do |method|
       assert_respond_to @files, method
     end
   end
 
+  def test_collection_attributes
+    attributes = [
+      :directory,
+      :delimiter,
+      :marker,
+      :max_results,
+      :next_marker,
+      :prefix
+    ]
+    attributes.each do |attribute|
+      assert_respond_to @files, attribute
+    end
+  end
+
   def test_all_method
-    @service.stub :list_blobs, @list_results do
-      assert_instance_of Fog::Storage::AzureRM::Files, @files.all
-      assert @files.all.size >= 1
-      @files.all.each do |file|
-        assert_instance_of Fog::Storage::AzureRM::File, file
+    @service.stub :get_container_properties, @container do
+      @service.stub :list_blobs, @blob_list do
+        files = @files.all
+        assert_instance_of Fog::Storage::AzureRM::Files, files
+        assert_equal @blob_list[:blobs].size, files.size
+      end
+    end
+  end
+
+  def test_all_method_without_directory_exception
+    assert_raises(ArgumentError) do
+      @files.attributes.delete(:directory)
+      @files.all
+    end
+  end
+
+  def test_all_method_container_not_found
+    @directory.collection.stub :get, nil do
+      assert @files.all.nil?
+    end
+  end
+
+  def test_each_method_with_all
+    @blob_list[:next_marker] = nil
+    @blob_list[:marker] = nil
+    @service.stub :get_container_properties, @container do
+      @service.stub :list_blobs, @blob_list do
+        @files.next_marker = nil
+        j = 0
+        @files.each do |file|
+          assert_instance_of Fog::Storage::AzureRM::File, file
+          j += 1
+        end
+        assert_equal @blob_list[:blobs].size, j
       end
     end
   end
 
-  def test_all_method_http_exception
-    http_exception = -> (_container_name, _option) { raise Azure::Core::Http::HTTPError.new(@mocked_response) }
-    @blob_client.stub :list_blobs, http_exception do
-      assert_raises(RuntimeError) do
-        @files.all
+  def test_each_method_with_parts
+    assert_instance_of Fog::Storage::AzureRM::Files, @files.each
+
+    multiple_values = lambda do |_container_name, options|
+      if options[:marker] == 'marker'
+        return {
+          next_marker: nil,
+          blobs: @blob_list[:blobs][2, 2]
+        }
+      end
+      return {
+        next_marker: 'marker',
+        blobs: @blob_list[:blobs][0, 2]
+      }
+    end
+
+    @service.stub :get_container_properties, @container do
+      @service.stub :list_blobs, multiple_values do
+        j = 0
+        @files.each do |file|
+          assert_instance_of Fog::Storage::AzureRM::File, file
+          j += 1
+        end
+        assert_equal @blob_list[:blobs].size, j
+      end
+    end
+  end
+
+  def test_get_method_success
+    content = 'data'
+    @service.stub :get_blob, [@blob, content] do
+      file = @files.get('test_blob')
+      assert_instance_of Fog::Storage::AzureRM::File, file
+      assert_equal content, file.body
+    end
+  end
+
+  def test_get_method_without_directory_exception
+    assert_raises(ArgumentError) do
+      @files.attributes.delete(:directory)
+      @files.get('test_blob')
+    end
+  end
+
+  def test_get_method_blob_not_found
+    exception = ->(*) { raise 'NotFound' }
+    @service.stub :get_blob, exception do
+      assert @files.get('test_blob').nil?
+    end
+  end
+
+  def test_get_method_http_exception
+    http_exception = ->(*) { raise Azure::Core::Http::HTTPError.new(@mocked_response) }
+    @service.stub :get_blob, http_exception do
+      assert_raises(Azure::Core::Http::HTTPError) do
+        @files.get('test_blob')
       end
     end
   end
 
-  def test_all_method_mock
-    assert_instance_of Fog::Storage::AzureRM::Files, @mock_files.all
-    assert @mock_files.all.size >= 1
-    @mock_files.all.each do |file|
+  def test_get_url_method_success
+    @files.stub :get_https_url, @blob_https_url do
+      assert_equal @blob_https_url, @files.get_url('test_blob', Time.now + 3600)
+    end
+
+    options = { scheme: 'http' }
+    http_url = @blob_https_url.gsub('https:', 'http:')
+    @files.stub :get_http_url, http_url do
+      assert_equal http_url, @files.get_url('test_blob', Time.now + 3600, options)
+    end
+  end
+
+  def test_get_url_method_without_directory_exception
+    assert_raises(ArgumentError) do
+      @files.attributes.delete(:directory)
+      @files.get_url('test_blob', Time.now + 3600)
+    end
+  end
+
+  def test_get_http_url_method_success
+    http_url = @blob_https_url.gsub('https:', 'http:')
+    @service.stub :get_blob_http_url, http_url do
+      assert_equal http_url, @files.get_http_url('test_blob', Time.now + 3600)
+    end
+  end
+
+  def test_get_http_url_method_without_directory_exception
+    assert_raises(ArgumentError) do
+      @files.attributes.delete(:directory)
+      @files.get_http_url('test_blob', Time.now + 3600)
+    end
+  end
+
+  def test_get_https_url_method_success
+    @service.stub :get_blob_https_url, @blob_https_url do
+      assert_equal @blob_https_url, @files.get_https_url('test_blob', Time.now + 3600)
+    end
+  end
+
+  def test_get_https_url_method_without_directory_exception
+    assert_raises(ArgumentError) do
+      @files.attributes.delete(:directory)
+      @files.get_https_url('test_blob', Time.now + 3600)
+    end
+  end
+
+  def test_head_method_success
+    @service.stub :get_blob_properties, @blob do
+      file = @files.head('test_blob')
       assert_instance_of Fog::Storage::AzureRM::File, file
+      assert file.attributes[:body].nil?
     end
   end
 
-  def test_get_method
-    @service.stub :list_blobs, @list_results do
-      assert_instance_of Fog::Storage::AzureRM::File, @files.get('test-container', 'testblob1')
-      assert !@files.get('test-container', 'wrong-name').nil?
+  def test_head_method_without_directory_exception
+    assert_raises(ArgumentError) do
+      @files.attributes.delete(:directory)
+      @files.head('test_blob')
     end
   end
 
-  def test_get_method_mock
-    @service.stub :list_blobs, @list_results do
-      assert_instance_of Fog::Storage::AzureRM::File, @mock_files.get('test-container', 'testblob1')
-      assert !@mock_files.get('test-container', 'wrong-name').nil?
+  def test_head_method_blob_not_found
+    exception = ->(*) { raise 'NotFound' }
+    @service.stub :get_blob_properties, exception do
+      assert @files.head('test_blob').nil?
+    end
+  end
+
+  def test_head_method_http_exception
+    http_exception = ->(*) { raise Azure::Core::Http::HTTPError.new(@mocked_response) }
+    @service.stub :get_blob_properties, http_exception do
+      assert_raises(Azure::Core::Http::HTTPError) do
+        @files.head('test_blob')
+      end
+    end
+  end
+
+  def test_new_method_success
+    assert_instance_of Fog::Storage::AzureRM::File, @files.new
+  end
+
+  def test_new_method_without_directory_exception
+    assert_raises(ArgumentError) do
+      @files.attributes.delete(:directory)
+      @files.new
     end
   end
 end

data/test/models/storage/test_storage_account.rb

@@ -66,7 +66,7 @@ class TestStorageAccount < Minitest::Test
   def test_get_access_keys_method_response
     key1 = Azure::ARM::Storage::Models::StorageAccountKey.new
     key1.key_name = 'key1'
-    key1.value = 'sfhyuiafhhfids0943'
+    key1.value = 'YWNjZXNzLWtleQ=='
     key1.permissions = 'Full'
     response = [key1]
     @service.stub :get_storage_access_keys, response do

data/test/requests/compute/test_attach_data_disk_to_vm.rb

@@ -14,11 +14,15 @@ class TestAttachDataDiskToVM < Minitest::Test
   end
 
   def test_attach_data_disk_to_vm_success
-    @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
-          @service.stub :check_blob_exist, true do
-            assert_equal @service.attach_data_disk_to_vm('fog-test-rg', 'fog-test-vm', 'disk1', 1, 'mystorage1'), @update_vm_response
+    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
@@ -30,10 +34,8 @@ class TestAttachDataDiskToVM < Minitest::Test
     @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
-          @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
+          assert_raises RuntimeError do
+            @service.attach_data_disk_to_vm('fog-test-rg', 'fog-test-vm', 'disk1', 1, 'mystorage1')
           end
         end
       end
@@ -45,10 +47,8 @@ class TestAttachDataDiskToVM < Minitest::Test
     @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
-          @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
+          assert_raises RuntimeError do
+            @service.attach_data_disk_to_vm('fog-test-rg', 'fog-test-vm', 'disk1', 1, 'mystorage1')
           end
         end
       end
@@ -60,10 +60,8 @@ class TestAttachDataDiskToVM < Minitest::Test
     @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
-          @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
+          assert_raises RuntimeError do
+            @service.attach_data_disk_to_vm('fog-test-rg', 'fog-test-vm', 'disk1', 1, 'mystorage1')
           end
         end
       end

data/test/requests/storage/test_acquire_blob_lease.rb

@@ -1,17 +1,36 @@
 require File.expand_path '../../test_helper', __dir__
 
 # Storage Blob Class
-class TestListBlobs < Minitest::Test
+class TestAcquireBlobLease < Minitest::Test
   # This class posesses the test cases for the requests of acquire 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)
+
+    @blob_lease_id = ApiStub::Requests::Storage::File.blob_lease_id
   end
 
   def test_acquire_blob_lease_success
-    lease_id = { 'leaseId' => 'abc123' }
-    @blob_client.stub :acquire_blob_lease, lease_id do
-      assert @service.acquire_blob_lease('test_container', 'test_blob'), lease_id
+    @blob_client.stub :acquire_blob_lease, @blob_lease_id do
+      assert_equal @blob_lease_id, @service.acquire_blob_lease('test_container', 'test_blob')
     end
   end
+
+  def test_acquire_blob_lease_http_exception
+    http_exception = ->(*) { raise Azure::Core::Http::HTTPError.new(@mocked_response) }
+    @blob_client.stub :acquire_blob_lease, http_exception do
+      assert_raises(RuntimeError) do
+        @service.acquire_blob_lease('test_container', 'test_blob')
+      end
+    end
+  end
+
+  def test_acquire_blob_lease_mock
+    assert_equal @blob_lease_id, @mock_service.acquire_blob_lease('test_container', 'test_blob')
+  end
 end

data/test/requests/storage/test_acquire_container_lease.rb

@@ -1,17 +1,36 @@
 require File.expand_path '../../test_helper', __dir__
 
-# Storage Blob Class
-class TestListBlobs < Minitest::Test
+# Storage Container Class
+class TestAcquireContainerLease < Minitest::Test
   # This class posesses the test cases for the requests of acquire 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)
+
+    @container_lease_id = ApiStub::Requests::Storage::Directory.container_lease_id
   end
 
   def test_acquire_container_lease_success
-    lease_id = { 'leaseId' => 'abc123' }
-    @blob_client.stub :acquire_container_lease, lease_id do
-      assert @service.acquire_container_lease('test_container'), lease_id
+    @blob_client.stub :acquire_container_lease, @container_lease_id do
+      assert_equal @container_lease_id, @service.acquire_container_lease('test_container')
     end
   end
+
+  def test_acquire_container_lease_http_exception
+    http_exception = ->(*) { raise Azure::Core::Http::HTTPError.new(@mocked_response) }
+    @blob_client.stub :acquire_container_lease, http_exception do
+      assert_raises(RuntimeError) do
+        @service.acquire_container_lease('test_container')
+      end
+    end
+  end
+
+  def test_acquire_container_lease_mock
+    assert_equal @container_lease_id, @mock_service.acquire_container_lease('test_container')
+  end
 end

data/test/requests/storage/test_commit_blob_blocks.rb

@@ -0,0 +1,34 @@
+require File.expand_path '../../test_helper', __dir__
+
+# Storage Blob Class
+class TestCommitBlobBlocks < Minitest::Test
+  # This class posesses the test cases for the requests of committing blob blocks.
+  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_commit_blob_blocks_success
+    @blob_client.stub :commit_blob_blocks, true do
+      assert @service.commit_blob_blocks('test_container', 'test_blob', [])
+    end
+  end
+
+  def test_commit_blob_blocks_http_exception
+    http_exception = ->(*) { raise Azure::Core::Http::HTTPError.new(@mocked_response) }
+    @blob_client.stub :commit_blob_blocks, http_exception do
+      assert_raises(RuntimeError) do
+        @service.commit_blob_blocks('test_container', 'test_blob', [])
+      end
+    end
+  end
+
+  def test_commit_blob_blocks_mock
+    assert @mock_service.commit_blob_blocks('test_container', 'test_blob', [])
+  end
+end