mirror of
https://github.com/logos-messaging/logos-delivery.git
synced 2026-05-03 21:43:50 +00:00
118 lines
3.9 KiB
Nim
118 lines
3.9 KiB
Nim
{.used.}
|
|
|
|
import
|
|
std/[tempfiles, osproc],
|
|
testutils/unittests,
|
|
presto,
|
|
presto/client as presto_client,
|
|
libp2p/peerinfo,
|
|
libp2p/multiaddress,
|
|
libp2p/crypto/crypto
|
|
import
|
|
waku/[
|
|
common/waku_protocol,
|
|
waku_node,
|
|
node/waku_node as waku_node2,
|
|
# TODO: Remove after moving `git_version` to the app code.
|
|
rest_api/endpoint/server,
|
|
rest_api/endpoint/client,
|
|
rest_api/endpoint/responses,
|
|
rest_api/endpoint/health/handlers as health_rest_interface,
|
|
rest_api/endpoint/health/client as health_rest_client,
|
|
waku_rln_relay,
|
|
node/health_monitor,
|
|
],
|
|
../testlib/common,
|
|
../testlib/wakucore,
|
|
../testlib/wakunode,
|
|
../waku_rln_relay/[rln/waku_rln_relay_utils, utils_onchain]
|
|
|
|
proc testWakuNode(): WakuNode =
|
|
let
|
|
privkey = crypto.PrivateKey.random(Secp256k1, rng[]).tryGet()
|
|
bindIp = parseIpAddress("0.0.0.0")
|
|
extIp = parseIpAddress("127.0.0.1")
|
|
port = Port(0)
|
|
|
|
newTestWakuNode(privkey, bindIp, port, some(extIp), some(port))
|
|
|
|
suite "Waku v2 REST API - health":
|
|
# TODO: better test for health
|
|
var anvilProc {.threadVar.}: Process
|
|
var manager {.threadVar.}: OnchainGroupManager
|
|
|
|
setup:
|
|
anvilProc = runAnvil(stateFile = some(DEFAULT_ANVIL_STATE_PATH))
|
|
manager = waitFor setupOnchainGroupManager(deployContracts = false)
|
|
|
|
teardown:
|
|
stopAnvil(anvilProc)
|
|
|
|
asyncTest "Get node health info - GET /health":
|
|
# Given
|
|
let node = testWakuNode()
|
|
await node.start()
|
|
(await node.mountRelay()).isOkOr:
|
|
assert false, "Failed to mount relay"
|
|
|
|
var restPort = Port(0)
|
|
let restAddress = parseIpAddress("0.0.0.0")
|
|
let restServer = WakuRestServerRef.init(restAddress, restPort).tryGet()
|
|
restPort = restServer.httpServer.address.port # update with bound port for client use
|
|
|
|
let healthMonitor = NodeHealthMonitor.new(node)
|
|
|
|
installHealthApiHandler(restServer.router, healthMonitor)
|
|
restServer.start()
|
|
let client = newRestHttpClient(initTAddress(restAddress, restPort))
|
|
|
|
# kick in rln (currently the only check for health)
|
|
await node.mountRlnRelay(
|
|
getWakuRlnConfig(manager = manager, index = MembershipIndex(1))
|
|
)
|
|
|
|
node.mountLightPushClient()
|
|
await node.mountFilterClient()
|
|
|
|
# We don't have a Waku, so we need to set the overall health to READY here in its behalf
|
|
healthMonitor.setOverallHealth(HealthStatus.READY)
|
|
|
|
# When
|
|
var response = await client.healthCheck()
|
|
let report = response.data
|
|
|
|
# Then
|
|
check:
|
|
response.status == 200
|
|
$response.contentType == $MIMETYPE_JSON
|
|
report.nodeHealth == HealthStatus.READY
|
|
report.protocolsHealth.len() == 13
|
|
|
|
report.getHealth(RelayProtocol).health == HealthStatus.NOT_READY
|
|
report.getHealth(RelayProtocol).desc == some("No connected peers")
|
|
|
|
report.getHealth(RlnRelayProtocol).health == HealthStatus.READY
|
|
|
|
report.getHealth(LightpushProtocol).health == HealthStatus.NOT_MOUNTED
|
|
report.getHealth(LegacyLightpushProtocol).health == HealthStatus.NOT_MOUNTED
|
|
report.getHealth(FilterProtocol).health == HealthStatus.NOT_MOUNTED
|
|
report.getHealth(StoreProtocol).health == HealthStatus.NOT_MOUNTED
|
|
report.getHealth(PeerExchangeProtocol).health == HealthStatus.NOT_MOUNTED
|
|
report.getHealth(RendezvousProtocol).health == HealthStatus.NOT_MOUNTED
|
|
report.getHealth(MixProtocol).health == HealthStatus.NOT_MOUNTED
|
|
|
|
report.getHealth(LightpushClientProtocol).health == HealthStatus.NOT_READY
|
|
report.getHealth(LightpushClientProtocol).desc ==
|
|
some("No Lightpush service peer available yet")
|
|
|
|
report.getHealth(LegacyLightpushClientProtocol).health == HealthStatus.NOT_MOUNTED
|
|
report.getHealth(StoreClientProtocol).health == HealthStatus.NOT_MOUNTED
|
|
|
|
report.getHealth(FilterClientProtocol).health == HealthStatus.NOT_READY
|
|
report.getHealth(FilterClientProtocol).desc ==
|
|
some("No Filter service peer available yet")
|
|
|
|
await restServer.stop()
|
|
await restServer.closeWait()
|
|
await node.stop()
|