2 # Copyright (C) 2019 EfficiOS Inc.
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
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.
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.
20 from utils
import run_in_component_init
23 class StreamTestCase(unittest
.TestCase
):
26 return comp_self
._create
_trace
_class
()
28 self
._tc
= run_in_component_init(f
)
29 self
._sc
= self
._tc
.create_stream_class(assigns_automatic_stream_id
=True)
32 def test_create_default(self
):
33 stream
= self
._tr
.create_stream(self
._sc
)
34 self
.assertIsNone(stream
.name
)
35 self
.assertEqual(len(stream
.user_attributes
), 0)
38 stream
= self
._tr
.create_stream(self
._sc
, name
='équidistant')
39 self
.assertEqual(stream
.name
, 'équidistant')
41 def test_invalid_name(self
):
42 with self
.assertRaises(TypeError):
43 self
._tr
.create_stream(self
._sc
, name
=22)
45 def test_create_user_attributes(self
):
46 stream
= self
._tr
.create_stream(self
._sc
, user_attributes
={'salut': 23})
47 self
.assertEqual(stream
.user_attributes
, {'salut': 23})
49 def test_create_invalid_user_attributes(self
):
50 with self
.assertRaises(TypeError):
51 self
._tr
.create_stream(self
._sc
, user_attributes
=object())
53 def test_create_invalid_user_attributes_value_type(self
):
54 with self
.assertRaises(TypeError):
55 self
._tr
.create_stream(self
._sc
, user_attributes
=23)
57 def test_stream_class(self
):
58 stream
= self
._tr
.create_stream(self
._sc
)
59 self
.assertEqual(stream
.cls
, self
._sc
)
62 stream
= self
._tr
.create_stream(self
._sc
)
63 self
.assertEqual(stream
.trace
.addr
, self
._tr
.addr
)
65 def test_invalid_id(self
):
66 sc
= self
._tc
.create_stream_class(assigns_automatic_stream_id
=False)
68 with self
.assertRaises(TypeError):
69 self
._tr
.create_stream(sc
, id='string')