96 lines
2.9 KiB
Go
96 lines
2.9 KiB
Go
// Copyright (c) HashiCorp, Inc.
|
|
// SPDX-License-Identifier: BUSL-1.1
|
|
|
|
package plugin
|
|
|
|
import (
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/hashicorp/boundary/internal/db/timestamp"
|
|
"github.com/hashicorp/boundary/internal/pagination"
|
|
"github.com/stretchr/testify/assert"
|
|
"google.golang.org/protobuf/types/known/structpb"
|
|
)
|
|
|
|
type fakeItem struct {
|
|
pagination.Item
|
|
publicId string
|
|
updateTime time.Time
|
|
}
|
|
|
|
func (p *fakeItem) GetPublicId() string {
|
|
return p.publicId
|
|
}
|
|
|
|
func (p *fakeItem) GetUpdateTime() *timestamp.Timestamp {
|
|
return timestamp.New(p.updateTime)
|
|
}
|
|
|
|
func Test_GetOpts(t *testing.T) {
|
|
t.Parallel()
|
|
t.Run("WithChunkSize", func(t *testing.T) {
|
|
t.Parallel()
|
|
opts := getOpts(WithChunkSize(1024))
|
|
testOpts := getDefaultOptions()
|
|
testOpts.withChunkSize = 1024
|
|
assert.Equal(t, opts, testOpts)
|
|
})
|
|
t.Run("WithAttributes", func(t *testing.T) {
|
|
t.Parallel()
|
|
opts := getOpts(WithAttributes(&structpb.Struct{Fields: map[string]*structpb.Value{"foo": structpb.NewStringValue("bar")}}))
|
|
testOpts := getDefaultOptions()
|
|
testOpts.withAttributes = &structpb.Struct{Fields: map[string]*structpb.Value{"foo": structpb.NewStringValue("bar")}}
|
|
assert.Equal(t, opts, testOpts)
|
|
})
|
|
t.Run("WithName", func(t *testing.T) {
|
|
t.Parallel()
|
|
opts := getOpts(WithName("test"))
|
|
testOpts := getDefaultOptions()
|
|
testOpts.withName = "test"
|
|
assert.Equal(t, opts, testOpts)
|
|
})
|
|
t.Run("WithDescription", func(t *testing.T) {
|
|
t.Parallel()
|
|
opts := getOpts(WithDescription("test desc"))
|
|
testOpts := getDefaultOptions()
|
|
testOpts.withDescription = "test desc"
|
|
assert.Equal(t, opts, testOpts)
|
|
})
|
|
t.Run("WithSecrets", func(t *testing.T) {
|
|
t.Parallel()
|
|
opts := getOpts(WithSecrets(&structpb.Struct{Fields: map[string]*structpb.Value{"foo": structpb.NewStringValue("bar")}}))
|
|
testOpts := getDefaultOptions()
|
|
testOpts.withSecrets = &structpb.Struct{Fields: map[string]*structpb.Value{"foo": structpb.NewStringValue("bar")}}
|
|
assert.Equal(t, opts, testOpts)
|
|
})
|
|
t.Run("WithBucketPrefix", func(t *testing.T) {
|
|
t.Parallel()
|
|
opts := getOpts(WithBucketPrefix("test prefix"))
|
|
testOpts := getDefaultOptions()
|
|
testOpts.withBucketPrefix = "test prefix"
|
|
assert.Equal(t, opts, testOpts)
|
|
})
|
|
t.Run("WithWorkerFilter", func(t *testing.T) {
|
|
t.Parallel()
|
|
opts := getOpts(WithWorkerFilter("test filter"))
|
|
testOpts := getDefaultOptions()
|
|
testOpts.withWorkerFilter = "test filter"
|
|
assert.Equal(t, opts, testOpts)
|
|
})
|
|
t.Run("WithLimit", func(t *testing.T) {
|
|
t.Parallel()
|
|
opts := getOpts(WithLimit(12345))
|
|
testOpts := getDefaultOptions()
|
|
testOpts.withLimit = 12345
|
|
assert.Equal(t, opts, testOpts)
|
|
})
|
|
t.Run("WithStartPageAfterItem", func(t *testing.T) {
|
|
t.Parallel()
|
|
updateTime := time.Now()
|
|
opts := getOpts(WithStartPageAfterItem(&fakeItem{nil, "s_1", updateTime}))
|
|
assert.Equal(t, opts.withStartPageAfterItem.GetPublicId(), "s_1")
|
|
assert.Equal(t, opts.withStartPageAfterItem.GetUpdateTime(), timestamp.New(updateTime))
|
|
})
|
|
}
|