Add a basic test for the database notification hook (doesn't cover much but does prove that it works in the simplest case).

Author: gwynne

Package.swift

@@ -14,7 +14,12 @@ let package = Package(
     dependencies: [
         .package(url: "https://github.com/vapor/queues.git", from: "1.5.0"),
         .package(url: "https://github.com/vapor/fluent-kit.git", from: "1.7.0"),
-        .package(url: "https://github.com/vapor/sql-kit.git", from: "3.7.0")
+        .package(url: "https://github.com/vapor/sql-kit.git", from: "3.7.0"),
+        
+        // Test-only dependencies
+        .package(url: "https://github.com/vapor/fluent-sqlite-driver.git", from: "4.0.0"),
+        .package(url: "https://github.com/vapor/vapor.git", from: "4.0.0"),
+        .package(url: "https://github.com/vapor/fluent.git", from: "4.0.0"),
     ],
     targets: [
         .target(
@@ -26,6 +31,12 @@ let package = Package(
             ]),
         .testTarget(
             name: "QueuesDatabaseHooksTests",
-            dependencies: ["QueuesDatabaseHooks"]),
+            dependencies: [
+                .target(name: "QueuesDatabaseHooks"),
+                .product(name: "FluentSQLiteDriver", package: "fluent-sqlite-driver"),
+                .product(name: "XCTQueues", package: "queues"),
+                .product(name: "XCTVapor", package: "vapor"),
+                .product(name: "Fluent", package: "fluent"),
+            ]),
     ]
 )

Tests/QueuesDatabaseHooksTests/DatabaseHooksTests.swift

@@ -1,8 +1,59 @@
-//
-//  File.swift
-//  
-//
-//  Created by Jimmy McDermott on 10/10/20.
-//
+import QueuesDatabaseHooks
+import Queues
+import XCTQueues
+import XCTVapor
+import Fluent
+import FluentSQLiteDriver
+import XCTest
 
-import Foundation
+final class QueuesDatabaseHooksTests: XCTestCase {
+    var app: Application!
+    
+    override func setUpWithError() throws {
+        self.app = Application(.testing)
+        self.app.databases.use(.sqlite(.memory, maxConnectionsPerEventLoop: 1), as: .sqlite)
+        self.app.migrations.add(QueueDatabaseEntryMigration())
+        try self.app.migrator.setupIfNeeded().wait()
+        try self.app.migrator.prepareBatch().wait()
+        self.app.queues.use(.test)
+        self.app.queues.add(QueuesDatabaseNotificationHook(db: self.app.db, errorClosure: { $0.localizedDescription }, payloadClosure: { $0 }))
+    }
+    
+    override func tearDownWithError() throws {
+        self.app.shutdown()
+    }
+    
+    func testJobLifecycle() throws {
+        struct Foo: Job {
+            struct Payload: Codable {}
+            
+            func dequeue(_ context: QueueContext, _: Payload) -> EventLoopFuture<Void> {
+                context.eventLoop.makeSucceededVoidFuture()
+            }
+            
+            func error(_ context: QueueContext, _: Error, _: Payload) -> EventLoopFuture<Void> {
+                context.eventLoop.makeSucceededVoidFuture()
+            }
+        }
+        
+        self.app.queues.add(Foo())
+        self.app.get("foo") { $0.queue.dispatch(Foo.self, .init()).map { _ in "done" } }
+        try app.testable().test(.GET, "/foo") { XCTAssertEqual($0.body.string, "done") }
+        
+        let entries1 = try QueueDatabaseEntry.query(on: self.app.db).all().wait()
+        XCTAssertEqual(entries1.count, 1)
+        XCTAssertEqual(entries1.first?.status, .queued)
+        XCTAssertNil(entries1.first?.dequeuedAt)
+        XCTAssertNil(entries1.first?.completedAt)
+        XCTAssertNil(entries1.first?.errorString)
+        
+        try app.queues.queue.worker.run().wait()
+        
+        let entries2 = try QueueDatabaseEntry.query(on: self.app.db).all().wait()
+        XCTAssertEqual(entries2.count, 1)
+        XCTAssertEqual(entries2.first?.status, .success)
+        XCTAssertNotNil(entries2.first?.dequeuedAt)
+        XCTAssertNotNil(entries2.first?.completedAt)
+        XCTAssertNil(entries2.first?.errorString)
+    }
+}