fix: add missing imports - DIA-46202 (#69)

Author: hadrien

fastapi_sqla/__init__.py

@@ -5,6 +5,8 @@
 from fastapi_sqla import sqla
 from fastapi_sqla.sqla import (
     Base,
+    Collection,
+    Item,
     Page,
     Paginate,
     PaginateSignature,
@@ -15,6 +17,8 @@
 
 __all__ = [
     "Base",
+    "Collection",
+    "Item",
     "Page",
     "Paginate",
     "PaginateSignature",

tests/test_import.py

@@ -1,11 +1,25 @@
 from pytest import mark
 
 
-def test_import_fastapi_sqla():
-    import fastapi_sqla  # noqa
+def test_import_sync_api():
+    from fastapi_sqla import (  # noqa
+        Base,
+        Collection,
+        Item,
+        Page,
+        Paginate,
+        Session,
+        open_session,
+        setup,
+    )
 
 
 @mark.require_asyncpg
 @mark.sqlalchemy("1.4")
-def test_import_async_session():
-    from fastapi_sqla import AsyncSession  # noqa
+def test_import_async_api():
+    from fastapi_sqla import (  # noqa
+        AsyncPaginate,
+        AsyncPagination,
+        AsyncSession,
+        open_async_session,
+    )