mirror of
https://github.com/apache/superset.git
synced 2026-04-18 23:55:00 +00:00
chore: upgrade black (#19410)
This commit is contained in:
@@ -39,7 +39,10 @@ def key_value_entry() -> KeyValueEntry:
|
||||
from superset.key_value.models import KeyValueEntry
|
||||
|
||||
entry = KeyValueEntry(
|
||||
id=ID_KEY, uuid=UUID_KEY, resource=RESOURCE, value=pickle.dumps(VALUE),
|
||||
id=ID_KEY,
|
||||
uuid=UUID_KEY,
|
||||
resource=RESOURCE,
|
||||
value=pickle.dumps(VALUE),
|
||||
)
|
||||
db.session.add(entry)
|
||||
db.session.commit()
|
||||
@@ -47,7 +50,9 @@ def key_value_entry() -> KeyValueEntry:
|
||||
|
||||
|
||||
def test_delete_id_entry(
|
||||
app_context: AppContext, admin: User, key_value_entry: KeyValueEntry,
|
||||
app_context: AppContext,
|
||||
admin: User,
|
||||
key_value_entry: KeyValueEntry,
|
||||
) -> None:
|
||||
from superset.key_value.commands.delete import DeleteKeyValueCommand
|
||||
from superset.key_value.models import KeyValueEntry
|
||||
@@ -56,7 +61,9 @@ def test_delete_id_entry(
|
||||
|
||||
|
||||
def test_delete_uuid_entry(
|
||||
app_context: AppContext, admin: User, key_value_entry: KeyValueEntry,
|
||||
app_context: AppContext,
|
||||
admin: User,
|
||||
key_value_entry: KeyValueEntry,
|
||||
) -> None:
|
||||
from superset.key_value.commands.delete import DeleteKeyValueCommand
|
||||
from superset.key_value.models import KeyValueEntry
|
||||
@@ -65,7 +72,9 @@ def test_delete_uuid_entry(
|
||||
|
||||
|
||||
def test_delete_entry_missing(
|
||||
app_context: AppContext, admin: User, key_value_entry: KeyValueEntry,
|
||||
app_context: AppContext,
|
||||
admin: User,
|
||||
key_value_entry: KeyValueEntry,
|
||||
) -> None:
|
||||
from superset.key_value.commands.delete import DeleteKeyValueCommand
|
||||
from superset.key_value.models import KeyValueEntry
|
||||
|
||||
@@ -43,7 +43,10 @@ def key_value_entry() -> Generator[KeyValueEntry, None, None]:
|
||||
from superset.key_value.models import KeyValueEntry
|
||||
|
||||
entry = KeyValueEntry(
|
||||
id=ID_KEY, uuid=UUID_KEY, resource=RESOURCE, value=pickle.dumps(VALUE),
|
||||
id=ID_KEY,
|
||||
uuid=UUID_KEY,
|
||||
resource=RESOURCE,
|
||||
value=pickle.dumps(VALUE),
|
||||
)
|
||||
db.session.add(entry)
|
||||
db.session.commit()
|
||||
|
||||
@@ -53,7 +53,8 @@ def test_get_uuid_entry(
|
||||
|
||||
|
||||
def test_get_id_entry_missing(
|
||||
app_context: AppContext, key_value_entry: KeyValueEntry,
|
||||
app_context: AppContext,
|
||||
key_value_entry: KeyValueEntry,
|
||||
) -> None:
|
||||
from superset.key_value.commands.get import GetKeyValueCommand
|
||||
|
||||
|
||||
@@ -40,13 +40,18 @@ NEW_VALUE = "new value"
|
||||
|
||||
|
||||
def test_update_id_entry(
|
||||
app_context: AppContext, admin: User, key_value_entry: KeyValueEntry,
|
||||
app_context: AppContext,
|
||||
admin: User,
|
||||
key_value_entry: KeyValueEntry,
|
||||
) -> None:
|
||||
from superset.key_value.commands.update import UpdateKeyValueCommand
|
||||
from superset.key_value.models import KeyValueEntry
|
||||
|
||||
key = UpdateKeyValueCommand(
|
||||
actor=admin, resource=RESOURCE, key=ID_KEY, value=NEW_VALUE,
|
||||
actor=admin,
|
||||
resource=RESOURCE,
|
||||
key=ID_KEY,
|
||||
value=NEW_VALUE,
|
||||
).run()
|
||||
assert key.id == ID_KEY
|
||||
entry = db.session.query(KeyValueEntry).filter_by(id=ID_KEY).autoflush(False).one()
|
||||
@@ -55,13 +60,18 @@ def test_update_id_entry(
|
||||
|
||||
|
||||
def test_update_uuid_entry(
|
||||
app_context: AppContext, admin: User, key_value_entry: KeyValueEntry,
|
||||
app_context: AppContext,
|
||||
admin: User,
|
||||
key_value_entry: KeyValueEntry,
|
||||
) -> None:
|
||||
from superset.key_value.commands.update import UpdateKeyValueCommand
|
||||
from superset.key_value.models import KeyValueEntry
|
||||
|
||||
key = UpdateKeyValueCommand(
|
||||
actor=admin, resource=RESOURCE, key=UUID_KEY, value=NEW_VALUE,
|
||||
actor=admin,
|
||||
resource=RESOURCE,
|
||||
key=UUID_KEY,
|
||||
value=NEW_VALUE,
|
||||
).run()
|
||||
assert key.uuid == UUID_KEY
|
||||
entry = (
|
||||
@@ -75,6 +85,9 @@ def test_update_missing_entry(app_context: AppContext, admin: User) -> None:
|
||||
from superset.key_value.commands.update import UpdateKeyValueCommand
|
||||
|
||||
key = UpdateKeyValueCommand(
|
||||
actor=admin, resource=RESOURCE, key=456, value=NEW_VALUE,
|
||||
actor=admin,
|
||||
resource=RESOURCE,
|
||||
key=456,
|
||||
value=NEW_VALUE,
|
||||
).run()
|
||||
assert key is None
|
||||
|
||||
@@ -40,13 +40,18 @@ NEW_VALUE = "new value"
|
||||
|
||||
|
||||
def test_upsert_id_entry(
|
||||
app_context: AppContext, admin: User, key_value_entry: KeyValueEntry,
|
||||
app_context: AppContext,
|
||||
admin: User,
|
||||
key_value_entry: KeyValueEntry,
|
||||
) -> None:
|
||||
from superset.key_value.commands.upsert import UpsertKeyValueCommand
|
||||
from superset.key_value.models import KeyValueEntry
|
||||
|
||||
key = UpsertKeyValueCommand(
|
||||
actor=admin, resource=RESOURCE, key=ID_KEY, value=NEW_VALUE,
|
||||
actor=admin,
|
||||
resource=RESOURCE,
|
||||
key=ID_KEY,
|
||||
value=NEW_VALUE,
|
||||
).run()
|
||||
assert key.id == ID_KEY
|
||||
entry = (
|
||||
@@ -57,13 +62,18 @@ def test_upsert_id_entry(
|
||||
|
||||
|
||||
def test_upsert_uuid_entry(
|
||||
app_context: AppContext, admin: User, key_value_entry: KeyValueEntry,
|
||||
app_context: AppContext,
|
||||
admin: User,
|
||||
key_value_entry: KeyValueEntry,
|
||||
) -> None:
|
||||
from superset.key_value.commands.upsert import UpsertKeyValueCommand
|
||||
from superset.key_value.models import KeyValueEntry
|
||||
|
||||
key = UpsertKeyValueCommand(
|
||||
actor=admin, resource=RESOURCE, key=UUID_KEY, value=NEW_VALUE,
|
||||
actor=admin,
|
||||
resource=RESOURCE,
|
||||
key=UUID_KEY,
|
||||
value=NEW_VALUE,
|
||||
).run()
|
||||
assert key.uuid == UUID_KEY
|
||||
entry = (
|
||||
@@ -78,7 +88,10 @@ def test_upsert_missing_entry(app_context: AppContext, admin: User) -> None:
|
||||
from superset.key_value.models import KeyValueEntry
|
||||
|
||||
key = UpsertKeyValueCommand(
|
||||
actor=admin, resource=RESOURCE, key=456, value=NEW_VALUE,
|
||||
actor=admin,
|
||||
resource=RESOURCE,
|
||||
key=456,
|
||||
value=NEW_VALUE,
|
||||
).run()
|
||||
assert key.id == 456
|
||||
db.session.query(KeyValueEntry).filter_by(id=456).delete()
|
||||
|
||||
Reference in New Issue
Block a user