Merge pull request #35 from nodes-vapor/abort-sugar

Added report and doNotReport extensions to Abort. Register configurab…

Sources/Bugsnag/Abort+Report.swift

@@ -0,0 +1,30 @@
+import Vapor
+
+extension Abort {
+    public func report() -> Abort {
+        return addReportMetadata(true)
+    }
+
+    public func doNotReport() -> Abort {
+        return addReportMetadata(false)
+    }
+
+    func addReportMetadata(_ shouldReport: Bool) -> Abort {
+        var metadata = self.metadata ?? Node.object([:])
+        try? metadata.set("report", shouldReport)
+
+        let result = Abort.init(
+            status,
+            metadata: metadata,
+            reason: reason,
+            identifier: identifier,
+            possibleCauses: possibleCauses,
+            suggestedFixes: suggestedFixes,
+            documentationLinks: documentationLinks,
+            stackOverflowQuestions: stackOverflowQuestions,
+            gitHubIssues: gitHubIssues
+        )
+
+        return result
+    }
+}

Sources/Bugsnag/Provider.swift

@@ -32,6 +32,7 @@ public final class Provider: Vapor.Provider {
         }
 
         self.config = try BugsnagConfig(config)
+        try config.addConfigurable(middleware: Middleware(config: config), name: "bugsnag")
     }
 
     // is automatically called directly after boot()

Tests/BugsnagTests/PayloadTransformerTests.swift

@@ -18,7 +18,8 @@ class PayloadTransformerTests: XCTestCase {
         ("testThatQueryParametersGetsFiltered", testThatQueryParametersGetsFiltered),
         ("testThatFormParametersGetsFiltered", testThatFormParametersGetsFiltered),
         ("testThatJsonParametersGetsFiltered", testThatJsonParametersGetsFiltered),
-        ("testThatStackTraceSizeIsWorking", testThatStackTraceSizeIsWorking)
+        ("testThatStackTraceSizeIsWorking", testThatStackTraceSizeIsWorking),
+        ("testAbortExtensions", testAbortExtensions)
     ]
 
     override func setUp() {
@@ -192,4 +193,19 @@ class PayloadTransformerTests: XCTestCase {
         XCTAssertEqual(FrameAddressMock.lastStackSize, 99)
 
     }
+
+    func testAbortExtensions() {
+        let errorReported = Abort.serverError.report()
+        let errorNotReported = Abort.serverError.doNotReport()
+
+        // ensure status code is the same
+        XCTAssertEqual(errorReported.status, Abort.serverError.status)
+        XCTAssertEqual(errorNotReported.status, Abort.serverError.status)
+
+        XCTAssertNotNil(errorReported.metadata)
+        XCTAssertNotNil(errorNotReported.metadata)
+
+        XCTAssertEqual(errorReported.metadata?["report"]?.bool, true)
+        XCTAssertEqual(errorNotReported.metadata?["report"]?.bool, false)
+    }
 }

Tests/LinuxMain.swift

@@ -5,6 +5,5 @@ XCTMain([
      testCase(MiddlewareTests.allTests),
      testCase(PayloadTransformerTests.allTests),
      testCase(ReporterTests.allTests),
-     testCase(ConnectionManagerTests.allTests),
      testCase(ConfigurationTests.allTests)
 ])