From 9bc3e785b362480fa7380abca3f926558545e465 Mon Sep 17 00:00:00 2001 From: Philippe Proulx Date: Fri, 14 Jun 2019 21:44:11 -0400 Subject: [PATCH] test_component.py: add `logging_level` property tests Signed-off-by: Philippe Proulx Change-Id: Ibc7dbbbb3d497efe8b8c237488fd0733c094f242 Reviewed-on: https://review.lttng.org/c/babeltrace/+/1455 Tested-by: jenkins Reviewed-by: Francis Deslauriers --- tests/bindings/python/bt2/test_component.py | 26 +++++++++++++++++---- 1 file changed, 22 insertions(+), 4 deletions(-) diff --git a/tests/bindings/python/bt2/test_component.py b/tests/bindings/python/bt2/test_component.py index a01404e4..24f056f1 100644 --- a/tests/bindings/python/bt2/test_component.py +++ b/tests/bindings/python/bt2/test_component.py @@ -6,13 +6,13 @@ import bt2 class UserComponentTestCase(unittest.TestCase): @staticmethod - def _create_comp(comp_cls, name=None): + def _create_comp(comp_cls, name=None, log_level=bt2.LoggingLevel.NONE): graph = bt2.Graph() if name is None: name = 'comp' - return graph.add_component(comp_cls, name) + return graph.add_component(comp_cls, name, logging_level=log_level) def test_name(self): class MySink(bt2._UserSinkComponent): @@ -24,6 +24,16 @@ class UserComponentTestCase(unittest.TestCase): comp = self._create_comp(MySink, 'yaes') + def test_logging_level(self): + class MySink(bt2._UserSinkComponent): + def __init__(comp_self, params): + self.assertEqual(comp_self.logging_level, bt2.LoggingLevel.INFO) + + def _consume(self): + pass + + comp = self._create_comp(MySink, 'yaes', bt2.LoggingLevel.INFO) + def test_class(self): class MySink(bt2._UserSinkComponent): def __init__(comp_self, params): @@ -66,13 +76,13 @@ class UserComponentTestCase(unittest.TestCase): class GenericComponentTestCase(unittest.TestCase): @staticmethod - def _create_comp(comp_cls, name=None): + def _create_comp(comp_cls, name=None, log_level=bt2.LoggingLevel.NONE): graph = bt2.Graph() if name is None: name = 'comp' - return graph.add_component(comp_cls, name) + return graph.add_component(comp_cls, name, logging_level=log_level) def test_name(self): class MySink(bt2._UserSinkComponent): @@ -82,6 +92,14 @@ class GenericComponentTestCase(unittest.TestCase): comp = self._create_comp(MySink, 'yaes') self.assertEqual(comp.name, 'yaes') + def test_logging_level(self): + class MySink(bt2._UserSinkComponent): + def _consume(self): + pass + + comp = self._create_comp(MySink, 'yaes', bt2.LoggingLevel.WARN) + self.assertEqual(comp.logging_level, bt2.LoggingLevel.WARN) + def test_class(self): class MySink(bt2._UserSinkComponent): def _consume(self): -- 2.34.1