Browse Source

Added simple test for BuildVolume

Jaime van Kessel 5 years ago
parent
commit
f85095f5d8
2 changed files with 30 additions and 6 deletions
  1. 3 6
      cura/BuildVolume.py
  2. 27 0
      tests/TestBuildVolume.py

+ 3 - 6
cura/BuildVolume.py

@@ -165,16 +165,13 @@ class BuildVolume(SceneNode):
             active_extruder_changed.connect(self._updateDisallowedAreasAndRebuild)
 
     def setWidth(self, width: float) -> None:
-        if width is not None:
-            self._width = width
+        self._width = width
 
     def setHeight(self, height: float) -> None:
-        if height is not None:
-            self._height = height
+        self._height = height
 
     def setDepth(self, depth: float) -> None:
-        if depth is not None:
-            self._depth = depth
+        self._depth = depth
 
     def setShape(self, shape: str) -> None:
         if shape:

+ 27 - 0
tests/TestBuildVolume.py

@@ -0,0 +1,27 @@
+from unittest.mock import MagicMock, patch
+
+import pytest
+
+from cura.BuildVolume import BuildVolume
+
+@pytest.fixture
+def build_volume():
+    mocked_application = MagicMock()
+    mocked_platform = MagicMock(name="platform")
+    with patch("cura.BuildVolume.Platform", mocked_platform):
+        return BuildVolume(mocked_application)
+
+
+def test_buildVolumeSetSizes(build_volume):
+    build_volume.setWidth(10)
+    assert build_volume.getDiagonalSize() == 10
+
+    build_volume.setWidth(0)
+    build_volume.setHeight(100)
+    assert build_volume.getDiagonalSize() == 100
+
+    build_volume.setHeight(0)
+    build_volume.setDepth(200)
+    assert build_volume.getDiagonalSize() == 200
+
+