def test_no_alerts_fig_raises_httpnotfound(self, m_available):
request = get_current_request()
m_available.return_value = False
with pytest.raises(HTTPNotFound):
salishseacast._data_for_publish_template(
request,
"nowcast",
arrow.get("2016-11-04"),
salishseacast.publish_figures,
salishseacast.publish_tides_max_ssh_figure_group,
arrow.get("2016-11-04"),
)
def test_figures(self, m_available):
request = get_current_request()
m_available.return_value = True
data = salishseacast._data_for_publish_template(
request,
"forecast",
arrow.get("2016-11-04"),
salishseacast.publish_figures,
salishseacast.publish_tides_max_ssh_figure_group,
arrow.get("2016-11-03"),
)
assert data["figures"] == salishseacast.publish_figures
def test_run_type_title(self, m_available, run_type, expected):
request = get_current_request()
m_available.return_value = True
data = salishseacast._data_for_publish_template(
request,
run_type,
arrow.get("2016-11-04"),
salishseacast.publish_figures,
salishseacast.publish_tides_max_ssh_figure_group,
arrow.get("2016-11-04"),
)
assert data["run_type_title"] == expected
def test_missing_figures(self, m_available):
request = get_current_request()
m_available.side_effect = (
[True, True]
+ [False] * (len(salishseacast.publish_figures) - 1)
+ [True] * len(salishseacast.publish_tides_max_ssh_figure_group.figures)
)
data = salishseacast._data_for_publish_template(
request,
"forecast",
arrow.get("2016-11-04"),
salishseacast.publish_figures,
salishseacast.publish_tides_max_ssh_figure_group,
arrow.get("2016-11-03"),
)
assert data["figures"] == [salishseacast.publish_figures[0]]