|
32 | 32 | import unittest |
33 | 33 |
|
34 | 34 | try: |
35 | | - from robotmbt.visualise.models import ScenarioInfo, StateInfo, TraceInfo, ModelSpace |
| 35 | + from robotmbt.visualise.models import ScenarioInfo, StateInfo, TraceInfo |
36 | 36 |
|
37 | 37 | VISUALISE = True |
38 | 38 | except ImportError: |
@@ -88,40 +88,35 @@ def test_split_name_multiple_words_need_split(self): |
88 | 88 | self.assertIn('\n', result) |
89 | 89 | self.assertLessEqual(max([len(line) for line in result.split('\n')]), 20) |
90 | 90 |
|
91 | | - |
92 | 91 | class TestStateInfo(unittest.TestCase): |
93 | 92 | def test_stateInfo_empty(self): |
94 | | - s = StateInfo(ModelSpace()) |
| 93 | + modelspacestub = SimpleNamespace(ref_id=None, props={}) |
| 94 | + s = StateInfo(modelspacestub) |
95 | 95 | self.assertEqual(str(s), '') |
96 | 96 |
|
97 | 97 | def test_stateInfo_prop_empty(self): |
98 | | - space = ModelSpace() |
99 | | - space.props['prop1'] = ModelSpace() |
100 | | - s = StateInfo(space) |
| 98 | + modelspacestub = SimpleNamespace(ref_id=None, props={}) |
| 99 | + s = StateInfo(modelspacestub) |
101 | 100 | self.assertEqual(str(s), '') |
102 | 101 |
|
103 | 102 | def test_stateInfo_prop_val(self): |
104 | | - space = ModelSpace() |
105 | | - prop1 = ModelSpace() |
106 | | - prop1.value = 1 |
107 | | - space.props['prop1'] = prop1 |
108 | | - s = StateInfo(space) |
| 103 | + modelspacestub = SimpleNamespace(ref_id=None, props=dict(prop1=SimpleNamespace(value=1))) |
| 104 | + s = StateInfo(modelspacestub) |
109 | 105 | self.assertTrue('prop1:' in str(s)) |
110 | 106 | self.assertTrue('value=1' in str(s)) |
111 | 107 |
|
112 | 108 | def test_stateInfo_prop_val_empty(self): |
113 | | - space = ModelSpace() |
114 | | - prop1 = ModelSpace() |
115 | | - prop1.value = 1 |
116 | | - prop2 = ModelSpace() |
117 | | - space.props['prop1'] = prop1 |
118 | | - space.props['prop2'] = prop2 |
119 | | - s = StateInfo(space) |
| 109 | + class EmptyProp: |
| 110 | + def __dir__(self): |
| 111 | + return {} |
| 112 | + |
| 113 | + modelspacestub = SimpleNamespace(ref_id=None, props=dict(prop1=SimpleNamespace(value=1), |
| 114 | + prop2=EmptyProp())) |
| 115 | + s = StateInfo(modelspacestub) |
120 | 116 | self.assertTrue('prop1:' in str(s)) |
121 | 117 | self.assertTrue('value=1' in str(s)) |
122 | 118 | self.assertFalse('prop2:' in str(s)) |
123 | 119 |
|
124 | | - |
125 | 120 | class TestTraceInfo(unittest.TestCase): |
126 | 121 | def test_trace_info_update_normal(self): |
127 | 122 | info = TraceInfo() |
|
0 commit comments