add a test for it

xmtp · Sep 13, 2023 · 7023ae4 · 7023ae4
1 parent 379f3d4
commit 7023ae4
Show file tree

Hide file tree

Showing 2 changed files with 15 additions and 0 deletions.
diff --git a/Sources/XMTPTestHelpers/TestHelpers.swift b/Sources/XMTPTestHelpers/TestHelpers.swift
@@ -197,6 +197,15 @@ public class FakeApiClient: ApiClient {
 				}
 			}
 		}
+
+        if let direction = pagination?.direction {
+            switch direction {
+            case .ascending:
+                result = Array(result.reversed())
+            default:
+                break
+            }
+        }
 
 		var queryResponse = QueryResponse()
 		queryResponse.envelopes = result

diff --git a/Tests/XMTPTests/IntegrationTests.swift b/Tests/XMTPTests/IntegrationTests.swift
@@ -414,6 +414,12 @@ final class IntegrationTests: XCTestCase {
 		XCTAssertEqual(1, messages3.count)
 		let nowMessage2 = messages3[0]
 		XCTAssertEqual("now", nowMessage2.body)
+
+        let messagesAsc = try await convo.messages(direction: .ascending)
+        XCTAssertEqual("10 seconds ago", messagesAsc[0].body)
+
+        let messagesDesc = try await convo.messages(direction: .descending)
+        XCTAssertEqual("now", messagesDesc[0].body)
 	}
 
     func testStreamingMessagesShouldBeReceived() async throws {