Add more tests for auth middleware

vapor-community · Aug 20, 2024 · acd12dc · acd12dc
1 parent 6684b88
commit acd12dc
Show file tree

Hide file tree

Showing 2 changed files with 32 additions and 0 deletions.
diff --git a/Tests/OrdersTests/OrdersTests.swift b/Tests/OrdersTests/OrdersTests.swift
@@ -86,6 +86,18 @@ final class OrdersTests: XCTestCase {
  let deviceLibraryIdentifier = "abcdefg"
  let pushToken = "1234567890"
 
+ // Test registration without authentication token
+ try await app.test(
+ .POST,
+ "\(ordersURI)devices/\(deviceLibraryIdentifier)/registrations/\(order.orderTypeIdentifier)/\(order.requireID())",
+ beforeRequest: { req async throws in
+ try req.content.encode(RegistrationDTO(pushToken: pushToken))
+ },
+ afterResponse: { res async throws in
+ XCTAssertEqual(res.status, .unauthorized)
+ }
+ )
+
  try await app.test(
  .POST,
  "\(ordersURI)devices/\(deviceLibraryIdentifier)/registrations/\(order.orderTypeIdentifier)/\(order.requireID())",
@@ -98,6 +110,7 @@ final class OrdersTests: XCTestCase {
  }
  )
 
+ // Test registration of an already registered device
  try await app.test(
  .POST,
  "\(ordersURI)devices/\(deviceLibraryIdentifier)/registrations/\(order.orderTypeIdentifier)/\(order.requireID())",
@@ -169,6 +182,9 @@ final class OrdersTests: XCTestCase {
  let orderData = OrderData(title: "Test Order")
  try await orderData.create(on: app.db)
  let order = try await orderData._$order.get(on: app.db)
+
+ try await ordersService.sendPushNotificationsForOrder(id: order.requireID(), of: order.orderTypeIdentifier, on: app.db)
+
  let deviceLibraryIdentifier = "abcdefg"
  let pushToken = "1234567890"
 

diff --git a/Tests/PassesTests/PassesTests.swift b/Tests/PassesTests/PassesTests.swift
@@ -172,6 +172,18 @@ final class PassesTests: XCTestCase {
  let deviceLibraryIdentifier = "abcdefg"
  let pushToken = "1234567890"
 
+ // Test registration without authentication token
+ try await app.test(
+ .POST,
+ "\(passesURI)devices/\(deviceLibraryIdentifier)/registrations/\(pass.passTypeIdentifier)/\(pass.requireID())",
+ beforeRequest: { req async throws in
+ try req.content.encode(RegistrationDTO(pushToken: pushToken))
+ },
+ afterResponse: { res async throws in
+ XCTAssertEqual(res.status, .unauthorized)
+ }
+ )
+
  try await app.test(
  .POST,
  "\(passesURI)devices/\(deviceLibraryIdentifier)/registrations/\(pass.passTypeIdentifier)/\(pass.requireID())",
@@ -184,6 +196,7 @@ final class PassesTests: XCTestCase {
  }
  )
 
+ // Test registration of an already registered device
  try await app.test(
  .POST,
  "\(passesURI)devices/\(deviceLibraryIdentifier)/registrations/\(pass.passTypeIdentifier)/\(pass.requireID())",
@@ -255,6 +268,9 @@ final class PassesTests: XCTestCase {
  let passData = PassData(title: "Test Pass")
  try await passData.create(on: app.db)
  let pass = try await passData._$pass.get(on: app.db)
+
+ try await passesService.sendPushNotificationsForPass(id: pass.requireID(), of: pass.passTypeIdentifier, on: app.db)
+
  let deviceLibraryIdentifier = "abcdefg"
  let pushToken = "1234567890"