TST: add restore page smoke test

superscore/tests/test_page.py

@@ -5,9 +5,10 @@
 from qtpy import QtCore
 
 from superscore.client import Client
-from superscore.model import Collection, Parameter
+from superscore.model import Collection, Parameter, Snapshot
 from superscore.widgets.page.collection_builder import CollectionBuilderPage
 from superscore.widgets.page.entry import CollectionPage
+from superscore.widgets.page.restore import Restore
 from superscore.widgets.page.search import SearchPage
 
 
@@ -35,9 +36,17 @@ def collection_builder_page(qtbot: QtBot, sample_client: Client):
     qtbot.waitUntil(lambda: page.sub_pv_table_view._model._poll_thread.isFinished())
 
 
+@pytest.fixture
+def restore_page(qtbot: QtBot, sample_client: Client, simple_snapshot: Snapshot):
+    page = Restore(data=simple_snapshot, client=sample_client)
+    qtbot.addWidget(page)
+    yield page
+    page.close()
+
+
 @pytest.mark.parametrize(
     'page',
-    ["collection_page", "search_page", "collection_builder_page"]
+    ["collection_page", "search_page", "collection_builder_page", "restore_page"]
 )
 def test_page_smoke(page: str, request: pytest.FixtureRequest):
     """smoke test, just create each page and see if they fail"""
@@ -114,3 +123,16 @@ def test_coll_builder_edit(
 
     coll_model.setData(first_index, 'anothername', role=QtCore.Qt.EditRole)
     qtbot.waitUntil(lambda: "anothername" in page.data.children[1].title)
+
+
+def test_restore_page_toggle_live(restore_page):
+    tableView = restore_page.tableView
+    live_columns = tableView.live_headers
+    toggle_live_button = restore_page.compareLiveButton
+
+    toggle_live_button.click()
+    assert tableView._model._poll_thread.running
+    assert all((not tableView.isColumnHidden(column) for column in live_columns))
+
+    toggle_live_button.click()
+    assert all((tableView.isColumnHidden(column) for column in live_columns))