Skip to content

Commit 56261df

Browse files
committed
test: stabilize open menu refresh test
1 parent 24b96c0 commit 56261df

1 file changed

Lines changed: 8 additions & 2 deletions

File tree

Tests/CodexBarTests/StatusMenuOpenRefreshTests.swift

Lines changed: 8 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -88,14 +88,20 @@ extension StatusMenuTests {
8888
defer { StatusItemController.resetMenuRefreshEnabledForTesting() }
8989

9090
let openedVersion = controller.menuVersions[key]
91+
var rebuildCount = 0
92+
controller._test_openMenuRebuildObserver = { _ in
93+
rebuildCount += 1
94+
}
95+
defer { controller._test_openMenuRebuildObserver = nil }
9196

9297
controller.refreshOpenMenusAfterExplicitStoreAction()
93-
for _ in 0..<20 where controller.menuVersions[key] != controller.menuContentVersion {
98+
for _ in 0..<20 where rebuildCount == 0 {
9499
await Task.yield()
95100
}
96101

97102
#expect(controller.menuContentVersion != openedVersion)
98-
#expect(controller.menuVersions[key] == controller.menuContentVersion)
103+
#expect(rebuildCount == 1)
104+
#expect(controller.menuVersions[key] != openedVersion)
99105
}
100106

101107
@Test

0 commit comments

Comments
 (0)