mirror of
https://github.com/coder/coder.git
synced 2026-06-03 21:18:24 +00:00
Thomas Kosiewski
106 lines
2.4 KiB
Go
106 lines
2.4 KiB
Go
package chatdebug_test
|
|
|
|
import (
|
|
"context"
|
|
"testing"
|
|
|
|
"github.com/google/uuid"
|
|
"github.com/stretchr/testify/require"
|
|
|
|
"github.com/coder/coder/v2/coderd/x/chatd/chatdebug"
|
|
)
|
|
|
|
func TestContextWithRunRoundTrip(t *testing.T) {
|
|
t.Parallel()
|
|
|
|
rc := &chatdebug.RunContext{
|
|
RunID: uuid.New(),
|
|
ChatID: uuid.New(),
|
|
RootChatID: uuid.New(),
|
|
ParentChatID: uuid.New(),
|
|
ModelConfigID: uuid.New(),
|
|
TriggerMessageID: 11,
|
|
HistoryTipMessageID: 22,
|
|
Kind: chatdebug.KindChatTurn,
|
|
Provider: "anthropic",
|
|
Model: "claude-sonnet",
|
|
}
|
|
|
|
ctx := chatdebug.ContextWithRun(context.Background(), rc)
|
|
got, ok := chatdebug.RunFromContext(ctx)
|
|
require.True(t, ok)
|
|
require.Same(t, rc, got)
|
|
require.Equal(t, *rc, *got)
|
|
}
|
|
|
|
func TestRunFromContextAbsent(t *testing.T) {
|
|
t.Parallel()
|
|
|
|
got, ok := chatdebug.RunFromContext(context.Background())
|
|
require.False(t, ok)
|
|
require.Nil(t, got)
|
|
}
|
|
|
|
func TestContextWithStepRoundTrip(t *testing.T) {
|
|
t.Parallel()
|
|
|
|
sc := &chatdebug.StepContext{
|
|
StepID: uuid.New(),
|
|
RunID: uuid.New(),
|
|
ChatID: uuid.New(),
|
|
StepNumber: 7,
|
|
Operation: chatdebug.OperationStream,
|
|
HistoryTipMessageID: 33,
|
|
}
|
|
|
|
ctx := chatdebug.ContextWithStep(context.Background(), sc)
|
|
got, ok := chatdebug.StepFromContext(ctx)
|
|
require.True(t, ok)
|
|
require.Same(t, sc, got)
|
|
require.Equal(t, *sc, *got)
|
|
}
|
|
|
|
func TestStepFromContextAbsent(t *testing.T) {
|
|
t.Parallel()
|
|
|
|
got, ok := chatdebug.StepFromContext(context.Background())
|
|
require.False(t, ok)
|
|
require.Nil(t, got)
|
|
}
|
|
|
|
func TestContextWithRunAndStep(t *testing.T) {
|
|
t.Parallel()
|
|
|
|
rc := &chatdebug.RunContext{RunID: uuid.New(), ChatID: uuid.New()}
|
|
sc := &chatdebug.StepContext{StepID: uuid.New(), RunID: rc.RunID, ChatID: rc.ChatID}
|
|
|
|
ctx := chatdebug.ContextWithStep(
|
|
chatdebug.ContextWithRun(context.Background(), rc),
|
|
sc,
|
|
)
|
|
|
|
gotRun, ok := chatdebug.RunFromContext(ctx)
|
|
require.True(t, ok)
|
|
require.Same(t, rc, gotRun)
|
|
|
|
gotStep, ok := chatdebug.StepFromContext(ctx)
|
|
require.True(t, ok)
|
|
require.Same(t, sc, gotStep)
|
|
}
|
|
|
|
func TestContextWithRunPanicsOnNil(t *testing.T) {
|
|
t.Parallel()
|
|
|
|
require.Panics(t, func() {
|
|
_ = chatdebug.ContextWithRun(context.Background(), nil)
|
|
})
|
|
}
|
|
|
|
func TestContextWithStepPanicsOnNil(t *testing.T) {
|
|
t.Parallel()
|
|
|
|
require.Panics(t, func() {
|
|
_ = chatdebug.ContextWithStep(context.Background(), nil)
|
|
})
|
|
}
|