mirror of
https://github.com/apache/superset.git
synced 2026-04-10 20:06:13 +00:00
92 lines
3.5 KiB
Python
92 lines
3.5 KiB
Python
# Licensed to the Apache Software Foundation (ASF) under one
|
|
# or more contributor license agreements. See the NOTICE file
|
|
# distributed with this work for additional information
|
|
# regarding copyright ownership. The ASF licenses this file
|
|
# to you under the Apache License, Version 2.0 (the
|
|
# "License"); you may not use this file except in compliance
|
|
# with the License. You may obtain a copy of the License at
|
|
#
|
|
# http://www.apache.org/licenses/LICENSE-2.0
|
|
#
|
|
# Unless required by applicable law or agreed to in writing,
|
|
# software distributed under the License is distributed on an
|
|
# "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
|
|
# KIND, either express or implied. See the License for the
|
|
# specific language governing permissions and limitations
|
|
# under the License.
|
|
from unittest.mock import MagicMock, patch
|
|
|
|
|
|
@patch("superset.daos.report.get_user_id", return_value=1)
|
|
@patch("superset.daos.report.db")
|
|
def test_validate_unique_creation_method_duplicate_returns_false(
|
|
mock_db: MagicMock,
|
|
mock_uid: MagicMock,
|
|
) -> None:
|
|
from superset.daos.report import ReportScheduleDAO
|
|
|
|
# Simulate that a matching report already exists
|
|
mock_db.session.query.return_value.filter_by.return_value.filter.return_value = (
|
|
MagicMock()
|
|
)
|
|
mock_db.session.query.return_value.scalar.return_value = True
|
|
assert ReportScheduleDAO.validate_unique_creation_method(dashboard_id=1) is False
|
|
|
|
|
|
@patch("superset.daos.report.get_user_id", return_value=1)
|
|
@patch("superset.daos.report.db")
|
|
def test_validate_unique_creation_method_no_duplicate_returns_true(
|
|
mock_db: MagicMock,
|
|
mock_uid: MagicMock,
|
|
) -> None:
|
|
from superset.daos.report import ReportScheduleDAO
|
|
|
|
mock_db.session.query.return_value.filter_by.return_value.filter.return_value = (
|
|
MagicMock()
|
|
)
|
|
mock_db.session.query.return_value.scalar.return_value = False
|
|
assert ReportScheduleDAO.validate_unique_creation_method(dashboard_id=1) is True
|
|
|
|
|
|
@patch("superset.daos.report.db")
|
|
def test_find_last_error_notification_returns_none_after_success(
|
|
mock_db: MagicMock,
|
|
) -> None:
|
|
from superset.daos.report import ReportScheduleDAO
|
|
|
|
schedule = MagicMock()
|
|
error_log = MagicMock()
|
|
success_log = MagicMock()
|
|
|
|
# Build the query chain so each .query().filter().order_by().first() call
|
|
# returns a different result. The DAO calls db.session.query() twice:
|
|
# 1st call finds the error marker log
|
|
# 2nd call finds a non-error log after it (success happened since last error email)
|
|
query_mock = MagicMock()
|
|
mock_db.session.query.return_value = query_mock
|
|
chain = query_mock.filter.return_value.order_by.return_value
|
|
chain.first.side_effect = [error_log, success_log]
|
|
|
|
result = ReportScheduleDAO.find_last_error_notification(schedule)
|
|
# Success log exists after error → should return None (no re-notification needed)
|
|
assert result is None
|
|
|
|
|
|
@patch("superset.daos.report.db")
|
|
def test_find_last_error_notification_returns_log_when_only_errors(
|
|
mock_db: MagicMock,
|
|
) -> None:
|
|
from superset.daos.report import ReportScheduleDAO
|
|
|
|
schedule = MagicMock()
|
|
error_log = MagicMock()
|
|
|
|
query_mock = MagicMock()
|
|
mock_db.session.query.return_value = query_mock
|
|
chain = query_mock.filter.return_value.order_by.return_value
|
|
# 1st call: error marker log found; 2nd call: no success log after it
|
|
chain.first.side_effect = [error_log, None]
|
|
|
|
result = ReportScheduleDAO.find_last_error_notification(schedule)
|
|
assert result is error_log
|