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

/*
 * test_plugin.c
 *
 * CTF IR Reference Count test
 *
 * 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 <babeltrace/plugin/plugin.h>
#include <babeltrace/ref.h>
#include <babeltrace/values.h>
#include <babeltrace/graph/component.h>
#include <babeltrace/graph/graph.h>
#include <stdlib.h>
#include <string.h>
#include <stdio.h>
#include <assert.h>
#include <glib.h>
#include "tap/tap.h"
#include "common.h"

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

/* Those symbols are written to by some test plugins */
int test_plugin_init_called;
int test_plugin_exit_called;

static void reset_test_plugin_symbols(void)
{
	test_plugin_init_called = 0;
	test_plugin_exit_called = 0;
}

static char *get_test_plugin_path(const char *plugin_dir,
		const char *plugin_name)
{
	GString *path = g_string_new(plugin_dir);
	char *ret;

	assert(path);
	g_string_append_printf(path, "/plugin-%s.so", plugin_name);
	ret = path->str;
	g_string_free(path, FALSE);
	return ret;
}

static void test_invalid(const char *plugin_dir)
{
	struct bt_plugin_set *plugin_set;

	plugin_set = bt_plugin_create_all_from_file(NON_EXISTING_PATH);
	ok(!plugin_set, "bt_plugin_create_all_from_file() fails with a non-existing file");

	plugin_set = bt_plugin_create_all_from_file(plugin_dir);
	ok(!plugin_set, "bt_plugin_create_all_from_file() fails with a directory");

	ok(!bt_plugin_create_all_from_file(NULL),
		"bt_plugin_create_all_from_file() handles NULL correctly");
	ok(!bt_plugin_create_all_from_dir(NULL, BT_FALSE),
		"bt_plugin_create_all_from_dir() handles NULL correctly");
	ok(!bt_plugin_get_name(NULL),
		"bt_plugin_get_name() handles NULL correctly");
	ok(!bt_plugin_get_description(NULL),
		"bt_plugin_get_description() handles NULL correctly");
	ok(bt_plugin_get_version(NULL, NULL, NULL, NULL, NULL) !=
		BT_PLUGIN_STATUS_OK,
		"bt_plugin_get_version() handles NULL correctly");
	ok(!bt_plugin_get_author(NULL),
		"bt_plugin_get_author() handles NULL correctly");
	ok(!bt_plugin_get_license(NULL),
		"bt_plugin_get_license() handles NULL correctly");
	ok(!bt_plugin_get_path(NULL),
		"bt_plugin_get_path() handles NULL correctly");
	ok(bt_plugin_get_component_class_count(NULL) < 0,
		"bt_plugin_get_component_class_count() handles NULL correctly");
	ok(!bt_plugin_get_component_class_by_index(NULL, 0),
		"bt_plugin_get_component_class_by_index() handles NULL correctly");
	ok(!bt_plugin_get_component_class_by_name_and_type(NULL, NULL, 0),
		"bt_plugin_get_component_class_by_name_and_type() handles NULL correctly");
}

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

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

	reset_test_plugin_symbols();
	plugin_set = bt_plugin_create_all_from_file(minimal_path);
	ok(plugin_set && bt_plugin_set_get_plugin_count(plugin_set) == 1,
		"bt_plugin_create_all_from_file() succeeds with a valid file");
	ok(test_plugin_init_called, "plugin's initialization function is called during bt_plugin_create_all_from_file()");
	ok(bt_plugin_set_get_plugin_count(plugin_set) == 1,
		"bt_plugin_create_all_from_file() returns the expected number of plugins");
	plugin = bt_plugin_set_get_plugin(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_PLUGIN_STATUS_OK,
		"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_component_class_count(plugin) == 0,
		"bt_plugin_get_component_class_count() returns the expected value");
	bt_put(plugin);
	bt_put(plugin_set);
	ok(test_plugin_exit_called, "plugin's exit function is called when the plugin is destroyed");

	free(minimal_path);
}

