From: Matthias Kruk <m@m10k.eu>
Date: Sun, 15 Aug 2021 05:48:29 +0000 (+0900)
Subject: dist: Add test script for distbot
X-Git-Url: https://git.corax.cc/?a=commitdiff_plain;h=6ed43ebf1ca1ee59b3ad026cfa9d9392bbf9d9aa;p=foundry

dist: Add test script for distbot

To make sure distbot is working correctly, this commit adds a script
that may be used to inject distrequest messages into the build system.
This script may also be used to manually distribute packages via
distbot.
---

diff --git a/dist.sh b/dist.sh
new file mode 100644
index 0000000..f537abf
--- /dev/null
+++ b/dist.sh
@@ -0,0 +1,177 @@
+#!/bin/bash
+
+check_result() {
+	local endpoint="$1"
+	local context="$2"
+
+	local start
+	local timelimit
+
+	start=$(date +"%s")
+	timelimit=$((5 * 60))
+
+	while (( ( $(date +"%s") - start ) < timelimit )); do
+		local msg
+		local fmsg
+		local msgtype
+		local ctx
+
+		if ! msg=$(ipc_endpoint_recv "$endpoint" 5); then
+			continue
+		fi
+
+		if ! fmsg=$(ipc_msg_get_data "$msg"); then
+			log_warn "Dropping invalid message (no data)"
+			continue
+		fi
+
+		if ! msgtype=$(foundry_msg_get_type "$fmsg"); then
+			log_warn "Dropping message with unexpected type \"$msgtype\""
+			continue
+		fi
+
+		if ! ctx=$(foundry_msg_get_context "$fmsg"); then
+			log_warn "Dropping message without context"
+			continue
+		fi
+
+		if [[ "$ctx" != "$context" ]]; then
+			# Unrelated
+			continue
+		fi
+
+		foundry_context_get_logs "$context" | log_highlight "$context logs"
+		return 0
+	done
+
+	return 1
+}
+
+prepare_context() {
+	local packages=("$@")
+
+	local name
+	local context
+	local package
+
+	if (( ${#packages[@]} < 1 )); then
+		log_error "No packages to distribute"
+		return 1
+	fi
+
+	name="${packages[0]}"
+	name="${name%%_*}"
+
+	if ! context=$(foundry_context_new "$name"); then
+		log_error "Could not create new context for $name"
+		return 1
+	fi
+
+	for package in "${package[@]}"; do
+		if ! foundry_context_add_file "$context" "$package"; then
+			log_error "Could not add $package to context $context"
+			return 1
+		fi
+	done
+
+	echo "$context"
+	return 0
+}
+
+send_dist_request() {
+	local endpoint="$1"
+	local context="$2"
+
+	local distreq
+
+	if ! distreq=$(foundry_msg_distrequest_new "$context"); then
+		log_error "Could not make distrequest message"
+		return 1
+	fi
+
+	if ! ipc_endpoint_send "$endpoint" "pub/distbot" "$distreq"; then
+		log_error "Could not send distrequest message to pub/distbot"
+		return 1
+	fi
+
+	return 0
+}
+
+test_distbot() {
+	local packages=("$@")
+
+	local context
+	local package
+	local endpoint
+	local name
+
+	if ! endpoint=$(ipc_endpoint_open); then
+		log_error "Could not open an IPC endpoint"
+		return 1
+	fi
+
+	if ! ipc_endpoint_subscribe "$endpoint" "dists"; then
+		log_error "Could not subscribe $endpoint to dists"
+		return 1
+	fi
+
+	if ! context=$(prepare_context "$name" "${packages[@]}"); then
+		log_error "Could not prepare context"
+		return 1
+	fi
+
+	if ! send_dist_request "$endpoint" "$context"; then
+		log_error "Could not send distrequest message"
+		return 1
+	fi
+
+	if ! check_result "$endpoint" "$context"; then
+		return 1
+	fi
+
+	return 0
+}
+
+_add_package() {
+	local name="$1"
+	local value="$2"
+
+	if [[ "$name" != "package" ]]; then
+		return 1
+	fi
+
+	packages+=("$value")
+	return 0
+}
+
+main() {
+	local packages
+
+	packages=()
+
+	opt_add_arg "p" "package" "rv" "" "A package to add to distribute" "" _add_package
+
+	if ! opt_parse "$@"; then
+		return 1
+	fi
+
+	if ! test_distbot "${packages[@]}"; then
+		echo "Test failed"
+		return 1
+	fi
+
+	return 0
+}
+
+{
+	if ! . toolbox.sh; then
+		exit 1
+	fi
+
+	if ! include "log" "opt" "ipc" "foundry/msg" "foundry/context"; then
+		exit 1
+	fi
+
+	main "$@"
+	exit "$?"
+}