cli: improve command dispatch a bit

1 files changed, 44 insertions, 0 deletions

diff --git a/nihil.util/test_capture_stream.cc b/nihil.util/test_capture_stream.cc
new file mode 100644
index 0000000..27c8596
--- /dev/null
+++ b/nihil.util/test_capture_stream.cc
@@ -0,0 +1,44 @@
+/*
+ * This source code is released into the public domain.
+ */
+
+#include <iostream>
+
+#include <catch2/catch_test_macros.hpp>
+
+import nihil.util;
+
+TEST_CASE("nihil.util: capture", "[nihil][nihil.util]")
+{
+	SECTION("std::cout with release()") {
+		auto cap = nihil::capture_stream(std::cout);
+
+		std::cout << 1 << '+' << 1 << '=' << (1 + 1) << '\n';
+		REQUIRE(cap.str() == "1+1=2\n");
+		
+		cap.release();
+		REQUIRE(cap.str() == "1+1=2\n");
+	}
+
+	SECTION("std::cout with dtor") {
+		auto cap = nihil::capture_stream(std::cout);
+		std::cout << 1 << '+' << 1 << '=' << (1 + 1) << '\n';
+		REQUIRE(cap.str() == "1+1=2\n");
+	}
+
+	SECTION("std::cerr with release()") {
+		auto cap = nihil::capture_stream(std::cerr);
+
+		std::cerr << 1 << '+' << 1 << '=' << (1 + 1) << '\n';
+		REQUIRE(cap.str() == "1+1=2\n");
+		
+		cap.release();
+		REQUIRE(cap.str() == "1+1=2\n");
+	}
+
+	SECTION("std::cerr with dtor") {
+		auto cap = nihil::capture_stream(std::cerr);
+		std::cerr << 1 << '+' << 1 << '=' << (1 + 1) << '\n';
+		REQUIRE(cap.str() == "1+1=2\n");
+	}
+}