chore: upgrade black (#19410)

This commit is contained in:
Ville Brofeldt
2022-03-29 20:03:09 +03:00
committed by GitHub
parent 816a2c3e1e
commit a619cb4ea9
204 changed files with 2125 additions and 608 deletions

View File

@@ -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

View File

@@ -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()

View File

@@ -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

View File

@@ -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

View File

@@ -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()