1 files changed, 513 insertions, 0 deletions
diff --git a/ctdb/tests/src/conf_test.c b/ctdb/tests/src/conf_test.c
new file mode 100644
index 0000000..9b3bd8f
--- /dev/null
+++ b/ctdb/tests/src/conf_test.c
@@ -0,0 +1,513 @@
+/*
+   Configuration file handling on top of tini
+
+   Copyright (C) Amitay Isaacs  2017
+
+   This program is free software; you can redistribute it and/or modify
+   it under the terms of the GNU General Public License as published by
+   the Free Software Foundation; either version 3 of the License, or
+   (at your option) any later version.
+
+   This program is distributed in the hope that it will be useful,
+   but WITHOUT ANY WARRANTY; without even the implied warranty of
+   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+   GNU General Public License for more details.
+
+   You should have received a copy of the GNU General Public License
+   along with this program; if not, see <http://www.gnu.org/licenses/>.
+*/
+
+#include "replace.h"
+
+#include <assert.h>
+
+#include "common/conf.c"
+
+static void test1(void)
+{
+	TALLOC_CTX *mem_ctx = talloc_new(NULL);
+	struct conf_context *conf;
+	int ret;
+	bool status;
+
+	ret = conf_init(mem_ctx, &conf);
+	assert(ret == 0);
+	assert(conf != NULL);
+
+	conf_define_section(conf, "section1", NULL);
+	status = conf_valid(conf);
+	assert(status == true);
+
+	conf_define_section(conf, NULL, NULL);
+	status = conf_valid(conf);
+	assert(status == false);
+
+	talloc_free(mem_ctx);
+}
+
+static void test2(void)
+{
+	TALLOC_CTX *mem_ctx = talloc_new(NULL);
+	struct conf_context *conf;
+	int ret;
+	bool status;
+
+	ret = conf_init(mem_ctx, &conf);
+	assert(ret == 0);
+	assert(conf != NULL);
+
+	conf_define_string(conf, "section1", "key1", "default", NULL);
+	status = conf_valid(conf);
+	assert(status == false);
+
+	talloc_free(mem_ctx);
+}
+
+static void test3(void)
+{
+	TALLOC_CTX *mem_ctx = talloc_new(NULL);
+	struct conf_context *conf;
+	int ret;
+	bool status;
+
+	ret = conf_init(mem_ctx, &conf);
+	assert(ret == 0);
+	assert(conf != NULL);
+
+	conf_define_section(conf, "section1", NULL);
+	status = conf_valid(conf);
+	assert(status == true);
+
+	conf_define_string(conf, "section1", "key1", NULL, NULL);
+	status = conf_valid(conf);
+	assert(status == true);
+
+	conf_define_string(conf, "section1", "key1", "value1", NULL);
+	status = conf_valid(conf);
+	assert(status == false);
+
+	talloc_free(mem_ctx);
+}
+
+static void test4(void)
+{
+	TALLOC_CTX *mem_ctx = talloc_new(NULL);
+	struct conf_context *conf;
+	int ret;
+	bool status;
+
+	ret = conf_init(mem_ctx, &conf);
+	assert(ret == 0);
+	assert(conf != NULL);
+
+	conf_define_section(conf, "section1", NULL);
+	status = conf_valid(conf);
+	assert(status == true);
+
+	conf_define_string(conf, "section1", "key1", NULL, NULL);
+	status = conf_valid(conf);
+	assert(status == true);
+
+	conf_define_integer(conf, "section1", "key1", 10, NULL);
+	status = conf_valid(conf);
+	assert(status == false);
+
+	talloc_free(mem_ctx);
+}
+
+static void test5(void)
+{
+	TALLOC_CTX *mem_ctx = talloc_new(NULL);
+	struct conf_context *conf;
+	enum conf_type type;
+	int ret;
+	bool status;
+	const char *s_val;
+	int i_val;
+	bool b_val;
+
+	ret = conf_init(mem_ctx, &conf);
+	assert(ret == 0);
+	assert(conf != NULL);
+
+	conf_define_section(conf, "section1", NULL);
+	status = conf_valid(conf);
+	assert(status == true);
+
+	conf_define_string(conf, "section1", "key1", "value1", NULL);
+	conf_define_integer(conf, "section1", "key2", 10, NULL);
+	conf_define_boolean(conf, "section1", "key3", true, NULL);
+
+	conf_assign_string_pointer(conf, "section1", "key1", &s_val);
+	conf_assign_integer_pointer(conf, "section1", "key2", &i_val);
+	conf_assign_boolean_pointer(conf, "section1", "key3", &b_val);
+
+	status = conf_valid(conf);
+	assert(status == true);
+
+	status = conf_query(conf, "section1", "key1", &type);
+	assert(status == true);
+	assert(type == CONF_STRING);
+
+	status = conf_query(conf, "section1", "key2", &type);
+	assert(status == true);
+	assert(type == CONF_INTEGER);
+
+	status = conf_query(conf, "section1", "key3", &type);
+	assert(status == true);
+	assert(type == CONF_BOOLEAN);
+
+	assert(strcmp(s_val, "value1") == 0);
+	assert(i_val == 10);
+	assert(b_val == true);
+
+	conf_set_defaults(conf);
+
+	assert(strcmp(s_val, "value1") == 0);
+	assert(i_val == 10);
+	assert(b_val == true);
+
+	talloc_free(mem_ctx);
+}
+
+static void test6(void)
+{
+	TALLOC_CTX *mem_ctx = talloc_new(NULL);
+	struct conf_context *conf;
+	int ret;
+	bool status;
+	const char *s_val, *s2_val;
+	int i_val, i2_val;
+	bool b_val, b2_val, is_default;
+
+	ret = conf_init(mem_ctx, &conf);
+	assert(ret == 0);
+	assert(conf != NULL);
+
+	conf_define_section(conf, "section1", NULL);
+	status = conf_valid(conf);
+	assert(status == true);
+
+	conf_define_string(conf, "section1", "key1", "default", NULL);
+	conf_define_integer(conf, "section1", "key2", 10, NULL);
+	conf_define_boolean(conf, "section1", "key3", true, NULL);
+
+	conf_assign_string_pointer(conf, "section1", "key1", &s_val);
+	conf_assign_integer_pointer(conf, "section1", "key2", &i_val);
+	conf_assign_boolean_pointer(conf, "section1", "key3", &b_val);
+
+	status = conf_valid(conf);
+	assert(status == true);
+
+	is_default = false;
+	ret = conf_get_string(conf, "section1", "key1", &s2_val, &is_default);
+	assert(ret == 0);
+	assert(strcmp(s2_val, "default") == 0);
+	assert(is_default == true);
+
+	is_default = false;
+	ret = conf_get_integer(conf, "section1", "key2", &i2_val, &is_default);
+	assert(ret == 0);
+	assert(i2_val == 10);
+	assert(is_default == true);
+
+	is_default = false;
+	ret = conf_get_boolean(conf, "section1", "key3", &b2_val, &is_default);
+	assert(ret == 0);
+	assert(b2_val == true);
+	assert(is_default == true);
+
+	ret = conf_set_string(conf, "section1", "key1", "foobar");
+	assert(ret == 0);
+
+	ret = conf_set_integer(conf, "section1", "key2", 20);
+	assert(ret == 0);
+
+	ret = conf_set_boolean(conf, "section1", "key3", false);
+	assert(ret == 0);
+
+	assert(strcmp(s_val, "foobar") == 0);
+	assert(i_val == 20);
+	assert(b_val == false);
+
+	is_default = true;
+	ret = conf_get_string(conf, "section1", "key1", &s2_val, &is_default);
+	assert(ret == 0);
+	assert(strcmp(s2_val, "foobar") == 0);
+	assert(is_default == false);
+
+	is_default = true;
+	ret = conf_get_integer(conf, "section1", "key2", &i2_val, &is_default);
+	assert(ret == 0);
+	assert(i2_val == 20);
+	assert(is_default == false);
+
+	is_default = true;
+	ret = conf_get_boolean(conf, "section1", "key3", &b2_val, &is_default);
+	assert(ret == 0);
+	assert(b2_val == false);
+	assert(is_default == false);
+
+	conf_dump(conf, stdout);
+
+	conf_set_defaults(conf);
+
+	assert(strcmp(s_val, "default") == 0);
+	assert(i_val == 10);
+	assert(b_val == true);
+
+	talloc_free(mem_ctx);
+}
+
+static bool test7_validate_string(const char *key,
+				  const char *old_value, const char *new_value,
+				  enum conf_update_mode mode)
+{
+	return false;
+}
+
+static bool test7_validate_integer(const char *key,
+				   int old_value, int new_value,
+				   enum conf_update_mode mode)
+{
+	return false;
+}
+
+static bool test7_validate_boolean(const char *key,
+				   bool old_value, bool new_value,
+				   enum conf_update_mode mode)
+{
+	return false;
+}
+
+static void test7(void)
+{
+	TALLOC_CTX *mem_ctx = talloc_new(NULL);
+	struct conf_context *conf;
+	int ret;
+	bool status;
+	const char *s_val, *s2_val;
+	int i_val, i2_val;
+	bool b_val, b2_val;
+
+	ret = conf_init(mem_ctx, &conf);
+	assert(ret == 0);
+	assert(conf != NULL);
+
+	conf_define_section(conf, "section1", NULL);
+	status = conf_valid(conf);
+	assert(status == true);
+
+	conf_define_string(conf, "section1", "key1", "default",
+			   test7_validate_string);
+	conf_define_integer(conf, "section1", "key2", 10,
+			    test7_validate_integer);
+	conf_define_boolean(conf, "section1", "key3", true,
+			    test7_validate_boolean);
+
+	conf_assign_string_pointer(conf, "section1", "key1", &s_val);
+	conf_assign_integer_pointer(conf, "section1", "key2", &i_val);
+	conf_assign_boolean_pointer(conf, "section1", "key3", &b_val);
+
+	status = conf_valid(conf);
+	assert(status == true);
+
+	ret = conf_set_string(conf, "section1", "key1", "default");
+	assert(ret == 0);
+
+	ret = conf_set_string(conf, "section1", "key1", "foobar");
+	assert(ret == EINVAL);
+
+	ret = conf_set_integer(conf, "section1", "key2", 10);
+	assert(ret == 0);
+
+	ret = conf_set_integer(conf, "section1", "key2", 20);
+	assert(ret == EINVAL);
+
+	ret = conf_set_boolean(conf, "section1", "key3", true);
+	assert(ret == 0);
+
+	ret = conf_set_boolean(conf, "section1", "key3", false);
+	assert(ret == EINVAL);
+
+	assert(strcmp(s_val, "default") == 0);
+	assert(i_val == 10);
+	assert(b_val == true);
+
+	ret = conf_get_string(conf, "section1", "key2", &s2_val, NULL);
+	assert(ret == EINVAL);
+
+	ret = conf_get_integer(conf, "section1", "key3", &i2_val, NULL);
+	assert(ret == EINVAL);
+
+	ret = conf_get_boolean(conf, "section1", "key1", &b2_val, NULL);
+	assert(ret == EINVAL);
+
+	talloc_free(mem_ctx);
+}
+
+static bool test8_validate(struct conf_context *conf,
+			   const char *section,
+			   enum conf_update_mode mode)
+{
+	return false;
+}
+
+static void test8(const char *filename)
+{
+	TALLOC_CTX *mem_ctx = talloc_new(NULL);
+	struct conf_context *conf;
+	int ret;
+	bool status;
+
+	ret = conf_init(mem_ctx, &conf);
+	assert(ret == 0);
+	assert(conf != NULL);
+
+	conf_define_section(conf, "section1", test8_validate);
+	status = conf_valid(conf);
+	assert(status == true);
+
+	conf_define_string(conf, "section1", "key1", "default", NULL);
+
+	status = conf_valid(conf);
+	assert(status == true);
+
+	ret = conf_load(conf, filename, true);
+	conf_dump(conf, stdout);
+
+	talloc_free(mem_ctx);
+	exit(ret);
+}
+
+static void test9(const char *filename, bool ignore_unknown)
+{
+	TALLOC_CTX *mem_ctx = talloc_new(NULL);
+	struct conf_context *conf;
+	int ret;
+	bool status;
+
+	ret = conf_init(mem_ctx, &conf);
+	assert(ret == 0);
+	assert(conf != NULL);
+
+	conf_define_section(conf, "section1", NULL);
+
+	conf_define_string(conf, "section1", "key1", "value1", NULL);
+	conf_define_integer(conf, "section1", "key2", 10, NULL);
+	conf_define_boolean(conf, "section1", "key3", true, NULL);
+
+	status = conf_valid(conf);
+	assert(status == true);
+
+	conf_set_boolean(conf, "section1", "key3", false);
+
+	ret = conf_load(conf, filename, ignore_unknown);
+	conf_dump(conf, stdout);
+
+	talloc_free(mem_ctx);
+	exit(ret);
+}
+
+static void test11(const char *filename)
+{
+	TALLOC_CTX *mem_ctx = talloc_new(NULL);
+	char reload[PATH_MAX];
+	struct conf_context *conf;
+	int ret;
+	bool status;
+
+	ret = snprintf(reload, sizeof(reload), "%s.reload", filename);
+	assert((size_t)ret < sizeof(reload));
+
+	ret = conf_init(mem_ctx, &conf);
+	assert(ret == 0);
+	assert(conf != NULL);
+
+	conf_define_section(conf, "section1", NULL);
+
+	conf_define_string(conf, "section1", "key1", "value1", NULL);
+	conf_define_integer(conf, "section1", "key2", 10, NULL);
+	conf_define_boolean(conf, "section1", "key3", true, NULL);
+
+	status = conf_valid(conf);
+	assert(status == true);
+
+	ret = conf_load(conf, filename, false);
+	assert(ret == 0);
+
+	ret = rename(reload, filename);
+	assert(ret == 0);
+
+	ret = conf_reload(conf);
+	assert(ret == 0);
+
+	conf_dump(conf, stdout);
+
+	talloc_free(mem_ctx);
+	exit(ret);
+}
+
+int main(int argc, const char **argv)
+{
+	int num;
+
+	if (argc < 2) {
+		fprintf(stderr, "Usage: %s <testnum> [<config>]\n", argv[0]);
+		exit(1);
+	}
+
+	num = atoi(argv[1]);
+	if (num > 7 && argc != 3) {
+		fprintf(stderr, "Usage: %s <testnum> [<config>]\n", argv[0]);
+		exit(1);
+	}
+
+	switch (num) {
+	case 1:
+		test1();
+		break;
+
+	case 2:
+		test2();
+		break;
+
+	case 3:
+		test3();
+		break;
+
+	case 4:
+		test4();
+		break;
+
+	case 5:
+		test5();
+		break;
+
+	case 6:
+		test6();
+		break;
+
+	case 7:
+		test7();
+		break;
+
+	case 8:
+		test8(argv[2]);
+		break;
+
+	case 9:
+		test9(argv[2], true);
+		break;
+
+	case 10:
+		test9(argv[2], false);
+		break;
+
+	case 11:
+		test11(argv[2]);
+		break;
+	}
+
+	return 0;
+}