From a34500e2ab47a260378ee8a60653ddb7f644ed64 Mon Sep 17 00:00:00 2001
From: Tibor Arpas <tibor.arpas@infinit.sk>
Date: Tue, 31 Oct 2023 14:02:33 +0100
Subject: [PATCH] Open testmondata in readonly mode in xdist workers. This
 hopefully fixes re #203 sqlite3.OperationalError: database is locked .

---
 testmon/__init__.py       |  2 +-
 testmon/db.py             | 90 ++++++++++++++++++++++-----------------
 testmon/pytest_testmon.py |  1 +
 testmon/testmon_core.py   |  5 ++-
 4 files changed, 58 insertions(+), 40 deletions(-)

diff --git a/testmon/__init__.py b/testmon/__init__.py
index 738ddcd..c048349 100644
--- a/testmon/__init__.py
+++ b/testmon/__init__.py
@@ -1,2 +1,2 @@
 """PYTEST_DONT_REWRITE"""
-VERSION = "2.0.13"
+VERSION = "2.0.14.dev0"
diff --git a/testmon/db.py b/testmon/db.py
index 39e7048..12bec46 100644
--- a/testmon/db.py
+++ b/testmon/db.py
@@ -44,10 +44,11 @@ def check_fingerprint_db(files_methods_checksums, record):
 
 
 class DB:
-    def __init__(self, datafile):
+    def __init__(self, datafile, readonly=False):
+        self._readonly = readonly
         new_db = not os.path.exists(datafile)
 
-        connection = connect(datafile)
+        connection = connect(datafile, readonly)
         self.con = connection
         old_format = self._check_data_version(datafile)
 
@@ -483,10 +484,11 @@ def delete_filenames(self, con):
 
     def determine_tests(self, exec_id, files_mhashes):
         with self.con as con:
-            con.execute(
-                f"UPDATE test_execution set forced = NULL WHERE {self._test_execution_fk_column()} = ?",
-                [exec_id],
-            )
+            if not self._readonly:
+                con.execute(
+                    f"UPDATE test_execution set forced = NULL WHERE {self._test_execution_fk_column()} = ?",
+                    [exec_id],
+                )
             self.delete_filenames(con)
             con.executemany(
                 "INSERT INTO changed_files_mhashes VALUES (?, ?, ?)",
@@ -634,40 +636,52 @@ def fetch_or_create_environment(
         self, environment_name, system_packages, python_version
     ):
         with self.con as con:
-            try:
-                cursor = con.cursor()
-                cursor.execute(
-                    """
-                    INSERT INTO environment VALUES (?, ?, ?, ?)
-                    """,
-                    (
-                        None,
-                        environment_name,
-                        system_packages,
-                        python_version,
-                    ),
-                )
-                environment_id = cursor.lastrowid
-                count = cursor.execute(
-                    """
-                    SELECT count(*) as count FROM environment WHERE environment_name = ?
-                    """,
-                    (environment_name,),
-                ).fetchone()
-                packages_changed = count["count"] > 1
-            except sqlite3.IntegrityError:
-                environment = con.execute(
-                    """
-                    SELECT
-                    id as id, environment_name as name, system_packages as packages
-                    FROM environment
-                    WHERE environment_name = ?
-                    """,
-                    (environment_name,),
-                ).fetchone()
+            cursor = con.cursor()
+            environment = cursor.execute(
+                """
+                SELECT
+                id, environment_name, system_packages, python_version
+                FROM environment
+                WHERE environment_name = ?
+                """,
+                (environment_name,),
+            ).fetchone()
+
+            if environment:
                 environment_id = environment["id"]
+                packages_changed = (
+                    environment["system_packages"] != system_packages
+                    or environment["python_version"] != python_version
+                )
+            else:
                 packages_changed = False
-        return environment_id, packages_changed
+            if not environment or packages_changed:
+                try:
+                    cursor.execute(
+                        """
+                        INSERT INTO environment (environment_name, system_packages, python_version)
+                        VALUES (?, ?, ?)
+                        """,
+                        (
+                            environment_name,
+                            system_packages,
+                            python_version,
+                        ),
+                    )
+                    environment_id = cursor.lastrowid
+                except sqlite3.IntegrityError:
+                    environment = con.execute(
+                        """
+                        SELECT
+                        id as id, environment_name as name, system_packages as packages
+                        FROM environment
+                        WHERE environment_name = ?
+                        """,
+                        (environment_name,),
+                    ).fetchone()
+                    environment_id = environment["id"]
+
+            return environment_id, packages_changed
 
     def initiate_execution(
         self,
diff --git a/testmon/pytest_testmon.py b/testmon/pytest_testmon.py
index bec086c..35abf5f 100644
--- a/testmon/pytest_testmon.py
+++ b/testmon/pytest_testmon.py
@@ -175,6 +175,7 @@ def init_testmon_data(config):
         database=rpc_proxy,
         environment=environment,
         system_packages=system_packages,
+        readonly=parallelism_status(config) == "worker",
     )
     testmon_data.determine_stable(bool(rpc_proxy))
     config.testmon_data = testmon_data
diff --git a/testmon/testmon_core.py b/testmon/testmon_core.py
index 36db6a0..ac4a47e 100644
--- a/testmon/testmon_core.py
+++ b/testmon/testmon_core.py
@@ -141,6 +141,7 @@ def __init__(
         environment=None,
         system_packages=None,
         python_version=None,
+        readonly=False,
     ):
         self.rootdir = rootdir
         self.environment = environment if environment else "default"
@@ -154,7 +155,9 @@ def __init__(
         if database:
             self.db = database
         else:
-            self.db = db.DB(os.path.join(self.rootdir, get_data_file_path()))
+            self.db = db.DB(
+                os.path.join(self.rootdir, get_data_file_path()), readonly=readonly
+            )
 
         try:
             result = self.db.initiate_execution(