setools/tests-gui/widgets/test_constraintquery.py
Chris PeBenito d6dee948ce Add new ConstraintQueryTab implementation.
Signed-off-by: Chris PeBenito <pebenito@ieee.org>
2024-02-14 09:11:35 -05:00

69 lines
2.5 KiB
Python

# SPDX-License-Identifier: GPL-2.0-only
import typing
from PyQt6 import QtWidgets
from pytestqt.qtbot import QtBot
import setools
from setoolsgui.widgets.constraintquery import ConstraintQueryTab
from setoolsgui.widgets import models
from .criteria.util import build_mock_policy
def test_docs(qtbot: QtBot) -> None:
"""Check that docs are provided for the widget."""
mock_policy = build_mock_policy()
widget = ConstraintQueryTab(mock_policy, None)
qtbot.addWidget(widget)
assert widget.whatsThis()
assert widget.table_results.whatsThis()
assert widget.raw_results.whatsThis()
for w in widget.criteria:
assert w.toolTip()
assert w.whatsThis()
results = typing.cast(QtWidgets.QTabWidget, widget.results)
for index in range(results.count()):
assert results.tabWhatsThis(index)
def test_layout(qtbot: QtBot) -> None:
"""Test the layout of the criteria frame."""
mock_policy = build_mock_policy()
widget = ConstraintQueryTab(mock_policy, None)
qtbot.addWidget(widget)
rt, user, role, type_, tclass, perms = widget.criteria
assert widget.criteria_frame_layout.columnCount() == 2
assert widget.criteria_frame_layout.rowCount() == 4
assert widget.criteria_frame_layout.itemAtPosition(0, 0).widget() == rt
assert widget.criteria_frame_layout.itemAtPosition(0, 1).widget() == user
assert widget.criteria_frame_layout.itemAtPosition(1, 0).widget() == role
assert widget.criteria_frame_layout.itemAtPosition(1, 1).widget() == type_
assert widget.criteria_frame_layout.itemAtPosition(2, 0).widget() == tclass
assert widget.criteria_frame_layout.itemAtPosition(2, 1).widget() == perms
assert widget.criteria_frame_layout.itemAtPosition(3, 0).widget() == widget.buttonBox
assert widget.criteria_frame_layout.itemAtPosition(3, 1).widget() == widget.buttonBox
def test_criteria_mapping(qtbot: QtBot) -> None:
"""Test that widgets save to the correct query fields."""
mock_policy = build_mock_policy()
widget = ConstraintQueryTab(mock_policy, None)
qtbot.addWidget(widget)
rt, user, role, type_, tclass, perms = widget.criteria
assert isinstance(widget.query, setools.ConstraintQuery)
assert isinstance(widget.table_results_model, models.ConstraintTable)
assert rt.attrname == "ruletype"
assert user.attrname == "user"
assert role.attrname == "role"
assert type_.attrname == "type_"
assert tclass.attrname == "tclass"
assert perms.attrname == "perms"