Tests: Fix MySQL tests (timezone issues)

Author: erezsh

data_diff/databases/base.py

@@ -267,6 +267,9 @@ def query(self, sql_ast: Union[Expr, Generator], res_type: type = list):
             if res is None:  # May happen due to sum() of 0 items
                 return None
             return int(res)
+        elif res_type is datetime:
+            res = _one(_one(res))
+            return res  # XXX parse timestamp?
         elif res_type is tuple:
             assert len(res) == 1, (sql_code, res)
             return res[0]

tests/test_cli.py

@@ -37,7 +37,10 @@ def setUp(self) -> None:
 
         src_table = table(table_src_name, schema={"id": int, "datetime": datetime, "text_comment": str})
         self.conn.query(src_table.create())
-        self.now = now = arrow.get(datetime.now())
+
+        self.conn.query("SET @@session.time_zone='+00:00'")
+        db_time = self.conn.query("select now()", datetime)
+        self.now = now = arrow.get(db_time)
 
         rows = [
             (now, "now"),