Skip to content

Commit 5d36cf3

Browse files
committed
Remove old customizaion from tests
1 parent 55b32ee commit 5d36cf3

File tree

2 files changed

+8
-5
lines changed

2 files changed

+8
-5
lines changed

tests/ext/test_cassandra.py

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -7,6 +7,7 @@
77

88
from fastapi_pagination import add_pagination
99
from fastapi_pagination.cursor import CursorPage as BaseCursorPage
10+
from fastapi_pagination.customization import CustomizedPage, UseParamsFields
1011

1112
from ..schemas import UserOut
1213

@@ -15,7 +16,10 @@
1516
except ImportError:
1617
paginate = None
1718

18-
CursorPage = BaseCursorPage.with_custom_options(str_cursor=False)
19+
CursorPage = CustomizedPage[
20+
BaseCursorPage,
21+
UseParamsFields(str_cursor=False),
22+
]
1923

2024

2125
class User(models.Model):

tests/utils.py

Lines changed: 3 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -4,8 +4,7 @@
44
from pydantic import BaseModel
55

66
from fastapi_pagination import LimitOffsetPage, Page
7-
from fastapi_pagination.default import OptionalParams
8-
from fastapi_pagination.limit_offset import OptionalLimitOffsetParams
7+
from fastapi_pagination.customization import CustomizedPage, UseOptionalParams
98
from fastapi_pagination.utils import IS_PYDANTIC_V2
109

1110
faker = Faker()
@@ -32,8 +31,8 @@ def normalize(model: Type[T], *models: Any) -> List[T]:
3231
return [parse_obj(model, m) for m in models]
3332

3433

35-
OptionalPage = Page.with_params(OptionalParams)
36-
OptionalLimitOffsetPage = LimitOffsetPage.with_params(OptionalLimitOffsetParams)
34+
OptionalPage = CustomizedPage[Page, UseOptionalParams()]
35+
OptionalLimitOffsetPage = CustomizedPage[LimitOffsetPage, UseOptionalParams()]
3736

3837

3938
__all__ = [

0 commit comments

Comments
 (0)