ci(iast): fix postgres connexion in integrations_packages (#13297)

In this previous pull request, we reduced the timeout for the suite
specs because some tests were migrated to integrations_packages.

https://github.com/DataDog/dd-trace-py/pull/13261/files#diff-f2d9ffc2ed86aa35a3c226322703b2f640a1806b97dfade074d3b6bed795061aR119

For some reason, the PostgreSQL database tests are having trouble
connecting to the database, which causes timeouts and makes those tests
flaky

## Checklist
- [x] PR author has checked that all the criteria below are met
- The PR description includes an overview of the change
- The PR description articulates the motivation for the change
- The change includes tests OR the PR description describes a testing
strategy
- The PR description notes risks associated with the change, if any
- Newly-added code is easy to change
- The change follows the [library release note
guidelines](https://ddtrace.readthedocs.io/en/stable/releasenotes.html)
- The change includes or references documentation updates if necessary
- Backport labels are set (if
[applicable](https://ddtrace.readthedocs.io/en/latest/contributing.html#backporting))

## Reviewer Checklist
- [x] Reviewer has checked that all the criteria below are met 
- Title is accurate
- All changes are related to the pull request's stated goal
- Avoids breaking
[API](https://ddtrace.readthedocs.io/en/stable/versioning.html#interfaces)
changes
- Testing strategy adequately addresses listed risks
- Newly-added code is easy to change
- Release note makes sense to a user of the library
- If necessary, author has acknowledged and discussed the performance
implications of this PR as reported in the benchmarks PR comment
- Backport labels are set in a manner that is consistent with the
[release branch maintenance
policy](https://ddtrace.readthedocs.io/en/latest/contributing.html#backporting)

Author: avara1986

hatch.toml

@@ -454,7 +454,7 @@ DD_IAST_DEDUPLICATION_ENABLED = "false"
 test = [
     "uname -a",
     "pip freeze",
-    "python -m pytest --no-ddtrace {args:tests/appsec/integrations/packages_tests/}",
+    "python -m pytest -vvv -s --no-cov --no-ddtrace {args:tests/appsec/integrations/packages_tests/}",
 ]
 
 [[envs.appsec_integrations_packages.matrix]]

tests/appsec/integrations/fixtures/sql_injection_mysqldb.py

@@ -1,13 +1,9 @@
-import os
-
 import MySQLdb
 from MySQLdb import OperationalError
 
 from ddtrace.appsec._iast._taint_tracking._taint_objects import get_tainted_ranges
 from ddtrace.appsec._iast._taint_tracking._taint_objects import is_pyobject_tainted
-
-
-MYSQL_HOST = os.getenv("TEST_MYSQL_HOST", "127.0.0.1")
+from tests.appsec.integrations.packages_tests.db_utils import MYSQL_HOST
 
 
 def get_connection():

tests/appsec/integrations/fixtures/sql_injection_psycopg2.py

@@ -1,4 +1,6 @@
 from psycopg2.errors import DuplicateTable
+from psycopg2.errors import InFailedSqlTransaction
+from psycopg2.errors import QueryCanceled
 
 from ddtrace.appsec._iast._taint_tracking._taint_objects import get_tainted_ranges
 from ddtrace.appsec._iast._taint_tracking._taint_objects import is_pyobject_tainted
@@ -10,9 +12,15 @@ def sqli_simple(table):
     cur = connection.cursor()
     try:
         cur.execute("CREATE TABLE students (name TEXT, addr TEXT, city TEXT, pin TEXT)")
-    except DuplicateTable:
-        connection.rollback()
-    # label test_sql_injection
-    cur.execute("SELECT 1 FROM " + table)
-    rows = cur.fetchone()
+    except (DuplicateTable, QueryCanceled):
+        pass
+
+    rows = []
+    try:
+        # label test_sql_injection
+        cur.execute("SELECT 1 FROM " + table)
+        rows = cur.fetchone()
+    except (QueryCanceled, InFailedSqlTransaction):
+        pass
+
     return {"result": rows, "tainted": is_pyobject_tainted(table), "ranges": str(get_tainted_ranges(table))}

tests/appsec/integrations/fixtures/sql_injection_sqlalchemy.py

@@ -4,11 +4,13 @@
 
 from ddtrace.appsec._iast._taint_tracking._taint_objects import get_tainted_ranges
 from ddtrace.appsec._iast._taint_tracking._taint_objects import is_pyobject_tainted
+from tests.appsec.integrations.packages_tests.db_utils import POSTGRES_HOST
 
 
 def sqli_simple(table):
-    engine = create_engine("postgresql://postgres:postgres@127.0.0.1/postgres")
+    engine = create_engine(f"postgresql://postgres:postgres@{POSTGRES_HOST}/postgres")
     with engine.connect() as connection:
+        connection.execute(text("SET statement_timeout = 1000"))
         try:
             connection.execute(text("CREATE TABLE students (name TEXT, addr TEXT, city TEXT, pin TEXT)"))
         except ProgrammingError:

tests/appsec/integrations/packages_tests/db_utils.py

@@ -10,7 +10,12 @@
 
 def get_psycopg2_connection():
     connection = psycopg2.connect(
-        user="postgres", password="postgres", host=POSTGRES_HOST, port="5432", database="postgres"
+        user="postgres",
+        password="postgres",
+        host=POSTGRES_HOST,
+        port=5432,
+        database="postgres",
+        options="-c statement_timeout=1000",
     )
     return connection