Todd
2ecd550e27
* Remove the server_id from session table, change it to worker_id on session_connection This is step 1 in updating the worker table. The worker is becoming a public resource so it must have a public id. We drop the server_id column on the session_connection table and add the worker_id column so we can use the wt_public_key type. Since db migrations are applied while the controller is offline, it should be safe to assume all session connections are not in a running state, thus allowing us to null out all the worker ids in the session_connection table. For now we don't have very good enforcement of the wt_public_id contraints, since the id is currently defined in the worker config, but this will soon change and the worker's id will be generated by the controller. * Updating server_id references to controller_id. * Adding worker tag db writing test. * Move the server table back to the top. * Update comment for the worker object. * Update comment for the worker table. * Update the newWorkerId call to take a context and fix its error. * Correct comment for the wt_network_address domain.
69 lines
1.8 KiB
Go
69 lines
1.8 KiB
Go
package servers
|
|
|
|
import (
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
)
|
|
|
|
// Test_GetOpts provides unit tests for GetOpts and all the options
|
|
func Test_GetOpts(t *testing.T) {
|
|
t.Parallel()
|
|
opts := getOpts()
|
|
assert.Equal(t, options{}, opts)
|
|
|
|
t.Run("WithName", func(t *testing.T) {
|
|
opts := getOpts(WithName("test"))
|
|
testOpts := getDefaultOptions()
|
|
testOpts.withName = "test"
|
|
assert.Equal(t, opts, testOpts)
|
|
})
|
|
t.Run("WithDescription", func(t *testing.T) {
|
|
opts := getOpts(WithDescription("test desc"))
|
|
testOpts := getDefaultOptions()
|
|
testOpts.withDescription = "test desc"
|
|
assert.Equal(t, opts, testOpts)
|
|
})
|
|
t.Run("WithLimit", func(t *testing.T) {
|
|
opts := getOpts(WithLimit(5))
|
|
testOpts := getDefaultOptions()
|
|
testOpts.withLimit = 5
|
|
assert.Equal(t, opts, testOpts)
|
|
})
|
|
t.Run("WithAddress", func(t *testing.T) {
|
|
opts := getOpts(WithAddress("test"))
|
|
testOpts := getDefaultOptions()
|
|
testOpts.withAddress = "test"
|
|
assert.Equal(t, opts, testOpts)
|
|
})
|
|
t.Run("WithPublicId", func(t *testing.T) {
|
|
opts := getOpts(WithPublicId("test"))
|
|
testOpts := getDefaultOptions()
|
|
testOpts.withPublicId = "test"
|
|
assert.Equal(t, opts, testOpts)
|
|
})
|
|
t.Run("WithLiveness", func(t *testing.T) {
|
|
opts := getOpts(WithLiveness(time.Hour))
|
|
testOpts := getDefaultOptions()
|
|
testOpts.withLiveness = time.Hour
|
|
assert.Equal(t, opts, testOpts)
|
|
})
|
|
t.Run("WithUpdateTags", func(t *testing.T) {
|
|
opts := getOpts(WithUpdateTags(true))
|
|
testOpts := getDefaultOptions()
|
|
testOpts.withUpdateTags = true
|
|
assert.Equal(t, opts, testOpts)
|
|
})
|
|
t.Run("WithWorkerTags", func(t *testing.T) {
|
|
tags := []*Tag{
|
|
{Key: "key1", Value: "val1"},
|
|
{Key: "key2", Value: "val2"},
|
|
}
|
|
opts := getOpts(WithWorkerTags(tags...))
|
|
testOpts := getDefaultOptions()
|
|
testOpts.withWorkerTags = tags
|
|
assert.Equal(t, opts, testOpts)
|
|
})
|
|
}
|