From 0af0424b4f0b2b4e4c2f8e527dc22cbb602d74fb Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?St=C3=A9phane=20Bidoul?= <stephane.bidoul@acsone.eu>
Date: Thu, 23 Mar 2023 20:09:59 +0100
Subject: [PATCH] session_db: add a few tests

---
 session_db/tests/__init__.py              |  1 +
 session_db/tests/test_pg_session_store.py | 70 +++++++++++++++++++++++
 2 files changed, 71 insertions(+)
 create mode 100644 session_db/tests/__init__.py
 create mode 100644 session_db/tests/test_pg_session_store.py

diff --git a/session_db/tests/__init__.py b/session_db/tests/__init__.py
new file mode 100644
index 000000000..22a56c889
--- /dev/null
+++ b/session_db/tests/__init__.py
@@ -0,0 +1 @@
+from . import test_pg_session_store
diff --git a/session_db/tests/test_pg_session_store.py b/session_db/tests/test_pg_session_store.py
new file mode 100644
index 000000000..8e57ce53b
--- /dev/null
+++ b/session_db/tests/test_pg_session_store.py
@@ -0,0 +1,70 @@
+from unittest import mock
+
+import psycopg2
+
+from odoo import http
+from odoo.sql_db import connection_info_for
+from odoo.tests.common import TransactionCase
+from odoo.tools import config
+
+from odoo.addons.session_db.pg_session_store import PGSessionStore
+
+
+def _make_postgres_uri(
+    login=None, password=None, host=None, port=None, database=None, **kwargs
+):
+    uri = ["postgres://"]
+    if login:
+        uri.append(login)
+        if password:
+            uri.append(f":{password}")
+        uri.append("@")
+    if host:
+        uri.append(host)
+        if port:
+            uri.append(f":{port}")
+    uri.append("/")
+    if database:
+        uri.append(database)
+    return "".join(uri)
+
+
+class TestPGSessionStore(TransactionCase):
+    def setUp(self):
+        super().setUp()
+        _, connection_info = connection_info_for(config["db_name"])
+        self.session_store = PGSessionStore(
+            _make_postgres_uri(**connection_info), session_class=http.Session
+        )
+
+    def test_session_crud(self):
+        session = self.session_store.new()
+        session["test"] = "test"
+        self.session_store.save(session)
+        assert session.sid is not None
+        assert self.session_store.get(session.sid)["test"] == "test"
+        self.session_store.delete(session)
+        assert self.session_store.get(session.sid).get("test") is None
+
+    def test_retry(self):
+        """Test that session operations are retried before failing"""
+        with mock.patch("odoo.sql_db.Cursor.execute") as mock_execute:
+            mock_execute.side_effect = psycopg2.OperationalError()
+            with self.assertRaises(psycopg2.OperationalError):
+                self.session_store.get("abc")
+            assert mock_execute.call_count == 5
+        # when the error is resolved, it works again
+        self.session_store.get("abc")
+
+    def test_retry_connect_fail(self):
+        with mock.patch("odoo.sql_db.Cursor.execute") as mock_execute, mock.patch(
+            "odoo.sql_db.db_connect"
+        ) as mock_db_connect:
+            mock_execute.side_effect = psycopg2.OperationalError()
+            mock_db_connect.side_effect = RuntimeError("connection failed")
+            # get fails, and a RuntimeError is raised when trying to reconnect
+            with self.assertRaises(RuntimeError):
+                self.session_store.get("abc")
+            assert mock_execute.call_count == 1
+        # when the error is resolved, it works again
+        self.session_store.get("abc")