From 761e1890bdacd5e9e53c3aa052259b79698b45bd Mon Sep 17 00:00:00 2001 From: Philippe Proulx Date: Sat, 15 Jun 2019 00:44:52 -0400 Subject: [PATCH] tests: bt2: add query log level tests Signed-off-by: Philippe Proulx Change-Id: I725d3b7d64b4a820da1f3762c3217af9a88a0ca3 Reviewed-on: https://review.lttng.org/c/babeltrace/+/1462 Tested-by: jenkins Reviewed-by: Francis Deslauriers --- .../python/bt2/test_component_class.py | 16 ++++++++ .../python/bt2/test_query_executor.py | 40 +++++++++++++++++++ 2 files changed, 56 insertions(+) diff --git a/tests/bindings/python/bt2/test_component_class.py b/tests/bindings/python/bt2/test_component_class.py index c2b2acd6..228ef9c7 100644 --- a/tests/bindings/python/bt2/test_component_class.py +++ b/tests/bindings/python/bt2/test_component_class.py @@ -211,6 +211,22 @@ class UserComponentClassTestCase(unittest.TestCase): self.assertIsNone(res) del query_params + def test_query_logging_level(self): + class MySink(bt2._UserSinkComponent): + def _consume(self): + pass + + @classmethod + def _query(cls, query_exec, obj, params, log_level): + nonlocal query_log_level + query_log_level = log_level + + query_log_level = None + res = bt2.QueryExecutor().query(MySink, 'obj', None, + bt2.LoggingLevel.WARN) + self.assertEqual(query_log_level, bt2.LoggingLevel.WARN) + del query_log_level + def test_query_simple(self): class MySink(bt2._UserSinkComponent): def _consume(self): diff --git a/tests/bindings/python/bt2/test_query_executor.py b/tests/bindings/python/bt2/test_query_executor.py index 4fc0f863..89ddf838 100644 --- a/tests/bindings/python/bt2/test_query_executor.py +++ b/tests/bindings/python/bt2/test_query_executor.py @@ -53,6 +53,22 @@ class QueryExecutorTestCase(unittest.TestCase): self.assertEqual(query_params, None) del query_params + def test_query_logging_level(self): + class MySink(bt2._UserSinkComponent): + def _consume(self): + pass + + @classmethod + def _query(cls, query_exec, obj, params, log_level): + nonlocal query_log_level + query_log_level = log_level + + query_log_level = None + res = bt2.QueryExecutor().query(MySink, 'obj', None, + bt2.LoggingLevel.INFO) + self.assertEqual(query_log_level, bt2.LoggingLevel.INFO) + del query_log_level + def test_query_gen_error(self): class MySink(bt2._UserSinkComponent): def _consume(self): @@ -77,6 +93,30 @@ class QueryExecutorTestCase(unittest.TestCase): with self.assertRaises(bt2.InvalidQueryObject): res = bt2.QueryExecutor().query(MySink, 'obj', [17, 23]) + def test_query_logging_level_invalid_type(self): + class MySink(bt2._UserSinkComponent): + def _consume(self): + pass + + @classmethod + def _query(cls, query_exec, obj, params, log_level): + pass + + with self.assertRaises(TypeError): + res = bt2.QueryExecutor().query(MySink, 'obj', [17, 23], 'yeah') + + def test_query_logging_level_invalid_value(self): + class MySink(bt2._UserSinkComponent): + def _consume(self): + pass + + @classmethod + def _query(cls, query_exec, obj, params, log_level): + pass + + with self.assertRaises(ValueError): + res = bt2.QueryExecutor().query(MySink, 'obj', [17, 23], 12345) + def test_query_invalid_params(self): class MySink(bt2._UserSinkComponent): def _consume(self): -- 2.34.1