[babeltrace.git] / tests / lib / test_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 		35
#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;
	const bt_plugin *plugin;
	char *minimal_path = get_test_plugin_path(plugin_dir, "minimal");

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

	reset_test_plugin_env_vars();
	plugin_set = bt_plugin_find_all_from_file(minimal_path);
	ok(plugin_set && bt_plugin_set_get_plugin_count(plugin_set) == 1,
		"bt_plugin_find_all_from_file() succeeds with a valid file");
	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;
	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_status graph_ret;
	bt_query_executor *query_exec = bt_query_executor_create();
	int ret;

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

	plugin_set = bt_plugin_find_all_from_file(sfs_path);
	BT_ASSERT(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, &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, &sink_component);
	ok(graph_ret == BT_GRAPH_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;

	diag("create from all test below");

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

	plugin_set = bt_plugin_find_all_from_dir(plugin_dir, BT_FALSE);
	ok(plugin_set, "bt_plugin_find_all_from_dir() succeeds 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;

	ok(!bt_plugin_find(NON_EXISTING_PATH),
		"bt_plugin_find() returns NULL 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 = bt_plugin_find("test_minimal");
	ok(plugin,
		"bt_plugin_find() succeeds with a plugin name it can find");
	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
d94d92ac	27	#define NR_TESTS 35
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	{
b19ff26f PP	67	const bt_plugin_set *plugin_set;
b19ff26f PP	68	const bt_plugin *plugin;
cbb9e0b1 PP	69	char *minimal_path = get_test_plugin_path(plugin_dir, "minimal");
cbb9e0b1 PP	70
25583cd0	71	BT_ASSERT(minimal_path);
cbb9e0b1 PP	72	diag("minimal plugin test below");
cbb9e0b1 PP	73
df5b5d01	74	reset_test_plugin_env_vars();
c8db3219	75	plugin_set = bt_plugin_find_all_from_file(minimal_path);
a8ff38ef	76	ok(plugin_set && bt_plugin_set_get_plugin_count(plugin_set) == 1,
c8db3219	77	"bt_plugin_find_all_from_file() succeeds with a valid file");
df5b5d01	78	ok(check_env_var("BT_TEST_PLUGIN_INIT_CALLED") == 1,
c8db3219	79	"plugin's initialization function is called during bt_plugin_find_all_from_file()");
a8ff38ef	80	ok(bt_plugin_set_get_plugin_count(plugin_set) == 1,
c8db3219	81	"bt_plugin_find_all_from_file() returns the expected number of plugins");
92fed4e1	82	plugin = bt_plugin_set_borrow_plugin_by_index_const(plugin_set, 0);
6ba0b073	83	ok(strcmp(bt_plugin_get_name(plugin), "test_minimal") == 0,
cbb9e0b1 PP	84	"bt_plugin_get_name() returns the expected name");
	85	ok(strcmp(bt_plugin_get_description(plugin),
	86	"Minimal Babeltrace plugin with no component classes") == 0,
	87	"bt_plugin_get_description() returns the expected description");
d94d92ac PP	88	ok(bt_plugin_get_version(plugin, NULL, NULL, NULL, NULL) ==
d94d92ac PP	89	BT_PROPERTY_AVAILABILITY_NOT_AVAILABLE,
458e8e1d	90	"bt_plugin_get_version() fails when there's no version");
cbb9e0b1 PP	91	ok(strcmp(bt_plugin_get_author(plugin), "Janine Sutto") == 0,
	92	"bt_plugin_get_author() returns the expected author");
	93	ok(strcmp(bt_plugin_get_license(plugin), "Beerware") == 0,
	94	"bt_plugin_get_license() returns the expected license");
	95	ok(strcmp(bt_plugin_get_path(plugin), minimal_path) == 0,
	96	"bt_plugin_get_path() returns the expected path");
d94d92ac PP	97	ok(bt_plugin_get_source_component_class_count(plugin) == 0,
	98	"bt_plugin_get_source_component_class_count() returns the expected value");
	99	ok(bt_plugin_get_filter_component_class_count(plugin) == 0,
	100	"bt_plugin_get_filter_component_class_count() returns the expected value");
	101	ok(bt_plugin_get_sink_component_class_count(plugin) == 0,
	102	"bt_plugin_get_sink_component_class_count() returns the expected value");
c5b9b441	103	bt_plugin_set_put_ref(plugin_set);
df5b5d01 PP	104	ok(check_env_var("BT_TEST_PLUGIN_EXIT_CALLED") == 1,
df5b5d01 PP	105	"plugin's exit function is called when the plugin is destroyed");
cbb9e0b1 PP	106
	107	free(minimal_path);
	108	}
	109
	110	static void test_sfs(const char *plugin_dir)
	111	{
b19ff26f PP	112	const bt_plugin_set *plugin_set;
	113	const bt_plugin *plugin;
	114	const bt_component_class_sink *sink_comp_class;
	115	const bt_component_class_source *source_comp_class;
	116	const bt_component_class_filter *filter_comp_class;
	117	const bt_component_sink *sink_component;
cbb9e0b1	118	char *sfs_path = get_test_plugin_path(plugin_dir, "sfs");
458e8e1d PP	119	unsigned int major, minor, patch;
458e8e1d PP	120	const char *extra;
b19ff26f PP	121	bt_value *params;
	122	const bt_value *results;
	123	const bt_value *object;
	124	const bt_value *res_params;
	125	bt_graph *graph;
a67681c1	126	const char *object_str;
4cdfc5e8	127	bt_graph_status graph_ret;
b19ff26f	128	bt_query_executor *query_exec = bt_query_executor_create();
c7eee084	129	int ret;
cbb9e0b1	130
25583cd0 PP	131	BT_ASSERT(query_exec);
25583cd0 PP	132	BT_ASSERT(sfs_path);
cbb9e0b1 PP	133	diag("sfs plugin test below");
cbb9e0b1 PP	134
c8db3219	135	plugin_set = bt_plugin_find_all_from_file(sfs_path);
25583cd0	136	BT_ASSERT(plugin_set && bt_plugin_set_get_plugin_count(plugin_set) == 1);
92fed4e1	137	plugin = bt_plugin_set_borrow_plugin_by_index_const(plugin_set, 0);
458e8e1d	138	ok(bt_plugin_get_version(plugin, &major, &minor, &patch, &extra) ==
d94d92ac	139	BT_PROPERTY_AVAILABILITY_AVAILABLE,
458e8e1d PP	140	"bt_plugin_get_version() succeeds when there's a version");
	141	ok(major == 1,
	142	"bt_plugin_get_version() returns the expected major version");
	143	ok(minor == 2,
	144	"bt_plugin_get_version() returns the expected minor version");
	145	ok(patch == 3,
	146	"bt_plugin_get_version() returns the expected patch version");
	147	ok(strcmp(extra, "yes") == 0,
	148	"bt_plugin_get_version() returns the expected extra version");
d94d92ac PP	149	ok(bt_plugin_get_source_component_class_count(plugin) == 1,
	150	"bt_plugin_get_source_component_class_count() returns the expected value");
	151	ok(bt_plugin_get_filter_component_class_count(plugin) == 1,
	152	"bt_plugin_get_filter_component_class_count() returns the expected value");
	153	ok(bt_plugin_get_sink_component_class_count(plugin) == 1,
	154	"bt_plugin_get_sink_component_class_count() returns the expected value");
	155
92fed4e1	156	source_comp_class = bt_plugin_borrow_source_component_class_by_name_const(
d94d92ac	157	plugin, "source");
cbb9e0b1	158	ok(source_comp_class,
92fed4e1	159	"bt_plugin_borrow_source_component_class_by_name_const() finds a source component class");
cbb9e0b1	160
92fed4e1	161	sink_comp_class = bt_plugin_borrow_sink_component_class_by_name_const(
d94d92ac	162	plugin, "sink");
cbb9e0b1	163	ok(sink_comp_class,
92fed4e1	164	"bt_plugin_borrow_sink_component_class_by_name_const() finds a sink component class");
0d72b8c3 PP	165	ok(strcmp(bt_component_class_get_help(bt_component_class_sink_as_component_class_const(sink_comp_class)),
	166	"Bacon ipsum dolor amet strip steak cupim pastrami venison shoulder.\n"
	167	"Prosciutto beef ribs flank meatloaf pancetta brisket kielbasa drumstick\n"
	168	"venison tenderloin cow tail. Beef short loin shoulder meatball, sirloin\n"
	169	"ground round brisket salami cupim pork bresaola turkey bacon boudin.\n") == 0,
a889b89f PP	170	"bt_component_class_get_help() returns the expected help text");
a889b89f PP	171
92fed4e1	172	filter_comp_class = bt_plugin_borrow_filter_component_class_by_name_const(
d94d92ac	173	plugin, "filter");
cbb9e0b1	174	ok(filter_comp_class,
92fed4e1	175	"bt_plugin_borrow_filter_component_class_by_name_const() finds a filter component class");
fdd3a2da	176	params = bt_value_signed_integer_create_init(23);
25583cd0	177	BT_ASSERT(params);
0d72b8c3 PP	178	ret = bt_query_executor_query(query_exec,
0d72b8c3 PP	179	bt_component_class_filter_as_component_class_const(filter_comp_class),
05e21286	180	"get-something", params, &results);
0d72b8c3	181	ok(ret == 0 && results, "bt_query_executor_query() succeeds");
07208d85	182	BT_ASSERT(bt_value_is_array(results) && bt_value_array_get_size(results) == 2);
05e21286	183	object = bt_value_array_borrow_element_by_index_const(results, 0);
25583cd0	184	BT_ASSERT(object && bt_value_is_string(object));
601b0d3c	185	object_str = bt_value_string_get(object);
a67681c1 PP	186	ok(strcmp(object_str, "get-something") == 0,
a67681c1 PP	187	"bt_component_class_query() receives the expected object name");
05e21286 PP	188	res_params = bt_value_array_borrow_element_by_index_const(results, 1);
05e21286 PP	189	ok(bt_value_compare(res_params, params),
a67681c1	190	"bt_component_class_query() receives the expected parameters");
cbb9e0b1	191
c5b9b441	192	bt_component_class_sink_get_ref(sink_comp_class);
d94d92ac	193	diag("> putting the plugin set object here");
c5b9b441	194	BT_PLUGIN_SET_PUT_REF_AND_RESET(plugin_set);
0d72b8c3	195	graph = bt_graph_create();
25583cd0	196	BT_ASSERT(graph);
0d72b8c3 PP	197	graph_ret = bt_graph_add_sink_component(graph, sink_comp_class,
0d72b8c3 PP	198	"the-sink", NULL, &sink_component);
c3c69044	199	ok(graph_ret == BT_GRAPH_STATUS_OK && sink_component,
0d72b8c3	200	"bt_graph_add_sink_component() still works after the plugin object is destroyed");
c5b9b441 PP	201	BT_COMPONENT_SINK_PUT_REF_AND_RESET(sink_component);
c5b9b441 PP	202	bt_graph_put_ref(graph);
cbb9e0b1 PP	203
cbb9e0b1 PP	204	free(sfs_path);
c5b9b441 PP	205	bt_component_class_sink_put_ref(sink_comp_class);
	206	bt_value_put_ref(results);
	207	bt_value_put_ref(params);
	208	bt_query_executor_put_ref(query_exec);
cbb9e0b1 PP	209	}
	210
	211	static void test_create_all_from_dir(const char *plugin_dir)
	212	{
b19ff26f	213	const bt_plugin_set *plugin_set;
cbb9e0b1 PP	214
	215	diag("create from all test below");
	216
c8db3219	217	plugin_set = bt_plugin_find_all_from_dir(NON_EXISTING_PATH, BT_FALSE);
a8ff38ef	218	ok(!plugin_set,
c8db3219	219	"bt_plugin_find_all_from_dir() fails with an invalid path");
cbb9e0b1	220
c8db3219 PP	221	plugin_set = bt_plugin_find_all_from_dir(plugin_dir, BT_FALSE);
c8db3219 PP	222	ok(plugin_set, "bt_plugin_find_all_from_dir() succeeds with a valid path");
cbb9e0b1 PP	223
cbb9e0b1 PP	224	/* 2 or 4, if `.la` files are considered or not */
a8ff38ef PP	225	ok(bt_plugin_set_get_plugin_count(plugin_set) == 2 \|\|
a8ff38ef PP	226	bt_plugin_set_get_plugin_count(plugin_set) == 4,
c8db3219	227	"bt_plugin_find_all_from_dir() returns the expected number of plugin objects");
cbb9e0b1	228
c5b9b441	229	bt_plugin_set_put_ref(plugin_set);
cbb9e0b1 PP	230	}
cbb9e0b1 PP	231
2b43acf9	232	static void test_find(const char *plugin_dir)
a8b3f23b	233	{
8a6001c5	234	int ret;
b19ff26f	235	const bt_plugin *plugin;
a8b3f23b PP	236	char *plugin_path;
a8b3f23b PP	237
2b43acf9 PP	238	ok(!bt_plugin_find(NON_EXISTING_PATH),
2b43acf9 PP	239	"bt_plugin_find() returns NULL with an unknown plugin name");
84095ea4 MJ	240	ret = asprintf(&plugin_path, "%s" G_SEARCHPATH_SEPARATOR_S
	241	G_DIR_SEPARATOR_S "ec1d09e5-696c-442e-b1c3-f9c6cf7f5958"
	242	G_SEARCHPATH_SEPARATOR_S G_SEARCHPATH_SEPARATOR_S
	243	G_SEARCHPATH_SEPARATOR_S "%s" G_SEARCHPATH_SEPARATOR_S
	244	"8db46494-a398-466a-9649-c765ae077629"
	245	G_SEARCHPATH_SEPARATOR_S,
a8b3f23b	246	NON_EXISTING_PATH, plugin_dir);
25583cd0	247	BT_ASSERT(ret > 0 && plugin_path);
aacfaf40	248	g_setenv("BABELTRACE_PLUGIN_PATH", plugin_path, 1);
2b43acf9	249	plugin = bt_plugin_find("test_minimal");
a8b3f23b	250	ok(plugin,
2b43acf9	251	"bt_plugin_find() succeeds with a plugin name it can find");
a8b3f23b	252	ok(strcmp(bt_plugin_get_author(plugin), "Janine Sutto") == 0,
2b43acf9	253	"bt_plugin_find() finds the correct plugin for a given name");
c5b9b441	254	BT_PLUGIN_PUT_REF_AND_RESET(plugin);
a8b3f23b PP	255	free(plugin_path);
	256	}
	257
cbb9e0b1 PP	258	int main(int argc, char **argv)
	259	{
	260	int ret;
	261	const char *plugin_dir;
	262
	263	if (argc != 2) {
	264	puts("Usage: test_plugin plugin_directory");
	265	ret = 1;
	266	goto end;
	267	}
	268
	269	plugin_dir = argv[1];
	270	plan_tests(NR_TESTS);
cbb9e0b1 PP	271	test_minimal(plugin_dir);
	272	test_sfs(plugin_dir);
	273	test_create_all_from_dir(plugin_dir);
2b43acf9	274	test_find(plugin_dir);
cbb9e0b1 PP	275	ret = exit_status();
	276	end:
	277	return ret;
	278	}