chore: import upstream snapshot with attribution
This commit is contained in:
wehub-resource-sync
@@ -0,0 +1,430 @@
|
||||
#!/usr/bin/env bats
|
||||
load $BATS_TEST_DIRNAME/setup/common.bash
|
||||
|
||||
setup() {
|
||||
setup_common
|
||||
}
|
||||
|
||||
teardown() {
|
||||
teardown_common
|
||||
}
|
||||
|
||||
@test 'doltgres: --help' {
|
||||
# just a smoke test
|
||||
doltgres --help
|
||||
}
|
||||
|
||||
@test 'doltgres: --config-help' {
|
||||
# just a smoke test
|
||||
doltgres --config-help
|
||||
}
|
||||
|
||||
@test 'doltgres: no arguments' {
|
||||
PORT=5432
|
||||
mkdir test-home
|
||||
# TODO: DOLT_ROOT_PATH behavior overrides the HOME behavior, which is confusing and not
|
||||
# applicable to Doltgres, fix it
|
||||
HOME=test-home DOLTGRES_DATA_DIR='' DOLT_ROOT_PATH='' doltgres > server.out 2>&1 &
|
||||
SERVER_PID=$!
|
||||
run wait_for_connection $PORT 7500
|
||||
|
||||
cat server.out
|
||||
echo "$output"
|
||||
[ "$status" -eq 0 ]
|
||||
|
||||
query_server -c "create table t1 (a int primary key, b int)"
|
||||
query_server -c "insert into t1 values (1,2)"
|
||||
|
||||
run query_server -c "select * from t1" -t
|
||||
[ "$status" -eq 0 ]
|
||||
[[ "$output" =~ "1 | 2" ]] || false
|
||||
|
||||
# databases should get created in home/doltgres/databases by default
|
||||
[ -d test-home/doltgres/databases/postgres ]
|
||||
}
|
||||
|
||||
@test 'doltgres: data-dir param' {
|
||||
PORT=5432
|
||||
DOLTGRES_DATA_DIR=fake doltgres --data-dir test > server.out 2>&1 &
|
||||
SERVER_PID=$!
|
||||
run wait_for_connection $PORT 7500
|
||||
|
||||
cat server.out
|
||||
echo "$output"
|
||||
[ "$status" -eq 0 ]
|
||||
|
||||
query_server -c "create table t1 (a int primary key, b int)"
|
||||
query_server -c "insert into t1 values (1,2)"
|
||||
|
||||
run query_server -c "select * from t1" -t
|
||||
[ "$status" -eq 0 ]
|
||||
[[ "$output" =~ "1 | 2" ]] || false
|
||||
|
||||
[ -d test/postgres ]
|
||||
}
|
||||
|
||||
@test 'doltgres: data dir in env var' {
|
||||
PORT=5432
|
||||
DOLTGRES_DATA_DIR=test doltgres > server.out 2>&1 &
|
||||
SERVER_PID=$!
|
||||
run wait_for_connection $PORT 7500
|
||||
|
||||
cat server.out
|
||||
echo "$output"
|
||||
[ "$status" -eq 0 ]
|
||||
|
||||
query_server -c "create table t1 (a int primary key, b int)"
|
||||
query_server -c "insert into t1 values (1,2)"
|
||||
|
||||
run query_server -c "select * from t1" -t
|
||||
[ "$status" -eq 0 ]
|
||||
[[ "$output" =~ "1 | 2" ]] || false
|
||||
|
||||
[ -d test/postgres ]
|
||||
}
|
||||
|
||||
@test 'doltgres: implicit config.yaml' {
|
||||
PORT=5434
|
||||
|
||||
cat > config.yaml <<EOF
|
||||
log_level: info
|
||||
|
||||
behavior:
|
||||
read_only: false
|
||||
disable_client_multi_statements: false
|
||||
dolt_transaction_commit: false
|
||||
|
||||
user:
|
||||
name: "postgres"
|
||||
password: "password"
|
||||
|
||||
listener:
|
||||
host: localhost
|
||||
port: $PORT
|
||||
read_timeout_millis: 28800000
|
||||
write_timeout_millis: 28800000
|
||||
|
||||
data_dir: test
|
||||
EOF
|
||||
|
||||
doltgres > server.out 2>&1 &
|
||||
SERVER_PID=$!
|
||||
run wait_for_connection $PORT 7500
|
||||
|
||||
cat server.out
|
||||
echo "$output"
|
||||
[ "$status" -eq 0 ]
|
||||
|
||||
query_server -c "create table t1 (a int primary key, b int)"
|
||||
query_server -c "insert into t1 values (1,2)"
|
||||
|
||||
run query_server -c "select * from t1" -t
|
||||
[ "$status" -eq 0 ]
|
||||
[[ "$output" =~ "1 | 2" ]] || false
|
||||
|
||||
[ -d test/postgres ]
|
||||
}
|
||||
|
||||
@test 'doltgres: config.yaml without data dir' {
|
||||
PORT=5434
|
||||
|
||||
cat > config.yaml <<EOF
|
||||
log_level: info
|
||||
|
||||
behavior:
|
||||
read_only: false
|
||||
disable_client_multi_statements: false
|
||||
dolt_transaction_commit: false
|
||||
|
||||
user:
|
||||
name: "postgres"
|
||||
password: "password"
|
||||
|
||||
listener:
|
||||
host: localhost
|
||||
port: $PORT
|
||||
read_timeout_millis: 28800000
|
||||
write_timeout_millis: 28800000
|
||||
|
||||
EOF
|
||||
|
||||
mkdir test-home
|
||||
HOME=test-home DOLTGRES_DATA_DIR='' DOLT_ROOT_PATH='' doltgres --config config.yaml > server.out 2>&1 &
|
||||
SERVER_PID=$!
|
||||
run wait_for_connection $PORT 7500
|
||||
|
||||
cat server.out
|
||||
echo "$output"
|
||||
[ "$status" -eq 0 ]
|
||||
|
||||
query_server -c "create table t1 (a int primary key, b int)"
|
||||
query_server -c "insert into t1 values (1,2)"
|
||||
|
||||
run query_server -c "select * from t1" -t
|
||||
[ "$status" -eq 0 ]
|
||||
[[ "$output" =~ "1 | 2" ]] || false
|
||||
|
||||
[ -d test-home/doltgres/databases/postgres ]
|
||||
}
|
||||
|
||||
@test 'doltgres: config file override with explicit config.yaml' {
|
||||
PORT=5434
|
||||
|
||||
cat > config-test.yaml <<EOF
|
||||
log_level: info
|
||||
|
||||
behavior:
|
||||
read_only: false
|
||||
disable_client_multi_statements: false
|
||||
dolt_transaction_commit: false
|
||||
|
||||
user:
|
||||
name: "postgres"
|
||||
password: "password"
|
||||
|
||||
listener:
|
||||
host: localhost
|
||||
port: $PORT
|
||||
read_timeout_millis: 28800000
|
||||
write_timeout_millis: 28800000
|
||||
|
||||
data_dir: test
|
||||
EOF
|
||||
|
||||
# The only supported override right now is the data dir, add more here as we add more overrides
|
||||
doltgres --config config-test.yaml --data-dir local-override > server.out 2>&1 &
|
||||
SERVER_PID=$!
|
||||
run wait_for_connection $PORT 7500
|
||||
|
||||
cat server.out
|
||||
echo "$output"
|
||||
[ "$status" -eq 0 ]
|
||||
|
||||
query_server -c "create table t1 (a int primary key, b int)"
|
||||
query_server -c "insert into t1 values (1,2)"
|
||||
|
||||
run query_server -c "select * from t1" -t
|
||||
[ "$status" -eq 0 ]
|
||||
[[ "$output" =~ "1 | 2" ]] || false
|
||||
|
||||
[ ! -d test/postgres ]
|
||||
[ -d local-override/postgres ]
|
||||
}
|
||||
|
||||
@test 'doltgres: config file override with implicit config.yaml' {
|
||||
PORT=5434
|
||||
|
||||
cat > config.yaml <<EOF
|
||||
log_level: info
|
||||
|
||||
behavior:
|
||||
read_only: false
|
||||
disable_client_multi_statements: false
|
||||
dolt_transaction_commit: false
|
||||
|
||||
user:
|
||||
name: "postgres"
|
||||
password: "password"
|
||||
|
||||
listener:
|
||||
host: localhost
|
||||
port: $PORT
|
||||
read_timeout_millis: 28800000
|
||||
write_timeout_millis: 28800000
|
||||
|
||||
data_dir: test
|
||||
EOF
|
||||
|
||||
# The only supported override right now is the data dir, add more here as we add more overrides
|
||||
doltgres --data-dir local-override > server.out 2>&1 &
|
||||
SERVER_PID=$!
|
||||
run wait_for_connection $PORT 7500
|
||||
|
||||
cat server.out
|
||||
echo "$output"
|
||||
[ "$status" -eq 0 ]
|
||||
|
||||
query_server -c "create table t1 (a int primary key, b int)"
|
||||
query_server -c "insert into t1 values (1,2)"
|
||||
|
||||
run query_server -c "select * from t1" -t
|
||||
[ "$status" -eq 0 ]
|
||||
[[ "$output" =~ "1 | 2" ]] || false
|
||||
|
||||
[ ! -d test/postgres ]
|
||||
[ -d local-override/postgres ]
|
||||
}
|
||||
|
||||
@test 'doltgres: config file' {
|
||||
PORT=$( definePORT )
|
||||
CONFIG=$( defineCONFIG $PORT )
|
||||
echo "$CONFIG" > config.yaml
|
||||
|
||||
cat config.yaml
|
||||
start_sql_server_with_args -config config.yaml > log.txt 2>&1
|
||||
|
||||
run cat log.txt
|
||||
[[ ! "$output" =~ "Author identity unknown" ]] || false
|
||||
[ -d "postgres" ]
|
||||
|
||||
query_server -c "create table t1 (a int primary key, b int)"
|
||||
query_server -c "insert into t1 values (1,2)"
|
||||
|
||||
run query_server -c "select * from t1" -t
|
||||
[ "$status" -eq 0 ]
|
||||
[[ "$output" =~ "1 | 2" ]] || false
|
||||
}
|
||||
|
||||
@test 'doltgres: config file with all options' {
|
||||
PORT=$( definePORT )
|
||||
cat > config.yaml <<EOF
|
||||
log_level: info
|
||||
|
||||
behavior:
|
||||
read_only: false
|
||||
disable_client_multi_statements: false
|
||||
dolt_transaction_commit: false
|
||||
|
||||
user:
|
||||
name: "postgres"
|
||||
password: "password"
|
||||
|
||||
listener:
|
||||
host: localhost
|
||||
port: $PORT
|
||||
read_timeout_millis: 28800000
|
||||
write_timeout_millis: 28800000
|
||||
tls_key: null
|
||||
tls_cert: null
|
||||
require_secure_transport: null
|
||||
allow_cleartext_passwords: null
|
||||
|
||||
performance:
|
||||
query_parallelism: null
|
||||
|
||||
data_dir: .
|
||||
|
||||
cfg_dir: .doltcfg
|
||||
|
||||
metrics:
|
||||
labels: {}
|
||||
host: null
|
||||
port: -1
|
||||
|
||||
remotesapi: {}
|
||||
|
||||
privilege_file: .doltcfg/privileges.db
|
||||
|
||||
auth_file: .doltcfg/auth.db
|
||||
|
||||
branch_control_file: .doltcfg/branch_control.db
|
||||
|
||||
user_session_vars: []
|
||||
|
||||
jwks: []
|
||||
EOF
|
||||
|
||||
cat config.yaml
|
||||
|
||||
start_sql_server_with_args -config config.yaml
|
||||
|
||||
query_server -c "create table t1 (a int primary key, b int)"
|
||||
query_server -c "insert into t1 values (1,2)"
|
||||
|
||||
run query_server -c "select * from t1" -t
|
||||
[ "$status" -eq 0 ]
|
||||
[[ "$output" =~ "1 | 2" ]] || false
|
||||
|
||||
run test -f ".doltcfg/auth.db"
|
||||
[ "$status" -eq 0 ]
|
||||
}
|
||||
|
||||
@test 'doltgres: DOLTGRES_DATA_DIR set to current dir' {
|
||||
[ ! -d "postgres" ]
|
||||
export DOLTGRES_DATA_DIR="$(pwd)"
|
||||
start_sql_server > log.txt 2>&1
|
||||
|
||||
run cat log.txt
|
||||
[[ ! "$output" =~ "Author identity unknown" ]] || false
|
||||
[ -d "postgres" ]
|
||||
|
||||
run query_server -c "\l"
|
||||
[ "$status" -eq 0 ]
|
||||
[[ "$output" =~ "postgres" ]] || false
|
||||
}
|
||||
|
||||
@test 'doltgres: user name and pass via env' {
|
||||
export DOLTGRES_USER="myuser"
|
||||
export DOLTGRES_PASSWORD="mypass"
|
||||
|
||||
[ ! -d "auth.db" ]
|
||||
|
||||
start_sql_server "" log.txt myuser mypass
|
||||
cat log.txt
|
||||
|
||||
# db matches user name since DOLTGRES_DB was not set
|
||||
query_server_for_user_and_pass myuser mypass myuser -c "create table myTable (a int);"
|
||||
|
||||
run query_server_for_user_and_pass myuser mypass myuser -c "insert into mytable values (1), (2)"
|
||||
[ "$status" -eq 0 ]
|
||||
[[ "$output" =~ "INSERT" ]] || false
|
||||
|
||||
run query_server_for_user_and_pass postgres password myuser -c "insert into mytable values (1), (2)"
|
||||
[ "$status" -ne 0 ]
|
||||
}
|
||||
|
||||
@test 'doltgres: default db via env' {
|
||||
[ ! -d "auth.db" ]
|
||||
|
||||
start_sql_server mydb log.txt myuser
|
||||
cat log.txt
|
||||
|
||||
query_server_for_user_and_pass myuser password mydb -c "create table myTable (a int);"
|
||||
|
||||
run query_server_for_user_and_pass myuser password mydb -c "insert into mytable values (1), (2)"
|
||||
[ "$status" -eq 0 ]
|
||||
[[ "$output" =~ "INSERT" ]] || false
|
||||
|
||||
run query_server_for_user_and_pass postgres password mydb -c "insert into mytable values (1), (2)"
|
||||
[ "$status" -ne 0 ]
|
||||
}
|
||||
|
||||
query_server_for_user_and_pass() {
|
||||
user=$1
|
||||
pass=$2
|
||||
db=$3
|
||||
shift
|
||||
shift
|
||||
shift
|
||||
|
||||
nativevar PGPASSWORD "$pass" /w
|
||||
psql -U "$user" -h localhost -p $PORT "$@" $db
|
||||
}
|
||||
|
||||
# Test for https://github.com/dolthub/doltgresql/issues/1863
|
||||
@test 'doltgres: connection to non-existent database fails' {
|
||||
start_sql_server
|
||||
|
||||
# Connecting to the default postgres database should work
|
||||
run query_server -c "SELECT 1"
|
||||
[ "$status" -eq 0 ]
|
||||
|
||||
# Connecting to a non-existent database should fail
|
||||
nativevar PGPASSWORD "password" /w
|
||||
run psql -U postgres -h localhost -p $PORT -c "SELECT 1" nonexistent_db
|
||||
[ "$status" -ne 0 ]
|
||||
[[ "$output" =~ "does not exist" ]] || false
|
||||
}
|
||||
|
||||
@test 'doltgres: CREATE SCHEMA works on valid database' {
|
||||
start_sql_server
|
||||
|
||||
# CREATE SCHEMA should work on a valid database
|
||||
run query_server -c "CREATE SCHEMA test_schema_bats"
|
||||
[ "$status" -eq 0 ]
|
||||
|
||||
# Verify schema was created
|
||||
run query_server -c "SELECT schema_name FROM information_schema.schemata WHERE schema_name = 'test_schema_bats'"
|
||||
[ "$status" -eq 0 ]
|
||||
[[ "$output" =~ "test_schema_bats" ]] || false
|
||||
}
|
||||
Reference in New Issue
Block a user