Fix PostgrestClient.schema() not work. Add tests for PostgrestClient

Author: lqmanh

CHANGELOG.md

@@ -11,6 +11,10 @@
 
 - Remove `RequestBuilder.filter_in()` and `RequestBuilder.filter_out()`
 
+#### Fixed
+
+- Fix `PostgrestClient.schema()` not actually work
+
 ### v0.2.0
 
 #### Added

README.md

@@ -14,12 +14,18 @@ Status: **Unstable**
 
 ### Instructions
 
-#### With Poetry
+#### With Poetry (recommended)
 
 ```sh
 $ poetry add postgrest-py
 ```
 
+#### With Pip
+
+```sh
+$ pip install postgrest-py
+```
+
 ## USAGE
 
 ### Getting started
@@ -65,6 +71,20 @@ await client.from_("countries").eq("name", "Việt Nam").delete().execute()
 
 ### Stored procedures (RPC)
 
+## DEVELOPMENT
+
+```sh
+$ git clone https://github.com/lqmanh/postgrest-py.git
+$ cd postgrest-py
+$ poetry install
+```
+
+### Testing
+
+```sh
+$ poetry run pytest
+```
+
 ## CHANGELOG
 
 Read more [here](https://github.com/lqmanh/postgrest-py/blob/master/CHANGELOG.md).

TODO.md

@@ -3,5 +3,4 @@
 - [ ] Exact match filtering
 - [ ] AND and OR filters
 - [ ] Counting
-- [ ] Tests
 - [ ] CI/CD

poetry.lock

@@ -1,3 +1,5 @@
+from typing import Union
+
 from deprecation import deprecated
 from httpx import AsyncClient, Response
 
@@ -15,7 +17,7 @@ def __init__(self, base_url: str, *, schema="public") -> None:
             "Accept-Profile": schema,
             "Content-Profile": schema,
         }
-        self.session = AsyncClient(base_url=base_url, params={}, headers=headers)
+        self.session = AsyncClient(base_url=base_url, headers=headers)
 
     async def __aenter__(self):
         return self
@@ -26,7 +28,13 @@ async def __aexit__(self, exc_type, exc, tb) -> None:
     async def aclose(self) -> None:
         await self.session.aclose()
 
-    def auth(self, token: str, *, username: str = None, password: str = None):
+    def auth(
+        self,
+        token: str,
+        *,
+        username: Union[str, bytes] = None,
+        password: Union[str, bytes] = None,
+    ):
         """Authenticate the client with either bearer token or basic authentication."""
         if username:
             self.session.auth = (username, password)
@@ -36,7 +44,7 @@ def auth(self, token: str, *, username: str = None, password: str = None):
 
     def schema(self, schema: str):
         """Switch to another schema."""
-        self.session.merge_headers(
+        self.session.headers.update(
             {"Accept-Profile": schema, "Content-Profile": schema}
         )
         return self

postgrest_py/client.py

@@ -12,6 +12,7 @@ deprecation = "^2.1.0"
 
 [tool.poetry.dev-dependencies]
 pytest = "^6.0.1"
+pytest-asyncio = "^0.14.0"
 black = {version = "^19.10b0", allow-prereleases = true}
 pylint = "^2.5.3"
 

pyproject.toml

@@ -0,0 +1,53 @@
+import pytest
+from httpx import AsyncClient, Headers
+
+from postgrest_py import PostgrestClient
+
+
+@pytest.fixture
+async def postgrest_client():
+    async with PostgrestClient("https://example.com") as client:
+        yield client
+
+
+@pytest.mark.asyncio
+def test_constructor(postgrest_client):
+    session = postgrest_client.session
+
+    assert session.base_url == "https://example.com"
+    assert session.headers == Headers(
+        {
+            "accept": "application/json",
+            "content-type": "application/json",
+            "accept-profile": "public",
+            "content-profile": "public",
+        }
+    )
+
+
+class TestAuth:
+    @pytest.mark.asyncio
+    def test_auth_token(self, postgrest_client):
+        postgrest_client.auth("s3cr3t")
+        session = postgrest_client.session
+
+        assert session.headers["Authorization"] == "Bearer s3cr3t"
+
+    @pytest.mark.asyncio
+    def test_auth_basic(self, postgrest_client):
+        postgrest_client.auth(None, username="admin", password="s3cr3t")
+        session = postgrest_client.session
+
+        assert session.auth == ("admin", "s3cr3t")
+
+
+@pytest.mark.asyncio
+def test_schema(postgrest_client: PostgrestClient):
+    postgrest_client.schema("private")
+    session = postgrest_client.session
+    subheaders = {
+        "accept-profile": "private",
+        "content-profile": "private",
+    }
+
+    assert subheaders.items() < dict(session.headers).items()