test/requests/sql/test_delete_database.rb in fog-azure-rm-0.1.2 vs test/requests/sql/test_delete_database.rb in fog-azure-rm-0.2.0

@@ -2,33 +2,22 @@
 
 # Test class for Delete Database
 class TestDeleteDatabase < Minitest::Test
   def setup
     @service = Fog::Sql::AzureRM.new(credentials)
-    @token_provider = Fog::Credentials::AzureRM.instance_variable_get(:@token_provider)
+    @sql_manager_client = @service.instance_variable_get(:@sql_mgmt_client)
+    @databases = @sql_manager_client.databases
   end
 
   def test_delete_database_success
-    @token_provider.stub :get_authentication_header, 'Bearer <some-token>' do
-      RestClient.stub :delete, true do
-        assert @service.delete_database('fog-test-rg', 'fog-test-server-name', 'database-name')
-      end
+    @databases.stub :delete, true do
+      assert @service.delete_database('fog-test-rg', 'fog-test-server-name', 'database-name')
     end
   end
 
   def test_delete_database_failure
-    @token_provider.stub :get_authentication_header, 'Bearer <some-token>' do
-      assert_raises ArgumentError do
-        @service.delete_database('fog-test-zone')
-      end
-    end
-  end
-
-  def test_delete_database_exception
     response = proc { raise MsRestAzure::AzureOperationError.new(nil, nil, 'error' => { 'message' => 'mocked exception' }) }
-    @token_provider.stub :get_authentication_header, response do
-      assert_raises Exception do
-        assert @service.delete_database('fog-test-rg', 'fog-test-server-name', 'database-name')
-      end
+    @databases.stub :delete, response do
+      assert_raises(RuntimeError) { @service.delete_database('fog-test-rg', 'fog-test-server-name', 'database-name') }
     end
   end
 end