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

74 lines
2.9 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.nodeconquery import NodeconQueryTab
from setoolsgui.widgets import criteria, 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 = NodeconQueryTab(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 = NodeconQueryTab(mock_policy, None)
qtbot.addWidget(widget)
net, proto, context = widget.criteria
assert widget.criteria_frame_layout.columnCount() == 4
assert widget.criteria_frame_layout.rowCount() == 3
assert widget.criteria_frame_layout.itemAtPosition(0, 0).widget() == net
assert widget.criteria_frame_layout.itemAtPosition(0, 1).widget() == net
assert widget.criteria_frame_layout.itemAtPosition(0, 2).widget() == proto
assert widget.criteria_frame_layout.itemAtPosition(0, 3).widget() == proto
assert widget.criteria_frame_layout.itemAtPosition(1, 0).widget() == context
assert widget.criteria_frame_layout.itemAtPosition(1, 1).widget() == context
assert widget.criteria_frame_layout.itemAtPosition(1, 2).widget() == context
assert widget.criteria_frame_layout.itemAtPosition(1, 3).widget() == context
assert widget.criteria_frame_layout.itemAtPosition(2, 0).widget() == widget.buttonBox
assert widget.criteria_frame_layout.itemAtPosition(2, 1).widget() == widget.buttonBox
assert widget.criteria_frame_layout.itemAtPosition(2, 2).widget() == widget.buttonBox
assert widget.criteria_frame_layout.itemAtPosition(2, 3).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 = NodeconQueryTab(mock_policy, None)
qtbot.addWidget(widget)
net, proto, context = widget.criteria
context = typing.cast(criteria.ContextMatch, context)
assert isinstance(widget.query, setools.NodeconQuery)
assert isinstance(widget.table_results_model, models.NodeconTable)
assert net.attrname == "network"
assert proto.attrname == "ip_version"
assert context.user_attrname == "user"
assert context.role_attrname == "role"
assert context.type_attrname == "type_"
assert context.range_attrname == "range_"