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
|
from UM.Settings.ContainerRegistry import ContainerRegistry
from UM.Settings.Models.ContainerPropertyProvider import ContainerPropertyProvider
import pytest
from unittest.mock import patch, MagicMock
test_validate_data_get_set = [
{"attribute": "containerId", "value": "YAY"},
{"attribute": "watchedProperties", "value": {"beep": "oh noes"}},
{"attribute": "key", "value": "YAY"},
]
@pytest.fixture()
def registry():
mocked_registry = MagicMock()
mocked_registry.isReadOnly = MagicMock(return_value=False)
mocked_container = MagicMock()
mocked_container.getProperty = MagicMock(return_value = "yay")
mocked_registry.findContainers = MagicMock(return_value = [mocked_container])
return mocked_registry
@pytest.mark.parametrize("data", test_validate_data_get_set)
def test_getAndSet(data):
model = ContainerPropertyProvider()
# Convert the first letter into a capital
attribute = list(data["attribute"])
attribute[0] = attribute[0].capitalize()
attribute = "".join(attribute)
# mock the correct emit
setattr(model, data["attribute"] + "Changed", MagicMock())
# Attempt to set the value
with patch("UM.Settings.ContainerRegistry.ContainerRegistry.getInstance"):
getattr(model, "set" + attribute)(data["value"])
# Check if signal fired.
signal = getattr(model, data["attribute"] + "Changed")
assert signal.emit.call_count == 1
# Ensure that the value got set
assert getattr(model, data["attribute"]) == data["value"]
# Attempt to set the value again
getattr(model, "set" + attribute)(data["value"])
# The signal should not fire again
assert signal.emit.call_count == 1
def test_setPropertyValueNoKey(registry):
model = ContainerPropertyProvider()
with patch("UM.Settings.ContainerRegistry.ContainerRegistry.getInstance", MagicMock(return_value = registry)):
model.setContainerId("bloop")
model.setPropertyValue("derp", "zomg")
# Check if the mocked container got it's setProperty targeted
assert ContainerRegistry.getInstance().findContainers()[0].setProperty.call_count == 0
def test_setPropertyValueNotWatchedProperty(registry):
model = ContainerPropertyProvider()
with patch("UM.Settings.ContainerRegistry.ContainerRegistry.getInstance", MagicMock(return_value = registry)):
model.setContainerId("bloop")
model.setKey("whatever")
model.setPropertyValue("derp", "zomg")
# Check if the mocked container got it's setProperty targeted
assert ContainerRegistry.getInstance().findContainers()[0].setProperty.call_count == 0
def test_setPropertyValue(registry):
model = ContainerPropertyProvider()
with patch("UM.Settings.ContainerRegistry.ContainerRegistry.getInstance", MagicMock(return_value = registry)):
model.setContainerId("bloop")
model.setWatchedProperties(["derp"])
model.setKey("whatever")
model.setPropertyValue("derp", "zomg")
# Check if the mocked container got it's setProperty targeted
assert ContainerRegistry.getInstance().findContainers()[0].setProperty.call_count == 1
def test_setPropertyValueCache(registry):
model = ContainerPropertyProvider()
with patch("UM.Settings.ContainerRegistry.ContainerRegistry.getInstance", MagicMock(return_value=registry)):
model.setContainerId("bloop")
model.setWatchedProperties(["derp", "yay"])
model.setKey("derp")
# Fake that we got a signal indicating the value is already set
model._onPropertyChanged("derp", "yay")
model.setPropertyValue("derp", "yay")
# Check if the mocked container got it's setProperty targeted
assert ContainerRegistry.getInstance().findContainers()[0].setProperty.call_count == 0
|
