You are viewing a plain text version of this content. The canonical link for it is here.
Posted to commits@superset.apache.org by be...@apache.org on 2021/12/22 17:29:30 UTC
[superset] 02/04: Update existing tests
This is an automated email from the ASF dual-hosted git repository.
beto pushed a commit to branch ch24621c
in repository https://gitbox.apache.org/repos/asf/superset.git
commit 60c560e08f9588bea2a710a4010304437ac3534a
Author: Beto Dealmeida <ro...@dealmeida.net>
AuthorDate: Thu Dec 9 18:38:07 2021 -0800
Update existing tests
---
tests/integration_tests/reports/commands_tests.py | 61 +++++++++++++++++------
1 file changed, 47 insertions(+), 14 deletions(-)
diff --git a/tests/integration_tests/reports/commands_tests.py b/tests/integration_tests/reports/commands_tests.py
index 9aa2ca4..cdc5036 100644
--- a/tests/integration_tests/reports/commands_tests.py
+++ b/tests/integration_tests/reports/commands_tests.py
@@ -382,7 +382,9 @@ def create_alert_slack_chart_success():
@pytest.fixture(
- params=["alert1",]
+ params=[
+ "alert1",
+ ]
)
def create_alert_slack_chart_grace(request):
param_config = {
@@ -660,7 +662,9 @@ def create_invalid_sql_alert_email_chart(request):
@patch("superset.reports.notifications.email.send_email_smtp")
@patch("superset.utils.screenshots.ChartScreenshot.get_screenshot")
def test_email_chart_report_schedule(
- screenshot_mock, email_mock, create_report_email_chart,
+ screenshot_mock,
+ email_mock,
+ create_report_email_chart,
):
"""
ExecuteReport Command: Test chart email report schedule with screenshot
@@ -700,7 +704,9 @@ def test_email_chart_report_schedule(
@patch("superset.reports.notifications.email.send_email_smtp")
@patch("superset.utils.screenshots.ChartScreenshot.get_screenshot")
def test_email_chart_report_schedule_force_screenshot(
- screenshot_mock, email_mock, create_report_email_chart_force_screenshot,
+ screenshot_mock,
+ email_mock,
+ create_report_email_chart_force_screenshot,
):
"""
ExecuteReport Command: Test chart email report schedule with screenshot
@@ -742,7 +748,9 @@ def test_email_chart_report_schedule_force_screenshot(
@patch("superset.reports.notifications.email.send_email_smtp")
@patch("superset.utils.screenshots.ChartScreenshot.get_screenshot")
def test_email_chart_alert_schedule(
- screenshot_mock, email_mock, create_alert_email_chart,
+ screenshot_mock,
+ email_mock,
+ create_alert_email_chart,
):
"""
ExecuteReport Command: Test chart email alert schedule with screenshot
@@ -779,7 +787,9 @@ def test_email_chart_alert_schedule(
@patch("superset.reports.notifications.email.send_email_smtp")
@patch("superset.utils.screenshots.ChartScreenshot.get_screenshot")
def test_email_chart_report_dry_run(
- screenshot_mock, email_mock, create_report_email_chart,
+ screenshot_mock,
+ email_mock,
+ create_report_email_chart,
):
"""
ExecuteReport Command: Test chart email report schedule dry run
@@ -804,7 +814,11 @@ def test_email_chart_report_dry_run(
@patch("superset.reports.notifications.email.send_email_smtp")
@patch("superset.utils.csv.get_chart_csv_data")
def test_email_chart_report_schedule_with_csv(
- csv_mock, email_mock, mock_open, mock_urlopen, create_report_email_chart_with_csv,
+ csv_mock,
+ email_mock,
+ mock_open,
+ mock_urlopen,
+ create_report_email_chart_with_csv,
):
"""
ExecuteReport Command: Test chart email report schedule with CSV
@@ -993,7 +1007,9 @@ def test_email_dashboard_report_schedule(
@patch("superset.reports.notifications.slack.WebClient.files_upload")
@patch("superset.utils.screenshots.ChartScreenshot.get_screenshot")
def test_slack_chart_report_schedule(
- screenshot_mock, file_upload_mock, create_report_slack_chart,
+ screenshot_mock,
+ file_upload_mock,
+ create_report_slack_chart,
):
"""
ExecuteReport Command: Test chart slack report schedule
@@ -1212,7 +1228,9 @@ def test_report_schedule_success_grace_end(
@patch("superset.reports.notifications.email.send_email_smtp")
@patch("superset.utils.screenshots.ChartScreenshot.get_screenshot")
def test_alert_limit_is_applied(
- screenshot_mock, email_mock, create_alert_email_chart,
+ screenshot_mock,
+ email_mock,
+ create_alert_email_chart,
):
"""
ExecuteReport Command: Test that all alerts apply a SQL limit to stmts
@@ -1268,7 +1286,9 @@ def test_email_dashboard_report_fails(
ALERTS_ATTACH_REPORTS=True,
)
def test_slack_chart_alert(
- screenshot_mock, email_mock, create_alert_email_chart,
+ screenshot_mock,
+ email_mock,
+ create_alert_email_chart,
):
"""
ExecuteReport Command: Test chart slack alert
@@ -1325,7 +1345,9 @@ def test_slack_chart_alert_no_attachment(email_mock, create_alert_email_chart):
@patch("superset.reports.notifications.slack.WebClient")
@patch("superset.utils.screenshots.ChartScreenshot.get_screenshot")
def test_slack_token_callable_chart_report(
- screenshot_mock, slack_client_mock_class, create_report_slack_chart,
+ screenshot_mock,
+ slack_client_mock_class,
+ create_report_slack_chart,
):
"""
ExecuteReport Command: Test chart slack alert (slack token callable)
@@ -1437,7 +1459,11 @@ def test_soft_timeout_screenshot(screenshot_mock, email_mock, create_alert_email
@patch("superset.reports.notifications.email.send_email_smtp")
@patch("superset.utils.csv.get_chart_csv_data")
def test_soft_timeout_csv(
- csv_mock, email_mock, mock_open, mock_urlopen, create_report_email_chart_with_csv,
+ csv_mock,
+ email_mock,
+ mock_open,
+ mock_urlopen,
+ create_report_email_chart_with_csv,
):
"""
ExecuteReport Command: Test fail on generating csv
@@ -1461,7 +1487,8 @@ def test_soft_timeout_csv(
assert email_mock.call_args[0][0] == OWNER_EMAIL
assert_log(
- ReportState.ERROR, error_message="A timeout occurred while generating a csv.",
+ ReportState.ERROR,
+ error_message="A timeout occurred while generating a csv.",
)
@@ -1473,7 +1500,11 @@ def test_soft_timeout_csv(
@patch("superset.reports.notifications.email.send_email_smtp")
@patch("superset.utils.csv.get_chart_csv_data")
def test_generate_no_csv(
- csv_mock, email_mock, mock_open, mock_urlopen, create_report_email_chart_with_csv,
+ csv_mock,
+ email_mock,
+ mock_open,
+ mock_urlopen,
+ create_report_email_chart_with_csv,
):
"""
ExecuteReport Command: Test fail on generating csv
@@ -1658,7 +1689,9 @@ def test_grace_period_error(email_mock, create_invalid_sql_alert_email_chart):
@patch("superset.reports.notifications.email.send_email_smtp")
@patch("superset.utils.screenshots.ChartScreenshot.get_screenshot")
def test_grace_period_error_flap(
- screenshot_mock, email_mock, create_invalid_sql_alert_email_chart,
+ screenshot_mock,
+ email_mock,
+ create_invalid_sql_alert_email_chart,
):
"""
ExecuteReport Command: Test alert grace period on error