From 0fcce96a175531ec6042cde1b11a0052aa261dd5 Mon Sep 17 00:00:00 2001 From: Daniel Baumann Date: Fri, 19 Apr 2024 19:43:34 +0200 Subject: Adding upstream version 1.3.2. Signed-off-by: Daniel Baumann --- tests/test_classificationmap.py | 46 +++++++++++++++++++++++++++++++++++++++++ 1 file changed, 46 insertions(+) create mode 100644 tests/test_classificationmap.py (limited to 'tests/test_classificationmap.py') diff --git a/tests/test_classificationmap.py b/tests/test_classificationmap.py new file mode 100644 index 0000000..fb3d205 --- /dev/null +++ b/tests/test_classificationmap.py @@ -0,0 +1,46 @@ +# Copyright (C) 2017 Open Information Security Foundation +# +# You can copy, redistribute or modify this Program under the terms of +# the GNU General Public License version 2 as published by the Free +# Software Foundation. +# +# 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 +# version 2 along with this program; if not, write to the Free Software +# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA +# 02110-1301, USA. + +import unittest + +from suricata.update.maps import ClassificationMap + +class ClassificationMapTestCase(unittest.TestCase): + + test_filename = "tests/classification1.config" + + def test_load_from_file(self): + m = ClassificationMap(open(self.test_filename)) + + # Classifications are indexed at 1. + self.assertEqual(None, m.get(0)) + + c = m.get(1) + self.assertEqual("not-suspicious", c["name"]) + self.assertEqual("Not Suspicious Traffic", c["description"]) + self.assertEqual(3, c["priority"]) + + c = m.get(34) + self.assertEqual("default-login-attempt", c["name"]) + self.assertEqual("Attempt to login by a default username and password", + c["description"]) + self.assertEqual(2, c["priority"]) + + c = m.get_by_name("unknown") + self.assertTrue(c is not None) + self.assertEqual("unknown", c["name"]) + self.assertEqual("Unknown Traffic", c["description"]) + self.assertEqual(3, c["priority"]) -- cgit v1.2.3