From c398d02380d5c5cc0f2a7b06afd3cbd593d09eb1 Mon Sep 17 00:00:00 2001
From: Federico Mon <federico.mon@datadoghq.com>
Date: Thu, 28 Nov 2024 10:30:47 +0100
Subject: [PATCH] add psycopg to iast tests

---
 riotfile.py | 14 ++++++++++++++
 1 file changed, 14 insertions(+)

diff --git a/riotfile.py b/riotfile.py
index 2133641a05..f860ca387e 100644
--- a/riotfile.py
+++ b/riotfile.py
@@ -159,6 +159,20 @@ def select_pys(min_version: str = MIN_PYTHON_VERSION, max_version: str = MAX_PYT
                 "DD_IAST_REQUEST_SAMPLING": "100",  # Override default 30% to analyze all IAST requests
                 "_DD_APPSEC_DEDUPLICATION_ENABLED": "false",
             },
+            venvs=[
+                Venv(
+                    pys=select_pys(min_version="3.10", max_version="3.12"),
+                    pkgs={
+                        "psycopg[binary]": latest,
+                    },
+                ),
+                Venv(
+                    pys=select_pys(min_version="3.7", max_version="3.9"),
+                    pkgs={
+                        "psycopg2-binary": "~=2.9.9",
+                    },
+                ),
+            ],
         ),
         Venv(
             name="appsec_iast_memcheck",