beeref/tests/test_main.py
2021-04-13 17:39:38 +02:00

32 lines
974 B
Python

from unittest.mock import patch
from PyQt6 import QtCore
from beeref.__main__ import BeeRefMainWindow, main
from beeref.assets import BeeAssets
from beeref.view import BeeGraphicsView
from .base import BeeTestCase
class BeeRefMainWindowTestCase(BeeTestCase):
@patch('PyQt6.QtWidgets.QWidget.show')
def test_init(self, show_mock):
window = BeeRefMainWindow(self.app)
assert window.windowTitle() == 'BeeRef'
assert BeeAssets().logo == BeeAssets().logo
assert window.windowIcon()
assert window.contentsMargins() == QtCore.QMargins(0, 0, 0, 0)
assert isinstance(window.view, BeeGraphicsView)
show_mock.assert_called_once()
class BeeRefMainTestCase(BeeTestCase):
@patch('PyQt6.QtWidgets.QApplication')
def test_run(self, app_mock):
app_mock.return_value = self.app
with patch.object(self.app, 'exec') as exec_mock:
main()
exec_mock.assert_called_once_with()