Add more timeline tests

This commit is contained in:
Thomas Ricouard 2023-12-30 09:51:34 +01:00
parent b0cc02541e
commit 6ea4888ae5
2 changed files with 96 additions and 9 deletions

View file

@ -18,15 +18,7 @@ import SwiftUI
didSet {
timelineTask?.cancel()
timelineTask = Task {
if timeline == .latest || timeline == .resume {
if oldValue == .home {
await clearHomeCache()
}
if timeline == .resume, let marker = await fetchMarker() {
self.marker = marker
}
timeline = oldValue
}
await handleLatestOrResume(oldValue)
if oldValue != timeline {
await reset()
@ -103,6 +95,18 @@ import SwiftUI
func reset() async {
await datasource.reset()
}
private func handleLatestOrResume(_ oldValue: TimelineFilter) async {
if timeline == .latest || timeline == .resume {
if oldValue == .home {
await clearHomeCache()
}
if timeline == .resume, let marker = await fetchMarker() {
self.marker = marker
}
timeline = oldValue
}
}
func handleEvent(event: any StreamEvent) async {
if let event = event as? StreamEventUpdate,

View file

@ -21,4 +21,87 @@ final class TimelineViewModelTests: XCTestCase {
count = await subject.datasource.count()
XCTAssertTrue(count == 2)
}
func testStreamEventInsertDuplicateStatus() async throws {
let subject = TimelineViewModel()
let client = Client(server: "localhost")
subject.client = client
subject.timeline = .home
subject.isTimelineVisible = true
let isEmpty = await subject.datasource.isEmpty
XCTAssertTrue(isEmpty)
let status = Status.placeholder()
await subject.datasource.append(status)
var count = await subject.datasource.count()
XCTAssertTrue(count == 1)
await subject.handleEvent(event: StreamEventUpdate(status: status))
count = await subject.datasource.count()
XCTAssertTrue(count == 1)
}
func testStreamEventRemove() async throws {
let subject = TimelineViewModel()
let client = Client(server: "localhost")
subject.client = client
subject.timeline = .home
subject.isTimelineVisible = true
let isEmpty = await subject.datasource.isEmpty
XCTAssertTrue(isEmpty)
let status = Status.placeholder()
await subject.datasource.append(status)
var count = await subject.datasource.count()
XCTAssertTrue(count == 1)
await subject.handleEvent(event: StreamEventDelete(status: status.id))
count = await subject.datasource.count()
XCTAssertTrue(count == 0)
}
func testStreamEventUpdateStatus() async throws {
let subject = TimelineViewModel()
let client = Client(server: "localhost")
subject.client = client
subject.timeline = .home
subject.isTimelineVisible = true
var status = Status.placeholder()
let isEmpty = await subject.datasource.isEmpty
XCTAssertTrue(isEmpty)
await subject.datasource.append(status)
var count = await subject.datasource.count()
XCTAssertTrue(count == 1)
status = .init(id: status.id,
content: .init(stringValue: "test"),
account: status.account,
createdAt: status.createdAt,
editedAt: status.editedAt,
reblog: status.reblog,
mediaAttachments: status.mediaAttachments,
mentions: status.mentions,
repliesCount: status.repliesCount,
reblogsCount: status.reblogsCount,
favouritesCount: status.favouritesCount,
card: status.card,
favourited: status.favourited,
reblogged: status.reblogged,
pinned: status.pinned,
bookmarked: status.bookmarked,
emojis: status.emojis,
url: status.url,
application: status.application,
inReplyToId: status.inReplyToId,
inReplyToAccountId: status.inReplyToAccountId,
visibility: status.visibility,
poll: status.poll,
spoilerText: status.spoilerText,
filtered: status.filtered,
sensitive: status.sensitive,
language: status.language)
await subject.handleEvent(event: StreamEventStatusUpdate(status: status))
let statuses = await subject.datasource.get()
count = await subject.datasource.count()
XCTAssertTrue(count == 1)
XCTAssertTrue(statuses.first?.content.asRawText == "test")
}
}