mirror of
https://github.com/apache/superset.git
synced 2024-09-19 20:19:37 -04:00
123 lines
3.4 KiB
Python
123 lines
3.4 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 contextlib import nullcontext
|
||
|
from typing import Any
|
||
|
|
||
|
import pytest
|
||
|
from marshmallow import Schema
|
||
|
|
||
|
from superset.dashboards.permalink.schemas import DashboardPermalinkSchema
|
||
|
from superset.key_value.exceptions import KeyValueCodecEncodeException
|
||
|
from superset.key_value.types import (
|
||
|
JsonKeyValueCodec,
|
||
|
MarshmallowKeyValueCodec,
|
||
|
PickleKeyValueCodec,
|
||
|
)
|
||
|
|
||
|
|
||
|
@pytest.mark.parametrize(
|
||
|
"input_,expected_result",
|
||
|
[
|
||
|
(
|
||
|
{"foo": "bar"},
|
||
|
{"foo": "bar"},
|
||
|
),
|
||
|
(
|
||
|
{"foo": (1, 2, 3)},
|
||
|
{"foo": [1, 2, 3]},
|
||
|
),
|
||
|
(
|
||
|
{1, 2, 3},
|
||
|
KeyValueCodecEncodeException(),
|
||
|
),
|
||
|
(
|
||
|
object(),
|
||
|
KeyValueCodecEncodeException(),
|
||
|
),
|
||
|
],
|
||
|
)
|
||
|
def test_json_codec(input_: Any, expected_result: Any):
|
||
|
cm = (
|
||
|
pytest.raises(type(expected_result))
|
||
|
if isinstance(expected_result, Exception)
|
||
|
else nullcontext()
|
||
|
)
|
||
|
with cm:
|
||
|
codec = JsonKeyValueCodec()
|
||
|
encoded_value = codec.encode(input_)
|
||
|
assert expected_result == codec.decode(encoded_value)
|
||
|
|
||
|
|
||
|
@pytest.mark.parametrize(
|
||
|
"schema,input_,expected_result",
|
||
|
[
|
||
|
(
|
||
|
DashboardPermalinkSchema(),
|
||
|
{
|
||
|
"dashboardId": "1",
|
||
|
"state": {
|
||
|
"urlParams": [["foo", "bar"], ["foo", "baz"]],
|
||
|
},
|
||
|
},
|
||
|
{
|
||
|
"dashboardId": "1",
|
||
|
"state": {
|
||
|
"urlParams": [("foo", "bar"), ("foo", "baz")],
|
||
|
},
|
||
|
},
|
||
|
),
|
||
|
(
|
||
|
DashboardPermalinkSchema(),
|
||
|
{"foo": "bar"},
|
||
|
KeyValueCodecEncodeException(),
|
||
|
),
|
||
|
],
|
||
|
)
|
||
|
def test_marshmallow_codec(schema: Schema, input_: Any, expected_result: Any):
|
||
|
cm = (
|
||
|
pytest.raises(type(expected_result))
|
||
|
if isinstance(expected_result, Exception)
|
||
|
else nullcontext()
|
||
|
)
|
||
|
with cm:
|
||
|
codec = MarshmallowKeyValueCodec(schema)
|
||
|
encoded_value = codec.encode(input_)
|
||
|
assert expected_result == codec.decode(encoded_value)
|
||
|
|
||
|
|
||
|
@pytest.mark.parametrize(
|
||
|
"input_,expected_result",
|
||
|
[
|
||
|
(
|
||
|
{1, 2, 3},
|
||
|
{1, 2, 3},
|
||
|
),
|
||
|
(
|
||
|
{"foo": 1, "bar": {1: (1, 2, 3)}, "baz": {1, 2, 3}},
|
||
|
{
|
||
|
"foo": 1,
|
||
|
"bar": {1: (1, 2, 3)},
|
||
|
"baz": {1, 2, 3},
|
||
|
},
|
||
|
),
|
||
|
],
|
||
|
)
|
||
|
def test_pickle_codec(input_: Any, expected_result: Any):
|
||
|
codec = PickleKeyValueCodec()
|
||
|
encoded_value = codec.encode(input_)
|
||
|
assert expected_result == codec.decode(encoded_value)
|