Overwriting Connector class is enough to operate with

[datasette-connectors.git] / tests / fixtures.py

diff --git a/tests/fixtures.py b/tests/fixtures.py
index 6b772c67ba258cc75f36b4bda7c8d963245dc4ff..4cb60cef297aaf5fe12e21ff3591e94f4c9ea613 100644 (file)
--- a/tests/fixtures.py
+++ b/tests/fixtures.py
@@ -1,13 +1,15 @@
 from datasette_connectors import monkey; monkey.patch_datasette()
-from datasette_connectors import connectors
-from . import dummy
-connectors.db_connectors['dummy'] = dummy
+from datasette_connectors.connectors import ConnectorList
+from .dummy import DummyConnector
+ConnectorList.add_connector('dummy', DummyConnector())
 
 from datasette.app import Datasette
+from datasette.utils.testing import TestClient
 import os
 import pytest
 import tempfile
 
+
 @pytest.fixture(scope='session')
 def app_client(max_returned_rows=None):
     with tempfile.TemporaryDirectory() as tmpdir:
@@ -20,7 +22,7 @@ def app_client(max_returned_rows=None):
                 'max_returned_rows': max_returned_rows or 1000,
             }
         )
-        client = ds.app().test_client
+        client = TestClient(ds.app())
         client.ds = ds
         yield client