mirror of
https://github.com/apache/superset.git
synced 2024-09-19 03:59:49 -04:00
b7a0559aaf
* rename key_value to temporary_cache * add migration * create new key_value package * add commands * lots of new stuff * fix schema reference * remove redundant filter state from bootstrap data * add missing license headers * fix pylint * fix dashboard permalink access * use valid json mocks for filter state tests * fix temporary cache tests * add anchors to dashboard state * lint * fix util test * fix url shortlink button tests * remove legacy shortner * remove unused imports * fix js tests * fix test * add native filter state to anchor link * add UPDATING.md section * address comments * address comments * lint * fix test * add utils tests + other test stubs * add key_value integration tests * add filter box state to permalink state * fully support persisting url parameters * lint, add redirects and a few integration tests * fix test + clean up trailing comma * fix anchor bug * change value to LargeBinary to support persisting binary values * fix urlParams type and simplify urlencode * lint * add optional entry expiration * fix incorrect chart id + add test
118 lines
3.7 KiB
Python
118 lines
3.7 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 __future__ import annotations
|
|
|
|
import json
|
|
from typing import TYPE_CHECKING
|
|
from unittest.mock import patch
|
|
from uuid import UUID
|
|
|
|
if TYPE_CHECKING:
|
|
from superset.key_value.models import KeyValueEntry
|
|
|
|
import pytest
|
|
from flask.ctx import AppContext
|
|
|
|
from superset.key_value.types import Key
|
|
|
|
RESOURCE = "my-resource"
|
|
UUID_KEY = "3e7a2ab8-bcaf-49b0-a5df-dfb432f291cc"
|
|
ID_KEY = "123"
|
|
|
|
|
|
@pytest.fixture
|
|
def key_value_entry(app_context: AppContext):
|
|
from superset.key_value.models import KeyValueEntry
|
|
|
|
return KeyValueEntry(
|
|
id=int(ID_KEY), uuid=UUID(UUID_KEY), value=json.dumps({"foo": "bar"}),
|
|
)
|
|
|
|
|
|
def test_parse_permalink_key_uuid_valid(app_context: AppContext) -> None:
|
|
from superset.key_value.utils import parse_permalink_key
|
|
|
|
assert parse_permalink_key(UUID_KEY) == Key(id=None, uuid=UUID(UUID_KEY))
|
|
|
|
|
|
def test_parse_permalink_key_id_invalid(app_context: AppContext) -> None:
|
|
from superset.key_value.utils import parse_permalink_key
|
|
|
|
with pytest.raises(ValueError):
|
|
parse_permalink_key(ID_KEY)
|
|
|
|
|
|
@patch("superset.key_value.utils.current_app.config", {"PERMALINK_KEY_TYPE": "id"})
|
|
def test_parse_permalink_key_id_valid(app_context: AppContext) -> None:
|
|
from superset.key_value.utils import parse_permalink_key
|
|
|
|
assert parse_permalink_key(ID_KEY) == Key(id=int(ID_KEY), uuid=None)
|
|
|
|
|
|
@patch("superset.key_value.utils.current_app.config", {"PERMALINK_KEY_TYPE": "id"})
|
|
def test_parse_permalink_key_uuid_invalid(app_context: AppContext) -> None:
|
|
from superset.key_value.utils import parse_permalink_key
|
|
|
|
with pytest.raises(ValueError):
|
|
parse_permalink_key(UUID_KEY)
|
|
|
|
|
|
def test_format_permalink_key_uuid(app_context: AppContext) -> None:
|
|
from superset.key_value.utils import format_permalink_key
|
|
|
|
assert format_permalink_key(Key(id=None, uuid=UUID(UUID_KEY))) == UUID_KEY
|
|
|
|
|
|
def test_format_permalink_key_id(app_context: AppContext) -> None:
|
|
from superset.key_value.utils import format_permalink_key
|
|
|
|
assert format_permalink_key(Key(id=int(ID_KEY), uuid=None)) == ID_KEY
|
|
|
|
|
|
def test_extract_key_uuid(
|
|
app_context: AppContext, key_value_entry: KeyValueEntry,
|
|
) -> None:
|
|
from superset.key_value.utils import extract_key
|
|
|
|
assert extract_key(key_value_entry, "id") == ID_KEY
|
|
|
|
|
|
def test_extract_key_id(
|
|
app_context: AppContext, key_value_entry: KeyValueEntry,
|
|
) -> None:
|
|
from superset.key_value.utils import extract_key
|
|
|
|
assert extract_key(key_value_entry, "uuid") == UUID_KEY
|
|
|
|
|
|
def test_get_filter_uuid(app_context: AppContext,) -> None:
|
|
from superset.key_value.utils import get_filter
|
|
|
|
assert get_filter(resource=RESOURCE, key=UUID_KEY, key_type="uuid",) == {
|
|
"resource": RESOURCE,
|
|
"uuid": UUID(UUID_KEY),
|
|
}
|
|
|
|
|
|
def test_get_filter_id(app_context: AppContext,) -> None:
|
|
from superset.key_value.utils import get_filter
|
|
|
|
assert get_filter(resource=RESOURCE, key=ID_KEY, key_type="id",) == {
|
|
"resource": RESOURCE,
|
|
"id": int(ID_KEY),
|
|
}
|