Timothy Messier
eac433f7ff
This moves all of the tcp target functionality into a tcp package. It also establishes a Register function in the target package to allow specific target types to be registered at initilazation time. When registering, a target type needs to provide its Subtype, as well as some RepositoryHooks that are used by the target Repository.
115 lines
3.2 KiB
Go
115 lines
3.2 KiB
Go
package target
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
)
|
|
|
|
// Test_GetOpts provides unit tests for GetOpts and all the options
|
|
func Test_GetOpts(t *testing.T) {
|
|
t.Parallel()
|
|
t.Run("WithName", func(t *testing.T) {
|
|
assert := assert.New(t)
|
|
opts := GetOpts(WithName("test"))
|
|
testOpts := getDefaultOptions()
|
|
testOpts.WithName = "test"
|
|
assert.Equal(opts, testOpts)
|
|
})
|
|
t.Run("WithDescription", func(t *testing.T) {
|
|
assert := assert.New(t)
|
|
opts := GetOpts(WithDescription("test desc"))
|
|
testOpts := getDefaultOptions()
|
|
testOpts.WithDescription = "test desc"
|
|
assert.Equal(opts, testOpts)
|
|
})
|
|
t.Run("WithLimit", func(t *testing.T) {
|
|
assert := assert.New(t)
|
|
// test default of 0
|
|
opts := GetOpts()
|
|
testOpts := getDefaultOptions()
|
|
testOpts.WithLimit = 0
|
|
assert.Equal(opts, testOpts)
|
|
|
|
opts = GetOpts(WithLimit(-1))
|
|
testOpts = getDefaultOptions()
|
|
testOpts.WithLimit = -1
|
|
assert.Equal(opts, testOpts)
|
|
|
|
opts = GetOpts(WithLimit(1))
|
|
testOpts = getDefaultOptions()
|
|
testOpts.WithLimit = 1
|
|
assert.Equal(opts, testOpts)
|
|
})
|
|
t.Run("WithDefaultPort", func(t *testing.T) {
|
|
assert := assert.New(t)
|
|
// test default of 0
|
|
opts := GetOpts()
|
|
testOpts := getDefaultOptions()
|
|
testOpts.WithDefaultPort = 0
|
|
assert.Equal(opts, testOpts)
|
|
|
|
opts = GetOpts(WithDefaultPort(22))
|
|
testOpts = getDefaultOptions()
|
|
testOpts.WithDefaultPort = uint32(22)
|
|
assert.Equal(opts, testOpts)
|
|
})
|
|
t.Run("WithUserId", func(t *testing.T) {
|
|
assert := assert.New(t)
|
|
opts := GetOpts(WithUserId("testId"))
|
|
testOpts := getDefaultOptions()
|
|
testOpts.WithUserId = "testId"
|
|
assert.Equal(opts, testOpts)
|
|
})
|
|
t.Run("WithScopeId", func(t *testing.T) {
|
|
assert := assert.New(t)
|
|
opts := GetOpts(WithScopeId("testId"))
|
|
testOpts := getDefaultOptions()
|
|
testOpts.WithScopeId = "testId"
|
|
assert.Equal(opts, testOpts)
|
|
})
|
|
t.Run("WithScopeName", func(t *testing.T) {
|
|
assert := assert.New(t)
|
|
opts := GetOpts(WithScopeName("testName"))
|
|
testOpts := getDefaultOptions()
|
|
testOpts.WithScopeName = "testName"
|
|
assert.Equal(opts, testOpts)
|
|
})
|
|
t.Run("WithPublicId", func(t *testing.T) {
|
|
assert := assert.New(t)
|
|
opts := GetOpts(WithPublicId("testId"))
|
|
testOpts := getDefaultOptions()
|
|
testOpts.WithPublicId = "testId"
|
|
assert.Equal(opts, testOpts)
|
|
})
|
|
t.Run("WithTargetType", func(t *testing.T) {
|
|
assert := assert.New(t)
|
|
opts := GetOpts(WithTargetType(TcpTargetType))
|
|
testOpts := getDefaultOptions()
|
|
target := TcpTargetType
|
|
testOpts.WithTargetType = &target
|
|
assert.Equal(opts, testOpts)
|
|
})
|
|
t.Run("WithHostSources", func(t *testing.T) {
|
|
assert := assert.New(t)
|
|
opts := GetOpts(WithHostSources([]string{"alice", "bob"}))
|
|
testOpts := getDefaultOptions()
|
|
testOpts.WithHostSources = []string{"alice", "bob"}
|
|
assert.Equal(opts, testOpts)
|
|
})
|
|
t.Run("WithWorkerFilter", func(t *testing.T) {
|
|
assert := assert.New(t)
|
|
opts := GetOpts(WithWorkerFilter(`"/foo" == "bar"`))
|
|
testOpts := getDefaultOptions()
|
|
testOpts.WithWorkerFilter = `"/foo" == "bar"`
|
|
assert.Equal(opts, testOpts)
|
|
})
|
|
t.Run("WithCredentialSources", func(t *testing.T) {
|
|
assert := assert.New(t)
|
|
opts := GetOpts(WithCredentialSources([]string{"alice", "bob"}))
|
|
testOpts := getDefaultOptions()
|
|
testOpts.WithCredentialSources = []string{"alice", "bob"}
|
|
assert.Equal(opts, testOpts)
|
|
})
|
|
}
|