File size: 3,101 Bytes
985c397 | 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 | # SPDX-License-Identifier: LGPL-2.1-or-later
# /****************************************************************************
# *
# Copyright (c) 2025 Weston Schmidt <weston_schmidt@alumni.purdue.edu> *
# *
# This file is part of FreeCAD. *
# *
# FreeCAD is free software: you can redistribute it and/or modify it *
# under the terms of the GNU Lesser General Public License as *
# published by the Free Software Foundation, either version 2.1 of the *
# License, or (at your option) any later version. *
# *
# FreeCAD is distributed in the hope that it will be useful, but *
# WITHOUT ANY WARRANTY; without even the implied warranty of *
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU *
# Lesser General Public License for more details. *
# *
# You should have received a copy of the GNU Lesser General Public *
# License along with FreeCAD. If not, see *
# <https://www.gnu.org/licenses/>. *
# *
# ***************************************************************************/
"""Mock classes for FreeCAD GUI testing."""
import builtins
from .MockGui import (
MockQIcon,
MockQTreeWidgetItem,
MockPySideUic,
MockGetDocument,
MockAddModule,
MockDoCommandSkip,
)
# Set up all FreeCAD GUI mocks for testing
# Always create mocks for consistent testing
Gui = type("MockGui", (), {})()
QtCore = type("MockQtCore", (), {})()
QtGui = type("MockQtGui", (), {})()
# Patch QtGui with our mock classes
QtGui.QIcon = MockQIcon
QtGui.QTreeWidgetItem = MockQTreeWidgetItem
# Mock the PySideUic if it doesn't exist
if not hasattr(Gui, "PySideUic"):
Gui.PySideUic = MockPySideUic
# Mock additional Gui methods that might be missing
if not hasattr(Gui, "getDocument"):
Gui.getDocument = MockGetDocument
# Mock Selection module
if not hasattr(Gui, "Selection"):
Gui.Selection = type(
"MockSelection",
(),
{
"clearSelection": lambda *args: None,
"addSelection": lambda *args: None,
"getSelection": lambda *args: [],
},
)()
# Mock addModule method
if not hasattr(Gui, "addModule"):
Gui.addModule = MockAddModule
# Mock doCommandSkip method
if not hasattr(Gui, "doCommandSkip"):
Gui.doCommandSkip = MockDoCommandSkip
# Make QtCore, QtGui and Gui available in the global namespace
builtins.QtCore = QtCore
builtins.QtGui = QtGui
builtins.Gui = Gui
builtins.QIcon = MockQIcon
__all__ = []
|