(t *testing.T, sess *session.Session, stream *mockStream)
| 229 | } |
| 230 | |
| 231 | func runSession(t *testing.T, sess *session.Session, stream *mockStream) []Event { |
| 232 | t.Helper() |
| 233 | |
| 234 | prov := &mockProvider{id: "test/mock-model", stream: stream} |
| 235 | root := agent.New("root", "You are a test agent", agent.WithModel(prov)) |
| 236 | tm := team.New(team.WithAgents(root)) |
| 237 | |
| 238 | rt, err := NewLocalRuntime(t.Context(), tm, WithSessionCompaction(false), WithModelStore(mockModelStore{})) |
| 239 | require.NoError(t, err) |
| 240 | |
| 241 | sess.Title = "Unit Test" |
| 242 | |
| 243 | evCh := rt.RunStream(t.Context(), sess) |
| 244 | |
| 245 | var events []Event |
| 246 | for ev := range evCh { |
| 247 | events = append(events, ev) |
| 248 | } |
| 249 | return events |
| 250 | } |
| 251 | |
| 252 | func hasEventType(t *testing.T, events []Event, target Event) bool { |
| 253 | t.Helper() |
no test coverage detected