github/supysonic
1
0
Fork 0
mirror of https://github.com/spl0k/supysonic.git synced 2024-12-22 17:06:17 +00:00
Code Issues Releases Wiki Activity

black

Browse Source
This commit is contained in:
Alban Féron 2023-01-15 16:28:27 +01:00
parent d5a4f1856c
commit be6b617e60
No known key found for this signature in database
GPG Key ID: 8CE0313646D16165
2 changed files with 9 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
supysonic/db.py
View File

@ -628,7 +628,9 @@ def init_database(database_uri):
if ext == ".sql": if ext == ".sql":
with db.atomic(): with db.atomic():
execute_sql_resource_script(f"schema/migration/{provider}/{migration}") execute_sql_resource_script(
f"schema/migration/{provider}/{migration}"
)
elif ext == ".py": elif ext == ".py":
m = importlib.import_module( m = importlib.import_module(
f".schema.migration.{provider}.{date}", __package__ f".schema.migration.{provider}.{date}", __package__

8
tests/managers/test_manager_user.py
View File

@ -104,7 +104,9 @@ class UserManagerTestCase(unittest.TestCase):
# Delete existing users # Delete existing users
for name in ["alice", "bob", "charlie"]: for name in ["alice", "bob", "charlie"]:
user = db.User.get(name=name) user = db.User.get(name=name)
db.ClientPrefs.create(user=user, client_name="tests") # test for FK handling db.ClientPrefs.create(
user=user, client_name="tests"
) # test for FK handling
UserManager.delete(user.id) UserManager.delete(user.id)
self.assertRaises(db.User.DoesNotExist, db.User.__getitem__, user.id) self.assertRaises(db.User.DoesNotExist, db.User.__getitem__, user.id)
self.assertEqual(db.User.select().count(), 0) self.assertEqual(db.User.select().count(), 0)
@ -115,7 +117,9 @@ class UserManagerTestCase(unittest.TestCase):
# Delete existing users # Delete existing users
for name in ["alice", "bob", "charlie"]: for name in ["alice", "bob", "charlie"]:
user = db.User.get(name=name) user = db.User.get(name=name)
db.ClientPrefs.create(user=user, client_name="tests") # test for FK handling db.ClientPrefs.create(
user=user, client_name="tests"
) # test for FK handling
UserManager.delete_by_name(name) UserManager.delete_by_name(name)
self.assertFalse(db.User.select().where(db.User.name == name).exists()) self.assertFalse(db.User.select().where(db.User.name == name).exists())