X-Git-Url: https://git.giacomolli.uk/onechat/blobdiff_plain/f59bb67ae1bc2c899ef4f6be9ce2ff6296c7ffcc..84fc815feb485b35eec53a8c7863c6770fc65e02:/turn_test.go?ds=inline

diff --git a/turn_test.go b/turn_test.go
index f1295e9..0de7dd7 100644
--- a/turn_test.go
+++ b/turn_test.go
@@ -30,7 +30,7 @@ func TestTurnRunsToolsUntilTheFinalAnswer(t *testing.T) {
 	}
 
 	test := newTestLoop(t, server)
-	_, err := test.loop.Submit("Run echo hi.")
+	_, err := test.loop.Submit(terminalSource, "Run echo hi.")
 	require.NoError(t, err)
 	test.events.WaitFor(t, "idle", 1)
 	require.Equal(t, 2, server.Calls())
@@ -74,7 +74,7 @@ func TestFirstRequestHoldsTheViewTheStateAndTheMessage(t *testing.T) {
 	_, err := test.memory.Append("user", "an earlier message")
 	require.NoError(t, err)
 
-	_, err = test.loop.Submit("The new message.")
+	_, err = test.loop.Submit(terminalSource, "The new message.")
 	require.NoError(t, err)
 	test.events.WaitFor(t, "idle", 1)
 
@@ -102,11 +102,11 @@ func TestMessageDuringATurnJoinsTheNextRequest(t *testing.T) {
 	}
 
 	test := newTestLoop(t, server)
-	_, err := test.loop.Submit("first")
+	_, err := test.loop.Submit(terminalSource, "first")
 	require.NoError(t, err)
 
 	<-arrived
-	_, err = test.loop.Submit("second, during the turn")
+	_, err = test.loop.Submit(terminalSource, "second, during the turn")
 	require.NoError(t, err)
 	close(release)
 
@@ -136,11 +136,11 @@ func TestMessageDuringTheFinalRequestStartsANewTurn(t *testing.T) {
 	}
 
 	test := newTestLoop(t, server)
-	_, err := test.loop.Submit("first")
+	_, err := test.loop.Submit(terminalSource, "first")
 	require.NoError(t, err)
 
 	<-arrived
-	_, err = test.loop.Submit("second")
+	_, err = test.loop.Submit(terminalSource, "second")
 	require.NoError(t, err)
 	close(release)
 
@@ -161,7 +161,7 @@ func TestTurnWaitsForTheSummaries(t *testing.T) {
 	_, err := test.memory.Append("echo", strings.Repeat("long output ", 100))
 	require.NoError(t, err)
 
-	_, err = test.loop.Submit("go on")
+	_, err = test.loop.Submit(terminalSource, "go on")
 	require.NoError(t, err)
 
 	time.Sleep(50 * time.Millisecond)
@@ -187,7 +187,7 @@ func TestInterruptEndsTheTurn(t *testing.T) {
 	}
 
 	test := newTestLoop(t, server)
-	_, err := test.loop.Submit("a long task")
+	_, err := test.loop.Submit(terminalSource, "a long task")
 	require.NoError(t, err)
 
 	<-arrived
@@ -199,7 +199,7 @@ func TestInterruptEndsTheTurn(t *testing.T) {
 	require.Equal(t, []string{"user: a long task"}, test.Log(t))
 	require.False(t, test.loop.Interrupt())
 
-	_, err = test.loop.Submit("again")
+	_, err = test.loop.Submit(terminalSource, "again")
 	require.NoError(t, err)
 	test.events.WaitFor(t, "idle", 2)
 	require.Equal(t, "reply: answer", test.Log(t)[2])
@@ -212,7 +212,7 @@ func TestFailedCallEndsTheTurnWithoutARetry(t *testing.T) {
 	}
 
 	test := newTestLoop(t, server)
-	_, err := test.loop.Submit("hello")
+	_, err := test.loop.Submit(terminalSource, "hello")
 	require.NoError(t, err)
 	test.events.WaitFor(t, "idle", 1)
 
@@ -225,7 +225,7 @@ func TestViewMessagesOnlyGrowBetweenTurns(t *testing.T) {
 	server := newFakeResponses(t)
 	test := newTestLoop(t, server)
 	for n, text := range []string{"one", "two", "three"} {
-		_, err := test.loop.Submit(text)
+		_, err := test.loop.Submit(terminalSource, text)
 		require.NoError(t, err)
 		test.events.WaitFor(t, "idle", n+1)
 	}