mirror of
https://github.com/sosedoff/pgweb.git
synced 2024-12-15 11:52:12 +03:00
47500bf92e
- Make ParseOptions func to take an arg and return struct so we can test it - Refactor and add more tests for options parsing - Run test suite on postgres 10.x branch - Change format for connection idle timer from float64 to in
51 lines
1.6 KiB
Go
51 lines
1.6 KiB
Go
package command
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
)
|
|
|
|
func TestParseOptions(t *testing.T) {
|
|
// Test default behavior
|
|
opts, err := ParseOptions([]string{})
|
|
assert.NoError(t, err)
|
|
assert.Equal(t, false, opts.Sessions)
|
|
assert.Equal(t, "", opts.Prefix)
|
|
assert.Equal(t, "", opts.ConnectToken)
|
|
assert.Equal(t, "", opts.ConnectHeaders)
|
|
assert.Equal(t, false, opts.DisableSSH)
|
|
assert.Equal(t, false, opts.DisablePrettyJson)
|
|
assert.Equal(t, false, opts.DisableConnectionIdleTimeout)
|
|
assert.Equal(t, 180, opts.ConnectionIdleTimeout)
|
|
assert.Equal(t, false, opts.Cors)
|
|
assert.Equal(t, "*", opts.CorsOrigin)
|
|
|
|
// Test sessions
|
|
opts, err = ParseOptions([]string{"--sessions", "1"})
|
|
assert.NoError(t, err)
|
|
assert.Equal(t, true, opts.Sessions)
|
|
|
|
opts, err = ParseOptions([]string{"--sessions", "1", "--bookmark", "test"})
|
|
assert.EqualError(t, err, "--bookmark is not allowed in multi-session mode")
|
|
|
|
// Test url prefix
|
|
opts, err = ParseOptions([]string{"--prefix", "pgweb"})
|
|
assert.NoError(t, err)
|
|
assert.Equal(t, "pgweb/", opts.Prefix)
|
|
|
|
opts, err = ParseOptions([]string{"--prefix", "pgweb/"})
|
|
assert.NoError(t, err)
|
|
assert.Equal(t, "pgweb/", opts.Prefix)
|
|
|
|
// Test connect backend options
|
|
opts, err = ParseOptions([]string{"--connect-backend", "test"})
|
|
assert.EqualError(t, err, "--sessions flag must be set")
|
|
|
|
opts, err = ParseOptions([]string{"--connect-backend", "test", "--sessions"})
|
|
assert.EqualError(t, err, "--connect-token flag must be set")
|
|
|
|
opts, err = ParseOptions([]string{"--connect-backend", "test", "--sessions", "--connect-token", "token"})
|
|
assert.NoError(t, err)
|
|
}
|