Add Repository.withRefDB(_:)

Author: swift-developer-tools-archive

Tests/Utilities/Repository.swift

@@ -1167,6 +1167,54 @@ internal extension Repository
 
 
 
+    /// Calls the given closure with a ``Repository`` instance and a pointer
+    /// to an opened reference database.
+    /// - Parameter body: The closure to call.
+    /// - Returns: The return value of the given closure.
+    /// - Throws: An error if an operation fails.
+    static func withRefDB<T>(
+        _ body: (Repository, OpaquePointer) throws -> T
+    ) throws -> T
+    {
+        return try Repository.withRepository
+        {
+            repository in
+            
+            var refDBPointer: OpaquePointer? = nil
+            
+            defer
+            {
+                gitRefDBFree(refDB: refDBPointer)
+            }
+            
+            
+            
+            let refDBOpenResult: GitErrorCode = gitRefDBOpen(
+                out:    &refDBPointer,
+                repo:   repository.pointer
+            )
+            
+            XCTAssertOK(refDBOpenResult)
+            
+            guard let refDBPointer: OpaquePointer = refDBPointer
+            else
+            {
+                throw NSError.makeError(
+                    "The reference database pointer was nil."
+                )
+            }
+            
+            
+            
+            return try body(
+                repository,
+                refDBPointer
+            )
+        }
+    }
+    
+    
+    
     /// Calls the given closure with a ``Repository`` instance and a pointer
     /// to a tree.
     /// - Parameter body: The closure to call.

Tests/swift-libgit2Tests/RefDB-Tests.swift

@@ -15,6 +15,21 @@ import XCTest
 
 final class RefDBTests: XCTestCaseStopOnFail
 {
+    func testGitRefDBCompress() throws
+    {
+        try Repository.withRefDB
+        {
+            _, refDBPointer in
+            
+            let refDBCompressResult: GitErrorCode
+                = gitRefDBCompress(out: refDBPointer)
+            
+            XCTAssertOK(refDBCompressResult)
+        }
+    }
+    
+    
+    
     func testGitRefDBFree() throws
     {
         gitRefDBFree(refDB: nil)
@@ -49,41 +64,11 @@ final class RefDBTests: XCTestCaseStopOnFail
 
 
 
-    func testGitRefDBOpenAndCompress() throws
+    func testGitRefDBOpen() throws
     {
-        try Repository.withRepository
+        try Repository.withRefDB
         {
-            repository in
-            
-            var refDBPointer: OpaquePointer? = nil
-            
-            defer
-            {
-                gitRefDBFree(refDB: refDBPointer)
-            }
-            
-            
-            
-            let refDBOpenResult: GitErrorCode = gitRefDBOpen(
-                out:    &refDBPointer,
-                repo:   repository.pointer
-            )
-            
-            XCTAssertOK(refDBOpenResult)
-            
-            guard let refDBPointer: OpaquePointer = refDBPointer
-            else
-            {
-                XCTFail("The reference database pointer was nil.")
-                return
-            }
-            
-            
-            
-            let refDBCompressResult: GitErrorCode
-                = gitRefDBCompress(out: refDBPointer)
-            
-            XCTAssertOK(refDBCompressResult)
+            _, _ in
         }
     }
 }