Merge pull request #114 from sroebert/bug/pivot

Added test for pivot database setting
vapor · Oct 17, 2016 · 5c30919 · 5c30919
2 parents edec439 + aea98ff
commit 5c30919
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/Tests/FluentTests/RelationTests.swift b/Tests/FluentTests/RelationTests.swift
@@ -5,6 +5,8 @@ class RelationTests: XCTestCase {
     static let allTests = [
         ("testHasMany", testHasMany),
         ("testBelongsToMany", testBelongsToMany),
+        ("testCustomForeignKey", testCustomForeignKey),
+        ("testPivotDatabase", testPivotDatabase),
     ]
 
     var memory: MemoryDriver!
@@ -69,4 +71,10 @@ class RelationTests: XCTestCase {
             print(error)
         }
     }
+
+    func testPivotDatabase() throws {
+        Pivot<Atom, Nucleus>.database = database
+        XCTAssertTrue(Pivot<Atom, Nucleus>.database === database)
+        XCTAssertTrue(Pivot<Nucleus, Atom>.database === database)
+    }
 }