mirror of
https://github.com/coder/coder.git
synced 2026-06-05 14:08:20 +00:00
Hugo Dutka
27 lines
677 B
Go
27 lines
677 B
Go
package chatd
|
|
|
|
import (
|
|
"context"
|
|
"sync"
|
|
|
|
"github.com/coder/coder/v2/coderd/database"
|
|
"github.com/coder/coder/v2/codersdk/workspacesdk"
|
|
)
|
|
|
|
type TurnWorkspaceContextForTest struct {
|
|
inner *turnWorkspaceContext
|
|
}
|
|
|
|
func NewTurnWorkspaceContextForTest(server *Server, chat database.Chat) *TurnWorkspaceContextForTest {
|
|
return &TurnWorkspaceContextForTest{inner: &turnWorkspaceContext{
|
|
server: server,
|
|
chatStateMu: &sync.Mutex{},
|
|
currentChat: &chat,
|
|
loadChatSnapshot: server.db.GetChatByID,
|
|
}}
|
|
}
|
|
|
|
func (c *TurnWorkspaceContextForTest) GetWorkspaceConn(ctx context.Context) (workspacesdk.AgentConn, error) {
|
|
return c.inner.getWorkspaceConn(ctx)
|
|
}
|