Added tests for current_timestamp + fixes for some of the dbs

Author: erezsh

data_diff/sqeleton/databases/base.py

@@ -302,7 +302,9 @@ def query(self, sql_ast: Union[Expr, Generator], res_type: type = list):
             return int(res)
         elif res_type is datetime:
             res = _one(_one(res))
-            return res  # XXX parse timestamp?
+            if isinstance(res, str):
+                res = datetime.fromisoformat(res[:23])  # TODO use a better parsing method
+            return res
         elif res_type is tuple:
             assert len(res) == 1, (sql_code, res)
             return res[0]

data_diff/sqeleton/databases/clickhouse.py

@@ -155,6 +155,9 @@ def _parse_type_repr(self, type_repr: str) -> Optional[Type[ColType]]:
     def set_timezone_to_utc(self) -> str:
         raise NotImplementedError()
 
+    def current_timestamp(self) -> str:
+        return "now()"
+
 
 class Clickhouse(ThreadedDatabase):
     dialect = Dialect()

data_diff/sqeleton/databases/duckdb.py

@@ -111,6 +111,8 @@ def parse_type(
     def set_timezone_to_utc(self) -> str:
         return "SET GLOBAL TimeZone='UTC'"
 
+    def current_timestamp(self) -> str:
+        return "current_timestamp"
 
 class DuckDB(Database):
     dialect = Dialect()

data_diff/sqeleton/databases/postgresql.py

@@ -90,6 +90,9 @@ def _convert_db_precision_to_digits(self, p: int) -> int:
     def set_timezone_to_utc(self) -> str:
         return "SET TIME ZONE 'UTC'"
 
+    def current_timestamp(self) -> str:
+        return "current_timestamp"
+
 
 class PostgreSQL(ThreadedDatabase):
     dialect = PostgresqlDialect()

data_diff/sqeleton/databases/presto.py

@@ -137,6 +137,8 @@ def parse_type(
     def set_timezone_to_utc(self) -> str:
         return "SET TIME ZONE '+00:00'"
 
+    def current_timestamp(self) -> str:
+        return "current_timestamp"
 
 class Presto(Database):
     dialect = Dialect()

data_diff/sqeleton/databases/vertica.py

@@ -144,6 +144,9 @@ def parse_type(
     def set_timezone_to_utc(self) -> str:
         return "SET TIME ZONE TO 'UTC'"
 
+    def current_timestamp(self) -> str:
+        return "current_timestamp(6)"
+
 
 class Vertica(ThreadedDatabase):
     dialect = Dialect()

data_diff/sqeleton/queries/__init__.py

@@ -18,6 +18,7 @@
     or_,
     leftjoin,
     rightjoin,
+    current_timestamp
 )
 from .ast_classes import Expr, ExprNode, Select, Count, BinOp, Explain, In, Code, Column
 from .extras import Checksum, NormalizeAsString, ApplyFuncAndNormalizeAsString

tests/sqeleton/test_database.py

@@ -1,10 +1,11 @@
 from typing import Callable, List
+from datetime import datetime
 import unittest
 
 from ..common import str_to_checksum, TEST_MYSQL_CONN_STRING
 from ..common import str_to_checksum, test_each_database_in_list, DiffTestCase, get_conn, random_table_suffix
 
-from data_diff.sqeleton.queries import table
+from data_diff.sqeleton.queries import table, current_timestamp
 
 from data_diff import databases as dbs
 from data_diff.databases import connect
@@ -63,3 +64,12 @@ def test_table_list(self):
 
         db.query(tbl.drop())
         assert not db.query(q)
+
+
+@test_each_database
+class TestQueries(unittest.TestCase):
+
+    def test_current_timestamp(self):
+        db = get_conn(self.db_cls)
+        res = db.query(current_timestamp(), datetime)
+        assert isinstance(res, datetime), (res, type(res))