Mercurial > irccd
diff tests/cmd-rule-edit/main.cpp @ 453:acb2d4990249
Irccdctl: implement rule-edit
author | David Demelier <markand@malikania.fr> |
---|---|
date | Thu, 20 Jul 2017 22:55:02 +0200 |
parents | |
children | 7e273b7f4f92 |
line wrap: on
line diff
--- /dev/null Thu Jan 01 00:00:00 1970 +0000 +++ b/tests/cmd-rule-edit/main.cpp Thu Jul 20 22:55:02 2017 +0200 @@ -0,0 +1,537 @@ +/* + * main.cpp -- test rule-info remote command + * + * Copyright (c) 2013-2017 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. + */ + +#include <command.hpp> +#include <command-tester.hpp> +#include <service.hpp> + +using namespace irccd; +using namespace irccd::command; + +class RuleEditCommandTest : public CommandTester { +protected: + nlohmann::json m_result; + + /* + * Rule sets are unordered so use this function to search a string in + * the JSON array. + */ + inline bool contains(const nlohmann::json &array, const std::string &str) + { + for (const auto &v : array) + if (v.is_string() && v == str) + return true; + + return false; + } + +public: + RuleEditCommandTest() + : CommandTester(std::make_unique<RuleEditCommand>()) + { + m_irccd.commands().add(std::make_unique<RuleInfoCommand>()); + m_irccd.rules().add(Rule( + { "s1", "s2" }, + { "c1", "c2" }, + { "o1", "o2" }, + { "p1", "p2" }, + { "onMessage", "onCommand" }, + RuleAction::Drop + )); + m_irccdctl.client().onMessage.connect([&] (auto result) { + m_result = result; + }); + } +}; + +TEST_F(RuleEditCommandTest, addServer) +{ + try { + m_irccdctl.client().request({ + { "command", "rule-edit" }, + { "add-servers", { "new-s3" } }, + { "index", 0 } + }); + + poll([&] () { + return m_result.is_object(); + }); + + ASSERT_TRUE(m_result.is_object()); + ASSERT_TRUE(m_result["status"].get<bool>()); + + m_result = nullptr; + m_irccdctl.client().request({ + { "command", "rule-info" }, + { "index", 0 } + }); + + poll([&] () { + return m_result.is_object(); + }); + + ASSERT_TRUE(m_result.is_object()); + ASSERT_TRUE(m_result["status"].get<bool>()); + ASSERT_TRUE(contains(m_result["servers"], "s1")); + ASSERT_TRUE(contains(m_result["servers"], "s2")); + ASSERT_TRUE(contains(m_result["servers"], "new-s3")); + ASSERT_TRUE(contains(m_result["channels"], "c1")); + ASSERT_TRUE(contains(m_result["channels"], "c2")); + ASSERT_TRUE(contains(m_result["plugins"], "p1")); + ASSERT_TRUE(contains(m_result["plugins"], "p2")); + ASSERT_TRUE(contains(m_result["events"], "onMessage")); + ASSERT_TRUE(contains(m_result["events"], "onCommand")); + ASSERT_EQ(m_result["action"].get<std::string>(), "drop"); + } catch (const std::exception &ex) { + FAIL() << ex.what(); + } +} + +TEST_F(RuleEditCommandTest, addChannel) +{ + try { + m_irccdctl.client().request({ + { "command", "rule-edit" }, + { "add-channels", { "new-c3" } }, + { "index", 0 } + }); + + poll([&] () { + return m_result.is_object(); + }); + + ASSERT_TRUE(m_result.is_object()); + ASSERT_TRUE(m_result["status"].get<bool>()); + + m_result = nullptr; + m_irccdctl.client().request({ + { "command", "rule-info" }, + { "index", 0 } + }); + + poll([&] () { + return m_result.is_object(); + }); + + ASSERT_TRUE(m_result.is_object()); + ASSERT_TRUE(m_result["status"].get<bool>()); + ASSERT_TRUE(contains(m_result["servers"], "s1")); + ASSERT_TRUE(contains(m_result["servers"], "s2")); + ASSERT_TRUE(contains(m_result["channels"], "c1")); + ASSERT_TRUE(contains(m_result["channels"], "c2")); + ASSERT_TRUE(contains(m_result["channels"], "new-c3")); + ASSERT_TRUE(contains(m_result["plugins"], "p1")); + ASSERT_TRUE(contains(m_result["plugins"], "p2")); + ASSERT_TRUE(contains(m_result["events"], "onMessage")); + ASSERT_TRUE(contains(m_result["events"], "onCommand")); + ASSERT_EQ(m_result["action"].get<std::string>(), "drop"); + } catch (const std::exception &ex) { + FAIL() << ex.what(); + } +} + +TEST_F(RuleEditCommandTest, addPlugin) +{ + try { + m_irccdctl.client().request({ + { "command", "rule-edit" }, + { "add-plugins", { "new-p3" } }, + { "index", 0 } + }); + + poll([&] () { + return m_result.is_object(); + }); + + ASSERT_TRUE(m_result.is_object()); + ASSERT_TRUE(m_result["status"].get<bool>()); + + m_result = nullptr; + m_irccdctl.client().request({ + { "command", "rule-info" }, + { "index", 0 } + }); + + poll([&] () { + return m_result.is_object(); + }); + + ASSERT_TRUE(m_result.is_object()); + ASSERT_TRUE(m_result["status"].get<bool>()); + ASSERT_TRUE(contains(m_result["servers"], "s1")); + ASSERT_TRUE(contains(m_result["servers"], "s2")); + ASSERT_TRUE(contains(m_result["channels"], "c1")); + ASSERT_TRUE(contains(m_result["channels"], "c2")); + ASSERT_TRUE(contains(m_result["plugins"], "p1")); + ASSERT_TRUE(contains(m_result["plugins"], "p2")); + ASSERT_TRUE(contains(m_result["plugins"], "new-p3")); + ASSERT_TRUE(contains(m_result["events"], "onMessage")); + ASSERT_TRUE(contains(m_result["events"], "onCommand")); + ASSERT_EQ(m_result["action"].get<std::string>(), "drop"); + } catch (const std::exception &ex) { + FAIL() << ex.what(); + } +} + +TEST_F(RuleEditCommandTest, addEvent) +{ + try { + m_irccdctl.client().request({ + { "command", "rule-edit" }, + { "add-events", { "onQuery" } }, + { "index", 0 } + }); + + poll([&] () { + return m_result.is_object(); + }); + + ASSERT_TRUE(m_result.is_object()); + ASSERT_TRUE(m_result["status"].get<bool>()); + + m_result = nullptr; + m_irccdctl.client().request({ + { "command", "rule-info" }, + { "index", 0 } + }); + + poll([&] () { + return m_result.is_object(); + }); + + ASSERT_TRUE(m_result.is_object()); + ASSERT_TRUE(m_result["status"].get<bool>()); + ASSERT_TRUE(contains(m_result["servers"], "s1")); + ASSERT_TRUE(contains(m_result["servers"], "s2")); + ASSERT_TRUE(contains(m_result["channels"], "c1")); + ASSERT_TRUE(contains(m_result["channels"], "c2")); + ASSERT_TRUE(contains(m_result["plugins"], "p1")); + ASSERT_TRUE(contains(m_result["plugins"], "p2")); + ASSERT_TRUE(contains(m_result["events"], "onMessage")); + ASSERT_TRUE(contains(m_result["events"], "onCommand")); + ASSERT_TRUE(contains(m_result["events"], "onQuery")); + ASSERT_EQ(m_result["action"].get<std::string>(), "drop"); + } catch (const std::exception &ex) { + FAIL() << ex.what(); + } +} + +TEST_F(RuleEditCommandTest, addEventAndServer) +{ + try { + m_irccdctl.client().request({ + { "command", "rule-edit" }, + { "add-servers", { "new-s3" } }, + { "add-events", { "onQuery" } }, + { "index", 0 } + }); + + poll([&] () { + return m_result.is_object(); + }); + + ASSERT_TRUE(m_result.is_object()); + ASSERT_TRUE(m_result["status"].get<bool>()); + + m_result = nullptr; + m_irccdctl.client().request({ + { "command", "rule-info" }, + { "index", 0 } + }); + + poll([&] () { + return m_result.is_object(); + }); + + ASSERT_TRUE(m_result.is_object()); + ASSERT_TRUE(m_result["status"].get<bool>()); + ASSERT_TRUE(contains(m_result["servers"], "s1")); + ASSERT_TRUE(contains(m_result["servers"], "s2")); + ASSERT_TRUE(contains(m_result["servers"], "new-s3")); + ASSERT_TRUE(contains(m_result["channels"], "c1")); + ASSERT_TRUE(contains(m_result["channels"], "c2")); + ASSERT_TRUE(contains(m_result["plugins"], "p1")); + ASSERT_TRUE(contains(m_result["plugins"], "p2")); + ASSERT_TRUE(contains(m_result["events"], "onMessage")); + ASSERT_TRUE(contains(m_result["events"], "onCommand")); + ASSERT_TRUE(contains(m_result["events"], "onQuery")); + ASSERT_EQ(m_result["action"].get<std::string>(), "drop"); + } catch (const std::exception &ex) { + FAIL() << ex.what(); + } +} + +TEST_F(RuleEditCommandTest, changeAction) +{ + try { + m_irccdctl.client().request({ + { "command", "rule-edit" }, + { "action", "accept" }, + { "index", 0 } + }); + + poll([&] () { + return m_result.is_object(); + }); + + ASSERT_TRUE(m_result.is_object()); + ASSERT_TRUE(m_result["status"].get<bool>()); + + m_result = nullptr; + m_irccdctl.client().request({ + { "command", "rule-info" }, + { "index", 0 } + }); + + poll([&] () { + return m_result.is_object(); + }); + + ASSERT_TRUE(m_result.is_object()); + ASSERT_TRUE(m_result["status"].get<bool>()); + ASSERT_TRUE(contains(m_result["servers"], "s1")); + ASSERT_TRUE(contains(m_result["servers"], "s2")); + ASSERT_TRUE(contains(m_result["channels"], "c1")); + ASSERT_TRUE(contains(m_result["channels"], "c2")); + ASSERT_TRUE(contains(m_result["plugins"], "p1")); + ASSERT_TRUE(contains(m_result["plugins"], "p2")); + ASSERT_TRUE(contains(m_result["events"], "onMessage")); + ASSERT_TRUE(contains(m_result["events"], "onCommand")); + ASSERT_EQ(m_result["action"].get<std::string>(), "accept"); + } catch (const std::exception &ex) { + FAIL() << ex.what(); + } +} + +TEST_F(RuleEditCommandTest, removeServer) +{ + try { + m_irccdctl.client().request({ + { "command", "rule-edit" }, + { "remove-servers", { "s2" } }, + { "index", 0 } + }); + + poll([&] () { + return m_result.is_object(); + }); + + ASSERT_TRUE(m_result.is_object()); + ASSERT_TRUE(m_result["status"].get<bool>()); + + m_result = nullptr; + m_irccdctl.client().request({ + { "command", "rule-info" }, + { "index", 0 } + }); + + poll([&] () { + return m_result.is_object(); + }); + + ASSERT_TRUE(m_result.is_object()); + ASSERT_TRUE(m_result["status"].get<bool>()); + ASSERT_TRUE(contains(m_result["servers"], "s1")); + ASSERT_FALSE(contains(m_result["servers"], "s2")); + ASSERT_TRUE(contains(m_result["channels"], "c1")); + ASSERT_TRUE(contains(m_result["channels"], "c2")); + ASSERT_TRUE(contains(m_result["plugins"], "p1")); + ASSERT_TRUE(contains(m_result["plugins"], "p2")); + ASSERT_TRUE(contains(m_result["events"], "onMessage")); + ASSERT_TRUE(contains(m_result["events"], "onCommand")); + ASSERT_EQ(m_result["action"].get<std::string>(), "drop"); + } catch (const std::exception &ex) { + FAIL() << ex.what(); + } +} + +TEST_F(RuleEditCommandTest, removeChannel) +{ + try { + m_irccdctl.client().request({ + { "command", "rule-edit" }, + { "remove-channels", { "c2" } }, + { "index", 0 } + }); + + poll([&] () { + return m_result.is_object(); + }); + + ASSERT_TRUE(m_result.is_object()); + ASSERT_TRUE(m_result["status"].get<bool>()); + + m_result = nullptr; + m_irccdctl.client().request({ + { "command", "rule-info" }, + { "index", 0 } + }); + + poll([&] () { + return m_result.is_object(); + }); + + ASSERT_TRUE(m_result.is_object()); + ASSERT_TRUE(m_result["status"].get<bool>()); + ASSERT_TRUE(contains(m_result["servers"], "s1")); + ASSERT_TRUE(contains(m_result["servers"], "s2")); + ASSERT_TRUE(contains(m_result["channels"], "c1")); + ASSERT_FALSE(contains(m_result["channels"], "c2")); + ASSERT_TRUE(contains(m_result["plugins"], "p1")); + ASSERT_TRUE(contains(m_result["plugins"], "p2")); + ASSERT_TRUE(contains(m_result["events"], "onMessage")); + ASSERT_TRUE(contains(m_result["events"], "onCommand")); + ASSERT_EQ(m_result["action"].get<std::string>(), "drop"); + } catch (const std::exception &ex) { + FAIL() << ex.what(); + } +} + +TEST_F(RuleEditCommandTest, removePlugin) +{ + try { + m_irccdctl.client().request({ + { "command", "rule-edit" }, + { "remove-plugins", { "p2" } }, + { "index", 0 } + }); + + poll([&] () { + return m_result.is_object(); + }); + + ASSERT_TRUE(m_result.is_object()); + ASSERT_TRUE(m_result["status"].get<bool>()); + + m_result = nullptr; + m_irccdctl.client().request({ + { "command", "rule-info" }, + { "index", 0 } + }); + + poll([&] () { + return m_result.is_object(); + }); + + ASSERT_TRUE(m_result.is_object()); + ASSERT_TRUE(m_result["status"].get<bool>()); + ASSERT_TRUE(contains(m_result["servers"], "s1")); + ASSERT_TRUE(contains(m_result["servers"], "s2")); + ASSERT_TRUE(contains(m_result["channels"], "c1")); + ASSERT_TRUE(contains(m_result["channels"], "c2")); + ASSERT_TRUE(contains(m_result["plugins"], "p1")); + ASSERT_FALSE(contains(m_result["plugins"], "p2")); + ASSERT_TRUE(contains(m_result["events"], "onMessage")); + ASSERT_TRUE(contains(m_result["events"], "onCommand")); + ASSERT_EQ(m_result["action"].get<std::string>(), "drop"); + } catch (const std::exception &ex) { + FAIL() << ex.what(); + } +} + +TEST_F(RuleEditCommandTest, removeEvent) +{ + try { + m_irccdctl.client().request({ + { "command", "rule-edit" }, + { "remove-events", { "onCommand" } }, + { "index", 0 } + }); + + poll([&] () { + return m_result.is_object(); + }); + + ASSERT_TRUE(m_result.is_object()); + ASSERT_TRUE(m_result["status"].get<bool>()); + + m_result = nullptr; + m_irccdctl.client().request({ + { "command", "rule-info" }, + { "index", 0 } + }); + + poll([&] () { + return m_result.is_object(); + }); + + ASSERT_TRUE(m_result.is_object()); + ASSERT_TRUE(m_result["status"].get<bool>()); + ASSERT_TRUE(contains(m_result["servers"], "s1")); + ASSERT_TRUE(contains(m_result["servers"], "s2")); + ASSERT_TRUE(contains(m_result["channels"], "c1")); + ASSERT_TRUE(contains(m_result["channels"], "c2")); + ASSERT_TRUE(contains(m_result["plugins"], "p1")); + ASSERT_TRUE(contains(m_result["plugins"], "p2")); + ASSERT_TRUE(contains(m_result["events"], "onMessage")); + ASSERT_FALSE(contains(m_result["events"], "onCommand")); + ASSERT_EQ(m_result["action"].get<std::string>(), "drop"); + } catch (const std::exception &ex) { + FAIL() << ex.what(); + } +} + +TEST_F(RuleEditCommandTest, removeEventAndServer) +{ + try { + m_irccdctl.client().request({ + { "command", "rule-edit" }, + { "remove-servers", { "s2" } }, + { "remove-events", { "onCommand" } }, + { "index", 0 } + }); + + poll([&] () { + return m_result.is_object(); + }); + + ASSERT_TRUE(m_result.is_object()); + ASSERT_TRUE(m_result["status"].get<bool>()); + + m_result = nullptr; + m_irccdctl.client().request({ + { "command", "rule-info" }, + { "index", 0 } + }); + + poll([&] () { + return m_result.is_object(); + }); + + ASSERT_TRUE(m_result.is_object()); + ASSERT_TRUE(m_result["status"].get<bool>()); + ASSERT_TRUE(contains(m_result["servers"], "s1")); + ASSERT_FALSE(contains(m_result["servers"], "s2")); + ASSERT_TRUE(contains(m_result["channels"], "c1")); + ASSERT_TRUE(contains(m_result["channels"], "c2")); + ASSERT_TRUE(contains(m_result["plugins"], "p1")); + ASSERT_TRUE(contains(m_result["plugins"], "p2")); + ASSERT_TRUE(contains(m_result["events"], "onMessage")); + ASSERT_FALSE(contains(m_result["events"], "onCommand")); + ASSERT_EQ(m_result["action"].get<std::string>(), "drop"); + } catch (const std::exception &ex) { + FAIL() << ex.what(); + } +} + +int main(int argc, char **argv) +{ + testing::InitGoogleTest(&argc, argv); + + return RUN_ALL_TESTS(); +}