Veronica Berglyd Olsen
109 lines
2.6 KiB
Python
109 lines
2.6 KiB
Python
"""
|
||
novelWriter – Test Suite Mocked Classes
|
||
=======================================
|
||
|
||
This file is a part of novelWriter
|
||
Copyright (C) 2019 Veronica Berglyd Olsen and novelWriter contributors
|
||
|
||
This program is free software: you can redistribute it and/or modify
|
||
it under the terms of the GNU General Public License as published by
|
||
the Free Software Foundation, either version 3 of the License, or
|
||
(at your option) any later version.
|
||
|
||
This program 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
|
||
General Public License for more details.
|
||
|
||
You should have received a copy of the GNU General Public License
|
||
along with this program. If not, see <https://www.gnu.org/licenses/>.
|
||
""" # noqa
|
||
from __future__ import annotations
|
||
|
||
from unittest.mock import MagicMock
|
||
|
||
from PyQt6.QtCore import QSize
|
||
from PyQt6.QtGui import QFont, QIcon, QPixmap
|
||
from PyQt6.QtWidgets import QWidget
|
||
|
||
from novelwriter.extensions.modified import NPushButton
|
||
|
||
|
||
class MockGuiMain(QWidget):
|
||
|
||
def __init__(self):
|
||
super().__init__()
|
||
self.mainStatus = MagicMock()
|
||
self.docEditor = MagicMock()
|
||
self.docViewer = MagicMock()
|
||
self.mainProgress = MagicMock()
|
||
self.projPath = ""
|
||
|
||
def postLaunchTasks(self, cmdOpen):
|
||
return
|
||
|
||
def setStatus(self, message):
|
||
return
|
||
|
||
def openProject(self, projPath):
|
||
self.projPath = projPath
|
||
|
||
def rebuildIndex(self):
|
||
return
|
||
|
||
def closeMain(self):
|
||
return "closeMain"
|
||
|
||
def close(self):
|
||
return "close"
|
||
|
||
|
||
class MockTheme:
|
||
|
||
def __init__(self):
|
||
self.baseIconHeight = 20
|
||
self.guiFont = QFont()
|
||
self.guiFontB = QFont()
|
||
self.guiFontBU = QFont()
|
||
|
||
def initThemes(self) -> None:
|
||
return
|
||
|
||
def getPixmap(self, *a) -> QPixmap:
|
||
return QPixmap()
|
||
|
||
def getHeaderDecoration(self, *a) -> QPixmap:
|
||
return QPixmap()
|
||
|
||
def getStandardButton(self, *a) -> NPushButton:
|
||
return NPushButton(None, "", QSize(1, 1)) # type: ignore
|
||
|
||
def getIcon(self, *a) -> QIcon:
|
||
return QIcon()
|
||
|
||
def getItemIcon(self, *a, **k) -> QIcon:
|
||
return QIcon()
|
||
|
||
|
||
class MockApp:
|
||
|
||
def __init__(self):
|
||
return
|
||
|
||
def installTranslator(self, language):
|
||
return
|
||
|
||
|
||
# Error Functions
|
||
# ===============
|
||
# Mock functions that will raise errors instead.
|
||
|
||
def causeOSError(*args, **kwargs):
|
||
"""Raise an OSError."""
|
||
raise OSError("Mock OSError")
|
||
|
||
|
||
def causeException(*args, **kwargs):
|
||
"""Raise an Exception."""
|
||
raise Exception("Mock Exception")
|