test/requests/storage/test_create_container.rb in fog-azure-rm-0.1.0 vs test/requests/storage/test_create_container.rb in fog-azure-rm-0.1.1

@@ -2,30 +2,35 @@
 
 # Storage Container Class
 class TestCreateContainer < Minitest::Test
   # This class posesses the test cases for the requests of creating storage containers.
   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)
-    @storage_container_object = ApiStub::Requests::Storage::Directory.create_container
+
+    @container = ApiStub::Requests::Storage::Directory.container
   end
 
-  def test_create_container_with_service_success
-    @blob_client.stub :create_container, @storage_container_object do
-      assert @service.create_container('testcontainer1')
+  def test_create_container_success
+    @blob_client.stub :create_container, @container do
+      assert_equal @container, @service.create_container('test_container')
     end
   end
 
-  def test_create_container_with_internal_client_success
-    @blob_client.stub :create_container, @storage_container_object do
-      assert @blob_client.create_container('testcontainer1')
+  def test_create_container_http_exception
+    http_exception = ->(*) { raise Azure::Core::Http::HTTPError.new(@mocked_response) }
+    @blob_client.stub :create_container, http_exception do
+      assert_raises(RuntimeError) do
+        @service.create_container('test_container')
+      end
     end
   end
 
-  def test_create_container_with_service_exception
-    assert_raises(URI::InvalidURIError) { @service.create_container('testcontainer1#@#@') }
-  end
-
-  def test_create_container_with_internal_client_exception
-    assert_raises(URI::InvalidURIError) { @blob_client.create_container('testcontainer1#@#@') }
+  def test_create_container_mock
+    assert_equal @container, @mock_service.create_container('test_container')
   end
 end