|
| 1 | +// |
| 2 | +// Copyright © 2025 Stream.io Inc. All rights reserved. |
| 3 | +// |
| 4 | + |
| 5 | +import CoreData |
| 6 | +import Foundation |
| 7 | +@testable import StreamChat |
| 8 | +@testable import StreamChatTestTools |
| 9 | +import XCTest |
| 10 | + |
| 11 | +final class ManualEventHandler_Tests: XCTestCase { |
| 12 | + var database: DatabaseContainer_Spy! |
| 13 | + var handler: ManualEventHandler! |
| 14 | + var cid: ChannelId! |
| 15 | + var cachedChannel: ChatChannel! |
| 16 | + |
| 17 | + override func setUp() { |
| 18 | + super.setUp() |
| 19 | + |
| 20 | + database = DatabaseContainer_Spy() |
| 21 | + cid = .unique |
| 22 | + |
| 23 | + // Setup database with channel and current user |
| 24 | + try! database.createChannel(cid: cid, withMessages: false) |
| 25 | + try! database.createCurrentUser() |
| 26 | + |
| 27 | + // Get the channel from database to use as cached channel |
| 28 | + cachedChannel = try! database.writableContext.channel(cid: cid)!.asModel() |
| 29 | + |
| 30 | + // Create handler with pre-cached channel to avoid registration requirements |
| 31 | + handler = ManualEventHandler( |
| 32 | + database: database, |
| 33 | + cachedChannels: [cid: cachedChannel] |
| 34 | + ) |
| 35 | + |
| 36 | + // Register the channel so events are processed |
| 37 | + handler.register(channelId: cid) |
| 38 | + } |
| 39 | + |
| 40 | + override func tearDown() { |
| 41 | + handler = nil |
| 42 | + database = nil |
| 43 | + cachedChannel = nil |
| 44 | + cid = nil |
| 45 | + super.tearDown() |
| 46 | + } |
| 47 | + |
| 48 | + // MARK: - Event Handling - Non-EventDTO |
| 49 | + |
| 50 | + func test_handle_nonEventDTO_returnsNil() { |
| 51 | + struct NonEventDTO: Event {} |
| 52 | + let event = NonEventDTO() |
| 53 | + |
| 54 | + let result = handler.handle(event) |
| 55 | + XCTAssertNil(result) |
| 56 | + } |
| 57 | + |
| 58 | + // MARK: - Event Handling - Missing CID |
| 59 | + |
| 60 | + func test_handle_eventWithoutCid_returnsNil() { |
| 61 | + // Create a simple event DTO that has no cid |
| 62 | + struct TestEventDTO: EventDTO { |
| 63 | + let payload: EventPayload = EventPayload( |
| 64 | + eventType: .healthCheck, |
| 65 | + connectionId: .unique |
| 66 | + ) |
| 67 | + } |
| 68 | + |
| 69 | + let eventDTO = TestEventDTO() |
| 70 | + let result = handler.handle(eventDTO) |
| 71 | + |
| 72 | + XCTAssertNil(result, "Events without cid should return nil") |
| 73 | + } |
| 74 | + |
| 75 | + // MARK: - Event Handling - Unregistered Channel |
| 76 | + |
| 77 | + func test_handle_unregisteredChannel_returnsNil() { |
| 78 | + let unregisteredCid: ChannelId = .unique |
| 79 | + let eventPayload = EventPayload( |
| 80 | + eventType: .messageNew, |
| 81 | + cid: unregisteredCid, |
| 82 | + user: .dummy(userId: .unique), |
| 83 | + message: .dummy(messageId: .unique, authorUserId: .unique), |
| 84 | + createdAt: .unique |
| 85 | + ) |
| 86 | + let eventDTO = try! MessageNewEventDTO(from: eventPayload) |
| 87 | + |
| 88 | + let result = handler.handle(eventDTO) |
| 89 | + XCTAssertNil(result) |
| 90 | + } |
| 91 | + |
| 92 | + // MARK: - Event Handling - Unsupported Event Type |
| 93 | + |
| 94 | + func test_handle_unsupportedEventType_returnsNil() { |
| 95 | + // Use a typing event which is not handled by ManualEventHandler |
| 96 | + let eventPayload = EventPayload( |
| 97 | + eventType: .userStartTyping, |
| 98 | + cid: cid, |
| 99 | + user: .dummy(userId: .unique), |
| 100 | + createdAt: .unique |
| 101 | + ) |
| 102 | + let eventDTO = try! TypingEventDTO(from: eventPayload) |
| 103 | + |
| 104 | + let result = handler.handle(eventDTO) |
| 105 | + XCTAssertNil(result, "Unsupported event types should return nil") |
| 106 | + } |
| 107 | + |
| 108 | + // MARK: - Message New Event |
| 109 | + |
| 110 | + func test_handle_messageNewEvent_withValidData_returnsEvent() throws { |
| 111 | + let userId: UserId = .unique |
| 112 | + let messageId: MessageId = .unique |
| 113 | + let createdAt = Date.unique |
| 114 | + |
| 115 | + let eventPayload = EventPayload( |
| 116 | + eventType: .messageNew, |
| 117 | + cid: cid, |
| 118 | + user: .dummy(userId: userId), |
| 119 | + message: .dummy(messageId: messageId, authorUserId: userId), |
| 120 | + watcherCount: 10, |
| 121 | + unreadCount: .init(channels: 1, messages: 2, threads: 0), |
| 122 | + createdAt: createdAt |
| 123 | + ) |
| 124 | + let eventDTO = try! MessageNewEventDTO(from: eventPayload) |
| 125 | + |
| 126 | + let result = handler.handle(eventDTO) |
| 127 | + |
| 128 | + let messageNewEvent = try XCTUnwrap(result as? MessageNewEvent) |
| 129 | + XCTAssertEqual(messageNewEvent.user.id, userId) |
| 130 | + XCTAssertEqual(messageNewEvent.message.id, messageId) |
| 131 | + XCTAssertEqual(messageNewEvent.cid, cid) |
| 132 | + XCTAssertEqual(messageNewEvent.watcherCount, 10) |
| 133 | + XCTAssertEqual(messageNewEvent.unreadCount?.messages, 2) |
| 134 | + XCTAssertEqual(messageNewEvent.createdAt, createdAt) |
| 135 | + } |
| 136 | + |
| 137 | + // MARK: - Message Updated Event |
| 138 | + |
| 139 | + func test_handle_messageUpdatedEvent_withValidData_returnsEvent() throws { |
| 140 | + let userId: UserId = .unique |
| 141 | + let messageId: MessageId = .unique |
| 142 | + let createdAt = Date.unique |
| 143 | + |
| 144 | + let eventPayload = EventPayload( |
| 145 | + eventType: .messageUpdated, |
| 146 | + cid: cid, |
| 147 | + user: .dummy(userId: userId), |
| 148 | + message: .dummy(messageId: messageId, authorUserId: userId), |
| 149 | + createdAt: createdAt |
| 150 | + ) |
| 151 | + let eventDTO = try! MessageUpdatedEventDTO(from: eventPayload) |
| 152 | + |
| 153 | + let result = handler.handle(eventDTO) |
| 154 | + |
| 155 | + let messageUpdatedEvent = try XCTUnwrap(result as? MessageUpdatedEvent) |
| 156 | + XCTAssertEqual(messageUpdatedEvent.user.id, userId) |
| 157 | + XCTAssertEqual(messageUpdatedEvent.message.id, messageId) |
| 158 | + XCTAssertEqual(messageUpdatedEvent.cid, cid) |
| 159 | + XCTAssertEqual(messageUpdatedEvent.createdAt, createdAt) |
| 160 | + } |
| 161 | + |
| 162 | + // MARK: - Message Deleted Event |
| 163 | + |
| 164 | + func test_handle_messageDeletedEvent_withValidData_returnsEvent() throws { |
| 165 | + let userId: UserId = .unique |
| 166 | + let messageId: MessageId = .unique |
| 167 | + let createdAt = Date.unique |
| 168 | + |
| 169 | + let eventPayload = EventPayload( |
| 170 | + eventType: .messageDeleted, |
| 171 | + cid: cid, |
| 172 | + user: .dummy(userId: userId), |
| 173 | + message: .dummy(messageId: messageId, authorUserId: userId), |
| 174 | + createdAt: createdAt, |
| 175 | + hardDelete: true |
| 176 | + ) |
| 177 | + let eventDTO = try! MessageDeletedEventDTO(from: eventPayload) |
| 178 | + |
| 179 | + let result = handler.handle(eventDTO) |
| 180 | + |
| 181 | + let messageDeletedEvent = try XCTUnwrap(result as? MessageDeletedEvent) |
| 182 | + XCTAssertEqual(messageDeletedEvent.user?.id, userId) |
| 183 | + XCTAssertEqual(messageDeletedEvent.message.id, messageId) |
| 184 | + XCTAssertEqual(messageDeletedEvent.cid, cid) |
| 185 | + XCTAssertEqual(messageDeletedEvent.isHardDelete, true) |
| 186 | + XCTAssertEqual(messageDeletedEvent.createdAt, createdAt) |
| 187 | + } |
| 188 | + |
| 189 | + func test_handle_messageDeletedEvent_withoutUser_returnsEvent() throws { |
| 190 | + let messageId: MessageId = .unique |
| 191 | + let createdAt = Date.unique |
| 192 | + |
| 193 | + let eventPayload = EventPayload( |
| 194 | + eventType: .messageDeleted, |
| 195 | + cid: cid, |
| 196 | + user: nil, |
| 197 | + message: .dummy(messageId: messageId, authorUserId: .unique), |
| 198 | + createdAt: createdAt, |
| 199 | + hardDelete: false |
| 200 | + ) |
| 201 | + let eventDTO = try! MessageDeletedEventDTO(from: eventPayload) |
| 202 | + |
| 203 | + let result = handler.handle(eventDTO) |
| 204 | + |
| 205 | + let messageDeletedEvent = try XCTUnwrap(result as? MessageDeletedEvent) |
| 206 | + XCTAssertNil(messageDeletedEvent.user) |
| 207 | + XCTAssertEqual(messageDeletedEvent.message.id, messageId) |
| 208 | + XCTAssertEqual(messageDeletedEvent.cid, cid) |
| 209 | + XCTAssertEqual(messageDeletedEvent.isHardDelete, false) |
| 210 | + XCTAssertEqual(messageDeletedEvent.createdAt, createdAt) |
| 211 | + } |
| 212 | + |
| 213 | + // MARK: - Reaction New Event |
| 214 | + |
| 215 | + func test_handle_reactionNewEvent_withValidData_returnsEvent() throws { |
| 216 | + let userId: UserId = .unique |
| 217 | + let messageId: MessageId = .unique |
| 218 | + let reactionType: MessageReactionType = "like" |
| 219 | + let createdAt = Date.unique |
| 220 | + |
| 221 | + let eventPayload = EventPayload( |
| 222 | + eventType: .reactionNew, |
| 223 | + cid: cid, |
| 224 | + user: .dummy(userId: userId), |
| 225 | + message: .dummy(messageId: messageId, authorUserId: userId), |
| 226 | + reaction: .dummy(type: reactionType, messageId: messageId, user: .dummy(userId: userId)), |
| 227 | + createdAt: createdAt |
| 228 | + ) |
| 229 | + let eventDTO = try! ReactionNewEventDTO(from: eventPayload) |
| 230 | + |
| 231 | + let result = handler.handle(eventDTO) |
| 232 | + |
| 233 | + let reactionNewEvent = try XCTUnwrap(result as? ReactionNewEvent) |
| 234 | + XCTAssertEqual(reactionNewEvent.user.id, userId) |
| 235 | + XCTAssertEqual(reactionNewEvent.message.id, messageId) |
| 236 | + XCTAssertEqual(reactionNewEvent.cid, cid) |
| 237 | + XCTAssertEqual(reactionNewEvent.reaction.type, reactionType) |
| 238 | + XCTAssertEqual(reactionNewEvent.createdAt, createdAt) |
| 239 | + } |
| 240 | + |
| 241 | + // MARK: - Reaction Updated Event |
| 242 | + |
| 243 | + func test_handle_reactionUpdatedEvent_withValidData_returnsEvent() throws { |
| 244 | + let userId: UserId = .unique |
| 245 | + let messageId: MessageId = .unique |
| 246 | + let reactionType: MessageReactionType = "love" |
| 247 | + let createdAt = Date.unique |
| 248 | + |
| 249 | + let eventPayload = EventPayload( |
| 250 | + eventType: .reactionUpdated, |
| 251 | + cid: cid, |
| 252 | + user: .dummy(userId: userId), |
| 253 | + message: .dummy(messageId: messageId, authorUserId: userId), |
| 254 | + reaction: .dummy(type: reactionType, messageId: messageId, user: .dummy(userId: userId)), |
| 255 | + createdAt: createdAt |
| 256 | + ) |
| 257 | + let eventDTO = try! ReactionUpdatedEventDTO(from: eventPayload) |
| 258 | + |
| 259 | + let result = handler.handle(eventDTO) |
| 260 | + |
| 261 | + let reactionUpdatedEvent = try XCTUnwrap(result as? ReactionUpdatedEvent) |
| 262 | + XCTAssertEqual(reactionUpdatedEvent.user.id, userId) |
| 263 | + XCTAssertEqual(reactionUpdatedEvent.message.id, messageId) |
| 264 | + XCTAssertEqual(reactionUpdatedEvent.cid, cid) |
| 265 | + XCTAssertEqual(reactionUpdatedEvent.reaction.type, reactionType) |
| 266 | + XCTAssertEqual(reactionUpdatedEvent.createdAt, createdAt) |
| 267 | + } |
| 268 | + |
| 269 | + // MARK: - Reaction Deleted Event |
| 270 | + |
| 271 | + func test_handle_reactionDeletedEvent_withValidData_returnsEvent() throws { |
| 272 | + let userId: UserId = .unique |
| 273 | + let messageId: MessageId = .unique |
| 274 | + let reactionType: MessageReactionType = "angry" |
| 275 | + let createdAt = Date.unique |
| 276 | + |
| 277 | + let eventPayload = EventPayload( |
| 278 | + eventType: .reactionDeleted, |
| 279 | + cid: cid, |
| 280 | + user: .dummy(userId: userId), |
| 281 | + message: .dummy(messageId: messageId, authorUserId: userId), |
| 282 | + reaction: .dummy(type: reactionType, messageId: messageId, user: .dummy(userId: userId)), |
| 283 | + createdAt: createdAt |
| 284 | + ) |
| 285 | + let eventDTO = try! ReactionDeletedEventDTO(from: eventPayload) |
| 286 | + |
| 287 | + let result = handler.handle(eventDTO) |
| 288 | + |
| 289 | + let reactionDeletedEvent = try XCTUnwrap(result as? ReactionDeletedEvent) |
| 290 | + XCTAssertEqual(reactionDeletedEvent.user.id, userId) |
| 291 | + XCTAssertEqual(reactionDeletedEvent.message.id, messageId) |
| 292 | + XCTAssertEqual(reactionDeletedEvent.cid, cid) |
| 293 | + XCTAssertEqual(reactionDeletedEvent.reaction.type, reactionType) |
| 294 | + XCTAssertEqual(reactionDeletedEvent.createdAt, createdAt) |
| 295 | + } |
| 296 | +} |
0 commit comments