tests: Add tests for libpdbg probe functions

Signed-off-by: Amitay Isaacs <amitay@ozlabs.org>

1 files changed, 253 insertions, 0 deletions

diff --git a/src/tests/libpdbg_probe_test.c b/src/tests/libpdbg_probe_test.c
new file mode 100644
index 0000000..8d7070b
--- /dev/null
+++ b/src/tests/libpdbg_probe_test.c
@@ -0,0 +1,253 @@
+/* Copyright 2018 IBM Corp.
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *      http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or
+ * implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+#include <stdio.h>
+#include <inttypes.h>
+#include <assert.h>
+
+#include <libpdbg.h>
+
+#include "fake.dt.h"
+
+static void for_each_target(struct pdbg_target *parent,
+			    void (*callback)(struct pdbg_target *target,
+					     enum pdbg_target_status status),
+			    enum pdbg_target_status status)
+{
+	struct pdbg_target *child;
+
+	assert(parent);
+
+	callback(parent, status);
+
+	pdbg_for_each_child_target(parent, child) {
+		for_each_target(child, callback, status);
+	}
+}
+
+static void for_target_to_root(struct pdbg_target *target,
+			       void (*callback)(struct pdbg_target *target,
+						enum pdbg_target_status status),
+			       enum pdbg_target_status status)
+{
+	assert(target);
+
+	do {
+		callback(target, status);
+		target = pdbg_target_parent(NULL, target);
+	} while (target != NULL);
+}
+
+static void check_status(struct pdbg_target *target,
+			 enum pdbg_target_status status)
+{
+	struct pdbg_target *root;
+
+	root = pdbg_target_root();
+	if (target == root) {
+		if (pdbg_target_status(target) != status) {
+			printf("root node: status=%u, expected=%u\n",
+			       pdbg_target_status(target), status);
+			assert(pdbg_target_status(target) == status);
+		}
+		return;
+	}
+
+	assert(pdbg_target_status(target) == status);
+}
+
+static void test1(void)
+{
+	struct pdbg_target *root, *target;
+
+	pdbg_targets_init(&_binary_fake_dtb_o_start);
+
+	root = pdbg_target_root();
+	assert(root);
+
+	for_each_target(root, check_status, PDBG_TARGET_UNKNOWN);
+
+	pdbg_target_probe_all(root);
+	for_each_target(root, check_status, PDBG_TARGET_ENABLED);
+
+	pdbg_for_each_class_target("core", target) {
+		pdbg_target_release(target);
+	}
+	pdbg_for_each_class_target("fsi", target) {
+		check_status(target, PDBG_TARGET_ENABLED);
+	}
+	pdbg_for_each_class_target("pib", target) {
+		check_status(target, PDBG_TARGET_ENABLED);
+	}
+	pdbg_for_each_class_target("core", target) {
+		for_each_target(target, check_status, PDBG_TARGET_RELEASED);
+	}
+
+	pdbg_for_each_class_target("pib", target) {
+		pdbg_target_release(target);
+	}
+	pdbg_for_each_class_target("fsi", target) {
+		check_status(target, PDBG_TARGET_ENABLED);
+	}
+	pdbg_for_each_class_target("pib", target) {
+		for_each_target(target, check_status, PDBG_TARGET_RELEASED);
+	}
+
+	pdbg_for_each_class_target("fsi", target) {
+		pdbg_target_release(target);
+	}
+	pdbg_for_each_class_target("fsi", target) {
+		for_each_target(target, check_status, PDBG_TARGET_RELEASED);
+	}
+	check_status(root, PDBG_TARGET_ENABLED);
+}
+
+static void test2(void)
+{
+	struct pdbg_target *root, *target;
+	enum pdbg_target_status status;
+
+	pdbg_targets_init(&_binary_fake_dtb_o_start);
+
+	root = pdbg_target_root();
+	assert(root);
+
+	for_each_target(root, check_status, PDBG_TARGET_UNKNOWN);
+
+	pdbg_for_each_class_target("pib", target) {
+		status = pdbg_target_probe(target);
+		assert(status == PDBG_TARGET_ENABLED);
+
+		for_target_to_root(target, check_status, PDBG_TARGET_ENABLED);
+	}
+	pdbg_for_each_class_target("core", target) {
+		for_each_target(target, check_status, PDBG_TARGET_UNKNOWN);
+	}
+
+	pdbg_for_each_class_target("core", target) {
+		status = pdbg_target_probe(target);
+		assert(status == PDBG_TARGET_ENABLED);
+
+		for_target_to_root(target, check_status, PDBG_TARGET_ENABLED);
+	}
+	pdbg_for_each_class_target("thread", target) {
+		for_each_target(target, check_status, PDBG_TARGET_UNKNOWN);
+	}
+
+	pdbg_for_each_class_target("core", target) {
+		pdbg_target_probe_all(target);
+	}
+	pdbg_for_each_class_target("thread", target) {
+		for_each_target(target, check_status, PDBG_TARGET_ENABLED);
+	}
+
+	pdbg_for_each_class_target("core", target) {
+		pdbg_target_release(target);
+	}
+	pdbg_for_each_class_target("fsi", target) {
+		check_status(target, PDBG_TARGET_ENABLED);
+	}
+	pdbg_for_each_class_target("pib", target) {
+		check_status(target, PDBG_TARGET_ENABLED);
+	}
+	pdbg_for_each_class_target("core", target) {
+		check_status(target, PDBG_TARGET_RELEASED);
+	}
+	pdbg_for_each_class_target("thread", target) {
+		check_status(target, PDBG_TARGET_RELEASED);
+	}
+
+	pdbg_for_each_class_target("fsi", target) {
+		pdbg_target_release(target);
+	}
+	pdbg_for_each_class_target("fsi", target) {
+		for_each_target(target, check_status, PDBG_TARGET_RELEASED);
+	}
+}
+
+static void test3(void)
+{
+	struct pdbg_target *root, *target;
+	enum pdbg_target_status status;
+
+	pdbg_targets_init(&_binary_fake_dtb_o_start);
+
+	root = pdbg_target_root();
+	assert(root);
+
+	for_each_target(root, check_status, PDBG_TARGET_UNKNOWN);
+
+	pdbg_for_each_class_target("core", target) {
+		pdbg_target_status_set(target, PDBG_TARGET_DISABLED);
+	}
+	pdbg_for_each_class_target("thread", target) {
+		check_status(target, PDBG_TARGET_UNKNOWN);
+	}
+
+	pdbg_for_each_class_target("thread", target) {
+		status = pdbg_target_probe(target);
+		assert(status == PDBG_TARGET_UNKNOWN);
+		check_status(target, PDBG_TARGET_UNKNOWN);
+	}
+
+	pdbg_target_probe_all(root);
+
+	pdbg_for_each_class_target("core", target) {
+		check_status(target, PDBG_TARGET_DISABLED);
+	}
+	pdbg_for_each_class_target("thread", target) {
+		check_status(target, PDBG_TARGET_UNKNOWN);
+	}
+
+	pdbg_for_each_class_target("pib", target) {
+		pdbg_target_release(target);
+	}
+	pdbg_for_each_class_target("fsi", target) {
+		for_target_to_root(target, check_status, PDBG_TARGET_ENABLED);
+	}
+	pdbg_for_each_class_target("pib", target) {
+		check_status(target, PDBG_TARGET_RELEASED);
+	}
+
+	pdbg_target_release(root);
+	pdbg_for_each_class_target("pib", target) {
+		for_target_to_root(target, check_status, PDBG_TARGET_RELEASED);
+	}
+	pdbg_for_each_class_target("core", target) {
+		check_status(target, PDBG_TARGET_DISABLED);
+	}
+	pdbg_for_each_class_target("thread", target) {
+		check_status(target, PDBG_TARGET_UNKNOWN);
+	}
+}
+
+int main(void)
+{
+	int test_id = TEST_ID;
+
+	if (test_id == 1) {
+		test1();
+	} else if (test_id == 2) {
+		test2();
+	} else if (test_id == 3) {
+		test3();
+	} else {
+		printf("No test for TEST_ID=%d\n", test_id);
+		return 1;
+	}
+
+	return 0;
+}