Add API tests

vapor-community · Aug 9, 2024 · 7e631ed · 7e631ed
1 parent 1a01afe
commit 7e631ed
Show file tree

Hide file tree

Showing 4 changed files with 89 additions and 5 deletions.
diff --git a/Sources/Orders/OrdersServiceCustom.swift b/Sources/Orders/OrdersServiceCustom.swift
@@ -453,7 +453,7 @@ extension OrdersServiceCustom {
         )
 
         let zipFile = tmp.appendingPathComponent("\(UUID().uuidString).order")
-        try await FileManager.default.zipItem(at: root, to: zipFile, shouldKeepParent: false)
+        try FileManager.default.zipItem(at: root, to: zipFile, shouldKeepParent: false)
         defer { _ = try? FileManager.default.removeItem(at: zipFile) }
 
         return try Data(contentsOf: zipFile)

diff --git a/Sources/Passes/PassesServiceCustom.swift b/Sources/Passes/PassesServiceCustom.swift
@@ -555,7 +555,7 @@ extension PassesServiceCustom {
         )
 
         let zipFile = tmp.appendingPathComponent("\(UUID().uuidString).pkpass")
-        try await FileManager.default.zipItem(at: root, to: zipFile, shouldKeepParent: false)
+        try FileManager.default.zipItem(at: root, to: zipFile, shouldKeepParent: false)
         defer { _ = try? FileManager.default.removeItem(at: zipFile) }
 
         return try Data(contentsOf: zipFile)
@@ -587,7 +587,7 @@ extension PassesServiceCustom {
         }
 
         let zipFile = tmp.appendingPathComponent("\(UUID().uuidString).pkpasses")
-        try await FileManager.default.zipItem(at: root, to: zipFile, shouldKeepParent: false)
+        try FileManager.default.zipItem(at: root, to: zipFile, shouldKeepParent: false)
         defer { _ = try? FileManager.default.removeItem(at: zipFile) }
 
         return try Data(contentsOf: zipFile)

diff --git a/Tests/OrdersTests/OrdersTests.swift b/Tests/OrdersTests/OrdersTests.swift
@@ -2,11 +2,13 @@ import XCTVapor
 import Fluent
 import FluentSQLiteDriver
 @testable import Orders
+@testable import PassKit
 
 final class OrdersTests: XCTestCase {
-    var app: Application!
     let orderDelegate = TestOrdersDelegate()
+    let ordersURI = "/api/orders/v1/"
     var ordersService: OrdersService!
+    var app: Application!
 
     override func setUp() async throws {
         self.app = try await Application.make(.testing)
@@ -27,4 +29,44 @@ final class OrdersTests: XCTestCase {
         let data = try await ordersService.generateOrderContent(for: order, on: app.db)
         XCTAssertNotNil(data)
     }
+
+    func testAPIDeviceRegistration() async throws {
+        let orderData = OrderData(title: "Test Order")
+        try await orderData.create(on: app.db)
+        let order = try await orderData.$order.get(on: app.db)
+        let deviceLibraryIdentifier = "abcdefg"
+
+        try await app.test(
+            .POST,
+            "\(ordersURI)devices/\(deviceLibraryIdentifier)/registrations/\(order.orderTypeIdentifier)/\(order.requireID())",
+            headers: ["Authorization": "AppleOrder \(order.authenticationToken)"],
+            beforeRequest: { req async throws in
+                try req.content.encode(RegistrationDTO(pushToken: "1234567890"))
+            },
+            afterResponse: { res async throws in
+                XCTAssertEqual(res.status, .created)
+            }
+        )
+
+        try await app.test(
+            .POST,
+            "\(ordersURI)devices/\(deviceLibraryIdentifier)/registrations/\(order.orderTypeIdentifier)/\(order.requireID())",
+            headers: ["Authorization": "AppleOrder \(order.authenticationToken)"],
+            beforeRequest: { req async throws in
+                try req.content.encode(RegistrationDTO(pushToken: "1234567890"))
+            },
+            afterResponse: { res async throws in
+                XCTAssertEqual(res.status, .ok)
+            }
+        )
+
+        try await app.test(
+            .DELETE,
+            "\(ordersURI)devices/\(deviceLibraryIdentifier)/registrations/\(order.orderTypeIdentifier)/\(order.requireID())",
+            headers: ["Authorization": "AppleOrder \(order.authenticationToken)"],
+            afterResponse: { res async throws in
+                XCTAssertEqual(res.status, .ok)
+            }
+        )
+    }
 }
diff --git a/Tests/PassesTests/PassesTests.swift b/Tests/PassesTests/PassesTests.swift
@@ -2,11 +2,13 @@ import XCTVapor
 import Fluent
 import FluentSQLiteDriver
 @testable import Passes
+@testable import PassKit
 
 final class PassesTests: XCTestCase {
-    var app: Application!
     let passDelegate = TestPassesDelegate()
+    let passesURI = "/api/passes/v1/"
     var passesService: PassesService!
+    var app: Application!
 
     override func setUp() async throws {
         self.app = try await Application.make(.testing)
@@ -54,4 +56,44 @@ final class PassesTests: XCTestCase {
 
         XCTAssertGreaterThan(dataPersonalize.count, data.count)
     }
+
+    func testAPIDeviceRegistration() async throws {
+        let passData = PassData(title: "Test Pass")
+        try await passData.create(on: app.db)
+        let pass = try await passData.$pass.get(on: app.db)
+        let deviceLibraryIdentifier = "abcdefg"
+
+        try await app.test(
+            .POST,
+            "\(passesURI)devices/\(deviceLibraryIdentifier)/registrations/\(pass.passTypeIdentifier)/\(pass.requireID())",
+            headers: ["Authorization": "ApplePass \(pass.authenticationToken)"],
+            beforeRequest: { req async throws in
+                try req.content.encode(RegistrationDTO(pushToken: "1234567890"))
+            },
+            afterResponse: { res async throws in
+                XCTAssertEqual(res.status, .created)
+            }
+        )
+
+        try await app.test(
+            .POST,
+            "\(passesURI)devices/\(deviceLibraryIdentifier)/registrations/\(pass.passTypeIdentifier)/\(pass.requireID())",
+            headers: ["Authorization": "ApplePass \(pass.authenticationToken)"],
+            beforeRequest: { req async throws in
+                try req.content.encode(RegistrationDTO(pushToken: "1234567890"))
+            },
+            afterResponse: { res async throws in
+                XCTAssertEqual(res.status, .ok)
+            }
+        )
+
+        try await app.test(
+            .DELETE,
+            "\(passesURI)devices/\(deviceLibraryIdentifier)/registrations/\(pass.passTypeIdentifier)/\(pass.requireID())",
+            headers: ["Authorization": "ApplePass \(pass.authenticationToken)"],
+            afterResponse: { res async throws in
+                XCTAssertEqual(res.status, .ok)
+            }
+        )
+    }
 }