add table name cache test

Author: tanner0101

Sources/PostgreSQL/Connection/PostgreSQLConnection+Query.swift

@@ -58,11 +58,13 @@ extension PostgreSQLConnection {
                 guard let row = currentRow else {
                     throw PostgreSQLError(identifier: "query", reason: "Unexpected PostgreSQLDataRow without preceding PostgreSQLRowDescription.", source: .capture())
                 }
+                print("before parse")
                 let parsed = try row.parse(
                     data: data,
                     formatCodes: resultFormat.formatCodes,
                     tableNameCache: self.tableNameCache
                 )
+                print("after parse")
                 try onRow(parsed)
             case .close: break
             default: throw PostgreSQLError(identifier: "query", reason: "Unexpected message during PostgreSQLParseRequest: \(message)", source: .capture())

Sources/PostgreSQL/Database/PostgreSQLDatabase.swift

@@ -12,9 +12,9 @@ public final class PostgreSQLDatabase: Database {
     internal var tableNameCache: PostgreSQLTableNameCache?
 
     /// Creates a new `PostgreSQLDatabase`.
-    public init(config: PostgreSQLDatabaseConfig) {
+    public init(config: PostgreSQLDatabaseConfig, on worker: Worker) {
         self.config = config
-        self.tableNameCache = PostgreSQLTableNameCache(connection: makeConnection(on: EmbeddedEventLoop()))
+        self.tableNameCache = PostgreSQLTableNameCache(connection: makeConnection(on: worker))
     }
 
     /// See `Database.makeConnection()`

Sources/PostgreSQL/PostgreSQLProvider.swift

@@ -33,6 +33,6 @@ extension PostgreSQLDatabaseConfig: ServiceType {
 extension PostgreSQLDatabase: ServiceType {
     /// See `ServiceType.makeService(for:)`
     public static func makeService(for worker: Container) throws -> PostgreSQLDatabase {
-        return try .init(config: worker.make(for: PostgreSQLDatabase.self))
+        return try .init(config: worker.make(for: PostgreSQLDatabase.self), on: worker)
     }
 }

Tests/PostgreSQLTests/PostgreSQLConnectionTests.swift

@@ -342,6 +342,34 @@ class PostgreSQLConnectionTests: XCTestCase {
         _ = try categories.wait()
     }
 
+    func testTableNameCache() throws {
+        let hostname: String
+        #if Xcode
+        hostname = try Process.execute("docker-machine", "ip")
+        #else
+        hostname = "localhost"
+        #endif
+        let config = PostgreSQLDatabaseConfig.init(
+            hostname: hostname,
+            port: 5432,
+            username: "vapor_username",
+            database: "vapor_database",
+            password: nil
+        )
+        let main = MultiThreadedEventLoopGroup(numThreads: 1)
+        defer { try! main.syncShutdownGracefully() }
+        let sub = MultiThreadedEventLoopGroup(numThreads: 1)
+        defer { try! sub.syncShutdownGracefully() }
+
+        let database = PostgreSQLDatabase(config: config, on: main)
+        let client = try database.makeConnection(on: sub).wait()
+        let query = """
+        select * from "pg_type" where "typlen" = $1 or "typlen" = $2
+        """
+        let rows = try client.query(query, [1, 2]).wait()
+        XCTAssertEqual(rows[0].keys.first?.table, "pg_type")
+    }
+
     static var allTests = [
         ("testVersion", testVersion),
         ("testSelectTypes", testSelectTypes),
@@ -351,14 +379,15 @@ class PostgreSQLConnectionTests: XCTestCase {
         ("testStruct", testStruct),
         ("testNull", testNull),
         ("testGH24", testGH24),
+        ("testTableNameCache", testTableNameCache),
     ]
 }
 
 extension PostgreSQLConnection {
     /// Creates a test event loop and psql client.
     static func makeTest() throws -> PostgreSQLConnection {
         let hostname: String
-        #if TEST_DOCKER_HOSTNAME
+        #if Xcode
         hostname = try Process.execute("docker-machine", "ip")
         #else
         hostname = "localhost"