fix upsert tests

vapor · Jun 22, 2018 · d8a17a9 · d8a17a9
1 parent e085d3d
commit d8a17a9
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 24 deletions.
diff --git a/Tests/PostgreSQLTests/PostgreSQLConnectionTests.swift b/Tests/PostgreSQLTests/PostgreSQLConnectionTests.swift
@@ -540,6 +540,21 @@ class PostgreSQLConnectionTests: XCTestCase {
             print(c)
         }
     }
+
+    func testUpsert() throws {
+        let values: [(PostgreSQLUpsert.Identifier, PostgreSQLUpsert.Expression)] = []
+
+        var upsert: PostgreSQLUpsert
+
+        upsert = PostgreSQLUpsert.upsert(nil, values)
+        XCTAssertEqual(upsert.columns, [PostgreSQLColumnIdentifier.column(nil, .identifier("id"))])
+
+        upsert = PostgreSQLUpsert.upsert([], values)
+        XCTAssertEqual(upsert.columns, [PostgreSQLColumnIdentifier.column(nil, .identifier("id"))])
+
+        upsert = PostgreSQLUpsert.upsert([.column(nil, .identifier("field"))], values)
+        XCTAssertEqual(upsert.columns, [PostgreSQLColumnIdentifier.column(nil, .identifier("field"))])
+    }
 
     static var allTests = [
         ("testBenchmark", testBenchmark),
@@ -560,6 +575,7 @@ class PostgreSQLConnectionTests: XCTestCase {
         ("testOrderBy", testOrderBy),
         ("testInvalidDate", testInvalidDate),
         ("testEmptyArray", testEmptyArray),
+        ("testUpsert", testUpsert),
     ]
 }
 

diff --git a/Tests/PostgreSQLTests/SqlTests/PostgreSQLUpsertTests.swift b/Tests/PostgreSQLTests/SqlTests/PostgreSQLUpsertTests.swift