TestCuraSceneNode.py

from UM.Math.Polygon import Polygon
from UM.Scene.SceneNodeDecorator import SceneNodeDecorator
from cura.Scene.CuraSceneNode import CuraSceneNode
import pytest

from unittest.mock import patch


class MockedConvexHullDecorator(SceneNodeDecorator):
    def __init__(self):
        super().__init__()

    def getConvexHull(self):
        return Polygon([[5, 5], [-5, 5], [-5, -5], [5, -5]])

    def getPrintingArea(self):
        return Polygon([[5, 5], [-5, 5], [-5, -5], [5, -5]])


class InvalidConvexHullDecorator(SceneNodeDecorator):
    def __init__(self):
        super().__init__()

    def getConvexHull(self):
        return Polygon()


@pytest.fixture()
def cura_scene_node():
    # Replace the SettingOverrideDecorator with an empty decorator
    with patch("cura.Scene.CuraSceneNode.SettingOverrideDecorator", SceneNodeDecorator):
        return CuraSceneNode()


class TestCollidesWithAreas:
    def test_noConvexHull(self, cura_scene_node):
        assert not cura_scene_node.collidesWithAreas([Polygon([[10, 10], [-10, 10], [-10, -10], [10, -10]])])

    def test_convexHullIntersects(self, cura_scene_node):
        cura_scene_node.addDecorator(MockedConvexHullDecorator())
        assert cura_scene_node.collidesWithAreas([Polygon([[10, 10], [-10, 10], [-10, -10], [10, -10]])])

    def test_convexHullNoIntersection(self, cura_scene_node):
        cura_scene_node.addDecorator(MockedConvexHullDecorator())

        assert not cura_scene_node.collidesWithAreas([Polygon([[60, 60], [40, 60], [40, 40], [60, 40]])])

    def test_invalidConvexHull(self, cura_scene_node):
        cura_scene_node.addDecorator(InvalidConvexHullDecorator())
        assert not cura_scene_node.collidesWithAreas([Polygon([[10, 10], [-10, 10], [-10, -10], [10, -10]])])


def test_outsideBuildArea(cura_scene_node):
    cura_scene_node.setOutsideBuildArea(True)
    assert cura_scene_node.isOutsideBuildArea