[babeltrace.git] / tests / lib / plugin.c

/*
 * Copyright (c) 2017 Philippe Proulx <pproulx@efficios.com>
 *
 * 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; under version 2 of the License.
 *
 * 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, write to the Free Software Foundation, Inc.,
 * 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
 */

#include <babeltrace2/babeltrace.h>
#include <stdlib.h>
#include <string.h>
#include <stdio.h>
#include "common/assert.h"
#include <glib.h>
#include "tap/tap.h"
#include "common.h"

#define NR_TESTS 		38
#define NON_EXISTING_PATH	"/this/hopefully/does/not/exist/5bc75f8d-0dba-4043-a509-d7984b97e42b.so"

/* Those symbols are written to by some test plugins */
static int check_env_var(const char *name)
{
	const char *val = getenv(name);

	if (!val) {
		return -1;
	}

	return atoi(val);
}

static void reset_test_plugin_env_vars(void)
{
	g_setenv("BT_TEST_PLUGIN_INIT_CALLED", "0", 1);
	g_setenv("BT_TEST_PLUGIN_EXIT_CALLED", "0", 1);
}

static char *get_test_plugin_path(const char *plugin_dir,
		const char *plugin_name)
{
	char *ret;
	char *plugin_file_name;

	if (asprintf(&plugin_file_name, "plugin-%s." G_MODULE_SUFFIX,
			plugin_name) == -1) {
		abort();
	}

	ret = g_build_filename(plugin_dir, plugin_file_name, NULL);
	free(plugin_file_name);

	return ret;
}

static void test_minimal(const char *plugin_dir)
{
	const bt_plugin_set *plugin_set = NULL;
	const bt_plugin *plugin;
	char *minimal_path = get_test_plugin_path(plugin_dir, "minimal");
	bt_plugin_find_all_from_file_status status;

	BT_ASSERT(minimal_path);
	diag("minimal plugin test below");

	reset_test_plugin_env_vars();
	status = bt_plugin_find_all_from_file(minimal_path, BT_FALSE,
		&plugin_set);
	ok(status == BT_PLUGIN_FIND_ALL_FROM_FILE_STATUS_OK,
		"bt_plugin_find_all_from_file() succeeds with a valid file");
	ok(plugin_set,
		"bt_plugin_find_all_from_file() returns a plugin set");
	ok(check_env_var("BT_TEST_PLUGIN_INIT_CALLED") == 1,
		"plugin's initialization function is called during bt_plugin_find_all_from_file()");
	ok(bt_plugin_set_get_plugin_count(plugin_set) == 1,
		"bt_plugin_find_all_from_file() returns the expected number of plugins");
	plugin = bt_plugin_set_borrow_plugin_by_index_const(plugin_set, 0);
	ok(strcmp(bt_plugin_get_name(plugin), "test_minimal") == 0,
		"bt_plugin_get_name() returns the expected name");
	ok(strcmp(bt_plugin_get_description(plugin),
		"Minimal Babeltrace plugin with no component classes") == 0,
		"bt_plugin_get_description() returns the expected description");
	ok(bt_plugin_get_version(plugin, NULL, NULL, NULL, NULL) ==
		BT_PROPERTY_AVAILABILITY_NOT_AVAILABLE,
		"bt_plugin_get_version() fails when there's no version");
	ok(strcmp(bt_plugin_get_author(plugin), "Janine Sutto") == 0,
		"bt_plugin_get_author() returns the expected author");
	ok(strcmp(bt_plugin_get_license(plugin), "Beerware") == 0,
		"bt_plugin_get_license() returns the expected license");
	ok(strcmp(bt_plugin_get_path(plugin), minimal_path) == 0,
		"bt_plugin_get_path() returns the expected path");
	ok(bt_plugin_get_source_component_class_count(plugin) == 0,
		"bt_plugin_get_source_component_class_count() returns the expected value");
	ok(bt_plugin_get_filter_component_class_count(plugin) == 0,
		"bt_plugin_get_filter_component_class_count() returns the expected value");
	ok(bt_plugin_get_sink_component_class_count(plugin) == 0,
		"bt_plugin_get_sink_component_class_count() returns the expected value");
	bt_plugin_set_put_ref(plugin_set);
	ok(check_env_var("BT_TEST_PLUGIN_EXIT_CALLED") == 1,
		"plugin's exit function is called when the plugin is destroyed");

	free(minimal_path);
}

