Add perf tests and pgbouncer in dev

core/settings/common.py

@@ -12,6 +12,7 @@
 # update
 
 import os
+import sys
 import json
 
 import environ
@@ -136,6 +137,11 @@ def get_installed_apps(proj_apps):
     }
 }
 
+# Skip pgbouncer for running tests
+if 'test' in sys.argv or 'test_coverage' in sys.argv:
+    DATABASES['default']['HOST'] = 'db'
+    DATABASES['default']['NAME'] = 'test_db'
+
 CONN_MAX_AGE = 600
 
 CACHES = {

docker-compose.yml

@@ -10,6 +10,20 @@ services:
       - ${DATE_DB_PORT}:5432
     environment:
       - POSTGRES_PASSWORD=${DATE_DB_PASSWORD}
+      - POSTGRES_MAX_CONNECTIONS=1000
+  pgbouncer:
+    image: edoburu/pgbouncer
+    depends_on:
+      - db
+    environment:
+      - DB_HOST=db
+      - DB_PORT=5432
+      - DB_USER=postgres
+      - DB_PASSWORD=${DATE_DB_PASSWORD}
+      - POOL_MODE=transaction
+      - AUTH_TYPE=scram-sha-256
+      - MAX_CLIENT_CONN=1000
+      - ADMIN_USERS=postgres,admin
   web:
     build: .
     restart: always
@@ -24,6 +38,7 @@ services:
       - DEBUG=${DATE_DEBUG}
       - DEVELOP=${DATE_DEVELOP}
       - PROJECT_NAME=${PROJECT_NAME}
+      - DB_HOST=pgbouncer
       - DB_PASSWORD=${DATE_DB_PASSWORD}
       - USE_S3=${USE_S3}
       - AWS_S3_ENDPOINT_URL=${S3_ENDPOINT_URL}
@@ -43,6 +58,7 @@ services:
       - CF_TURNSTILE_SITE_KEY=${CF_TURNSTILE_SITE_KEY}
       - CF_TURNSTILE_SECRET_KEY=${CF_TURNSTILE_SECRET_KEY}
     depends_on:
+      - pgbouncer
       - db
       - redis
   celery:

test/perf/test-event.js

@@ -0,0 +1,39 @@
+import http from 'k6/http';
+import { check } from 'k6';
+import { uuidv4 } from 'https://jslib.k6.io/k6-utils/1.4.0/index.js';
+
+export let options = {
+  stages: [
+    { duration: '5s', target: 500 }, // ramp-up to 10 users
+  ],
+};
+
+export default function () {
+  // Step 1: Get the CSRF token
+  let response = http.get('http://localhost:8000/events/perftest');
+  let csrfToken = response.html().find('input[name="csrfmiddlewaretoken"]').attr('value');
+
+  let uuid = uuidv4();
+
+  // Step 2: Post the form data with the CSRF token
+  let formData = {
+    csrfmiddlewaretoken: csrfToken,
+    // Add your form fields here
+    user: 'test',
+    email: uuid + '@test.datateknologerna.org',
+  };
+
+  let headers = {
+    'Content-Type': 'application/x-www-form-urlencoded',
+    'Referer': 'http://localhost:8000/events/perftest',
+    'Cookie': response.headers['Set-Cookie'],
+  };
+
+  let postResponse = http.post('http://localhost:8000/events/perftest/', formData, { headers: headers });
+
+  // Step 3: Check the response
+  check(postResponse, {
+    'status is 200': (r) => r.status === 200,
+    // Add more checks as needed
+  });
+}