test_directory.rb in fog-azure-rm-0.1.1

- old
+ new

@@ -1,28 +1,28 @@
 require File.expand_path '../../test_helper', __dir__
 
 # Test class for Storage Container Model
 class TestDirectory < Minitest::Test
   def setup
-    Fog.mock!
-    @mock_service = Fog::Storage::AzureRM.new(storage_account_credentials)
-    @mock_directory = storage_container(@mock_service)
-    Fog.unmock!
     @service = Fog::Storage::AzureRM.new(storage_account_credentials)
-    @directory = storage_container(@service)
-    @create_result = ApiStub::Models::Storage::Directory.create_container
-    @get_properties_result = ApiStub::Models::Storage::Directory.get_container_properties
-    @get_access_control_list_result = ApiStub::Models::Storage::Directory.get_container_access_control_list
-    @blob_client = @service.instance_variable_get(:@blob_client)
-    @mocked_response = mocked_storage_http_error
+    @directory = directory(@service)
+
+    @container = ApiStub::Models::Storage::Directory.container
+    @container_acl = ApiStub::Models::Storage::Directory.container_acl
+    @container_https_url = ApiStub::Models::Storage::Directory.container_https_url
   end
 
   def test_model_methods
     methods = [
+      :acl=,
+      :acl,
+      :destroy,
+      :files,
+      :public=,
+      :public_url,
       :save,
-      :create,
-      :destroy
+      :persisted?
     ]
     methods.each do |method|
       assert_respond_to @directory, method
     end
   end
@@ -35,91 +35,144 @@
       :lease_duration,
       :lease_state,
       :lease_status,
       :metadata
     ]
-    @service.stub :create_container, @create_result do
+    @service.stub :create_container, @container do
       attributes.each do |attribute|
         assert_respond_to @directory, attribute
       end
     end
   end
 
-  def test_save_method_response
-    @service.stub :create_container, @create_result do
-      assert_instance_of Fog::Storage::AzureRM::Directory, @directory.save
+  def test_set_acl_method_success
+    result = @directory.acl = 'container'
+    assert_equal 'container', result
+    assert_equal 'container', @directory.attributes[:acl]
+  end
+
+  def test_set_acl_method_exception
+    assert_raises(ArgumentError) do
+      @directory.acl = 'invalid-acl'
     end
-    @service.stub :create_container, @create_result do
-      assert_instance_of Fog::Storage::AzureRM::Directory, @directory.create
-    end
   end
 
-  def test_save_http_exception
-    http_exception = -> (_container_name, _option) { raise Azure::Core::Http::HTTPError.new(@mocked_response) }
-    @blob_client.stub :create_container, http_exception do
-      assert_raises(RuntimeError) do
-        @directory.save
+  def test_get_acl_method_success
+    # Return 'unknonw' when directory is not persist and acl is 'unknown'
+    @directory.acl = 'unknown'
+    @directory.stub :persisted?, false do
+      assert_equal 'unknown', @directory.acl
+    end
+
+    # Return actual value directly when directory is not persist and acl is not 'unknown'
+    @directory.acl = 'blob'
+    @directory.stub :persisted?, false do
+      assert_equal 'blob', @directory.acl
+    end
+
+    # Query actual value and return it when directory is persist and acl is 'unknown'
+    @directory.acl = 'unknown'
+    @directory.stub :persisted?, true do
+      @service.stub :get_container_acl, @container_acl do
+        assert_equal 'container', @directory.acl
+        assert_equal 'container', @directory.attributes[:acl]
       end
     end
   end
 
-  def test_save_mock
-    assert_instance_of Fog::Storage::AzureRM::Directory, @mock_directory.save
+  def test_get_acl_method_without_key_exception
+    @directory.attributes.delete(:key)
+    assert_raises(ArgumentError) do
+      @directory.acl
+    end
   end
 
-  def test_get_container_properties_method_response
-    @service.stub :get_container_properties, @get_properties_result do
-      assert_instance_of Fog::Storage::AzureRM::Directory, @directory.get_properties
+  def test_destroy_method_success
+    @service.stub :delete_container, true do
+      assert @directory.destroy
     end
   end
 
