|
2 | 2 | from typing import Generator
|
3 | 3 |
|
4 | 4 | import pytest
|
| 5 | +from litestar import Litestar, Request |
| 6 | +from litestar.testing import RequestFactory |
5 | 7 | from markupsafe import Markup
|
6 | 8 | from sqlalchemy import Boolean, Column, Enum, ForeignKey, Integer, String, select
|
7 | 9 | from sqlalchemy.dialects.postgresql import UUID
|
8 | 10 | from sqlalchemy.orm import declarative_base, relationship, sessionmaker
|
9 | 11 | from sqlalchemy.sql.expression import Select
|
10 |
| -from litestar import Litestar |
11 |
| -from litestar import Request |
12 |
| -from litestar.testing import RequestFactory |
13 | 12 |
|
14 | 13 | from sqladmin import Admin, ModelView
|
15 | 14 | from sqladmin.exceptions import InvalidModelError
|
@@ -178,7 +177,8 @@ class UserAdmin(ModelView, model=User):
|
178 | 177 |
|
179 | 178 |
|
180 | 179 | async def test_column_formatters_default() -> None:
|
181 |
| - class ProfileAdmin(ModelView, model=Profile): ... |
| 180 | + class ProfileAdmin(ModelView, model=Profile): |
| 181 | + ... |
182 | 182 |
|
183 | 183 | user = User(id=1, name="Long Name")
|
184 | 184 | profile = Profile(user=user, is_active=True)
|
@@ -341,7 +341,8 @@ class PostgresModel(Base):
|
341 | 341 |
|
342 | 342 |
|
343 | 343 | def test_model_default_sort() -> None:
|
344 |
| - class UserAdmin(ModelView, model=User): ... |
| 344 | + class UserAdmin(ModelView, model=User): |
| 345 | + ... |
345 | 346 |
|
346 | 347 | assert UserAdmin()._get_default_sort() == [("id", False)]
|
347 | 348 |
|
@@ -417,19 +418,24 @@ class UserAdmin(ModelView, model=User):
|
417 | 418 |
|
418 | 419 |
|
419 | 420 | def test_sort_query() -> None:
|
420 |
| - class AddressAdmin(ModelView, model=Address): ... |
| 421 | + class AddressAdmin(ModelView, model=Address): |
| 422 | + ... |
421 | 423 |
|
422 | 424 | query = select(Address)
|
423 | 425 |
|
424 |
| - request = RequestFactory().get("/", query_params={"sortBy":"id", "sort":"asc"}) |
| 426 | + request = RequestFactory().get("/", query_params={"sortBy": "id", "sort": "asc"}) |
425 | 427 | stmt = AddressAdmin().sort_query(query, request)
|
426 | 428 | assert "ORDER BY addresses.id ASC" in str(stmt)
|
427 | 429 |
|
428 |
| - request = RequestFactory().get("/", query_params={"sortBy":"user.name", "sort":"desc"}) |
| 430 | + request = RequestFactory().get( |
| 431 | + "/", query_params={"sortBy": "user.name", "sort": "desc"} |
| 432 | + ) |
429 | 433 | stmt = AddressAdmin().sort_query(query, request)
|
430 | 434 | assert "ORDER BY users.name DESC" in str(stmt)
|
431 | 435 |
|
432 |
| - request = RequestFactory().get("/", query_params={"sortBy":"user.profile.role", "sort":"asc"}) |
| 436 | + request = RequestFactory().get( |
| 437 | + "/", query_params={"sortBy": "user.profile.role", "sort": "asc"} |
| 438 | + ) |
433 | 439 | stmt = AddressAdmin().sort_query(query, request)
|
434 | 440 | assert "ORDER BY profiles.role ASC" in str(stmt)
|
435 | 441 |
|
|
0 commit comments