static void test_sfs(const char *plugin_dir)
{
	struct bt_plugin_set *plugin_set;
	struct bt_plugin *plugin;
	struct bt_component_class *sink_comp_class;
	struct bt_component_class *source_comp_class;
	struct bt_component_class *filter_comp_class;
	struct bt_component *sink_component;
	char *sfs_path = get_test_plugin_path(plugin_dir, "sfs");
	unsigned int major, minor, patch;
	const char *extra;
	struct bt_value *params;
	struct bt_value *results;
	struct bt_value *object;
	struct bt_value *res_params;
	struct bt_graph *graph;
	const char *object_str;
	int ret;

	assert(sfs_path);
	diag("sfs plugin test below");

	plugin_set = bt_plugin_create_all_from_file(sfs_path);
	assert(plugin_set && bt_plugin_set_get_plugin_count(plugin_set) == 1);
	plugin = bt_plugin_set_get_plugin(plugin_set, 0);
	ok(bt_plugin_get_version(plugin, &major, &minor, &patch, &extra) ==
		BT_PLUGIN_STATUS_OK,
		"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_component_class_count(plugin) == 3,
		"bt_plugin_get_component_class_count() returns the expected value");

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

	sink_comp_class = bt_plugin_get_component_class_by_name_and_type(
		plugin, "sink", BT_COMPONENT_CLASS_TYPE_SINK);
	ok(sink_comp_class,
		"bt_plugin_get_component_class_by_name_and_type() finds a sink component class");
	ok(strcmp(bt_component_class_get_help(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_get_component_class_by_name_and_type(
		plugin, "filter", BT_COMPONENT_CLASS_TYPE_FILTER);
	ok(filter_comp_class,
		"bt_plugin_get_component_class_by_name_and_type() finds a filter component class");
	ok(!bt_plugin_get_component_class_by_name_and_type(plugin, "filter",
		BT_COMPONENT_CLASS_TYPE_SOURCE),
		"bt_plugin_get_component_class_by_name_and_type() does not find a component class given the wrong type");
	params = bt_value_integer_create_init(23);
	assert(params);
	ok (!bt_component_class_query(NULL, "get-something", params),
		"bt_component_class_query() handles NULL (component class)");
	ok (!bt_component_class_query(filter_comp_class, NULL, params),
		"bt_component_class_query() handles NULL (object)");
	ok (!bt_component_class_query(filter_comp_class, "get-something", NULL),
		"bt_component_class_query() handles NULL (parameters)");
	results = bt_component_class_query(filter_comp_class,
		"get-something", params);
	ok(results, "bt_component_class_query() succeeds");
	assert(bt_value_is_array(results) && bt_value_array_size(results) == 2);
	object = bt_value_array_get(results, 0);
	assert(object && bt_value_is_string(object));
	ret = bt_value_string_get(object, &object_str);
	assert(ret == 0);
	ok(strcmp(object_str, "get-something") == 0,
		"bt_component_class_query() receives the expected object name");
	res_params = bt_value_array_get(results, 1);
	ok(res_params == params,
		"bt_component_class_query() receives the expected parameters");

	diag("> putting the plugin object here");
	BT_PUT(plugin);
	graph = bt_graph_create();
	assert(graph);
	ret = bt_graph_add_component(graph, sink_comp_class, "the-sink", NULL,
		&sink_component);
	ok(ret == 0 && sink_component,
		"bt_graph_add_component() still works after the plugin object is destroyed");
	BT_PUT(sink_component);
	BT_PUT(source_comp_class);
	bt_put(graph);
	graph = bt_graph_create();
	assert(graph);
	ret = bt_graph_add_component(graph, sink_comp_class, "the-sink", NULL,
		&sink_component);
	ok(ret == 0 && sink_component,
		"bt_graph_add_component() still works after the source component class object is destroyed");
	BT_PUT(sink_component);
	BT_PUT(filter_comp_class);
	bt_put(graph);
	graph = bt_graph_create();
	assert(graph);
	ret = bt_graph_add_component(graph, sink_comp_class, "the-sink", NULL,
		&sink_component);
	ok(ret == 0 && sink_component,
		"bt_graph_add_component() still works after the filter component class object is destroyed");
	BT_PUT(sink_comp_class);
	BT_PUT(sink_component);

	free(sfs_path);
	bt_put(graph);
	bt_put(plugin_set);
	bt_put(object);
	bt_put(res_params);
	bt_put(results);
	bt_put(params);
}

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

	diag("create from all test below");

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

	plugin_set = bt_plugin_create_all_from_dir(plugin_dir, BT_FALSE);
	ok(plugin_set, "bt_plugin_create_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_create_all_from_dir() returns the expected number of plugin objects");

	bt_put(plugin_set);
}

static void test_find(const char *plugin_dir)
{
	int ret;
	struct bt_plugin *plugin;
	struct bt_component_class *comp_cls_sink;
	struct bt_component_class *comp_cls_source;
	char *plugin_path;

	ok(!bt_plugin_find(NULL),
		"bt_plugin_find() handles NULL");
	ok(!bt_plugin_find(NON_EXISTING_PATH),
		"bt_plugin_find() returns NULL with an unknown plugin name");
	ret = asprintf(&plugin_path, "%s:/ec1d09e5-696c-442e-b1c3-f9c6cf7f5958:::%s:8db46494-a398-466a-9649-c765ae077629:",
		NON_EXISTING_PATH, plugin_dir);
	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_PUT(plugin);
	comp_cls_sink = bt_plugin_find_component_class(NULL, "sink",
		BT_COMPONENT_CLASS_TYPE_SINK);
	ok(!comp_cls_sink, "bt_plugin_find_component_class() handles NULL (plugin name)");
	comp_cls_sink = bt_plugin_find_component_class("test_sfs", NULL,
		BT_COMPONENT_CLASS_TYPE_SINK);
	ok(!comp_cls_sink, "bt_plugin_find_component_class() handles NULL (component class name)");
	comp_cls_sink = bt_plugin_find_component_class("test_sfs", "sink2",
		BT_COMPONENT_CLASS_TYPE_SINK);
	ok(!comp_cls_sink, "bt_plugin_find_component_class() fails with an unknown component class name");
	comp_cls_sink = bt_plugin_find_component_class("test_sfs", "sink",
		BT_COMPONENT_CLASS_TYPE_SINK);
	ok(comp_cls_sink, "bt_plugin_find_component_class() succeeds with valid parameters");
	ok(strcmp(bt_component_class_get_name(comp_cls_sink), "sink") == 0,
		"bt_plugin_find_component_class() returns the appropriate component class (sink)");
	comp_cls_source = bt_plugin_find_component_class("test_sfs", "source",
		BT_COMPONENT_CLASS_TYPE_SOURCE);
	ok(comp_cls_sink, "bt_plugin_find_component_class() succeeds with another component class name (same plugin)");
	ok(strcmp(bt_component_class_get_name(comp_cls_source), "source") == 0,
		"bt_plugin_find_component_class() returns the appropriate component class (source)");
	BT_PUT(comp_cls_sink);
	BT_PUT(comp_cls_source);
	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_invalid(plugin_dir);
	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 PP	1	/*
	2	* test_plugin.c
	3	*
	4	* CTF IR Reference Count test
	5	*
	6	* Copyright (c) 2017 Philippe Proulx <pproulx@efficios.com>
	7	*
	8	* This program is free software; you can redistribute it and/or modify
	9	* it under the terms of the GNU General Public License as published by
	10	* the Free Software Foundation; under version 2 of the License.
	11	*
	12	* This program is distributed in the hope that it will be useful,
	13	* but WITHOUT ANY WARRANTY; without even the implied warranty of
	14	* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
	15	* GNU General Public License for more details.
	16	*
	17	* You should have received a copy of the GNU General Public License along
	18	* with this program; if not, write to the Free Software Foundation, Inc.,
	19	* 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
	20	*/
	21
	22	#include <babeltrace/plugin/plugin.h>
	23	#include <babeltrace/ref.h>
	24	#include <babeltrace/values.h>
cc469c42	25	#include <babeltrace/graph/component.h>
36712f1d	26	#include <babeltrace/graph/graph.h>
cbb9e0b1 PP	27	#include <stdlib.h>
	28	#include <string.h>
	29	#include <stdio.h>
	30	#include <assert.h>
	31	#include <glib.h>
	32	#include "tap/tap.h"
	33	#include "common.h"
	34
2b43acf9	35	#define NR_TESTS 58
cbb9e0b1 PP	36	#define NON_EXISTING_PATH "/this/hopefully/does/not/exist/5bc75f8d-0dba-4043-a509-d7984b97e42b.so"
	37
	38	/* Those symbols are written to by some test plugins */
	39	int test_plugin_init_called;
	40	int test_plugin_exit_called;
	41
	42	static void reset_test_plugin_symbols(void)
	43	{
	44	test_plugin_init_called = 0;
	45	test_plugin_exit_called = 0;
	46	}
	47
	48	static char get_test_plugin_path(const char plugin_dir,
	49	const char *plugin_name)
	50	{
	51	GString *path = g_string_new(plugin_dir);
	52	char *ret;
	53
	54	assert(path);
	55	g_string_append_printf(path, "/plugin-%s.so", plugin_name);
	56	ret = path->str;
	57	g_string_free(path, FALSE);
	58	return ret;
	59	}
	60
	61	static void test_invalid(const char *plugin_dir)
	62	{
a8ff38ef	63	struct bt_plugin_set *plugin_set;
cbb9e0b1	64
a8ff38ef PP	65	plugin_set = bt_plugin_create_all_from_file(NON_EXISTING_PATH);
a8ff38ef PP	66	ok(!plugin_set, "bt_plugin_create_all_from_file() fails with a non-existing file");
cbb9e0b1	67
a8ff38ef PP	68	plugin_set = bt_plugin_create_all_from_file(plugin_dir);
a8ff38ef PP	69	ok(!plugin_set, "bt_plugin_create_all_from_file() fails with a directory");
cbb9e0b1	70
6ba0b073 PP	71	ok(!bt_plugin_create_all_from_file(NULL),
6ba0b073 PP	72	"bt_plugin_create_all_from_file() handles NULL correctly");
c55a9f58	73	ok(!bt_plugin_create_all_from_dir(NULL, BT_FALSE),
cbb9e0b1 PP	74	"bt_plugin_create_all_from_dir() handles NULL correctly");
	75	ok(!bt_plugin_get_name(NULL),
	76	"bt_plugin_get_name() handles NULL correctly");
	77	ok(!bt_plugin_get_description(NULL),
	78	"bt_plugin_get_description() handles NULL correctly");
458e8e1d PP	79	ok(bt_plugin_get_version(NULL, NULL, NULL, NULL, NULL) !=
	80	BT_PLUGIN_STATUS_OK,
	81	"bt_plugin_get_version() handles NULL correctly");
cbb9e0b1 PP	82	ok(!bt_plugin_get_author(NULL),
	83	"bt_plugin_get_author() handles NULL correctly");
	84	ok(!bt_plugin_get_license(NULL),
	85	"bt_plugin_get_license() handles NULL correctly");
	86	ok(!bt_plugin_get_path(NULL),
	87	"bt_plugin_get_path() handles NULL correctly");
	88	ok(bt_plugin_get_component_class_count(NULL) < 0,
	89	"bt_plugin_get_component_class_count() handles NULL correctly");
9ac68eb1 PP	90	ok(!bt_plugin_get_component_class_by_index(NULL, 0),
9ac68eb1 PP	91	"bt_plugin_get_component_class_by_index() handles NULL correctly");
cbb9e0b1 PP	92	ok(!bt_plugin_get_component_class_by_name_and_type(NULL, NULL, 0),
cbb9e0b1 PP	93	"bt_plugin_get_component_class_by_name_and_type() handles NULL correctly");
cbb9e0b1 PP	94	}
	95
	96	static void test_minimal(const char *plugin_dir)
	97	{
a8ff38ef	98	struct bt_plugin_set *plugin_set;
cbb9e0b1 PP	99	struct bt_plugin *plugin;
	100	char *minimal_path = get_test_plugin_path(plugin_dir, "minimal");
	101
	102	assert(minimal_path);
	103	diag("minimal plugin test below");
	104
	105	reset_test_plugin_symbols();
a8ff38ef PP	106	plugin_set = bt_plugin_create_all_from_file(minimal_path);
	107	ok(plugin_set && bt_plugin_set_get_plugin_count(plugin_set) == 1,
	108	"bt_plugin_create_all_from_file() succeeds with a valid file");
6ba0b073	109	ok(test_plugin_init_called, "plugin's initialization function is called during bt_plugin_create_all_from_file()");
a8ff38ef	110	ok(bt_plugin_set_get_plugin_count(plugin_set) == 1,
6ba0b073	111	"bt_plugin_create_all_from_file() returns the expected number of plugins");
a8ff38ef	112	plugin = bt_plugin_set_get_plugin(plugin_set, 0);
6ba0b073	113	ok(strcmp(bt_plugin_get_name(plugin), "test_minimal") == 0,
cbb9e0b1 PP	114	"bt_plugin_get_name() returns the expected name");
	115	ok(strcmp(bt_plugin_get_description(plugin),
	116	"Minimal Babeltrace plugin with no component classes") == 0,
	117	"bt_plugin_get_description() returns the expected description");
458e8e1d PP	118	ok(bt_plugin_get_version(plugin, NULL, NULL, NULL, NULL) !=
	119	BT_PLUGIN_STATUS_OK,
	120	"bt_plugin_get_version() fails when there's no version");
cbb9e0b1 PP	121	ok(strcmp(bt_plugin_get_author(plugin), "Janine Sutto") == 0,
	122	"bt_plugin_get_author() returns the expected author");
	123	ok(strcmp(bt_plugin_get_license(plugin), "Beerware") == 0,
	124	"bt_plugin_get_license() returns the expected license");
	125	ok(strcmp(bt_plugin_get_path(plugin), minimal_path) == 0,
	126	"bt_plugin_get_path() returns the expected path");
	127	ok(bt_plugin_get_component_class_count(plugin) == 0,
	128	"bt_plugin_get_component_class_count() returns the expected value");
a8ff38ef PP	129	bt_put(plugin);
a8ff38ef PP	130	bt_put(plugin_set);
cbb9e0b1 PP	131	ok(test_plugin_exit_called, "plugin's exit function is called when the plugin is destroyed");
	132
	133	free(minimal_path);
	134	}
	135
	136	static void test_sfs(const char *plugin_dir)
	137	{
a8ff38ef	138	struct bt_plugin_set *plugin_set;
cbb9e0b1 PP	139	struct bt_plugin *plugin;
	140	struct bt_component_class *sink_comp_class;
	141	struct bt_component_class *source_comp_class;
	142	struct bt_component_class *filter_comp_class;
	143	struct bt_component *sink_component;
	144	char *sfs_path = get_test_plugin_path(plugin_dir, "sfs");
458e8e1d PP	145	unsigned int major, minor, patch;
458e8e1d PP	146	const char *extra;
5933c0f2 PP	147	struct bt_value *params;
5933c0f2 PP	148	struct bt_value *results;
a67681c1	149	struct bt_value *object;
5933c0f2	150	struct bt_value *res_params;
36712f1d	151	struct bt_graph *graph;
a67681c1	152	const char *object_str;
5933c0f2	153	int ret;
cbb9e0b1 PP	154
	155	assert(sfs_path);
	156	diag("sfs plugin test below");
	157
a8ff38ef PP	158	plugin_set = bt_plugin_create_all_from_file(sfs_path);
	159	assert(plugin_set && bt_plugin_set_get_plugin_count(plugin_set) == 1);
	160	plugin = bt_plugin_set_get_plugin(plugin_set, 0);
458e8e1d PP	161	ok(bt_plugin_get_version(plugin, &major, &minor, &patch, &extra) ==
	162	BT_PLUGIN_STATUS_OK,
	163	"bt_plugin_get_version() succeeds when there's a version");
	164	ok(major == 1,
	165	"bt_plugin_get_version() returns the expected major version");
	166	ok(minor == 2,
	167	"bt_plugin_get_version() returns the expected minor version");
	168	ok(patch == 3,
	169	"bt_plugin_get_version() returns the expected patch version");
	170	ok(strcmp(extra, "yes") == 0,
	171	"bt_plugin_get_version() returns the expected extra version");
cbb9e0b1 PP	172	ok(bt_plugin_get_component_class_count(plugin) == 3,
	173	"bt_plugin_get_component_class_count() returns the expected value");
	174
	175	source_comp_class = bt_plugin_get_component_class_by_name_and_type(
d3e4dcd8	176	plugin, "source", BT_COMPONENT_CLASS_TYPE_SOURCE);
cbb9e0b1 PP	177	ok(source_comp_class,
	178	"bt_plugin_get_component_class_by_name_and_type() finds a source component class");
	179
	180	sink_comp_class = bt_plugin_get_component_class_by_name_and_type(
d3e4dcd8	181	plugin, "sink", BT_COMPONENT_CLASS_TYPE_SINK);
cbb9e0b1 PP	182	ok(sink_comp_class,
cbb9e0b1 PP	183	"bt_plugin_get_component_class_by_name_and_type() finds a sink component class");
a889b89f PP	184	ok(strcmp(bt_component_class_get_help(sink_comp_class),
	185	"Bacon ipsum dolor amet strip steak cupim pastrami venison shoulder.\n"
	186	"Prosciutto beef ribs flank meatloaf pancetta brisket kielbasa drumstick\n"
	187	"venison tenderloin cow tail. Beef short loin shoulder meatball, sirloin\n"
	188	"ground round brisket salami cupim pork bresaola turkey bacon boudin.\n") == 0,
	189	"bt_component_class_get_help() returns the expected help text");
	190
cbb9e0b1	191	filter_comp_class = bt_plugin_get_component_class_by_name_and_type(
d3e4dcd8	192	plugin, "filter", BT_COMPONENT_CLASS_TYPE_FILTER);
cbb9e0b1 PP	193	ok(filter_comp_class,
	194	"bt_plugin_get_component_class_by_name_and_type() finds a filter component class");
	195	ok(!bt_plugin_get_component_class_by_name_and_type(plugin, "filter",
d3e4dcd8	196	BT_COMPONENT_CLASS_TYPE_SOURCE),
cbb9e0b1	197	"bt_plugin_get_component_class_by_name_and_type() does not find a component class given the wrong type");
5933c0f2 PP	198	params = bt_value_integer_create_init(23);
5933c0f2 PP	199	assert(params);
a67681c1 PP	200	ok (!bt_component_class_query(NULL, "get-something", params),
	201	"bt_component_class_query() handles NULL (component class)");
	202	ok (!bt_component_class_query(filter_comp_class, NULL, params),
	203	"bt_component_class_query() handles NULL (object)");
	204	ok (!bt_component_class_query(filter_comp_class, "get-something", NULL),
	205	"bt_component_class_query() handles NULL (parameters)");
	206	results = bt_component_class_query(filter_comp_class,
5933c0f2	207	"get-something", params);
a67681c1	208	ok(results, "bt_component_class_query() succeeds");
5933c0f2	209	assert(bt_value_is_array(results) && bt_value_array_size(results) == 2);
a67681c1 PP	210	object = bt_value_array_get(results, 0);
	211	assert(object && bt_value_is_string(object));
	212	ret = bt_value_string_get(object, &object_str);
5933c0f2	213	assert(ret == 0);
a67681c1 PP	214	ok(strcmp(object_str, "get-something") == 0,
a67681c1 PP	215	"bt_component_class_query() receives the expected object name");
5933c0f2 PP	216	res_params = bt_value_array_get(results, 1);
5933c0f2 PP	217	ok(res_params == params,
a67681c1	218	"bt_component_class_query() receives the expected parameters");
cbb9e0b1 PP	219
	220	diag("> putting the plugin object here");
	221	BT_PUT(plugin);
36712f1d PP	222	graph = bt_graph_create();
	223	assert(graph);
	224	ret = bt_graph_add_component(graph, sink_comp_class, "the-sink", NULL,
	225	&sink_component);
	226	ok(ret == 0 && sink_component,
	227	"bt_graph_add_component() still works after the plugin object is destroyed");
cbb9e0b1 PP	228	BT_PUT(sink_component);
cbb9e0b1 PP	229	BT_PUT(source_comp_class);
36712f1d PP	230	bt_put(graph);
	231	graph = bt_graph_create();
	232	assert(graph);
	233	ret = bt_graph_add_component(graph, sink_comp_class, "the-sink", NULL,
	234	&sink_component);
	235	ok(ret == 0 && sink_component,
	236	"bt_graph_add_component() still works after the source component class object is destroyed");
cbb9e0b1 PP	237	BT_PUT(sink_component);
cbb9e0b1 PP	238	BT_PUT(filter_comp_class);
36712f1d PP	239	bt_put(graph);
	240	graph = bt_graph_create();
	241	assert(graph);
	242	ret = bt_graph_add_component(graph, sink_comp_class, "the-sink", NULL,
	243	&sink_component);
	244	ok(ret == 0 && sink_component,
	245	"bt_graph_add_component() still works after the filter component class object is destroyed");
cbb9e0b1 PP	246	BT_PUT(sink_comp_class);
	247	BT_PUT(sink_component);
	248
	249	free(sfs_path);
36712f1d	250	bt_put(graph);
a8ff38ef	251	bt_put(plugin_set);
a67681c1	252	bt_put(object);
5933c0f2 PP	253	bt_put(res_params);
	254	bt_put(results);
	255	bt_put(params);
cbb9e0b1 PP	256	}
	257
	258	static void test_create_all_from_dir(const char *plugin_dir)
	259	{
a8ff38ef	260	struct bt_plugin_set *plugin_set;
cbb9e0b1 PP	261
	262	diag("create from all test below");
	263
c55a9f58	264	plugin_set = bt_plugin_create_all_from_dir(NON_EXISTING_PATH, BT_FALSE);
a8ff38ef	265	ok(!plugin_set,
cbb9e0b1 PP	266	"bt_plugin_create_all_from_dir() fails with an invalid path");
cbb9e0b1 PP	267
c55a9f58	268	plugin_set = bt_plugin_create_all_from_dir(plugin_dir, BT_FALSE);
a8ff38ef	269	ok(plugin_set, "bt_plugin_create_all_from_dir() succeeds with a valid path");
cbb9e0b1 PP	270
cbb9e0b1 PP	271	/* 2 or 4, if `.la` files are considered or not */
a8ff38ef PP	272	ok(bt_plugin_set_get_plugin_count(plugin_set) == 2 \|\|
	273	bt_plugin_set_get_plugin_count(plugin_set) == 4,
	274	"bt_plugin_create_all_from_dir() returns the expected number of plugin objects");
cbb9e0b1	275
a8ff38ef	276	bt_put(plugin_set);
cbb9e0b1 PP	277	}
cbb9e0b1 PP	278
2b43acf9	279	static void test_find(const char *plugin_dir)
a8b3f23b	280	{
4a7062f7	281	int ret;
a8b3f23b	282	struct bt_plugin *plugin;
2b43acf9 PP	283	struct bt_component_class *comp_cls_sink;
2b43acf9 PP	284	struct bt_component_class *comp_cls_source;
a8b3f23b PP	285	char *plugin_path;
a8b3f23b PP	286
2b43acf9 PP	287	ok(!bt_plugin_find(NULL),
	288	"bt_plugin_find() handles NULL");
	289	ok(!bt_plugin_find(NON_EXISTING_PATH),
	290	"bt_plugin_find() returns NULL with an unknown plugin name");
4a7062f7	291	ret = asprintf(&plugin_path, "%s:/ec1d09e5-696c-442e-b1c3-f9c6cf7f5958:::%s:8db46494-a398-466a-9649-c765ae077629:",
a8b3f23b	292	NON_EXISTING_PATH, plugin_dir);
4a7062f7	293	assert(ret > 0 && plugin_path);
aacfaf40	294	g_setenv("BABELTRACE_PLUGIN_PATH", plugin_path, 1);
2b43acf9	295	plugin = bt_plugin_find("test_minimal");
a8b3f23b	296	ok(plugin,
2b43acf9	297	"bt_plugin_find() succeeds with a plugin name it can find");
a8b3f23b	298	ok(strcmp(bt_plugin_get_author(plugin), "Janine Sutto") == 0,
2b43acf9	299	"bt_plugin_find() finds the correct plugin for a given name");
a8b3f23b	300	BT_PUT(plugin);
2b43acf9 PP	301	comp_cls_sink = bt_plugin_find_component_class(NULL, "sink",
	302	BT_COMPONENT_CLASS_TYPE_SINK);
	303	ok(!comp_cls_sink, "bt_plugin_find_component_class() handles NULL (plugin name)");
	304	comp_cls_sink = bt_plugin_find_component_class("test_sfs", NULL,
	305	BT_COMPONENT_CLASS_TYPE_SINK);
	306	ok(!comp_cls_sink, "bt_plugin_find_component_class() handles NULL (component class name)");
	307	comp_cls_sink = bt_plugin_find_component_class("test_sfs", "sink2",
	308	BT_COMPONENT_CLASS_TYPE_SINK);
	309	ok(!comp_cls_sink, "bt_plugin_find_component_class() fails with an unknown component class name");
	310	comp_cls_sink = bt_plugin_find_component_class("test_sfs", "sink",
	311	BT_COMPONENT_CLASS_TYPE_SINK);
	312	ok(comp_cls_sink, "bt_plugin_find_component_class() succeeds with valid parameters");
	313	ok(strcmp(bt_component_class_get_name(comp_cls_sink), "sink") == 0,
	314	"bt_plugin_find_component_class() returns the appropriate component class (sink)");
	315	comp_cls_source = bt_plugin_find_component_class("test_sfs", "source",
	316	BT_COMPONENT_CLASS_TYPE_SOURCE);
	317	ok(comp_cls_sink, "bt_plugin_find_component_class() succeeds with another component class name (same plugin)");
	318	ok(strcmp(bt_component_class_get_name(comp_cls_source), "source") == 0,
	319	"bt_plugin_find_component_class() returns the appropriate component class (source)");
	320	BT_PUT(comp_cls_sink);
	321	BT_PUT(comp_cls_source);
a8b3f23b PP	322	free(plugin_path);
	323	}
	324
cbb9e0b1 PP	325	int main(int argc, char **argv)
	326	{
	327	int ret;
	328	const char *plugin_dir;
	329
	330	if (argc != 2) {
	331	puts("Usage: test_plugin plugin_directory");
	332	ret = 1;
	333	goto end;
	334	}
	335
	336	plugin_dir = argv[1];
	337	plan_tests(NR_TESTS);
	338	test_invalid(plugin_dir);
	339	test_minimal(plugin_dir);
	340	test_sfs(plugin_dir);
	341	test_create_all_from_dir(plugin_dir);
2b43acf9	342	test_find(plugin_dir);
cbb9e0b1 PP	343	ret = exit_status();
	344	end:
	345	return ret;
	346	}