[PATCH] libtracefs: Unit test for the tracing filter API

[Sameeruddin shaik <sameeruddin.shaik8@xxxxxxxxx>]

tracefs_function_filter();

Signed-off-by: Sameeruddin shaik <sameeruddin.shaik8@xxxxxxxxx>

diff --git a/utest/tracefs-utest.c b/utest/tracefs-utest.c
index ed2693b..09e564f 100644
--- a/utest/tracefs-utest.c
+++ b/utest/tracefs-utest.c
@@ -11,6 +11,7 @@
 #include <time.h>
 #include <dirent.h>
 #include <ftw.h>
+#include <errno.h>
 
 #include <CUnit/CUnit.h>
 #include <CUnit/Basic.h>
@@ -1020,6 +1021,80 @@ static void test_custom_trace_dir(void)
 	free(dname);
 }
 
+static int test_instance_filter(struct tracefs_instance *instance,
+				const char **filters, const char *module,
+				int flags)
+{
+	int ret;
+	int i;
+
+	if (filters) {
+		for (i = 0; filters[i]; i++) {
+			ret = tracefs_function_filter(instance, filters[i],
+						      module, flags);
+			if (ret) {
+				if (errno == EINVAL)
+					printf("Filter %s did not match\n",
+					       filters[i]);
+				else
+					printf("Failed writing %s\n",
+					       filters[i]);
+			}
+		}
+	} else {
+		ret = tracefs_function_filter(instance, NULL, module, flags);
+	}
+	return ret;
+}
+
+static void test_tracefs_function_filter(void)
+{
+	const char *filter[] = {"run_init_process", "ufs*", "^ext4.*$", NULL};
+	const char *future_filter[] = {"write_dummy", "read_dummy", NULL};
+	const char *future_module = "dummy";
+	struct tracefs_instance *instance;
+	const char *module = "btrtl";
+	int ret;
+
+	instance = tracefs_instance_create(TEST_INSTANCE_NAME);
+	CU_TEST(instance != NULL);
+
+	ret = tracefs_function_filter(instance, NULL, NULL,
+				      TRACEFS_FL_RESET | TRACEFS_FL_CONTINUE);
+	if (ret) {
+		printf("Failed to reset the filter\n");
+		if (ret < 0)
+			tracefs_function_filter(instance, NULL, NULL, 0);
+	}
+	/* Test string, kernel glob and regex for specific_instance*/
+	ret = test_instance_filter(instance, filter, NULL, TRACEFS_FL_CONTINUE);
+	CU_ASSERT(ret == 0);
+	/* Test Module only with no filters*/
+	ret = test_instance_filter(instance, NULL, module, TRACEFS_FL_CONTINUE);
+	CU_ASSERT(ret == 0);
+	ret = test_instance_filter(instance, future_filter, future_module,
+				   TRACEFS_FL_CONTINUE | TRACEFS_FL_FUTURE);
+	CU_ASSERT(ret == 0);
+	tracefs_function_filter(instance, NULL, NULL, 0);
+
+	ret = tracefs_function_filter(NULL, NULL, NULL,
+				      TRACEFS_FL_RESET | TRACEFS_FL_CONTINUE);
+	if (ret) {
+		printf("Failed to reset the filter\n");
+		if (ret < 0)
+			tracefs_function_filter(instance, NULL, NULL, 0);
+	}
+	/* Test top instance*/
+	ret = test_instance_filter(NULL, filter, NULL, TRACEFS_FL_CONTINUE);
+	CU_ASSERT(ret == 0);
+	ret = test_instance_filter(NULL, NULL, module, TRACEFS_FL_CONTINUE);
+	CU_ASSERT(ret == 0);
+	ret = test_instance_filter(NULL, future_filter, future_module,
+				   TRACEFS_FL_CONTINUE | TRACEFS_FL_FUTURE);
+	CU_ASSERT(ret == 0);
+	tracefs_function_filter(NULL, NULL, NULL, 0);
+}
+
 static int test_suite_destroy(void)
 {
 	tracefs_instance_destroy(test_instance);
@@ -1075,4 +1150,6 @@ void test_tracefs_lib(void)
 		    test_tracing_options);
 	CU_add_test(suite, "custom system directory",
 		    test_custom_trace_dir);
+	CU_add_test(suite, "Set Filter API",
+		    test_tracefs_function_filter);
 }
-- 
2.7.4