view tests/src/libirccd/command-plugin-reload/main.cpp @ 777:1b17b2af5a6b

Plugin links: add query and fix use-after-free
author David Demelier <markand@malikania.fr>
date Wed, 31 Oct 2018 13:43:25 +0100
parents 8c44bbcbbab9
children
line wrap: on
line source

/*
 * main.cpp -- test plugin-reload remote command
 *
 * Copyright (c) 2013-2018 David Demelier <markand@malikania.fr>
 *
 * Permission to use, copy, modify, and/or distribute this software for any
 * purpose with or without fee is hereby granted, provided that the above
 * copyright notice and this permission notice appear in all copies.
 *
 * THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
 * WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
 * MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
 * ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
 * WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
 * ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
 * OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
 */

#define BOOST_TEST_MODULE "plugin-reload"
#include <boost/test/unit_test.hpp>

#include <irccd/test/command_fixture.hpp>

using namespace irccd::test;

namespace irccd {

namespace {

class broken_plugin : public plugin {
public:
	broken_plugin()
		: plugin("broken")
	{
	}

	auto get_name() const noexcept -> std::string_view override
	{
		return "broken";
	}

	void handle_reload(irccd&) override
	{
		throw std::runtime_error("broken");
	}
};

class plugin_reload_fixture : public command_fixture {
protected:
	std::shared_ptr<mock_plugin> plugin_;

	plugin_reload_fixture()
		: plugin_(std::make_shared<mock_plugin>("test"))
	{
		irccd_.plugins().clear();
		irccd_.plugins().add(plugin_);
		irccd_.plugins().add(std::make_unique<broken_plugin>());
	}
};

BOOST_FIXTURE_TEST_SUITE(plugin_reload_fixture_suite, plugin_reload_fixture)

BOOST_AUTO_TEST_CASE(basic)
{
	const auto [json, code] = request({
		{ "command",    "plugin-reload" },
		{ "plugin",     "test"          }
	});

	BOOST_TEST(!code);
	BOOST_TEST(plugin_->find("handle_reload").size() == 1U);
}

BOOST_AUTO_TEST_SUITE(errors)

BOOST_AUTO_TEST_CASE(invalid_identifier)
{
	const auto [json, code] = request({
		{ "command", "plugin-reload" }
	});

	BOOST_TEST(code == plugin_error::invalid_identifier);
	BOOST_TEST(json["error"].get<int>() == plugin_error::invalid_identifier);
	BOOST_TEST(json["errorCategory"].get<std::string>() == "plugin");
}

BOOST_AUTO_TEST_CASE(not_found)
{
	const auto [json, code] = request({
		{ "command",    "plugin-reload" },
		{ "plugin",     "unknown"       }
	});

	BOOST_TEST(code == plugin_error::not_found);
	BOOST_TEST(json["error"].get<int>() == plugin_error::not_found);
	BOOST_TEST(json["errorCategory"].get<std::string>() == "plugin");
}

BOOST_AUTO_TEST_CASE(exec_error)
{
	const auto [json, code] = request({
		{ "command",    "plugin-reload" },
		{ "plugin",     "broken"        }
	});

	BOOST_TEST(code == plugin_error::exec_error);
	BOOST_TEST(json["error"].get<int>() == plugin_error::exec_error);
	BOOST_TEST(json["errorCategory"].get<std::string>() == "plugin");
}

BOOST_AUTO_TEST_SUITE_END()

BOOST_AUTO_TEST_SUITE_END()

} // !namespace

} // !irccd