# Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA.
#
-import uuid
import unittest
-import bt2
from utils import run_in_component_init, get_default_trace_class
class TraceClassTestCase(unittest.TestCase):
-
def test_create_default(self):
def f(comp_self):
return comp_self._create_trace_class()
tc = run_in_component_init(f)
self.assertEqual(len(tc), 0)
- self.assertEqual(len(tc.env), 0)
- self.assertIsNone(tc.uuid)
self.assertTrue(tc.assigns_automatic_stream_class_id)
- def test_uuid(self):
- def f(comp_self):
- return comp_self._create_trace_class(uuid=uuid.UUID('da7d6b6f-3108-4706-89bd-ab554732611b'))
-
- tc = run_in_component_init(f)
-
- self.assertEqual(tc.uuid, uuid.UUID('da7d6b6f-3108-4706-89bd-ab554732611b'))
-
def test_automatic_stream_class_id(self):
def f(comp_self):
return comp_self._create_trace_class(assigns_automatic_stream_class_id=True)
def test_no_assigns_automatic_stream_class_id(self):
def f(comp_self):
- return comp_self._create_trace_class(assigns_automatic_stream_class_id=False)
+ return comp_self._create_trace_class(
+ assigns_automatic_stream_class_id=False
+ )
tc = run_in_component_init(f)
self.assertFalse(tc.assigns_automatic_stream_class_id)
def test_no_assigns_automatic_stream_class_id_raises(self):
def f(comp_self):
- return comp_self._create_trace_class(assigns_automatic_stream_class_id=False)
+ return comp_self._create_trace_class(
+ assigns_automatic_stream_class_id=False
+ )
tc = run_in_component_init(f)
self.assertFalse(tc.assigns_automatic_stream_class_id)
with self.assertRaises(ValueError):
tc.create_stream_class()
- def test_env_get(self):
- def f(comp_self):
- return comp_self._create_trace_class(env={'hello': 'you', 'foo': -5})
-
- tc = run_in_component_init(f)
-
- self.assertEqual(tc.env['hello'], 'you')
- self.assertEqual(tc.env['foo'], -5)
-
- def test_env_get_non_existent(self):
- def f(comp_self):
- return comp_self._create_trace_class(env={'hello': 'you', 'foo': -5})
-
- tc = run_in_component_init(f)
-
- with self.assertRaises(KeyError):
- tc.env['lel']
-
@staticmethod
def _create_trace_class_with_some_stream_classes():
def f(comp_self):
- return comp_self._create_trace_class(assigns_automatic_stream_class_id=False)
+ return comp_self._create_trace_class(
+ assigns_automatic_stream_class_id=False
+ )
tc = run_in_component_init(f)
sc1 = tc.create_stream_class(id=12)
tc, sc1, sc2, sc3 = self._create_trace_class_with_some_stream_classes()
for sc_id, stream_class in tc.items():
- self.assertIsInstance(stream_class, bt2.stream_class._StreamClass)
-
if sc_id == 12:
self.assertEqual(stream_class.addr, sc1.addr)
elif sc_id == 54: