tests/bindings/python/bt2/test_component.py

   1 from bt2 import values
   2 import unittest
   3 import copy
   4 import bt2
   5
   6
   7 @unittest.skip("this is broken")
   8 class UserComponentTestCase(unittest.TestCase):
   9     @staticmethod
  10     def _create_comp(comp_cls, name=None):
  11         graph = bt2.Graph()
  12
  13         if name is None:
  14             name = 'comp'
  15
  16         return graph.add_component(comp_cls, name)
  17
  18     def test_name(self):
  19         class MySink(bt2._UserSinkComponent):
  20             def __init__(comp_self, params):
  21                 self.assertEqual(comp_self.name, 'yaes')
  22
  23             def _consume(self):
  24                 pass
  25
  26         comp = self._create_comp(MySink, 'yaes')
  27
  28     def test_graph(self):
  29         class MySink(bt2._UserSinkComponent):
  30             def __init__(comp_self, params):
  31                 nonlocal graph
  32                 self.assertEqual(comp_self.graph, graph)
  33
  34             def _consume(self):
  35                 pass
  36
  37         graph = bt2.Graph()
  38         comp = graph.add_component(MySink, 'lel')
  39         del graph
  40
  41     def test_class(self):
  42         class MySink(bt2._UserSinkComponent):
  43             def __init__(comp_self, params):
  44                 self.assertEqual(comp_self.component_class, MySink)
  45
  46             def _consume(self):
  47                 pass
  48
  49         self._create_comp(MySink)
  50
  51     def test_addr(self):
  52         class MySink(bt2._UserSinkComponent):
  53             def __init__(comp_self, params):
  54                 self.assertIsInstance(comp_self.addr, int)
  55                 self.assertNotEqual(comp_self.addr, 0)
  56
  57             def _consume(self):
  58                 pass
  59
  60         self._create_comp(MySink)
  61
  62     def test_finalize(self):
  63         finalized = False
  64
  65         class MySink(bt2._UserSinkComponent):
  66             def _consume(self):
  67                 pass
  68
  69             def _finalize(comp_self):
  70                 nonlocal finalized
  71                 finalized = True
  72
  73         graph = bt2.Graph()
  74         comp = graph.add_component(MySink, 'lel')
  75         del graph
  76         del comp
  77         self.assertTrue(finalized)
  78
  79
  80 @unittest.skip("this is broken")
  81 class GenericComponentTestCase(unittest.TestCase):
  82     @staticmethod
  83     def _create_comp(comp_cls, name=None):
  84         graph = bt2.Graph()
  85
  86         if name is None:
  87             name = 'comp'
  88
  89         return graph.add_component(comp_cls, name)
  90
  91     def test_name(self):
  92         class MySink(bt2._UserSinkComponent):
  93             def _consume(self):
  94                 pass
  95
  96         comp = self._create_comp(MySink, 'yaes')
  97         self.assertEqual(comp.name, 'yaes')
  98
  99     def test_graph(self):
 100         class MySink(bt2._UserSinkComponent):
 101             def _consume(self):
 102                 pass
 103
 104         graph = bt2.Graph()
 105         comp = graph.add_component(MySink, 'lel')
 106         self.assertEqual(comp.graph, graph)
 107
 108     def test_class(self):
 109         class MySink(bt2._UserSinkComponent):
 110             def _consume(self):
 111                 pass
 112
 113         comp = self._create_comp(MySink)
 114         self.assertEqual(comp.component_class, MySink)
 115
 116     def test_addr(self):
 117         class MySink(bt2._UserSinkComponent):
 118             def _consume(self):
 119                 pass
 120
 121         comp = self._create_comp(MySink)
 122         self.assertIsInstance(comp.addr, int)
 123         self.assertNotEqual(comp.addr, 0)