fixes warning about sqlite3 conversion deprecation in python 3.12

mcfunley · Jun 14, 2024 · 0f86ed7 · 0f86ed7
1 parent b5d204c
commit 0f86ed7
Showing 1 changed file with 12 additions and 4 deletions.
diff --git a/tests/test_pugsql.py b/tests/test_pugsql.py
@@ -1,9 +1,11 @@
-import pugsql
-from pugsql import exceptions
-import pytest
 import threading
 from unittest import TestCase
 
+import pytest
+
+import pugsql
+from pugsql import exceptions
+
 
 def test_module():
     assert pugsql.module('tests/sql').sqlpaths == {'tests/sql',}
@@ -174,10 +176,16 @@ def test_pass_connect_args(self):
         import sqlite3
         from datetime import datetime
 
+        def _convert_timestamp(ts):
+            return datetime.strptime(ts.decode("utf-8"), "%Y-%m-%d %H:%M:%S")
+        sqlite3.register_converter('TIMESTAMP', _convert_timestamp)
+
         self.fixtures.disconnect()
         self.fixtures.connect(
             'sqlite:///./tests/data/fixtures.sqlite3',
-            connect_args={'detect_types': sqlite3.PARSE_DECLTYPES})
+            connect_args={
+                'detect_types': sqlite3.PARSE_DECLTYPES,
+            })
         date = self.fixtures.find_date(id=1)
         self.assertIs(datetime, type(date['created']))