static void test_sfs(const char *plugin_dir)
{
	const bt_plugin_set *plugin_set = NULL;
	const bt_plugin *plugin;
	const bt_component_class_sink *sink_comp_class;
	const bt_component_class_source *source_comp_class;
	const bt_component_class_filter *filter_comp_class;
	const bt_component_sink *sink_component;
	char *sfs_path = get_test_plugin_path(plugin_dir, "sfs");
	unsigned int major, minor, patch;
	const char *extra;
	bt_value *params;
	const bt_value *results;
	const bt_value *object;
	const bt_value *res_params;
	bt_graph *graph;
	const char *object_str;
	bt_graph_add_component_status graph_ret;
	bt_query_executor *query_exec = bt_query_executor_create();
	int ret;
	bt_plugin_find_all_from_file_status status;

	BT_ASSERT(query_exec);
	BT_ASSERT(sfs_path);
	diag("sfs plugin test below");

	status = bt_plugin_find_all_from_file(sfs_path, BT_FALSE, &plugin_set);
	BT_ASSERT(status == BT_PLUGIN_FIND_ALL_FROM_FILE_STATUS_OK &&
		plugin_set && bt_plugin_set_get_plugin_count(plugin_set) == 1);
	plugin = bt_plugin_set_borrow_plugin_by_index_const(plugin_set, 0);
	ok(bt_plugin_get_version(plugin, &major, &minor, &patch, &extra) ==
		BT_PROPERTY_AVAILABILITY_AVAILABLE,
		"bt_plugin_get_version() succeeds when there's a version");
	ok(major == 1,
		"bt_plugin_get_version() returns the expected major version");
	ok(minor == 2,
		"bt_plugin_get_version() returns the expected minor version");
	ok(patch == 3,
		"bt_plugin_get_version() returns the expected patch version");
	ok(strcmp(extra, "yes") == 0,
		"bt_plugin_get_version() returns the expected extra version");
	ok(bt_plugin_get_source_component_class_count(plugin) == 1,
		"bt_plugin_get_source_component_class_count() returns the expected value");
	ok(bt_plugin_get_filter_component_class_count(plugin) == 1,
		"bt_plugin_get_filter_component_class_count() returns the expected value");
	ok(bt_plugin_get_sink_component_class_count(plugin) == 1,
		"bt_plugin_get_sink_component_class_count() returns the expected value");

	source_comp_class = bt_plugin_borrow_source_component_class_by_name_const(
		plugin, "source");
	ok(source_comp_class,
		"bt_plugin_borrow_source_component_class_by_name_const() finds a source component class");

	sink_comp_class = bt_plugin_borrow_sink_component_class_by_name_const(
		plugin, "sink");
	ok(sink_comp_class,
		"bt_plugin_borrow_sink_component_class_by_name_const() finds a sink component class");
	ok(strcmp(bt_component_class_get_help(bt_component_class_sink_as_component_class_const(sink_comp_class)),
		  "Bacon ipsum dolor amet strip steak cupim pastrami venison shoulder.\n"
		  "Prosciutto beef ribs flank meatloaf pancetta brisket kielbasa drumstick\n"
		  "venison tenderloin cow tail. Beef short loin shoulder meatball, sirloin\n"
		  "ground round brisket salami cupim pork bresaola turkey bacon boudin.\n") == 0,
		"bt_component_class_get_help() returns the expected help text");

	filter_comp_class = bt_plugin_borrow_filter_component_class_by_name_const(
		plugin, "filter");
	ok(filter_comp_class,
		"bt_plugin_borrow_filter_component_class_by_name_const() finds a filter component class");
	params = bt_value_signed_integer_create_init(23);
	BT_ASSERT(params);
	ret = bt_query_executor_query(query_exec,
		bt_component_class_filter_as_component_class_const(filter_comp_class),
		"get-something", params, BT_LOGGING_LEVEL_NONE, &results);
	ok(ret == 0 && results, "bt_query_executor_query() succeeds");
	BT_ASSERT(bt_value_is_array(results) && bt_value_array_get_size(results) == 2);
	object = bt_value_array_borrow_element_by_index_const(results, 0);
	BT_ASSERT(object && bt_value_is_string(object));
	object_str = bt_value_string_get(object);
	ok(strcmp(object_str, "get-something") == 0,
		"bt_component_class_query() receives the expected object name");
	res_params = bt_value_array_borrow_element_by_index_const(results, 1);
	ok(bt_value_compare(res_params, params),
		"bt_component_class_query() receives the expected parameters");

	bt_component_class_sink_get_ref(sink_comp_class);
	diag("> putting the plugin set object here");
	BT_PLUGIN_SET_PUT_REF_AND_RESET(plugin_set);
	graph = bt_graph_create();
	BT_ASSERT(graph);
	graph_ret = bt_graph_add_sink_component(graph, sink_comp_class,
		"the-sink", NULL, BT_LOGGING_LEVEL_NONE, &sink_component);
	ok(graph_ret == BT_GRAPH_ADD_COMPONENT_STATUS_OK && sink_component,
		"bt_graph_add_sink_component() still works after the plugin object is destroyed");
	BT_COMPONENT_SINK_PUT_REF_AND_RESET(sink_component);
	bt_graph_put_ref(graph);

	free(sfs_path);
	bt_component_class_sink_put_ref(sink_comp_class);
	bt_value_put_ref(results);
	bt_value_put_ref(params);
	bt_query_executor_put_ref(query_exec);
}

