Skip to content

Commit 7007750

Browse files
committed
Fix tests
1 parent d30c37c commit 7007750

File tree

1 file changed

+6
-7
lines changed

1 file changed

+6
-7
lines changed

VoiceAgentTests/ChatViewModelTests.swift renamed to VoiceAgentTests/AgentSessionTests.swift

Lines changed: 6 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -2,7 +2,7 @@ import Testing
22
@testable import VoiceAgent
33

44
@MainActor
5-
struct ChatViewModelTests {
5+
struct AgentSessionTests {
66
@Test func multipleReceivers() async throws {
77
let receiver1 = MockMessageReceiver()
88
let receiver2 = MockMessageReceiver()
@@ -18,20 +18,19 @@ struct ChatViewModelTests {
1818
content: .agentTranscript("Hi there")
1919
)
2020

21-
Dependencies.shared.messageReceivers = [receiver1, receiver2]
22-
let viewModel = ChatViewModel()
21+
let session = AgentSession(environment: .sandbox(id: ""), receivers: [receiver1, receiver2])
2322

2423
try await Task.sleep(for: .milliseconds(100))
2524
await receiver1.postMessage(message1)
2625
try await Task.sleep(for: .milliseconds(100))
2726
await receiver2.postMessage(message2)
2827
try await Task.sleep(for: .milliseconds(100))
2928

30-
#expect(viewModel.messages.count == 2)
31-
#expect(viewModel.messages["1"]?.content == .userTranscript("Hello"))
32-
#expect(viewModel.messages["2"]?.content == .agentTranscript("Hi there"))
29+
#expect(session.messages.count == 2)
30+
#expect(session.messages["1"]?.content == .userTranscript("Hello"))
31+
#expect(session.messages["2"]?.content == .agentTranscript("Hi there"))
3332

34-
let orderedMessages = Array(viewModel.messages.values)
33+
let orderedMessages = session.getMessageHistory()
3534
#expect(orderedMessages.count == 2)
3635
#expect(orderedMessages[0].id == "1")
3736
#expect(orderedMessages[1].id == "2")

0 commit comments

Comments
 (0)