From 2a9cae17d87208ae887c277ddcfce37de35b6f59 Mon Sep 17 00:00:00 2001
From: Daniel Stone <daniels@collabora.com>
Date: Tue, 12 Jul 2022 15:40:35 +0100
Subject: [PATCH] custom-env: Add tests for environment handling

Test the basic stuff: initialising from a known environment, setting a
new variable, overwriting a previous variable, and getting the resulting
array to pass to execve.

Signed-off-by: Daniel Stone <daniels@collabora.com>
---
 tests/custom-env-test.c | 87 +++++++++++++++++++++++++++++++++++++++++
 tests/meson.build       |  1 +
 2 files changed, 88 insertions(+)
 create mode 100644 tests/custom-env-test.c

diff --git a/tests/custom-env-test.c b/tests/custom-env-test.c
new file mode 100644
index 00000000..49adb022
--- /dev/null
+++ b/tests/custom-env-test.c
@@ -0,0 +1,87 @@
+/*
+ * Copyright 2022 Collabora, Ltd.
+ *
+ * Permission is hereby granted, free of charge, to any person obtaining
+ * a copy of this software and associated documentation files (the
+ * "Software"), to deal in the Software without restriction, including
+ * without limitation the rights to use, copy, modify, merge, publish,
+ * distribute, sublicense, and/or sell copies of the Software, and to
+ * permit persons to whom the Software is furnished to do so, subject to
+ * the following conditions:
+ *
+ * The above copyright notice and this permission notice (including the
+ * next paragraph) shall be included in all copies or substantial
+ * portions of the Software.
+ *
+ * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
+ * EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
+ * MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
+ * NONINFRINGEMENT.  IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS
+ * BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN
+ * ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN
+ * CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+ * SOFTWARE.
+ */
+
+#include "config.h"
+
+#include <assert.h>
+#include <stdio.h>
+#include <stdlib.h>
+#include <string.h>
+#include <unistd.h>
+
+#include <wayland-util.h>
+
+#include "shared/helpers.h"
+#include "shared/os-compatibility.h"
+#include "shared/process-util.h"
+#include "shared/string-helpers.h"
+
+#include "weston-test-runner.h"
+
+#define ASSERT_STR_MATCH(_as, _bs) do { \
+	const char *as = _as; \
+	const char *bs = _bs; \
+	assert(!!as == !!bs); \
+	assert(!as || strcmp(as, bs) == 0); \
+} while (0)
+
+#define ASSERT_STR_ARRAY_MATCH(_name, _aa, _ba) do { \
+	char * const *aa = _aa; \
+	char * const *ba = _ba; \
+	testlog("\tcomparing " _name ":\n"); \
+	for (int _i = 0; aa[_i] || ba[_i]; _i++) { \
+		testlog("\t\t[%d] '%s' == '%s'?\n", _i, aa[_i], ba[_i]); \
+		ASSERT_STR_MATCH(aa[_i], ba[_i]); \
+	} \
+	testlog("\tsuccessfully compared " _name "\n"); \
+} while (0)
+
+static enum test_result_code
+setup_env(struct weston_test_harness *harness)
+{
+	/* as this is a standalone test, we can clear the environment here */
+	clearenv();
+
+	putenv("ENV1=one");
+	setenv("ENV2", "two", 1);
+	setenv("ENV3", "three", 1);
+
+	return weston_test_harness_execute_standalone(harness);
+}
+
+DECLARE_FIXTURE_SETUP(setup_env);
+
+TEST(basic_env)
+{
+	struct custom_env env;
+	char *const envp[] = { "ENV1=one", "ENV2=two", "ENV3=four", "ENV5=five", NULL };
+
+	custom_env_init_from_environ(&env);
+	custom_env_set_env_var(&env, "ENV5", "five");
+	custom_env_set_env_var(&env, "ENV3", "four");
+	ASSERT_STR_ARRAY_MATCH("envp", custom_env_get_envp(&env), envp);
+	assert(env.env_finalized);
+	custom_env_fini(&env);
+}
diff --git a/tests/meson.build b/tests/meson.build
index fe107334..50722f95 100644
--- a/tests/meson.build
+++ b/tests/meson.build
@@ -146,6 +146,7 @@ tests = [
 		'dep_objs': dep_libexec_weston,
 	},
 	{	'name': 'color-manager', },
+        {       'name': 'custom-env', },
 	{	'name': 'devices', },
 	{
 		'name': 'drm-formats',