static void test_create_all_from_dir(const char *plugin_dir)
{
	const bt_plugin_set *plugin_set;
	bt_plugin_find_all_from_dir_status status;

	diag("create from all test below");

	status = bt_plugin_find_all_from_dir(NON_EXISTING_PATH, BT_FALSE,
		BT_FALSE, &plugin_set);
	ok(status == BT_PLUGIN_FIND_ALL_FROM_DIR_STATUS_ERROR,
		"bt_plugin_find_all_from_dir() fails with an invalid path");

	plugin_set = NULL;
	status = bt_plugin_find_all_from_dir(plugin_dir, BT_FALSE, BT_FALSE,
		&plugin_set);
	ok(status == BT_PLUGIN_FIND_ALL_FROM_DIR_STATUS_OK,
		"bt_plugin_find_all_from_dir() succeeds with a valid path");
	ok(plugin_set,
		"bt_plugin_find_all_from_dir() returns a plugin set with a valid path");

	/* 2 or 4, if `.la` files are considered or not */
	ok(bt_plugin_set_get_plugin_count(plugin_set) == 2 ||
		bt_plugin_set_get_plugin_count(plugin_set) == 4,
		"bt_plugin_find_all_from_dir() returns the expected number of plugin objects");

	bt_plugin_set_put_ref(plugin_set);
}

