From 636b3e296c3e614c72eca160f79719c28432af5a Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Anders=20Fredrik=20Ki=C3=A6r?= Date: Mon, 24 Feb 2020 14:42:38 +0000 Subject: [PATCH] Add webviz pytest fixture --- setup.py | 5 +- tests/test_table_plotter.py | 117 ++++++++++++++------------------ webviz_config/testing/plugin.py | 26 +++++++ 3 files changed, 82 insertions(+), 66 deletions(-) create mode 100644 webviz_config/testing/plugin.py diff --git a/setup.py b/setup.py index 6a7a4e47..4e03e510 100644 --- a/setup.py +++ b/setup.py @@ -30,7 +30,10 @@ "themes/default_assets/*", ] }, - entry_points={"console_scripts": ["webviz=webviz_config.command_line:main"]}, + entry_points={ + "console_scripts": ["webviz=webviz_config.command_line:main"], + "pytest11": ["webviz = webviz_config.testing.plugin"], + }, install_requires=[ # Pinning dash to the 1.7-series as long as # https://github.com/plotly/dash-core-components/issues/746 is open diff --git a/tests/test_table_plotter.py b/tests/test_table_plotter.py index c7691a06..bb3ec314 100644 --- a/tests/test_table_plotter.py +++ b/tests/test_table_plotter.py @@ -3,108 +3,95 @@ from webviz_config.common_cache import CACHE from webviz_config.themes import default_theme -from webviz_config.plugins import _table_plotter +from webviz_config.plugins import TablePlotter +CSV_FILE = "./tests/data/example_data.csv" -def test_table_plotter(dash_duo): - app = dash.Dash(__name__) - app.config.suppress_callback_exceptions = True - CACHE.init_app(app.server) - app.webviz_settings = {"theme": default_theme} - csv_file = "./tests/data/example_data.csv" - page = _table_plotter.TablePlotter(app, csv_file) - app.layout = page.layout - dash_duo.start_server(app) +def test_table_plotter(webviz_single_plugin, dash_duo): + + app = webviz_single_plugin.app - # Wait for the app to render(there is probably a better way...) - time.sleep(5) + plugin = TablePlotter(app, csv_file=CSV_FILE) + app.layout = plugin.layout + dash_duo.start_server(app) # Checking that no plot options are defined - assert page.plot_options == {} + assert plugin.plot_options == {} + # Check that filter is not active - assert not page.use_filter + assert not plugin.use_filter # Checking that the correct plot type is initialized - plot_dd = dash_duo.find_element("#" + page.uuid("plottype")) + plot_dd = dash_duo.find_element("#" + plugin.uuid("plottype")) assert plot_dd.text == "scatter" # Checking that only the relevant options are shown - for plot_option in page.plot_args.keys(): - plot_option_dd = dash_duo.find_element("#" + page.uuid(f"div-{plot_option}")) - if plot_option not in page.plots["scatter"]: - assert plot_option_dd.get_attribute("style") == "display: none;" + for plot_option in plugin.plot_args.keys(): + if plot_option not in plugin.plots["scatter"]: + dash_duo.wait_for_style_to_equal( + "#" + plugin.uuid(f"div-{plot_option}"), style="display", val="none" + ) # Checking that options are initialized correctly for option in ["x", "y"]: - plot_option_dd = dash_duo.find_element("#" + page.uuid(f"dropdown-{option}")) + plot_option_dd = dash_duo.find_element("#" + plugin.uuid(f"dropdown-{option}")) assert plot_option_dd.text == "Well" -def test_table_plotter_filter(dash_duo): +def test_table_plotter_filter(webviz_single_plugin, dash_duo): - app = dash.Dash(__name__) - app.config.suppress_callback_exceptions = True - CACHE.init_app(app.server) - app.webviz_settings = {"theme": default_theme} - csv_file = "./tests/data/example_data.csv" - page = _table_plotter.TablePlotter(app, csv_file, filter_cols=["Well"]) - app.layout = page.layout - dash_duo.start_server(app) + app = webviz_single_plugin.app + + plugin = TablePlotter(app, csv_file=CSV_FILE, filter_cols=["Well"]) + app.layout = plugin.layout - # Wait for the app to render(there is probably a better way...) - time.sleep(5) + dash_duo.start_server(app) # Checking that no plot options are defined - assert page.plot_options == {} + assert plugin.plot_options == {} + # Check that filter is active - assert page.use_filter - assert page.filter_cols == ["Well"] + assert plugin.use_filter + assert plugin.filter_cols == ["Well"] # Checking that the correct plot type is initialized - plot_dd = dash_duo.find_element("#" + page.uuid("plottype")) + plot_dd = dash_duo.find_element("#" + plugin.uuid("plottype")) assert plot_dd.text == "scatter" # Checking that only the relevant options are shown - for plot_option in page.plot_args.keys(): - plot_option_dd = dash_duo.find_element("#" + page.uuid(f"div-{plot_option}")) - if plot_option not in page.plots["scatter"]: - assert "display: none;" in plot_option_dd.get_attribute("style") + for plot_option in plugin.plot_args.keys(): + if plot_option not in plugin.plots["scatter"]: + dash_duo.wait_for_style_to_equal( + "#" + plugin.uuid(f"div-{plot_option}"), style="display", val="none" + ) # Checking that options are initialized correctly for option in ["x", "y"]: - plot_option_dd = dash_duo.find_element("#" + page.uuid(f"dropdown-{option}")) + plot_option_dd = dash_duo.find_element("#" + plugin.uuid(f"dropdown-{option}")) assert plot_option_dd.text == "Well" -def test_initialized_table_plotter(dash_duo): - - app = dash.Dash(__name__) - app.css.config.serve_locally = True - app.scripts.config.serve_locally = True - app.config.suppress_callback_exceptions = True - CACHE.init_app(app.server) - app.webviz_settings = {"theme": default_theme} - csv_file = "./tests/data/example_data.csv" - plot_options = dict( - x="Well", - y="Initial reservoir pressure (bar)", - size="Average permeability (D)", - facet_col="Segment", - ) - - page = _table_plotter.TablePlotter( - app, csv_file, lock=True, plot_options=plot_options - ) - app.layout = page.layout - dash_duo.start_server(app) +def test_initialized_table_plotter(webviz_single_plugin, dash_duo): + + app = webviz_single_plugin.app - # Wait for the app to render(there is probably a better way...) + plot_options = { + "x": "Well", + "y": "Initial reservoir pressure (bar)", + "size": "Average permeability (D)", + "facet_col": "Segment", + } + + plugin = TablePlotter(app, csv_file=CSV_FILE, lock=True, plot_options=plot_options) + app.layout = plugin.layout + + dash_duo.start_server(app) # Checking that plot options are defined - assert page.plot_options == plot_options - assert page.lock + assert plugin.plot_options == plot_options + assert plugin.lock # Checking that the selectors are hidden - selector_row = dash_duo.find_element("#" + page.uuid("selector-row")) + selector_row = dash_duo.find_element("#" + plugin.uuid("selector-row")) assert "display: none;" in selector_row.get_attribute("style") diff --git a/webviz_config/testing/plugin.py b/webviz_config/testing/plugin.py new file mode 100644 index 00000000..34ddf834 --- /dev/null +++ b/webviz_config/testing/plugin.py @@ -0,0 +1,26 @@ +import dash +from dash_html_components import Div +import pytest + +from webviz_config.common_cache import CACHE +from webviz_config.themes import default_theme + + +class WebvizSinglePlugin: + def __init__(self): + self._app = dash.Dash(__name__) + self._configure_app() + + def _configure_app(self): + self._app.config.suppress_callback_exceptions = True + CACHE.init_app(self._app.server) + self._app.webviz_settings = {"theme": default_theme} + + @property + def app(self): + return self._app + + +@pytest.fixture +def webviz_single_plugin(): + return WebvizSinglePlugin()