Haotian
631c174f47
* Adds WorkerAuthStorage wrapper for worker credential encryption using nodeenrollment library. * Adds AuthStoragePath field in worker config and dev mode to specify directory to store credentials.
90 lines
2.1 KiB
Go
90 lines
2.1 KiB
Go
package worker
|
|
|
|
import (
|
|
"sync"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/hashicorp/boundary/internal/daemon/worker/session"
|
|
"github.com/hashicorp/boundary/internal/db"
|
|
pbs "github.com/hashicorp/boundary/internal/gen/controller/servers/services"
|
|
wrapping "github.com/hashicorp/go-kms-wrapping/v2"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func TestTestWorkerLookupSession(t *testing.T) {
|
|
require := require.New(t)
|
|
// This loads the golang reference time, see those docs for more details. We
|
|
// just use this as a stable non-zero time.
|
|
refTime, err := time.Parse(time.RFC3339, "2006-01-02T15:04:05+07:00")
|
|
require.NoError(err)
|
|
|
|
tw := new(TestWorker)
|
|
tw.w = &Worker{
|
|
sessionInfoMap: new(sync.Map),
|
|
}
|
|
tw.w.sessionInfoMap.Store("foo", &session.Info{
|
|
Id: "foo",
|
|
Status: pbs.SESSIONSTATUS_SESSIONSTATUS_ACTIVE,
|
|
ConnInfoMap: map[string]*session.ConnInfo{
|
|
"one": {
|
|
Id: "one",
|
|
Status: pbs.CONNECTIONSTATUS_CONNECTIONSTATUS_CLOSED,
|
|
CloseTime: refTime,
|
|
},
|
|
},
|
|
})
|
|
|
|
expected := TestSessionInfo{
|
|
Id: "foo",
|
|
Status: pbs.SESSIONSTATUS_SESSIONSTATUS_ACTIVE,
|
|
Connections: map[string]TestConnectionInfo{
|
|
"one": {
|
|
Id: "one",
|
|
Status: pbs.CONNECTIONSTATUS_CONNECTIONSTATUS_CLOSED,
|
|
CloseTime: refTime,
|
|
},
|
|
},
|
|
}
|
|
|
|
actual, ok := tw.LookupSession("foo")
|
|
require.True(ok)
|
|
require.Equal(expected, actual)
|
|
}
|
|
|
|
func TestTestWorkerLookupSessionMissing(t *testing.T) {
|
|
require := require.New(t)
|
|
tw := new(TestWorker)
|
|
tw.w = &Worker{
|
|
sessionInfoMap: new(sync.Map),
|
|
}
|
|
actual, ok := tw.LookupSession("missing")
|
|
require.False(ok)
|
|
require.Equal(TestSessionInfo{}, actual)
|
|
}
|
|
|
|
func TestTestWorker_WorkerAuthStorageKms(t *testing.T) {
|
|
tests := []struct {
|
|
name string
|
|
wrapper wrapping.Wrapper
|
|
}{
|
|
{
|
|
name: "Nil Wrapper",
|
|
wrapper: nil,
|
|
},
|
|
{
|
|
name: "Valid Wrapper",
|
|
wrapper: db.TestWrapper(t),
|
|
},
|
|
}
|
|
for _, tt := range tests {
|
|
t.Run(tt.name, func(t *testing.T) {
|
|
require := require.New(t)
|
|
tw := NewTestWorker(t, &TestWorkerOpts{
|
|
WorkerAuthStorageKms: tt.wrapper,
|
|
})
|
|
require.Equal(tt.wrapper, tw.Config().WorkerAuthStorageKms)
|
|
})
|
|
}
|
|
}
|