X-Git-Url: http://git.efficios.com/?a=blobdiff_plain;f=tests%2Fbindings%2Fpython%2Fbt2%2Ftest_plugin.py;h=7c766c6994f14fc8f29f96db26eee3077a0fe2b0;hb=f5567ea88d172767b34373bc6e402da8bfd85ef8;hp=68400ccbe7e2d4169eb25998502e54f68d3e73b3;hpb=577fa92f184fd2d75fb0697e879ab563de117c2e;p=babeltrace.git diff --git a/tests/bindings/python/bt2/test_plugin.py b/tests/bindings/python/bt2/test_plugin.py index 68400ccb..7c766c69 100644 --- a/tests/bindings/python/bt2/test_plugin.py +++ b/tests/bindings/python/bt2/test_plugin.py @@ -1,29 +1,15 @@ +# SPDX-License-Identifier: GPL-2.0-only # # Copyright (C) 2019 EfficiOS Inc. # -# 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; only 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., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA. -# import unittest import bt2 -import bt2.plugin import os -_TEST_PLUGIN_PLUGINS_PATH = os.environ['BT_PLUGINS_PATH'] -_TEST_PLUGIN_PLUGIN_EXTENSION_BY_OS = {'cygwin': 'dll', 'mingw': 'dll'} +_TEST_PLUGIN_PLUGINS_PATH = os.environ["BT_PLUGINS_PATH"] +_TEST_PLUGIN_PLUGIN_EXTENSION_BY_OS = {"cygwin": "dll", "mingw": "dll"} class PluginSetTestCase(unittest.TestCase): @@ -42,16 +28,16 @@ class PluginSetTestCase(unittest.TestCase): for plugin in pset: names.add(plugin.name) - self.assertTrue('ctf' in names) - self.assertTrue('utils' in names) - self.assertTrue('text' in names) + self.assertTrue("ctf" in names) + self.assertTrue("utils" in names) + self.assertTrue("text" in names) class FindPluginsTestCase(unittest.TestCase): def test_find_nonexistent_dir(self): with self.assertRaises(ValueError): bt2.find_plugins_in_path( - '/this/does/not/exist/246703df-cb85-46d5-8406-5e8dc4a88b41' + "/this/does/not/exist/246703df-cb85-46d5-8406-5e8dc4a88b41" ) def test_find_none_existing_dir(self): @@ -64,10 +50,10 @@ class FindPluginsTestCase(unittest.TestCase): def test_find_file(self): extension = _TEST_PLUGIN_PLUGIN_EXTENSION_BY_OS.get( - os.environ['BT_OS_TYPE'], 'so' + os.environ["BT_TESTS_OS_TYPE"], "so" ) - plugin_name = 'babeltrace-plugin-utils.{}'.format(extension) - path = os.path.join(_TEST_PLUGIN_PLUGINS_PATH, 'utils', '.libs', plugin_name) + plugin_name = "babeltrace-plugin-utils.{}".format(extension) + path = os.path.join(_TEST_PLUGIN_PLUGINS_PATH, "utils", ".libs", plugin_name) pset = bt2.find_plugins_in_path(path) self.assertTrue(len(pset) == 1) @@ -75,24 +61,26 @@ class FindPluginsTestCase(unittest.TestCase): class FindPluginTestCase(unittest.TestCase): def test_find_none(self): plugin = bt2.find_plugin( - 'this-does-not-exist-246703df-cb85-46d5-8406-5e8dc4a88b41' + "this-does-not-exist-246703df-cb85-46d5-8406-5e8dc4a88b41" ) self.assertIsNone(plugin) def test_find_existing(self): - plugin = bt2.find_plugin('ctf', find_in_user_dir=False, find_in_sys_dir=False) - self.assertIsInstance(plugin, bt2.plugin._Plugin) + plugin = bt2.find_plugin("ctf", find_in_user_dir=False, find_in_sys_dir=False) + self.assertIsNotNone(plugin) class PluginTestCase(unittest.TestCase): def setUp(self): - self._plugin = bt2.find_plugin('ctf') + self._plugin = bt2.find_plugin( + "ctf", find_in_user_dir=False, find_in_sys_dir=False + ) def tearDown(self): del self._plugin def test_name(self): - self.assertEqual(self._plugin.name, 'ctf') + self.assertEqual(self._plugin.name, "ctf") def test_path(self): plugin_path = os.path.abspath(os.path.normcase(self._plugin.path)) @@ -100,13 +88,13 @@ class PluginTestCase(unittest.TestCase): self.assertTrue(plugin_path.startswith(plugin_path_env)) def test_author(self): - self.assertTrue('Philippe Proulx' in self._plugin.author) + self.assertEqual(self._plugin.author, "EfficiOS ") def test_license(self): - self.assertTrue('MIT' in self._plugin.license) + self.assertEqual(self._plugin.license, "MIT") def test_description(self): - self.assertTrue('CTF source and sink support' in self._plugin.description) + self.assertEqual(self._plugin.description, "CTF input and output") def test_version(self): self.assertIsNone(self._plugin.version) @@ -115,11 +103,11 @@ class PluginTestCase(unittest.TestCase): self.assertEqual(len(self._plugin.source_component_classes), 2) def test_source_comp_classes_getitem(self): - self.assertEqual(self._plugin.source_component_classes['fs'].name, 'fs') + self.assertEqual(self._plugin.source_component_classes["fs"].name, "fs") def test_source_comp_classes_getitem_invalid(self): with self.assertRaises(KeyError): - self._plugin.source_component_classes['lol'] + self._plugin.source_component_classes["lol"] def test_source_comp_classes_iter(self): plugins = {} @@ -127,14 +115,18 @@ class PluginTestCase(unittest.TestCase): for cc_name, cc in self._plugin.source_component_classes.items(): plugins[cc_name] = cc - self.assertTrue('fs' in plugins) - self.assertTrue('lttng-live' in plugins) - self.assertEqual(plugins['fs'].name, 'fs') - self.assertEqual(plugins['lttng-live'].name, 'lttng-live') + self.assertTrue("fs" in plugins) + self.assertTrue("lttng-live" in plugins) + self.assertEqual(plugins["fs"].name, "fs") + self.assertEqual(plugins["lttng-live"].name, "lttng-live") def test_filter_comp_classes_len(self): - plugin = bt2.find_plugin('utils') + plugin = bt2.find_plugin("utils", find_in_user_dir=False, find_in_sys_dir=False) self.assertEqual(len(plugin.filter_component_classes), 2) def test_sink_comp_classes_len(self): self.assertEqual(len(self._plugin.sink_component_classes), 1) + + +if __name__ == "__main__": + unittest.main()