static void test_find(const char *plugin_dir)
{
	int ret;
	const bt_plugin *plugin;
	char *plugin_path;
	bt_plugin_find_status status;

	ok(bt_plugin_find(NON_EXISTING_PATH, BT_FALSE, &plugin) ==
		BT_PLUGIN_FIND_STATUS_NOT_FOUND,
		"bt_plugin_find() returns BT_PLUGIN_STATUS_NOT_FOUND with an unknown plugin name");
	ret = asprintf(&plugin_path, "%s" G_SEARCHPATH_SEPARATOR_S
			G_DIR_SEPARATOR_S "ec1d09e5-696c-442e-b1c3-f9c6cf7f5958"
			G_SEARCHPATH_SEPARATOR_S G_SEARCHPATH_SEPARATOR_S
			G_SEARCHPATH_SEPARATOR_S "%s" G_SEARCHPATH_SEPARATOR_S
			"8db46494-a398-466a-9649-c765ae077629"
			G_SEARCHPATH_SEPARATOR_S,
		NON_EXISTING_PATH, plugin_dir);
	BT_ASSERT(ret > 0 && plugin_path);
	g_setenv("BABELTRACE_PLUGIN_PATH", plugin_path, 1);
	plugin = NULL;
	status = bt_plugin_find("test_minimal", BT_FALSE, &plugin);
	ok(status == BT_PLUGIN_FIND_STATUS_OK,
		"bt_plugin_find() succeeds with a plugin name it can find");
	ok(plugin, "bt_plugin_find() returns a plugin object");
	ok(strcmp(bt_plugin_get_author(plugin), "Janine Sutto") == 0,
		"bt_plugin_find() finds the correct plugin for a given name");
	BT_PLUGIN_PUT_REF_AND_RESET(plugin);
	free(plugin_path);
}

int main(int argc, char **argv)
{
	int ret;
	const char *plugin_dir;

	if (argc != 2) {
		puts("Usage: test_plugin plugin_directory");
		ret = 1;
		goto end;
	}

	plugin_dir = argv[1];
	plan_tests(NR_TESTS);
	test_minimal(plugin_dir);
	test_sfs(plugin_dir);
	test_create_all_from_dir(plugin_dir);
	test_find(plugin_dir);
	ret = exit_status();
end:
	return ret;
}
Commit	Line	Data
cbb9e0b1	1	/*
cbb9e0b1 PP	2	* Copyright (c) 2017 Philippe Proulx <pproulx@efficios.com>
	3	*
	4	* This program is free software; you can redistribute it and/or modify
	5	* it under the terms of the GNU General Public License as published by
	6	* the Free Software Foundation; under version 2 of the License.
	7	*
	8	* This program is distributed in the hope that it will be useful,
	9	* but WITHOUT ANY WARRANTY; without even the implied warranty of
	10	* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
	11	* GNU General Public License for more details.
	12	*
	13	* You should have received a copy of the GNU General Public License along
	14	* with this program; if not, write to the Free Software Foundation, Inc.,
	15	* 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
	16	*/
	17
3fadfbc0	18	#include <babeltrace2/babeltrace.h>
cbb9e0b1 PP	19	#include <stdlib.h>
	20	#include <string.h>
	21	#include <stdio.h>
578e048b	22	#include "common/assert.h"
cbb9e0b1 PP	23	#include <glib.h>
	24	#include "tap/tap.h"
	25	#include "common.h"
	26
9736d991	27	#define NR_TESTS 38
cbb9e0b1 PP	28	#define NON_EXISTING_PATH "/this/hopefully/does/not/exist/5bc75f8d-0dba-4043-a509-d7984b97e42b.so"
	29
	30	/* Those symbols are written to by some test plugins */
df5b5d01 PP	31	static int check_env_var(const char *name)
	32	{
	33	const char *val = getenv(name);
	34
	35	if (!val) {
	36	return -1;
	37	}
	38
	39	return atoi(val);
	40	}
cbb9e0b1	41
df5b5d01	42	static void reset_test_plugin_env_vars(void)
cbb9e0b1	43	{
fbb2f0da MJ	44	g_setenv("BT_TEST_PLUGIN_INIT_CALLED", "0", 1);
fbb2f0da MJ	45	g_setenv("BT_TEST_PLUGIN_EXIT_CALLED", "0", 1);
cbb9e0b1 PP	46	}
	47
	48	static char get_test_plugin_path(const char plugin_dir,
	49	const char *plugin_name)
	50	{
cbb9e0b1	51	char *ret;
84095ea4 MJ	52	char *plugin_file_name;
	53
	54	if (asprintf(&plugin_file_name, "plugin-%s." G_MODULE_SUFFIX,
	55	plugin_name) == -1) {
	56	abort();
	57	}
	58
	59	ret = g_build_filename(plugin_dir, plugin_file_name, NULL);
	60	free(plugin_file_name);
cbb9e0b1	61
cbb9e0b1 PP	62	return ret;
	63	}
	64