-  def test_get_container_properties_http_exception
-    http_exception = -> (_container_name, _option) { raise Azure::Core::Http::HTTPError.new(@mocked_response) }
-    @blob_client.stub :get_container_properties, http_exception do
-      assert_raises(RuntimeError) do
-        @directory.get_properties
-      end
+  def test_destroy_method_without_key_exception
+    @directory.attributes.delete(:key)
+    assert_raises(ArgumentError) do
+      @directory.destroy
     end
   end
 
-  def test_get_container_properties_mock
-    assert_instance_of Fog::Storage::AzureRM::Directory, @mock_directory.get_properties
+  def test_files_method_success
+    assert_instance_of Fog::Storage::AzureRM::Files, @directory.files
   end
 
-  def test_get_access_control_list_method_response
-    @service.stub :get_container_access_control_list, @get_access_control_list_result do
-      assert_instance_of Fog::Storage::AzureRM::Directory, @directory.access_control_list
-    end
+  def test_set_public_method_success
+    # Set public
+    result = @directory.public = true
+    assert result
+    assert_equal 'container', @directory.attributes[:acl]
+
+    # Set private
+    result = @directory.public = false
+    assert !result
+    assert_equal nil, @directory.attributes[:acl]
   end
 
-  def test_get_access_control_list_http_exception
-    http_exception = -> (_container_name, _option) { raise Azure::Core::Http::HTTPError.new(@mocked_response) }
-    @blob_client.stub :get_container_acl, http_exception do
-      assert_raises(RuntimeError) do
-        @directory.access_control_list
+  def test_public_url_method_with_public_success
+    @directory.stub :acl, 'container' do
+      @service.stub :get_container_url, @container_https_url do
+        assert_equal @container_https_url, @directory.public_url
       end
     end
   end
 
-  def test_get_access_control_list_mock
-    assert_instance_of Fog::Storage::AzureRM::Directory, @mock_directory.access_control_list
+  def test_public_url_method_with_private_success
+    @directory.stub :acl, nil do
+      assert @directory.public_url.nil?
+    end
   end
 
-  def test_delete_method_true_response
-    @service.stub :delete_container, true do
-      assert @directory.destroy
+  def test_public_url_method_without_key_exception
+    @directory.attributes.delete(:key)
+    assert_raises(ArgumentError) do
+      @directory.public_url
     end
   end
 
-  def test_delete_method_exception
-    exception = -> (_container_name, _option) { raise Azure::Core::Http::HTTPError.new(@mocked_response) }
-    @blob_client.stub :delete_container, exception do
-      assert_raises(RuntimeError) do
-        assert @directory.destroy
+  def test_save_method_create_success
+    @service.stub :create_container, @container do
+      assert @directory.save
+      assert @directory.attributes[:is_persisted]
+    end
+  end
+
+  def test_save_method_update_success
+    options = { is_create: false }
+    @service.stub :put_container_acl, true do
+      @service.stub :put_container_metadata, true do
+        @service.stub :get_container_properties, @container do
+          assert @directory.save(options)
+          assert @directory.attributes[:is_persisted]
+        end
       end
     end
   end
 
-  def test_delete_method_response_mock
-    assert @mock_directory.destroy
+  def test_save_method_without_key_exception
+    @directory.attributes.delete(:key)
+    assert_raises(ArgumentError) do
+      @directory.save
+    end
+  end
+
+  def test_persisted_method_success
+    @directory.attributes[:is_persisted] = true
+    @directory.attributes[:last_modified] = nil
+    assert @directory.persisted?
+
+    @directory.attributes[:is_persisted] = false
+    @directory.attributes[:last_modified] = 'datetime'
+    assert @directory.persisted?
+
+    @directory.attributes[:is_persisted] = false
+    @directory.attributes[:last_modified] = nil
+    assert !@directory.persisted?
   end
 end