1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
|
"""Unit test for Scenes object."""
import unittest
from unittest.mock import MagicMock
from pyvlx import PyVLX, Scene, Scenes
# pylint: disable=too-many-public-methods,invalid-name
class TestScenes(unittest.TestCase):
"""Test class for scenes object."""
def test_get_item(self) -> None:
"""Test get_item."""
pyvlx = MagicMock(spec=PyVLX)
scenes = Scenes(pyvlx)
scene1 = Scene(pyvlx, 0, "Scene_1")
scenes.add(scene1)
scene2 = Scene(pyvlx, 1, "Scene_2")
scenes.add(scene2)
scene3 = Scene(pyvlx, 4, "Scene_3")
scenes.add(scene3)
self.assertEqual(scenes["Scene_1"], scene1)
self.assertEqual(scenes["Scene_2"], scene2)
self.assertEqual(scenes["Scene_3"], scene3)
self.assertEqual(scenes[0], scene1)
self.assertEqual(scenes[1], scene2)
self.assertEqual(scenes[4], scene3)
def test_get_item_failed(self) -> None:
"""Test get_item with non existing object."""
pyvlx = MagicMock(spec=PyVLX)
scenes = Scenes(pyvlx)
scene1 = Scene(pyvlx, 0, "Scene_1")
scenes.add(scene1)
with self.assertRaises(KeyError):
scenes["Scene_2"] # pylint: disable=pointless-statement
with self.assertRaises(KeyError):
scenes[1] # pylint: disable=pointless-statement
def test_iter(self) -> None:
"""Test iterator."""
pyvlx = MagicMock(spec=PyVLX)
scenes = Scenes(pyvlx)
scene1 = Scene(pyvlx, 0, "Scene_1")
scenes.add(scene1)
scene2 = Scene(pyvlx, 1, "Scene_2")
scenes.add(scene2)
scene3 = Scene(pyvlx, 2, "Scene_3")
scenes.add(scene3)
self.assertEqual(tuple(scenes.__iter__()), (scene1, scene2, scene3)) # pylint: disable=unnecessary-dunder-call
def test_len(self) -> None:
"""Test len."""
pyvlx = MagicMock(spec=PyVLX)
scenes = Scenes(pyvlx)
self.assertEqual(len(scenes), 0)
scenes.add(Scene(pyvlx, 0, "Scene_1"))
scenes.add(Scene(pyvlx, 1, "Scene_2"))
scenes.add(Scene(pyvlx, 2, "Scene_3"))
scenes.add(Scene(pyvlx, 3, "Scene_4"))
self.assertEqual(len(scenes), 4)
def test_add_same_object(self) -> None:
"""Test adding object with same scene_id."""
pyvlx = MagicMock(spec=PyVLX)
scenes = Scenes(pyvlx)
self.assertEqual(len(scenes), 0)
scenes.add(Scene(pyvlx, 0, "Scene_1"))
scenes.add(Scene(pyvlx, 1, "Scene_2"))
scenes.add(Scene(pyvlx, 2, "Scene_3"))
scenes.add(Scene(pyvlx, 1, "Scene_2_same_id"))
self.assertEqual(len(scenes), 3)
self.assertEqual(scenes[1].name, "Scene_2_same_id")
def test_add_item_failed(self) -> None:
"""Test add() with wrong type."""
pyvlx = MagicMock()
scenes = Scenes(pyvlx)
with self.assertRaises(TypeError):
scenes.add(scenes) # type: ignore
with self.assertRaises(TypeError):
scenes.add("scene") # type: ignore
def test_clear(self) -> None:
"""Test clear() method."""
pyvlx = MagicMock(spec=PyVLX)
scenes = Scenes(pyvlx)
self.assertEqual(len(scenes), 0)
scenes.add(Scene(pyvlx, 0, "Scene_1"))
scenes.add(Scene(pyvlx, 1, "Scene_2"))
scenes.clear()
self.assertEqual(len(scenes), 0)
|
