@@ -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