Add unit test for notification state/type filters

fixes #13225

diff --git a/test/CMakeLists.txt b/test/CMakeLists.txt
index 5a36e892d72457c187fc40b4a9ee00da29e13863..8a42917386ce3fa3d9ed62544cc2cb135c1284cc 100644 (file)
--- a/test/CMakeLists.txt
+++ b/test/CMakeLists.txt
@@ -25,6 +25,7 @@ set(base_test_SOURCES
   base-serialize.cpp base-shellescape.cpp base-stacktrace.cpp
   base-stream.cpp base-string.cpp base-timer.cpp base-type.cpp
   base-value.cpp config-ops.cpp icinga-checkresult.cpp icinga-macros.cpp
+  icinga-notification.cpp
   icinga-perfdata.cpp remote-base64.cpp remote-url.cpp
 )
 
@@ -99,6 +100,8 @@ add_boost_test(base
         icinga_checkresult/service_3attempts
        icinga_checkresult/host_flapping_notification
        icinga_checkresult/service_flapping_notification
+       icinga_notification/state_filter
+       icinga_notification/type_filter
         icinga_macros/simple
         icinga_perfdata/empty
         icinga_perfdata/simple

diff --git a/test/icinga-notification.cpp b/test/icinga-notification.cpp
new file mode 100644 (file)
index 0000000..2639717
--- /dev/null
+++ b/test/icinga-notification.cpp
@@ -0,0 +1,106 @@
+/******************************************************************************
+ * Icinga 2                                                                   *
+ * Copyright (C) 2012-2016 Icinga Development Team (https://www.icinga.org/)  *
+ *                                                                            *
+ * 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; either version 2             *
+ * of the License, or (at your option) any later version.                     *
+ *                                                                            *
+ * 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 St, Fifth Floor, Boston, MA 02110-1301, USA.             *
+ ******************************************************************************/
+
+#include "icinga/notification.hpp"
+#include <BoostTestTargetConfig.h>
+
+using namespace icinga;
+
+BOOST_AUTO_TEST_SUITE(icinga_notification)
+
+BOOST_AUTO_TEST_CASE(state_filter)
+{
+       unsigned long fstate;
+
+       Array::Ptr states = new Array();
+       states->Add("OK");
+       states->Add("Warning");
+
+       Notification::Ptr notification = new Notification();
+
+       notification->SetStateFilter(FilterArrayToInt(states, notification->GetStateFilterMap(), ~0));
+       notification->Activate();
+       notification->SetAuthority(true);
+
+       /* Test passing notification state */
+       fstate = StateFilterWarning;
+       std::cout << "#1 Notification state: " << fstate << " against " << notification->GetStateFilter() << " must pass. " << std::endl;
+       BOOST_CHECK(notification->GetStateFilter() & fstate);
+
+       /* Test filtered notification state */
+       fstate = StateFilterUnknown;
+       std::cout << "#2 Notification state: " << fstate << " against " << notification->GetStateFilter() << " must fail." << std::endl;
+       BOOST_CHECK(!(notification->GetStateFilter() & fstate));
+
+       /* Test unset states filter configuration */
+       notification->SetStateFilter(FilterArrayToInt(Array::Ptr(), notification->GetStateFilterMap(), ~0));
+
+       fstate = StateFilterOK;
+       std::cout << "#3 Notification state: " << fstate << " against " << notification->GetStateFilter() << " must pass." << std::endl;
+       BOOST_CHECK(notification->GetStateFilter() & fstate);
+
+       /* Test empty states filter configuration */
+       states->Clear();
+       notification->SetStateFilter(FilterArrayToInt(states, notification->GetStateFilterMap(), ~0));
+
+       fstate = StateFilterCritical;
+       std::cout << "#4 Notification state: " << fstate << " against " << notification->GetStateFilter() << " must fail." << std::endl;
+       BOOST_CHECK(!(notification->GetStateFilter() & fstate));
+}
+BOOST_AUTO_TEST_CASE(type_filter)
+{
+       unsigned long ftype;
+
+       Array::Ptr types = new Array();
+       types->Add("Problem");
+       types->Add("DowntimeStart");
+       types->Add("DowntimeEnd");
+
+       Notification::Ptr notification = new Notification();
+
+       notification->SetTypeFilter(FilterArrayToInt(types, notification->GetTypeFilterMap(), ~0));
+       notification->Activate();
+       notification->SetAuthority(true);
+
+       /* Test passing notification type */
+       ftype = NotificationProblem;
+       std::cout << "#1 Notification type: " << ftype << " against " << notification->GetTypeFilter() << " must pass." << std::endl;
+       BOOST_CHECK(notification->GetTypeFilter() & ftype);
+
+       /* Test filtered notification type */
+       ftype = NotificationCustom;
+       std::cout << "#2 Notification type: " << ftype << " against " << notification->GetTypeFilter() << " must fail." << std::endl;
+       BOOST_CHECK(!(notification->GetTypeFilter() & ftype));
+
+       /* Test unset types filter configuration */
+       notification->SetTypeFilter(FilterArrayToInt(Array::Ptr(), notification->GetTypeFilterMap(), ~0));
+
+       ftype = NotificationRecovery;
+       std::cout << "#3 Notification type: " << ftype << " against " << notification->GetTypeFilter() << " must pass." << std::endl;
+       BOOST_CHECK(notification->GetTypeFilter() & ftype);
+
+       /* Test empty types filter configuration */
+       types->Clear();
+       notification->SetTypeFilter(FilterArrayToInt(types, notification->GetTypeFilterMap(), ~0));
+
+       ftype = NotificationProblem;
+       std::cout << "#4 Notification type: " << ftype << " against " << notification->GetTypeFilter() << " must fail." << std::endl;
+       BOOST_CHECK(!(notification->GetTypeFilter() & ftype));
+}
+BOOST_AUTO_TEST_SUITE_END()