Tests now use connect() instead of connect_to_uri()

Author: erezsh

tests/test_database_types.py

@@ -25,7 +25,7 @@
 )
 
 
-CONNS = {k: db.connect_to_uri(v, N_THREADS) for k, v in CONN_STRINGS.items()}
+CONNS = {k: db.connect.connect(v, N_THREADS) for k, v in CONN_STRINGS.items()}
 
 CONNS[db.MySQL].query("SET @@session.time_zone='+00:00'", None)
 oracle.SESSION_TIME_ZONE = postgresql.SESSION_TIME_ZONE = "UTC"

tests/test_diff_tables.py

@@ -6,7 +6,7 @@
 import preql
 import arrow  # comes with preql
 
-from data_diff.databases import connect_to_uri
+from data_diff.databases import connect
 from data_diff.diff_tables import TableDiffer, TableSegment, split_space
 from data_diff import databases as db
 from data_diff.utils import ArithAlphanumeric
@@ -21,7 +21,7 @@
 )
 
 DATABASE_URIS = {k.__name__: v for k, v in CONN_STRINGS.items()}
-DATABASE_INSTANCES = {k.__name__: connect_to_uri(v, N_THREADS) for k, v in CONN_STRINGS.items()}
+DATABASE_INSTANCES = {k.__name__: connect(v, N_THREADS) for k, v in CONN_STRINGS.items()}
 
 TEST_DATABASES = {x.__name__ for x in (db.MySQL, db.PostgreSQL, db.Oracle, db.Redshift, db.Snowflake, db.BigQuery)}
 

tests/test_postgresql.py

@@ -1,13 +1,13 @@
 import unittest
 
-from data_diff.databases.connect import connect_to_uri
+from data_diff.databases.connect import connect
 from data_diff import TableSegment, TableDiffer
 from .common import TEST_POSTGRESQL_CONN_STRING, random_table_suffix
 
 
 class TestWithConnection(unittest.TestCase):
     def setUp(self) -> None:
-        self.connection = connect_to_uri(TEST_POSTGRESQL_CONN_STRING)
+        self.connection = connect(TEST_POSTGRESQL_CONN_STRING)
 
         self.connection.query('CREATE EXTENSION IF NOT EXISTS "uuid-ossp";', None)