1 files changed, 78 insertions, 5 deletions

diff --git a/test/execute_test.cc b/test/execute_test.cc
index 9da18dc..722b6ea 100644
--- a/test/execute_test.cc
+++ b/test/execute_test.cc
@@ -1,6 +1,15 @@
 #include <uunit.h>
 
+#include <fstream>
+#include <map>
+#include <vector>
+
 #include <execute.h>
+#include <util.h>
+#include <algorithm>
+
+#include "paths.h"
+#include "tmpfile.h"
 
 class ExecuteTest
 	: public uUnit
@@ -8,14 +17,78 @@ class ExecuteTest
 public:
 	ExecuteTest()
 	{
-		uTEST(ExecuteTest::runit);
+		uTEST(ExecuteTest::return_value);
+		uTEST(ExecuteTest::env);
+	}
+
+	void return_value()
+	{
+		ctor::settings s;
+		auto cur_path = std::filesystem::path(paths::argv_0).parent_path();
+		std::vector<std::string> paths{{cur_path.string()}};
+		auto cmd = locate("testprog", paths);
+		uASSERT(!cmd.empty());
+
+		auto value = execute(s, cmd, {"retval", "0"}, {}, false);
+		uASSERT_EQUAL(0, value);
+		value = execute(s, cmd, {"retval", "1"}, {}, false);
+		uASSERT_EQUAL(1, value);
+		value = execute(s, "no-such-binary", {}, {}, false);
+		uASSERT_EQUAL(1, value);
+		value = execute(s, cmd, {"segfault"}, {}, false);
+		uASSERT_EQUAL(11, value);
+		value = execute(s, cmd, {"throw"}, {}, false);
+		uASSERT_EQUAL(6, value);
+		value = execute(s, cmd, {"abort"}, {}, false);
+		uASSERT_EQUAL(6, value);
 	}
 
-	void runit()
+	void env()
 	{
-		uASSERT_EQUAL(0, execute("/bin/true", {}, false));
-		uASSERT_EQUAL(1, execute("/bin/false", {}, false));
-		uASSERT_EQUAL(1, execute("no-such-binary", {}, false));
+		using namespace std::string_literals;
+
+		ctor::settings s;
+		auto cur_path = std::filesystem::path(paths::argv_0).parent_path();
+		std::vector<std::string> paths{{cur_path.string()}};
+		auto cmd = locate("testprog", paths);
+		uASSERT(!cmd.empty());
+
+		tmp_file tmp;
+
+		std::map<std::string, std::string> env;
+
+		// New env vars
+		env["foo"] = "bar";
+		env["bar"] = "42";
+
+		// Overwrite the exiting LANG var
+		env["LANG"] = "foo";
+
+		auto value = execute(s, cmd, {"envdump", tmp.get()}, env, false);
+		uASSERT_EQUAL(0, value);
+
+		std::vector<std::string> vars;
+		{
+			std::ifstream infile(tmp.get());
+			std::string line;
+			while (std::getline(infile, line))
+			{
+				vars.push_back(line);
+			}
+		}
+
+		// Check the two explicitly set vars
+		auto chk = std::find(vars.begin(), vars.end(), "foo=bar"s);
+		uASSERT(chk != vars.end());
+		chk = std::find(vars.begin(), vars.end(), "bar=42"s);
+		uASSERT(chk != vars.end());
+
+		// Check the one that should have overwritten the existing one (probably LANG=en_US.UTF-8 or something)
+		chk = std::find(vars.begin(), vars.end(), "LANG=foo"s);
+		uASSERT(chk != vars.end());
+
+		// Check that other vars are also there (ie. the env wasn't cleared on entry)
+		uASSERT(vars.size() > 3);
 	}
 };