d25bfefee0
- Introduce new unit test targets for Workspace and Sandbox V2 integration in the Makefile, ensuring comprehensive testing of the new features. - Update CI workflows to support MongoDB service for Sandbox V2 tests and improve Docker image handling. - Implement Workspace as a first-class entity, allowing for persistent storage decoupled from container lifecycles. - Enhance the Box struct to manage workspace IDs and update related methods for improved functionality. - Refactor tests to accommodate new Workspace features, ensuring robust testing of the integration. These changes significantly improve the testing framework and functionality of the Sandbox V2, enhancing the overall architecture and user experience.
89 lines
2 KiB
Go
89 lines
2 KiB
Go
package workspace_test
|
|
|
|
import (
|
|
"context"
|
|
"os"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/yaoapp/yao/tai"
|
|
"github.com/yaoapp/yao/tai/volume"
|
|
"github.com/yaoapp/yao/workspace"
|
|
)
|
|
|
|
type poolConfig struct {
|
|
Name string
|
|
Addr string
|
|
}
|
|
|
|
func testPools() []poolConfig {
|
|
pools := []poolConfig{
|
|
{Name: "local", Addr: "local"},
|
|
}
|
|
if addr := os.Getenv("SANDBOX_TEST_REMOTE_ADDR"); addr != "" {
|
|
pools = append(pools, poolConfig{Name: "remote", Addr: addr})
|
|
}
|
|
return pools
|
|
}
|
|
|
|
func setupManagerForPool(tb testing.TB, pc poolConfig) *workspace.Manager {
|
|
tb.Helper()
|
|
client := clientForPool(tb, pc)
|
|
pools := map[string]*tai.Client{pc.Name: client}
|
|
return workspace.NewManager(pools)
|
|
}
|
|
|
|
func clientForPool(tb testing.TB, pc poolConfig) *tai.Client {
|
|
tb.Helper()
|
|
if pc.Addr == "local" {
|
|
return localClient(tb, tb.TempDir())
|
|
}
|
|
client, err := tai.New(pc.Addr)
|
|
if err != nil {
|
|
tb.Fatalf("tai.New(%s): %v", pc.Addr, err)
|
|
}
|
|
tb.Cleanup(func() { client.Close() })
|
|
return client
|
|
}
|
|
|
|
func localClient(tb testing.TB, dataDir string) *tai.Client {
|
|
tb.Helper()
|
|
vol := volume.NewLocal(dataDir)
|
|
client, err := tai.New("local", tai.WithVolume(vol), tai.WithDataDir(dataDir))
|
|
if err != nil {
|
|
tb.Fatalf("tai.New local: %v", err)
|
|
}
|
|
tb.Cleanup(func() { client.Close() })
|
|
return client
|
|
}
|
|
|
|
func setupManagerMultiNode(t *testing.T) *workspace.Manager {
|
|
t.Helper()
|
|
pools := map[string]*tai.Client{
|
|
"node-a": localClient(t, t.TempDir()),
|
|
"node-b": localClient(t, t.TempDir()),
|
|
}
|
|
return workspace.NewManager(pools)
|
|
}
|
|
|
|
func createWorkspace(tb testing.TB, m *workspace.Manager, node string, opts ...func(*workspace.CreateOptions)) *workspace.Workspace {
|
|
tb.Helper()
|
|
co := workspace.CreateOptions{
|
|
Name: "test-workspace",
|
|
Owner: "test-user",
|
|
Node: node,
|
|
}
|
|
for _, fn := range opts {
|
|
fn(&co)
|
|
}
|
|
ctx, cancel := context.WithTimeout(context.Background(), 10*time.Second)
|
|
defer cancel()
|
|
ws, err := m.Create(ctx, co)
|
|
if err != nil {
|
|
tb.Fatalf("Create workspace: %v", err)
|
|
}
|
|
tb.Cleanup(func() {
|
|
m.Delete(context.Background(), ws.ID, true)
|
|
})
|
|
return ws
|
|
}
|