public check method, needed for #389

stephencelis · Nov 16, 2016 · bb3b705 · bb3b705
1 parent e177a1d
commit bb3b705
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 1 deletion.
diff --git a/SQLite/Core/Connection.swift b/SQLite/Core/Connection.swift
@@ -674,7 +674,7 @@ public final class Connection {
         return success!
     }
 
-    @discardableResult func check(_ resultCode: Int32, statement: Statement? = nil) throws -> Int32 {
+    @discardableResult public func check(_ resultCode: Int32, statement: Statement? = nil) throws -> Int32 {
         guard let error = Result(errorCode: resultCode, connection: self, statement: statement) else {
             return resultCode
         }

diff --git a/SQLiteTests/ConnectionTests.swift b/SQLiteTests/ConnectionTests.swift
@@ -338,4 +338,11 @@ class ConnectionTests : SQLiteTestCase {
         AssertThrows(try stmt.run())
     }
 
+    func test_check_with_successCode_returnsCode() {
+        try! XCTAssertEqual(SQLITE_OK, db.check(SQLITE_OK))
+    }
+
+    func test_check_with_errorCode_throws() {
+        XCTAssertThrowsError(try db.check(SQLITE_CONSTRAINT))
+    }
 }