mirror of
https://github.com/jcreek/CosmicClash.git
synced 2026-09-10 16:04:04 +00:00
Josh Creek
64 lines
2.4 KiB
GDScript
64 lines
2.4 KiB
GDScript
extends SceneTree
|
|
|
|
const ServerControlScript = preload("res://scripts/server_control.gd")
|
|
const PORT := 18080
|
|
|
|
|
|
func _init() -> void:
|
|
var control = ServerControlScript.new()
|
|
root.add_child(control)
|
|
if control.start(PORT, "drain-secret") != OK:
|
|
printerr("server control failed to bind")
|
|
quit(1)
|
|
return
|
|
control.set_process_ready(true)
|
|
control.set_meta("admission_armed", false)
|
|
control.initial_connect_ready.connect(func() -> void: control.set_meta("admission_armed", true))
|
|
await process_frame
|
|
var ready_response := await _request("GET", "/ready", [])
|
|
if ready_response != 200:
|
|
printerr("ready response was %d" % ready_response)
|
|
quit(1)
|
|
return
|
|
var unauthorized := await _request("POST", "/drain", ["Authorization: Bearer wrong"])
|
|
if unauthorized != 401:
|
|
printerr("unauthorized drain response was %d" % unauthorized)
|
|
quit(1)
|
|
return
|
|
var unauthorized_admission := await _request("POST", "/initial-connect-ready", ["Authorization: Bearer wrong"])
|
|
if unauthorized_admission != 401 or bool(control.get_meta("admission_armed")):
|
|
printerr("unauthorized initial-connect response/state was %d/%s" % [unauthorized_admission, control.get_meta("admission_armed")])
|
|
quit(1)
|
|
return
|
|
var admitted := await _request("POST", "/initial-connect-ready", ["Authorization: Bearer drain-secret"])
|
|
if admitted != 202 or not bool(control.get_meta("admission_armed")):
|
|
printerr("authorized initial-connect response/state was %d/%s" % [admitted, control.get_meta("admission_armed")])
|
|
quit(1)
|
|
return
|
|
var drained := await _request("POST", "/drain", ["Authorization: Bearer drain-secret"])
|
|
if drained != 202 or not control.is_draining():
|
|
printerr("authorized drain response/state was %d/%s" % [drained, control.is_draining()])
|
|
quit(1)
|
|
return
|
|
var not_ready := await _request("GET", "/ready", [])
|
|
if not_ready != 503:
|
|
printerr("draining ready response was %d" % not_ready)
|
|
quit(1)
|
|
return
|
|
control.stop()
|
|
print("server control smoke passed")
|
|
quit(0)
|
|
|
|
|
|
func _request(method: String, path: String, headers: PackedStringArray) -> int:
|
|
var request := HTTPRequest.new()
|
|
root.add_child(request)
|
|
var http_method := HTTPClient.METHOD_GET if method == "GET" else HTTPClient.METHOD_POST
|
|
var err := request.request("http://127.0.0.1:%d%s" % [PORT, path], headers, http_method)
|
|
if err != OK:
|
|
request.queue_free()
|
|
return -1
|
|
var result = await request.request_completed
|
|
request.queue_free()
|
|
return int(result[1])
|