120 lines
2.9 KiB
Go
120 lines
2.9 KiB
Go
package workspace
|
|
|
|
import (
|
|
"testing"
|
|
|
|
core "dappco.re/go/core"
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
type stubCrypt struct {
|
|
key string
|
|
err error
|
|
}
|
|
|
|
func (s stubCrypt) CreateKeyPair(_, _ string) (string, error) {
|
|
if s.err != nil {
|
|
return "", s.err
|
|
}
|
|
return s.key, nil
|
|
}
|
|
|
|
func newTestService(t *testing.T) (*Service, string) {
|
|
t.Helper()
|
|
|
|
tempHome := t.TempDir()
|
|
t.Setenv("HOME", tempHome)
|
|
|
|
svc, err := New(Options{Core: core.New(), Crypt: stubCrypt{key: "private-key"}})
|
|
require.NoError(t, err)
|
|
return svc, tempHome
|
|
}
|
|
|
|
func TestService_Workspace_RoundTrip_Good(t *testing.T) {
|
|
s, tempHome := newTestService(t)
|
|
|
|
id, err := s.CreateWorkspace("test-user", "pass123")
|
|
require.NoError(t, err)
|
|
assert.NotEmpty(t, id)
|
|
|
|
wsPath := core.Path(tempHome, ".core", "workspaces", id)
|
|
assert.DirExists(t, wsPath)
|
|
assert.DirExists(t, core.Path(wsPath, "keys"))
|
|
assert.FileExists(t, core.Path(wsPath, "keys", "private.key"))
|
|
|
|
err = s.SwitchWorkspace(id)
|
|
require.NoError(t, err)
|
|
assert.Equal(t, id, s.activeWorkspace)
|
|
|
|
err = s.WorkspaceFileSet("secret.txt", "top secret info")
|
|
require.NoError(t, err)
|
|
|
|
got, err := s.WorkspaceFileGet("secret.txt")
|
|
require.NoError(t, err)
|
|
assert.Equal(t, "top secret info", got)
|
|
}
|
|
|
|
func TestService_SwitchWorkspace_TraversalBlocked_Bad(t *testing.T) {
|
|
s, tempHome := newTestService(t)
|
|
|
|
outside := core.Path(tempHome, ".core", "escaped")
|
|
require.NoError(t, s.medium.EnsureDir(outside))
|
|
|
|
err := s.SwitchWorkspace("../escaped")
|
|
require.Error(t, err)
|
|
assert.Empty(t, s.activeWorkspace)
|
|
}
|
|
|
|
func TestService_WorkspaceFileSet_TraversalBlocked_Bad(t *testing.T) {
|
|
s, tempHome := newTestService(t)
|
|
|
|
id, err := s.CreateWorkspace("test-user", "pass123")
|
|
require.NoError(t, err)
|
|
require.NoError(t, s.SwitchWorkspace(id))
|
|
|
|
keyPath := core.Path(tempHome, ".core", "workspaces", id, "keys", "private.key")
|
|
before, err := s.medium.Read(keyPath)
|
|
require.NoError(t, err)
|
|
|
|
err = s.WorkspaceFileSet("../keys/private.key", "hijack")
|
|
require.Error(t, err)
|
|
|
|
after, err := s.medium.Read(keyPath)
|
|
require.NoError(t, err)
|
|
assert.Equal(t, before, after)
|
|
|
|
_, err = s.WorkspaceFileGet("../keys/private.key")
|
|
require.Error(t, err)
|
|
}
|
|
|
|
func TestService_HandleIPCEvents_Good(t *testing.T) {
|
|
s, _ := newTestService(t)
|
|
|
|
create := s.HandleIPCEvents(core.New(), map[string]any{
|
|
"action": "workspace.create",
|
|
"identifier": "ipc-user",
|
|
"password": "pass123",
|
|
})
|
|
assert.True(t, create.OK)
|
|
|
|
id, ok := create.Value.(string)
|
|
require.True(t, ok)
|
|
require.NotEmpty(t, id)
|
|
|
|
switchResult := s.HandleIPCEvents(core.New(), map[string]any{
|
|
"action": "workspace.switch",
|
|
"name": id,
|
|
})
|
|
assert.True(t, switchResult.OK)
|
|
assert.Equal(t, id, s.activeWorkspace)
|
|
|
|
failedSwitch := s.HandleIPCEvents(core.New(), map[string]any{
|
|
"action": "workspace.switch",
|
|
"name": "missing",
|
|
})
|
|
assert.False(t, failedSwitch.OK)
|
|
|
|
unknown := s.HandleIPCEvents(core.New(), "noop")
|
|
assert.True(t, unknown.OK)
|
|
}
|