Merge pull request #156 from grillazz/152-implement-acyncpg-coon-pool-poc

add performance tests with locust

Author: grillazz

.github/workflows/build-and-test.yml

@@ -60,6 +60,6 @@ jobs:
         if: steps.cached-poetry-dependencies.outputs.cache-hit != 'true'
         run: poetry install --no-interaction --no-root
       - name: Test Code
-        run: poetry run pytest
+        run: poetry run pytest tests/
       - name: Lint Code
         run: poetry run ruff .

Makefile

@@ -27,7 +27,7 @@ docker-create-db-migration:  ## Create new alembic database migration aka databa
 
 .PHONY: docker-test
 docker-test:	## Run project tests
-	docker-compose -f docker-compose.yml -f docker-compose.test.yml  run --rm app pytest
+	docker-compose -f docker-compose.yml -f docker-compose.test.yml  run --rm app pytest tests
 
 .PHONY: docker-test-snapshot
 docker-test-snapshot:	## Run project tests with inline snapshot

app/api/stuff.py

@@ -1,5 +1,4 @@
 from fastapi import APIRouter, Depends, HTTPException, status, Request
-from fastapi.exceptions import ResponseValidationError
 from sqlalchemy.exc import SQLAlchemyError
 from sqlalchemy.ext.asyncio import AsyncSession
 

app/models/stuff.py

@@ -1,6 +1,5 @@
 import uuid
 
-from fastapi import HTTPException, status
 from sqlalchemy import String, select, ForeignKey
 from sqlalchemy.dialects.postgresql import UUID
 from sqlalchemy.ext.asyncio import AsyncSession

performance/__init__.py

@@ -0,0 +1,13 @@
+from locust import HttpUser, task, between
+
+
+class Stuff(HttpUser):
+    wait_time = between(1, 3)
+
+    @task
+    def find_stuff(self):
+        self.client.get("/v1/stuff/string")
+
+    @task
+    def find_stuff_with_pool(self):
+        self.client.get("/v1/stuff/pool/string")

performance/locustfile.py

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "fastapi-sqlalchemy-asyncpg"
-version = "0.0.12"
+version = "0.0.13"
 description = ""
 authors = ["Jakub Miazek <the@grillazz.com>"]
 packages = []
@@ -32,18 +32,16 @@ inline-snapshot = "^0.9.0"
 dirty-equals = "^0.7.1.post0"
 polyfactory = "^2.15.0"
 
-[tool.poetry.dev-dependencies]
+[tool.poetry.group.dev.dependencies]
 devtools = { extras = ["pygments"], version = "*" }
 safety = "*"
 pyupgrade = "*"
 ipython = "8.24.0"
 ruff = "0.4.3"
 psycopg2-binary = "*"
 sqlacodegen = "3.0.0rc5"
-
-
-[tool.poetry.group.dev.dependencies]
 tryceratops = "^2.3.2"
+locust = "^2.27.0"
 
 [build-system]
 requires = ["poetry-core>=1.0.0"]