Fixed pytest warning about TestClient class

simonw · May 4, 2020 · 450d2e2 · 450d2e2
1 parent b314e08
commit 450d2e2
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 5 deletions.
diff --git a/tests/test_config_dir.py b/tests/test_config_dir.py
@@ -3,7 +3,7 @@
 import sqlite3
 
 from datasette.app import Datasette
-from .fixtures import TestClient
+from .fixtures import TestClient as _TestClient
 
 PLUGIN = """
 from datasette import hookimpl
@@ -76,7 +76,7 @@ def config_dir_client(tmp_path_factory):
     )
 
     ds = Datasette([], config_dir=config_dir)
-    client = TestClient(ds.app())
+    client = _TestClient(ds.app())
     client.ds = ds
     yield client
 
@@ -137,7 +137,7 @@ def test_metadata_yaml(tmp_path_factory, filename):
     config_dir = tmp_path_factory.mktemp("yaml-config-dir")
     (config_dir / filename).write_text("title: Title from metadata", "utf-8")
     ds = Datasette([], config_dir=config_dir)
-    client = TestClient(ds.app())
+    client = _TestClient(ds.app())
     client.ds = ds
     response = client.get("/-/metadata.json")
     assert 200 == response.status

diff --git a/tests/test_plugins.py b/tests/test_plugins.py
@@ -4,7 +4,7 @@
     make_app_client,
     TABLES,
     TEMP_PLUGIN_SECRET_FILE,
-    TestClient,
+    TestClient as _TestClient,
 )  # noqa
 from datasette.app import Datasette
 from datasette.plugins import get_plugins, DEFAULT_PLUGINS
@@ -293,7 +293,7 @@ def extra_template_vars(view_name):
     db_path = str(tmpdir / "fixtures.db")
     conn = sqlite3.connect(db_path)
     conn.executescript(TABLES)
-    return TestClient(
+    return _TestClient(
         Datasette(
             [db_path], template_dir=str(templates), plugins_dir=str(plugins)
         ).app()