X-Git-Url: http://git.efficios.com/?a=blobdiff_plain;f=tests%2Fbindings%2Fpython%2Fbt2%2Ftest_plugin.py;h=4abf884625da628d60c41e5251a71fcedb81c41e;hb=ffecc00e2725dea0af6a7e0cfb10a95b8f56ee68;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..4abf8846 100644
--- a/tests/bindings/python/bt2/test_plugin.py
+++ b/tests/bindings/python/bt2/test_plugin.py
@@ -1,24 +1,10 @@
+# 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
 
 
@@ -64,7 +50,7 @@ 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)
@@ -81,12 +67,14 @@ class FindPluginTestCase(unittest.TestCase):
 
     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)
+        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
@@ -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 <https://www.efficios.com/>')
 
     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)
@@ -133,8 +121,12 @@ class PluginTestCase(unittest.TestCase):
         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()