lib: make can_seek_ns_from_origin logic use `can_seek_forward` property of iterator
[babeltrace.git] / tests / bindings / python / bt2 / test_plugin.py
CommitLineData
d2d857a8
MJ
1#
2# Copyright (C) 2019 EfficiOS Inc.
3#
4# This program is free software; you can redistribute it and/or
5# modify it under the terms of the GNU General Public License
6# as published by the Free Software Foundation; only version 2
7# of the License.
8#
9# This program is distributed in the hope that it will be useful,
10# but WITHOUT ANY WARRANTY; without even the implied warranty of
11# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
12# GNU General Public License for more details.
13#
14# You should have received a copy of the GNU General Public License
15# along with this program; if not, write to the Free Software
16# Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA.
17#
18
811644b8 19import unittest
811644b8 20import bt2
811644b8
PP
21import os
22
23
644e0364 24_TEST_PLUGIN_PLUGINS_PATH = os.environ['BT_PLUGINS_PATH']
d4b72099 25_TEST_PLUGIN_PLUGIN_EXTENSION_BY_OS = {'cygwin': 'dll', 'mingw': 'dll'}
811644b8
PP
26
27
28class PluginSetTestCase(unittest.TestCase):
29 def test_create(self):
577fa92f 30 pset = bt2.find_plugins_in_path(_TEST_PLUGIN_PLUGINS_PATH)
811644b8
PP
31 self.assertTrue(len(pset) >= 3)
32
33 def test_getitem(self):
577fa92f 34 pset = bt2.find_plugins_in_path(_TEST_PLUGIN_PLUGINS_PATH)
811644b8
PP
35 self.assertTrue(pset[0].path.startswith(_TEST_PLUGIN_PLUGINS_PATH))
36
37 def test_iter(self):
577fa92f 38 pset = bt2.find_plugins_in_path(_TEST_PLUGIN_PLUGINS_PATH)
811644b8
PP
39 names = set()
40
41 for plugin in pset:
42 names.add(plugin.name)
43
44 self.assertTrue('ctf' in names)
45 self.assertTrue('utils' in names)
46 self.assertTrue('text' in names)
47
48
49class FindPluginsTestCase(unittest.TestCase):
9736d991 50 def test_find_nonexistent_dir(self):
ce4923b0 51 with self.assertRaises(ValueError):
577fa92f 52 bt2.find_plugins_in_path(
cfbd7cf3
FD
53 '/this/does/not/exist/246703df-cb85-46d5-8406-5e8dc4a88b41'
54 )
9736d991
PP
55
56 def test_find_none_existing_dir(self):
577fa92f 57 plugins = bt2.find_plugins_in_path(_TEST_PLUGIN_PLUGINS_PATH, recurse=False)
9736d991 58 self.assertIsNone(plugins)
811644b8
PP
59
60 def test_find_dir(self):
577fa92f 61 pset = bt2.find_plugins_in_path(_TEST_PLUGIN_PLUGINS_PATH)
811644b8
PP
62 self.assertTrue(len(pset) >= 3)
63
ab1bca6c 64 def test_find_file(self):
cfbd7cf3
FD
65 extension = _TEST_PLUGIN_PLUGIN_EXTENSION_BY_OS.get(
66 os.environ['BT_OS_TYPE'], 'so'
67 )
d4b72099
JR
68 plugin_name = 'babeltrace-plugin-utils.{}'.format(extension)
69 path = os.path.join(_TEST_PLUGIN_PLUGINS_PATH, 'utils', '.libs', plugin_name)
577fa92f 70 pset = bt2.find_plugins_in_path(path)
ab1bca6c
SM
71 self.assertTrue(len(pset) == 1)
72
811644b8
PP
73
74class FindPluginTestCase(unittest.TestCase):
75 def test_find_none(self):
cfbd7cf3
FD
76 plugin = bt2.find_plugin(
77 'this-does-not-exist-246703df-cb85-46d5-8406-5e8dc4a88b41'
78 )
811644b8
PP
79 self.assertIsNone(plugin)
80
81 def test_find_existing(self):
577fa92f 82 plugin = bt2.find_plugin('ctf', find_in_user_dir=False, find_in_sys_dir=False)
082db648 83 self.assertIsNotNone(plugin)
811644b8
PP
84
85
86class PluginTestCase(unittest.TestCase):
87 def setUp(self):
1045f82b
PP
88 self._plugin = bt2.find_plugin(
89 'ctf', find_in_user_dir=False, find_in_sys_dir=False
90 )
811644b8
PP
91
92 def tearDown(self):
93 del self._plugin
94
95 def test_name(self):
96 self.assertEqual(self._plugin.name, 'ctf')
97
98 def test_path(self):
824cfbd8
JR
99 plugin_path = os.path.abspath(os.path.normcase(self._plugin.path))
100 plugin_path_env = os.path.abspath(os.path.normcase(_TEST_PLUGIN_PLUGINS_PATH))
9769184f 101 self.assertTrue(plugin_path.startswith(plugin_path_env))
811644b8
PP
102
103 def test_author(self):
602d20a2 104 self.assertEqual(self._plugin.author, 'EfficiOS <https://www.efficios.com/>')
811644b8
PP
105
106 def test_license(self):
602d20a2 107 self.assertEqual(self._plugin.license, 'MIT')
811644b8
PP
108
109 def test_description(self):
602d20a2 110 self.assertEqual(self._plugin.description, 'CTF input and output')
811644b8
PP
111
112 def test_version(self):
113 self.assertIsNone(self._plugin.version)
114
115 def test_source_comp_classes_len(self):
116 self.assertEqual(len(self._plugin.source_component_classes), 2)
117
118 def test_source_comp_classes_getitem(self):
119 self.assertEqual(self._plugin.source_component_classes['fs'].name, 'fs')
120
121 def test_source_comp_classes_getitem_invalid(self):
122 with self.assertRaises(KeyError):
123 self._plugin.source_component_classes['lol']
124
125 def test_source_comp_classes_iter(self):
126 plugins = {}
127
128 for cc_name, cc in self._plugin.source_component_classes.items():
129 plugins[cc_name] = cc
130
131 self.assertTrue('fs' in plugins)
132 self.assertTrue('lttng-live' in plugins)
133 self.assertEqual(plugins['fs'].name, 'fs')
134 self.assertEqual(plugins['lttng-live'].name, 'lttng-live')
135
136 def test_filter_comp_classes_len(self):
1045f82b 137 plugin = bt2.find_plugin('utils', find_in_user_dir=False, find_in_sys_dir=False)
811644b8
PP
138 self.assertEqual(len(plugin.filter_component_classes), 2)
139
140 def test_sink_comp_classes_len(self):
141 self.assertEqual(len(self._plugin.sink_component_classes), 1)
d14ddbba
SM
142
143
144if __name__ == '__main__':
145 unittest.main()
This page took 0.049442 seconds and 4 git commands to generate.