♻ Refactor items and services endpoints to return count and data, and add CI tests

.github/workflows/test.yaml

@@ -0,0 +1,30 @@
+name: Pytest Backend
+
+on:
+  push:
+    branches:
+      - master
+
+jobs:
+
+  pytest:
+    runs-on: ubuntu-latest
+    defaults:
+        run:
+            working-directory: src
+    steps:
+      - name: Checkout
+        uses: actions/checkout@v2
+
+      - name: Setup python 3.10
+        uses: actions/setup-python@v2
+        with:
+          python-version: '3.10'
+
+      - name: docker-compose setup
+        run: |
+          docker-compose up -d
+
+      - name: docker-compose run pytest
+        run: |
+          docker exec src_backend_1 pytest --cov=app --cov-report=term-missing app/tests

src/backend/app/app/api/api_v1/endpoints/items.py

@@ -1,33 +1,38 @@
 from typing import Any
 
 from fastapi import APIRouter, HTTPException
-from sqlmodel import select
+from sqlmodel import select, func
 
 from app.api.deps import CurrentUser, SessionDep
-from app.models import Item, ItemCreate, ItemOut, ItemUpdate, Message
+from app.models import Item, ItemCreate, ItemOut, ItemUpdate, Message, ItemsOut
 
 router = APIRouter()
 
 
-@router.get("/", response_model=list[ItemOut])
+@router.get("/", response_model=ItemsOut)
 def read_items(
     session: SessionDep, current_user: CurrentUser, skip: int = 0, limit: int = 100
 ) -> Any:
     """
     Retrieve items.
     """
 
+    statment = select(func.count()).select_from(Item)
+    count = session.exec(statment).one()
+
     if current_user.is_superuser:
         statement = select(Item).offset(skip).limit(limit)
-        return session.exec(statement).all()
+        items =  session.exec(statement).all()
     else:
         statement = (
             select(Item)
             .where(Item.owner_id == current_user.id)
             .offset(skip)
             .limit(limit)
         )
-        return session.exec(statement).all()
+        items = session.exec(statement).all()
+
+    return ItemsOut(data=items, count=count)
 
 
 @router.get("/{id}", response_model=ItemOut)

src/backend/app/app/api/api_v1/endpoints/users.py

@@ -1,7 +1,7 @@
 from typing import Any, List
 
 from fastapi import APIRouter, Depends, HTTPException
-from sqlmodel import select
+from sqlmodel import select, func
 
 from app import crud
 from app.api.deps import (
@@ -16,6 +16,7 @@
     UserCreate,
     UserCreateOpen,
     UserOut,
+    UsersOut,
     UserUpdate,
     UserUpdateMe,
 )
@@ -27,15 +28,20 @@
 @router.get(
     "/",
     dependencies=[Depends(get_current_active_superuser)],
-    response_model=List[UserOut],
+    response_model=UsersOut
 )
 def read_users(session: SessionDep, skip: int = 0, limit: int = 100) -> Any:
     """
     Retrieve users.
     """
+
+    statment = select(func.count()).select_from(User)
+    count = session.exec(statment).one()
+
     statement = select(User).offset(skip).limit(limit)
     users = session.exec(statement).all()
-    return users
+
+    return UsersOut(data=users, count=count)
 
 
 @router.post(

src/backend/app/app/models.py

@@ -46,6 +46,11 @@ class UserOut(UserBase):
     id: int
 
 
+class UsersOut(BaseModel):
+    data: list[UserOut]
+    count: int
+
+
 # Shared properties
 class ItemBase(SQLModel):
     title: str
@@ -75,6 +80,12 @@ class Item(ItemBase, table=True):
 # Properties to return via API, id is always required
 class ItemOut(ItemBase):
     id: int
+    owner_id: int
+
+
+class ItemsOut(BaseModel):
+    data: list[ItemOut]
+    count: int
 
 
 # Generic message

src/backend/app/app/tests/api/api_v1/test_users.py

@@ -110,6 +110,7 @@ def test_retrieve_users(
     r = client.get(f"{settings.API_V1_STR}/users/", headers=superuser_token_headers)
     all_users = r.json()
 
-    assert len(all_users) > 1
-    for item in all_users:
+    assert len(all_users["data"]) > 1
+    assert "count" in all_users
+    for item in all_users["data"]:
         assert "email" in item

src/docker-compose.override.yml

@@ -28,6 +28,10 @@ services:
       - traefik.http.routers.${STACK_NAME?Variable not set}-traefik-public-http.rule=Host(`${DOMAIN?Variable not set}`)
       - traefik.http.services.${STACK_NAME?Variable not set}-traefik-public.loadbalancer.server.port=80
 
+  db:
+    ports:
+      - "5432:5432"
+
   pgadmin:
     ports:
       - "5050:5050"
@@ -84,14 +88,14 @@ services:
       - traefik.http.routers.${STACK_NAME?Variable not set}-frontend-http.rule=Host(`old-frontend.localhost.tiangolo.com`)
       - traefik.http.services.${STACK_NAME?Variable not set}-frontend.loadbalancer.server.port=80
 
-  new-frontend:
-    build:
-      context: ./new-frontend
-    labels:
-      - traefik.enable=true
-      - traefik.constraint-label-stack=${TRAEFIK_TAG?Variable not set}
-      - traefik.http.routers.${STACK_NAME?Variable not set}-new-frontend-http.rule=PathPrefix(`/`)
-      - traefik.http.services.${STACK_NAME?Variable not set}-new-frontend.loadbalancer.server.port=80
+#  new-frontend:
+#    build:
+#      context: ./new-frontend
+#    labels:
+#      - traefik.enable=true
+#      - traefik.constraint-label-stack=${TRAEFIK_TAG?Variable not set}
+#      - traefik.http.routers.${STACK_NAME?Variable not set}-new-frontend-http.rule=PathPrefix(`/`)
+#      - traefik.http.services.${STACK_NAME?Variable not set}-new-frontend.loadbalancer.server.port=80
 
 networks:
   traefik-public:

src/docker-compose.yml

@@ -191,16 +191,16 @@ services:
         - traefik.http.routers.${STACK_NAME?Variable not set}-frontend-http.rule=PathPrefix(`/`)
         - traefik.http.services.${STACK_NAME?Variable not set}-frontend.loadbalancer.server.port=80
 
-  new-frontend:
-    image: '${DOCKER_IMAGE_NEW_FRONTEND?Variable not set}:${TAG-latest}'
-    build:
-      context: ./new-frontend
-    deploy:
-      labels:
-        - traefik.enable=true
-        - traefik.constraint-label-stack=${TRAEFIK_TAG?Variable not set}
-        - traefik.http.routers.${STACK_NAME?Variable not set}-new-frontend-http.rule=PathPrefix(`/`)
-        - traefik.http.services.${STACK_NAME?Variable not set}-new-frontend.loadbalancer.server.port=80
+#  new-frontend:
+#    image: '${DOCKER_IMAGE_NEW_FRONTEND?Variable not set}:${TAG-latest}'
+#    build:
+#      context: ./new-frontend
+#    deploy:
+#      labels:
+#        - traefik.enable=true
+#        - traefik.constraint-label-stack=${TRAEFIK_TAG?Variable not set}
+#        - traefik.http.routers.${STACK_NAME?Variable not set}-new-frontend-http.rule=PathPrefix(`/`)
+#        - traefik.http.services.${STACK_NAME?Variable not set}-new-frontend.loadbalancer.server.port=80
 
 
 volumes: