def test_createQualityChanges(quality_mocked_application): manager = QualityManager(quality_mocked_application) mocked_quality_changes = MagicMock() manager._createQualityChanges = MagicMock(return_value = mocked_quality_changes) manager.initialize() container_manager = MagicMock() manager._container_registry.addContainer = MagicMock() # Side effect we want to check. with patch("cura.Settings.ContainerManager.ContainerManager.getInstance", container_manager): manager.createQualityChanges("derp") assert manager._container_registry.addContainer.called_once_with(mocked_quality_changes)
def test_duplicateQualityChangesQualityOnly(quality_mocked_application): manager = QualityManager(quality_mocked_application) manager.initialize() mocked_quality = MagicMock() quality_group = MagicMock() quality_group.getAllNodes = MagicMock(return_value = mocked_quality) mocked_quality_changes = MagicMock() # Isolate what we want to test (in this case, we're not interested if createQualityChanges does it's job!) manager._createQualityChanges = MagicMock(return_value = mocked_quality_changes) manager._container_registry.addContainer = MagicMock() # Side effect we want to check. manager.duplicateQualityChanges("zomg", {"quality_group": quality_group, "quality_changes_group": None}) assert manager._container_registry.addContainer.called_once_with(mocked_quality_changes)