cbb9e0b1 PP	65	static void test_minimal(const char *plugin_dir)
cbb9e0b1 PP	66	{
9736d991	67	const bt_plugin_set *plugin_set = NULL;
b19ff26f	68	const bt_plugin *plugin;
cbb9e0b1	69	char *minimal_path = get_test_plugin_path(plugin_dir, "minimal");
d24d5663	70	bt_plugin_find_all_from_file_status status;
cbb9e0b1	71
25583cd0	72	BT_ASSERT(minimal_path);
cbb9e0b1 PP	73	diag("minimal plugin test below");
cbb9e0b1 PP	74
df5b5d01	75	reset_test_plugin_env_vars();
9736d991 PP	76	status = bt_plugin_find_all_from_file(minimal_path, BT_FALSE,
9736d991 PP	77	&plugin_set);
d24d5663	78	ok(status == BT_PLUGIN_FIND_ALL_FROM_FILE_STATUS_OK,
c8db3219	79	"bt_plugin_find_all_from_file() succeeds with a valid file");
9736d991 PP	80	ok(plugin_set,
9736d991 PP	81	"bt_plugin_find_all_from_file() returns a plugin set");
df5b5d01	82	ok(check_env_var("BT_TEST_PLUGIN_INIT_CALLED") == 1,
c8db3219	83	"plugin's initialization function is called during bt_plugin_find_all_from_file()");
a8ff38ef	84	ok(bt_plugin_set_get_plugin_count(plugin_set) == 1,
c8db3219	85	"bt_plugin_find_all_from_file() returns the expected number of plugins");
92fed4e1	86	plugin = bt_plugin_set_borrow_plugin_by_index_const(plugin_set, 0);
6ba0b073	87	ok(strcmp(bt_plugin_get_name(plugin), "test_minimal") == 0,
cbb9e0b1 PP	88	"bt_plugin_get_name() returns the expected name");
	89	ok(strcmp(bt_plugin_get_description(plugin),
	90	"Minimal Babeltrace plugin with no component classes") == 0,
	91	"bt_plugin_get_description() returns the expected description");
d94d92ac PP	92	ok(bt_plugin_get_version(plugin, NULL, NULL, NULL, NULL) ==
d94d92ac PP	93	BT_PROPERTY_AVAILABILITY_NOT_AVAILABLE,
458e8e1d	94	"bt_plugin_get_version() fails when there's no version");
cbb9e0b1 PP	95	ok(strcmp(bt_plugin_get_author(plugin), "Janine Sutto") == 0,
	96	"bt_plugin_get_author() returns the expected author");
	97	ok(strcmp(bt_plugin_get_license(plugin), "Beerware") == 0,
	98	"bt_plugin_get_license() returns the expected license");
	99	ok(strcmp(bt_plugin_get_path(plugin), minimal_path) == 0,
	100	"bt_plugin_get_path() returns the expected path");
d94d92ac PP	101	ok(bt_plugin_get_source_component_class_count(plugin) == 0,
	102	"bt_plugin_get_source_component_class_count() returns the expected value");
	103	ok(bt_plugin_get_filter_component_class_count(plugin) == 0,
	104	"bt_plugin_get_filter_component_class_count() returns the expected value");
	105	ok(bt_plugin_get_sink_component_class_count(plugin) == 0,
	106	"bt_plugin_get_sink_component_class_count() returns the expected value");
c5b9b441	107	bt_plugin_set_put_ref(plugin_set);
df5b5d01 PP	108	ok(check_env_var("BT_TEST_PLUGIN_EXIT_CALLED") == 1,
df5b5d01 PP	109	"plugin's exit function is called when the plugin is destroyed");
cbb9e0b1 PP	110
	111	free(minimal_path);
	112	}
	113
	114	static void test_sfs(const char *plugin_dir)
	115	{
9736d991	116	const bt_plugin_set *plugin_set = NULL;
b19ff26f PP	117	const bt_plugin *plugin;
	118	const bt_component_class_sink *sink_comp_class;
	119	const bt_component_class_source *source_comp_class;
	120	const bt_component_class_filter *filter_comp_class;
	121	const bt_component_sink *sink_component;
cbb9e0b1	122	char *sfs_path = get_test_plugin_path(plugin_dir, "sfs");
458e8e1d PP	123	unsigned int major, minor, patch;
458e8e1d PP	124	const char *extra;
b19ff26f PP	125	bt_value *params;
	126	const bt_value *results;
	127	const bt_value *object;
	128	const bt_value *res_params;
	129	bt_graph *graph;
a67681c1	130	const char *object_str;
d24d5663	131	bt_graph_add_component_status graph_ret;
b19ff26f	132	bt_query_executor *query_exec = bt_query_executor_create();
c7eee084	133	int ret;
d24d5663	134	bt_plugin_find_all_from_file_status status;
cbb9e0b1	135
25583cd0 PP	136	BT_ASSERT(query_exec);
25583cd0 PP	137	BT_ASSERT(sfs_path);
cbb9e0b1 PP	138	diag("sfs plugin test below");
cbb9e0b1 PP	139
9736d991	140	status = bt_plugin_find_all_from_file(sfs_path, BT_FALSE, &plugin_set);
d24d5663 PP	141	BT_ASSERT(status == BT_PLUGIN_FIND_ALL_FROM_FILE_STATUS_OK &&
d24d5663 PP	142	plugin_set && bt_plugin_set_get_plugin_count(plugin_set) == 1);
92fed4e1	143	plugin = bt_plugin_set_borrow_plugin_by_index_const(plugin_set, 0);
458e8e1d	144	ok(bt_plugin_get_version(plugin, &major, &minor, &patch, &extra) ==
d94d92ac	145	BT_PROPERTY_AVAILABILITY_AVAILABLE,
458e8e1d PP	146	"bt_plugin_get_version() succeeds when there's a version");
	147	ok(major == 1,
	148	"bt_plugin_get_version() returns the expected major version");
	149	ok(minor == 2,
	150	"bt_plugin_get_version() returns the expected minor version");
	151	ok(patch == 3,
	152	"bt_plugin_get_version() returns the expected patch version");
	153	ok(strcmp(extra, "yes") == 0,
	154	"bt_plugin_get_version() returns the expected extra version");
d94d92ac PP	155	ok(bt_plugin_get_source_component_class_count(plugin) == 1,
	156	"bt_plugin_get_source_component_class_count() returns the expected value");
	157	ok(bt_plugin_get_filter_component_class_count(plugin) == 1,
	158	"bt_plugin_get_filter_component_class_count() returns the expected value");
	159	ok(bt_plugin_get_sink_component_class_count(plugin) == 1,
	160	"bt_plugin_get_sink_component_class_count() returns the expected value");
	161
92fed4e1	162	source_comp_class = bt_plugin_borrow_source_component_class_by_name_const(
d94d92ac	163	plugin, "source");
cbb9e0b1	164	ok(source_comp_class,
92fed4e1	165	"bt_plugin_borrow_source_component_class_by_name_const() finds a source component class");
cbb9e0b1	166
92fed4e1	167	sink_comp_class = bt_plugin_borrow_sink_component_class_by_name_const(
d94d92ac	168	plugin, "sink");
cbb9e0b1	169	ok(sink_comp_class,
92fed4e1	170	"bt_plugin_borrow_sink_component_class_by_name_const() finds a sink component class");
0d72b8c3 PP	171	ok(strcmp(bt_component_class_get_help(bt_component_class_sink_as_component_class_const(sink_comp_class)),
	172	"Bacon ipsum dolor amet strip steak cupim pastrami venison shoulder.\n"
	173	"Prosciutto beef ribs flank meatloaf pancetta brisket kielbasa drumstick\n"
	174	"venison tenderloin cow tail. Beef short loin shoulder meatball, sirloin\n"
	175	"ground round brisket salami cupim pork bresaola turkey bacon boudin.\n") == 0,
a889b89f PP	176	"bt_component_class_get_help() returns the expected help text");
a889b89f PP	177
92fed4e1	178	filter_comp_class = bt_plugin_borrow_filter_component_class_by_name_const(
d94d92ac	179	plugin, "filter");
cbb9e0b1	180	ok(filter_comp_class,
92fed4e1	181	"bt_plugin_borrow_filter_component_class_by_name_const() finds a filter component class");
fdd3a2da	182	params = bt_value_signed_integer_create_init(23);
25583cd0	183	BT_ASSERT(params);
0d72b8c3 PP	184	ret = bt_query_executor_query(query_exec,
0d72b8c3 PP	185	bt_component_class_filter_as_component_class_const(filter_comp_class),
f4e38e70	186	"get-something", params, BT_LOGGING_LEVEL_NONE, &results);
0d72b8c3	187	ok(ret == 0 && results, "bt_query_executor_query() succeeds");
07208d85	188	BT_ASSERT(bt_value_is_array(results) && bt_value_array_get_size(results) == 2);
05e21286	189	object = bt_value_array_borrow_element_by_index_const(results, 0);
25583cd0	190	BT_ASSERT(object && bt_value_is_string(object));
601b0d3c	191	object_str = bt_value_string_get(object);
a67681c1 PP	192	ok(strcmp(object_str, "get-something") == 0,
a67681c1 PP	193	"bt_component_class_query() receives the expected object name");
05e21286 PP	194	res_params = bt_value_array_borrow_element_by_index_const(results, 1);
05e21286 PP	195	ok(bt_value_compare(res_params, params),
a67681c1	196	"bt_component_class_query() receives the expected parameters");
cbb9e0b1	197
c5b9b441	198	bt_component_class_sink_get_ref(sink_comp_class);
d94d92ac	199	diag("> putting the plugin set object here");
c5b9b441	200	BT_PLUGIN_SET_PUT_REF_AND_RESET(plugin_set);
0d72b8c3	201	graph = bt_graph_create();
25583cd0	202	BT_ASSERT(graph);
0d72b8c3	203	graph_ret = bt_graph_add_sink_component(graph, sink_comp_class,
e874da19	204	"the-sink", NULL, BT_LOGGING_LEVEL_NONE, &sink_component);
d24d5663	205	ok(graph_ret == BT_GRAPH_ADD_COMPONENT_STATUS_OK && sink_component,
0d72b8c3	206	"bt_graph_add_sink_component() still works after the plugin object is destroyed");
c5b9b441 PP	207	BT_COMPONENT_SINK_PUT_REF_AND_RESET(sink_component);
c5b9b441 PP	208	bt_graph_put_ref(graph);
cbb9e0b1 PP	209
cbb9e0b1 PP	210	free(sfs_path);
c5b9b441 PP	211	bt_component_class_sink_put_ref(sink_comp_class);
	212	bt_value_put_ref(results);
	213	bt_value_put_ref(params);
	214	bt_query_executor_put_ref(query_exec);
cbb9e0b1 PP	215	}
	216
	217	static void test_create_all_from_dir(const char *plugin_dir)
	218	{
b19ff26f	219	const bt_plugin_set *plugin_set;
d24d5663	220	bt_plugin_find_all_from_dir_status status;
cbb9e0b1 PP	221
	222	diag("create from all test below");
	223
9736d991 PP	224	status = bt_plugin_find_all_from_dir(NON_EXISTING_PATH, BT_FALSE,
9736d991 PP	225	BT_FALSE, &plugin_set);
d24d5663	226	ok(status == BT_PLUGIN_FIND_ALL_FROM_DIR_STATUS_ERROR,
c8db3219	227	"bt_plugin_find_all_from_dir() fails with an invalid path");
cbb9e0b1	228
9736d991 PP	229	plugin_set = NULL;
	230	status = bt_plugin_find_all_from_dir(plugin_dir, BT_FALSE, BT_FALSE,
	231	&plugin_set);
d24d5663	232	ok(status == BT_PLUGIN_FIND_ALL_FROM_DIR_STATUS_OK,
9736d991 PP	233	"bt_plugin_find_all_from_dir() succeeds with a valid path");
	234	ok(plugin_set,
	235	"bt_plugin_find_all_from_dir() returns a plugin set with a valid path");
cbb9e0b1 PP	236
cbb9e0b1 PP	237	/* 2 or 4, if `.la` files are considered or not */
a8ff38ef PP	238	ok(bt_plugin_set_get_plugin_count(plugin_set) == 2 \|\|
a8ff38ef PP	239	bt_plugin_set_get_plugin_count(plugin_set) == 4,
c8db3219	240	"bt_plugin_find_all_from_dir() returns the expected number of plugin objects");
cbb9e0b1	241
c5b9b441	242	bt_plugin_set_put_ref(plugin_set);
cbb9e0b1 PP	243	}
cbb9e0b1 PP	244
2b43acf9	245	static void test_find(const char *plugin_dir)
a8b3f23b	246	{
8a6001c5	247	int ret;
b19ff26f	248	const bt_plugin *plugin;
a8b3f23b	249	char *plugin_path;
d24d5663	250	bt_plugin_find_status status;
a8b3f23b	251
9736d991	252	ok(bt_plugin_find(NON_EXISTING_PATH, BT_FALSE, &plugin) ==
d24d5663	253	BT_PLUGIN_FIND_STATUS_NOT_FOUND,
9736d991	254	"bt_plugin_find() returns BT_PLUGIN_STATUS_NOT_FOUND with an unknown plugin name");
84095ea4 MJ	255	ret = asprintf(&plugin_path, "%s" G_SEARCHPATH_SEPARATOR_S
	256	G_DIR_SEPARATOR_S "ec1d09e5-696c-442e-b1c3-f9c6cf7f5958"
	257	G_SEARCHPATH_SEPARATOR_S G_SEARCHPATH_SEPARATOR_S
	258	G_SEARCHPATH_SEPARATOR_S "%s" G_SEARCHPATH_SEPARATOR_S
	259	"8db46494-a398-466a-9649-c765ae077629"
	260	G_SEARCHPATH_SEPARATOR_S,
a8b3f23b	261	NON_EXISTING_PATH, plugin_dir);
25583cd0	262	BT_ASSERT(ret > 0 && plugin_path);
aacfaf40	263	g_setenv("BABELTRACE_PLUGIN_PATH", plugin_path, 1);
9736d991 PP	264	plugin = NULL;
9736d991 PP	265	status = bt_plugin_find("test_minimal", BT_FALSE, &plugin);
d24d5663	266	ok(status == BT_PLUGIN_FIND_STATUS_OK,
2b43acf9	267	"bt_plugin_find() succeeds with a plugin name it can find");
9736d991	268	ok(plugin, "bt_plugin_find() returns a plugin object");
a8b3f23b	269	ok(strcmp(bt_plugin_get_author(plugin), "Janine Sutto") == 0,
2b43acf9	270	"bt_plugin_find() finds the correct plugin for a given name");
c5b9b441	271	BT_PLUGIN_PUT_REF_AND_RESET(plugin);
a8b3f23b PP	272	free(plugin_path);
	273	}
	274
cbb9e0b1 PP	275	int main(int argc, char **argv)
	276	{
	277	int ret;
	278	const char *plugin_dir;
	279
	280	if (argc != 2) {
	281	puts("Usage: test_plugin plugin_directory");
	282	ret = 1;
	283	goto end;
	284	}
	285
	286	plugin_dir = argv[1];
	287	plan_tests(NR_TESTS);
cbb9e0b1 PP	288	test_minimal(plugin_dir);
	289	test_sfs(plugin_dir);
	290	test_create_all_from_dir(plugin_dir);
2b43acf9	291	test_find(plugin_dir);
cbb9e0b1 PP	292	ret = exit_status();
	293	end:
	294	return ret;
	295	}