mirror of
https://github.com/Kong/go-pluginserver.git
synced 2025-10-06 00:27:01 +08:00
103 lines
2.5 KiB
Bash
Executable File
103 lines
2.5 KiB
Bash
Executable File
#!/bin/bash
|
|
|
|
if [ "$1" == "-v" ]; then
|
|
VERBOSE='true'
|
|
fi
|
|
|
|
SOCKET='sock'
|
|
|
|
if pgrep go-pluginserver -l; then
|
|
PREVIOUS_SERVER="yes"
|
|
else
|
|
echo "starting server..."
|
|
[ -S "$SOCKET" ] && rm "$SOCKET"
|
|
./go-pluginserver -socket "$SOCKET" &
|
|
pgrep go-pluginserver -l
|
|
fi
|
|
|
|
msg() {
|
|
query="$1"
|
|
[ -v VERBOSE ] && rq <<< "$query"
|
|
METHOD="$(rq <<< "$1" -- 'at([2])')"
|
|
response="$(rq <<< "$query" -M | nc -U "$SOCKET" -N | rq -m)"
|
|
[ -v VERBOSE ] && rq <<< "$response"
|
|
|
|
ERROR="$(rq <<< "$response" -- 'at [2] ')"
|
|
RESULT="$(rq <<< "$response" -- 'at [3] ')"
|
|
}
|
|
|
|
assert_noerr() {
|
|
if [ "$ERROR" != "null" ]; then
|
|
echo "query: $query"
|
|
echo "response: $response"
|
|
echo "$METHOD : $ERROR" > /dev/stderr
|
|
exit 1
|
|
fi
|
|
echo "$METHOD: ok"
|
|
}
|
|
|
|
assert_fld_match() {
|
|
fld="$1"
|
|
pattern="$2"
|
|
|
|
fld_v="$(query_result 'at "'$fld'"')"
|
|
if [[ "$fld_v" =~ "$pattern" ]]; then
|
|
echo "==> $fld_v : ok"
|
|
else
|
|
echo "==> $fld_v : no match '$pattern'"
|
|
exit 1
|
|
fi
|
|
}
|
|
|
|
query_result() {
|
|
# echo "$RESULT"
|
|
rq <<< "$RESULT" -- "$1"
|
|
}
|
|
|
|
msg '[0, 19, "plugin.SetPluginDir", ["/home/javier/devel/kong_dev/kong"]]'
|
|
assert_noerr
|
|
|
|
msg '[0, 19, "plugin.GetPluginInfo", ["go-log"]]'
|
|
assert_noerr
|
|
|
|
msg '[0, 19, "plugin.StartInstance", [{"Name":"go-log", "Config":"{\"reopen\":false, \"path\":\"/some/where/else/\"}"}]]'
|
|
assert_noerr
|
|
instanceID=$(query_result 'at "Id"')
|
|
echo "instanceID: $instanceID"
|
|
|
|
msg '[0, 19, "plugin.HandleEvent", [{"InstanceId": '$instanceID', "EventName": "access", "Params": [45, 23]}]]'
|
|
assert_noerr
|
|
eventId=$(query_result 'at "EventId"')
|
|
|
|
msg '[0, 19, "plugin.Step", [{"EventId": '$eventId', "Data": "access", "Params": [45, 23]}]]'
|
|
assert_noerr
|
|
callBack=$(query_result 'at "Data"')
|
|
assert_fld_match 'Data' 'kong.request.get_header:[\"host\"]'
|
|
# echo "callBack: $callBack" # get_header('host')
|
|
|
|
msg '[0, 19, "plugin.Step", [{"EventId": '$eventId', "Data": "example.com"}]]'
|
|
assert_noerr
|
|
# callBack=$(query_result 'at "Data"')
|
|
assert_fld_match 'Data' 'kong.response.set_header:[\"x-hello-go\",\"Go says hello to example.com (/some/where/else/)\"]'
|
|
# echo "callBack: $callBack" # set_header('x-hello-go', ....)
|
|
|
|
msg '[0, 19, "plugin.Step", [{"EventId": '$eventId', "Data": "ok"}]]'
|
|
assert_noerr
|
|
callBack=$(query_result 'at "Data"')
|
|
assert_fld_match 'Data' '"ret"'
|
|
# [ "$(query_result 'at "Data"')" == '"ret"' ] || exit 1
|
|
|
|
|
|
msg '[0, 19, "plugin.InstanceStatus", ['$instanceID']]'
|
|
assert_noerr
|
|
|
|
msg "[0, 19, \"plugin.CloseInstance\", [$instanceID]]"
|
|
assert_noerr
|
|
|
|
|
|
if [ ! -v PREVIOUS_SERVER ]; then
|
|
pkill go-pluginserver
|
|
rm "$SOCKET"
|
|
fi
|
|
|