tests/migrations/f1410ed7ec95_tests.py

# 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 copy import deepcopy

from superset.migrations.versions.f1410ed7ec95_migrate_native_filters_to_new_schema import (
    downgrade_dashboard,
    upgrade_dashboard,
)

dashboard_v1 = {
    "native_filter_configuration": [
        {
            "filterType": "filter_select",
            "cascadingFilters": True,
            "defaultValue": ["Albania", "Algeria"],
        },
    ],
    "filter_sets_configuration": [
        {
            "nativeFilters": {
                "FILTER": {
                    "filterType": "filter_select",
                    "cascadingFilters": True,
                    "defaultValue": ["Albania", "Algeria"],
                },
            },
        },
    ],
}


dashboard_v2 = {
    "native_filter_configuration": [
        {
            "filterType": "filter_select",
            "cascadingFilters": True,
            "defaultDataMask": {"filterState": {"value": ["Albania", "Algeria"],},},
        }
    ],
    "filter_sets_configuration": [
        {
            "nativeFilters": {
                "FILTER": {
                    "filterType": "filter_select",
                    "cascadingFilters": True,
                    "defaultDataMask": {
                        "filterState": {"value": ["Albania", "Algeria"],},
                    },
                },
            },
        },
    ],
}


def test_upgrade_dashboard():
    """
    ensure that dashboard upgrade operation produces a correct dashboard object
    """
    converted_dashboard = deepcopy(dashboard_v1)
    filters, filter_sets = upgrade_dashboard(converted_dashboard)
    assert filters == 1
    assert filter_sets == 1
    assert dashboard_v2 == converted_dashboard


def test_downgrade_dashboard():
    """
    ensure that dashboard downgrade operation produces a correct dashboard object
    """
    converted_dashboard = deepcopy(dashboard_v2)
    filters, filter_sets = downgrade_dashboard(converted_dashboard)
    assert filters == 1
    assert filter_sets == 1
    assert dashboard_v1 == converted_dashboard