-
-
Notifications
You must be signed in to change notification settings - Fork 123
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
test(realtime): add realtime integration tests (#333)
* Move to ObservationTokenTests * test(realtime): add integration tests for realtime v2 * Rename schema to key_value_storage * chore: move realtime integration tests to IntegrationTests target * test: comment out failing test * test: fix optional type for receivedMessage * chore: fix dot-env target * chore: fix load env script * test: use mock web socket * chore: swift format
- Loading branch information
Showing
21 changed files
with
499 additions
and
45 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,4 @@ | ||
| # Get these from your API settings: https://supabase.com/dashboard/project/_/settings/api | ||
|
|
||
| SUPABASE_URL=https://mysupabasereference.supabase.co | ||
| SUPABASE_ANON_KEY=my.supabase.anon.key |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,14 @@ | ||
| // | ||
| // AsyncSequence.swift | ||
| // | ||
| // | ||
| // Created by Guilherme Souza on 04/04/24. | ||
| // | ||
|
|
||
| import Foundation | ||
|
|
||
| extension AsyncSequence { | ||
| package func collect() async rethrows -> [Element] { | ||
| try await reduce(into: [Element]()) { $0.append($1) } | ||
| } | ||
| } |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,244 @@ | ||
| // | ||
| // RealtimeIntegrationTests.swift | ||
| // | ||
| // | ||
| // Created by Guilherme Souza on 27/03/24. | ||
| // | ||
|
|
||
| import ConcurrencyExtras | ||
| import CustomDump | ||
| import PostgREST | ||
| @testable import Realtime | ||
| import Supabase | ||
| import TestHelpers | ||
| import XCTest | ||
|
|
||
| struct Logger: SupabaseLogger { | ||
| func log(message: SupabaseLogMessage) { | ||
| print(message.description) | ||
| } | ||
| } | ||
|
|
||
| final class RealtimeIntegrationTests: XCTestCase { | ||
| let realtime = RealtimeClientV2( | ||
| config: RealtimeClientV2.Configuration( | ||
| url: URL(string: "\(DotEnv.SUPABASE_URL)/realtime/v1")!, | ||
| apiKey: DotEnv.SUPABASE_ANON_KEY, | ||
| logger: Logger() | ||
| ) | ||
| ) | ||
|
|
||
| let db = PostgrestClient( | ||
| url: URL(string: "\(DotEnv.SUPABASE_URL)/rest/v1")!, | ||
| headers: [ | ||
| "apikey": DotEnv.SUPABASE_ANON_KEY, | ||
| ], | ||
| logger: Logger() | ||
| ) | ||
|
|
||
| func testBroadcast() async throws { | ||
| let expectation = expectation(description: "receivedBroadcastMessages") | ||
| expectation.expectedFulfillmentCount = 3 | ||
|
|
||
| let channel = await realtime.channel("integration") { | ||
| $0.broadcast.receiveOwnBroadcasts = true | ||
| } | ||
|
|
||
| let receivedMessages = LockIsolated<[JSONObject]>([]) | ||
|
|
||
| Task { | ||
| for await message in await channel.broadcast(event: "test") { | ||
| receivedMessages.withValue { | ||
| $0.append(message) | ||
| } | ||
| expectation.fulfill() | ||
| } | ||
| } | ||
|
|
||
| await Task.megaYield() | ||
|
|
||
| await channel.subscribe() | ||
|
|
||
| struct Message: Codable { | ||
| var value: Int | ||
| } | ||
|
|
||
| try await channel.broadcast(event: "test", message: Message(value: 1)) | ||
| try await channel.broadcast(event: "test", message: Message(value: 2)) | ||
| try await channel.broadcast(event: "test", message: ["value": 3, "another_value": 42]) | ||
|
|
||
| await fulfillment(of: [expectation], timeout: 0.5) | ||
|
|
||
| XCTAssertNoDifference( | ||
| receivedMessages.value, | ||
| [ | ||
| [ | ||
| "event": "test", | ||
| "payload": [ | ||
| "value": 1, | ||
| ], | ||
| "type": "broadcast", | ||
| ], | ||
| [ | ||
| "event": "test", | ||
| "payload": [ | ||
| "value": 2, | ||
| ], | ||
| "type": "broadcast", | ||
| ], | ||
| [ | ||
| "event": "test", | ||
| "payload": [ | ||
| "value": 3, | ||
| "another_value": 42, | ||
| ], | ||
| "type": "broadcast", | ||
| ], | ||
| ] | ||
| ) | ||
|
|
||
| await channel.unsubscribe() | ||
| } | ||
|
|
||
| func testPresence() async throws { | ||
| let channel = await realtime.channel("integration") { | ||
| $0.broadcast.receiveOwnBroadcasts = true | ||
| } | ||
|
|
||
| let expectation = expectation(description: "presenceChange") | ||
| expectation.expectedFulfillmentCount = 4 | ||
|
|
||
| let receivedPresenceChanges = LockIsolated<[any PresenceAction]>([]) | ||
|
|
||
| Task { | ||
| for await presence in await channel.presenceChange() { | ||
| receivedPresenceChanges.withValue { | ||
| $0.append(presence) | ||
| } | ||
| expectation.fulfill() | ||
| } | ||
| } | ||
|
|
||
| await Task.megaYield() | ||
|
|
||
| await channel.subscribe() | ||
|
|
||
| struct UserState: Codable, Equatable { | ||
| let email: String | ||
| } | ||
|
|
||
| try await channel.track(UserState(email: "[email protected]")) | ||
| try await channel.track(["email": "[email protected]"]) | ||
|
|
||
| await channel.untrack() | ||
|
|
||
| await fulfillment(of: [expectation], timeout: 0.5) | ||
|
|
||
| let joins = try receivedPresenceChanges.value.map { try $0.decodeJoins(as: UserState.self) } | ||
| let leaves = try receivedPresenceChanges.value.map { try $0.decodeLeaves(as: UserState.self) } | ||
| XCTAssertNoDifference( | ||
| joins, | ||
| [ | ||
| [], // This is the first PRESENCE_STATE event. | ||
| [UserState(email: "[email protected]")], | ||
| [UserState(email: "[email protected]")], | ||
| [], | ||
| ] | ||
| ) | ||
|
|
||
| XCTAssertNoDifference( | ||
| leaves, | ||
| [ | ||
| [], // This is the first PRESENCE_STATE event. | ||
| [], | ||
| [UserState(email: "[email protected]")], | ||
| [UserState(email: "[email protected]")], | ||
| ] | ||
| ) | ||
|
|
||
| await channel.unsubscribe() | ||
| } | ||
|
|
||
| // FIXME: Test getting stuck | ||
| // func testPostgresChanges() async throws { | ||
| // let channel = await realtime.channel("db-changes") | ||
| // | ||
| // let receivedInsertActions = Task { | ||
| // await channel.postgresChange(InsertAction.self, schema: "public").prefix(1).collect() | ||
| // } | ||
| // | ||
| // let receivedUpdateActions = Task { | ||
| // await channel.postgresChange(UpdateAction.self, schema: "public").prefix(1).collect() | ||
| // } | ||
| // | ||
| // let receivedDeleteActions = Task { | ||
| // await channel.postgresChange(DeleteAction.self, schema: "public").prefix(1).collect() | ||
| // } | ||
| // | ||
| // let receivedAnyActionsTask = Task { | ||
| // await channel.postgresChange(AnyAction.self, schema: "public").prefix(3).collect() | ||
| // } | ||
| // | ||
| // await Task.megaYield() | ||
| // await channel.subscribe() | ||
| // | ||
| // struct Entry: Codable, Equatable { | ||
| // let key: String | ||
| // let value: AnyJSON | ||
| // } | ||
| // | ||
| // let key = try await ( | ||
| // db.from("key_value_storage") | ||
| // .insert(["key": AnyJSON.string(UUID().uuidString), "value": "value1"]).select().single() | ||
| // .execute().value as Entry | ||
| // ).key | ||
| // try await db.from("key_value_storage").update(["value": "value2"]).eq("key", value: key) | ||
| // .execute() | ||
| // try await db.from("key_value_storage").delete().eq("key", value: key).execute() | ||
| // | ||
| // let insertedEntries = try await receivedInsertActions.value.map { | ||
| // try $0.decodeRecord( | ||
| // as: Entry.self, | ||
| // decoder: JSONDecoder() | ||
| // ) | ||
| // } | ||
| // let updatedEntries = try await receivedUpdateActions.value.map { | ||
| // try $0.decodeRecord( | ||
| // as: Entry.self, | ||
| // decoder: JSONDecoder() | ||
| // ) | ||
| // } | ||
| // let deletedEntryIds = await receivedDeleteActions.value.compactMap { | ||
| // $0.oldRecord["key"]?.stringValue | ||
| // } | ||
| // | ||
| // XCTAssertNoDifference(insertedEntries, [Entry(key: key, value: "value1")]) | ||
| // XCTAssertNoDifference(updatedEntries, [Entry(key: key, value: "value2")]) | ||
| // XCTAssertNoDifference(deletedEntryIds, [key]) | ||
| // | ||
| // let receivedAnyActions = await receivedAnyActionsTask.value | ||
| // XCTAssertEqual(receivedAnyActions.count, 3) | ||
| // | ||
| // if case let .insert(action) = receivedAnyActions[0] { | ||
| // let record = try action.decodeRecord(as: Entry.self, decoder: JSONDecoder()) | ||
| // XCTAssertNoDifference(record, Entry(key: key, value: "value1")) | ||
| // } else { | ||
| // XCTFail("Expected a `AnyAction.insert` on `receivedAnyActions[0]`") | ||
| // } | ||
| // | ||
| // if case let .update(action) = receivedAnyActions[1] { | ||
| // let record = try action.decodeRecord(as: Entry.self, decoder: JSONDecoder()) | ||
| // XCTAssertNoDifference(record, Entry(key: key, value: "value2")) | ||
| // } else { | ||
| // XCTFail("Expected a `AnyAction.update` on `receivedAnyActions[1]`") | ||
| // } | ||
| // | ||
| // if case let .delete(action) = receivedAnyActions[2] { | ||
| // XCTAssertNoDifference(key, action.oldRecord["key"]?.stringValue) | ||
| // } else { | ||
| // XCTFail("Expected a `AnyAction.delete` on `receivedAnyActions[2]`") | ||
| // } | ||
| // | ||
| // await channel.unsubscribe() | ||
| // } | ||
| } |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Oops, something went wrong.