From 4823c4515895714c118f6add8c5642bdcec228d5 Mon Sep 17 00:00:00 2001
From: Nat Lasseter <nat.lasseter@york.ac.uk>
Date: Tue, 3 Nov 2020 15:51:58 +0000
Subject: Add --test flag for testing new raiders

---
 longboat | 12 +++++++++---
 1 file changed, 9 insertions(+), 3 deletions(-)

(limited to 'longboat')

diff --git a/longboat b/longboat
index b4f0dee..658ec27 100755
--- a/longboat
+++ b/longboat
@@ -9,7 +9,13 @@ config = Longboat::Config.parse!
 # Start collection
 collector = Longboat::Collector.new(config)
 raiders = Longboat::Raiders.new(collector, config)
-t = raiders.raid_every
 
-# Serve metrics on HTTP
-Longboat::Server.serve!(collector, config)
+if config.test
+  # We're in test mode, output metrics to stdout once and quit
+  raiders.raid!
+  puts collector.prometheus_metrics
+else
+  # Serve metrics on HTTP forever
+  raiders.raid_every
+  Longboat::Server.serve!(collector, config)
+end
-- 
cgit v1.2.3