ha-atomberg-integration 1.1.1__tar.gz

ha_atomberg_integration-1.1.1/LICENSE

@@ -0,0 +1,21 @@
+MIT License
+
+Copyright (c) 2026 Ratul BMG
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in all
+copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+SOFTWARE.

ha_atomberg_integration-1.1.1/PKG-INFO

@@ -0,0 +1,86 @@
+Metadata-Version: 2.4
+Name: ha-atomberg-integration
+Version: 1.1.1
+Summary: Async Python client for the Atomberg public cloud API (fans, water purifiers, smart locks)
+Author: Ratul BMG
+License-Expression: MIT
+Project-URL: Homepage, https://github.com/ratulbmg/ha-atomberg-integration
+Project-URL: Issues, https://github.com/ratulbmg/ha-atomberg-integration/issues
+Keywords: atomberg,home-assistant,smart-home,iot
+Classifier: Development Status :: 4 - Beta
+Classifier: Framework :: AsyncIO
+Classifier: Intended Audience :: Developers
+Classifier: Programming Language :: Python :: 3
+Classifier: Typing :: Typed
+Requires-Python: >=3.12
+Description-Content-Type: text/markdown
+License-File: LICENSE
+Requires-Dist: aiohttp>=3.9
+Provides-Extra: test
+Requires-Dist: pytest>=8; extra == "test"
+Requires-Dist: pytest-asyncio>=0.23; extra == "test"
+Dynamic: license-file
+
+<p align="center"><img src="https://raw.githubusercontent.com/ratulbmg/ha-atomberg-integration/main/docs/logo.png" width="128" alt="ha-atomberg-integration logo"></p>
+
+# ha-atomberg-integration
+
+Async Python client for the [Atomberg public cloud API](https://developer.atomberg-iot.com/).
+It supports Atomberg fans, water purifiers, and smart locks, and is used by the
+Home Assistant `atomberg` integration.
+
+This is an unofficial community library. It is not affiliated with or endorsed by Atomberg.
+
+## Installation
+
+```bash
+pip install ha-atomberg-integration
+```
+
+## Usage
+
+You need an API key and a refresh token, which you generate in the Atomberg Home app
+(see the [developer portal](https://developer.atomberg-iot.com/)). Keep them private.
+
+```python
+import asyncio
+
+import aiohttp
+
+from atomberg_api import AtombergClient
+
+
+async def main() -> None:
+    async with aiohttp.ClientSession() as session:
+        client = AtombergClient(session, "YOUR_API_KEY", "YOUR_REFRESH_TOKEN")
+        for device in await client.async_get_devices():
+            print(device.name, device.device_type)
+        states = await client.async_get_states()
+        print(states)
+
+
+asyncio.run(main())
+```
+
+## API limits
+
+The public API allows 100 requests per day per account, shared with the Atomberg Home app
+and any other tools using the same account. One call to `async_get_states()` returns the
+state of every device.
+
+## Errors
+
+- `AtombergAuthError`: the API key or refresh token was rejected.
+- `AtombergConnectionError`: a network or server error.
+- `AtombergError`: any other unexpected response. The two errors above inherit from it.
+
+## Development
+
+```bash
+pip install -e ".[test]"
+pytest
+```
+
+## License
+
+MIT

ha_atomberg_integration-1.1.1/README.md

@@ -0,0 +1,63 @@
+<p align="center"><img src="https://raw.githubusercontent.com/ratulbmg/ha-atomberg-integration/main/docs/logo.png" width="128" alt="ha-atomberg-integration logo"></p>
+
+# ha-atomberg-integration
+
+Async Python client for the [Atomberg public cloud API](https://developer.atomberg-iot.com/).
+It supports Atomberg fans, water purifiers, and smart locks, and is used by the
+Home Assistant `atomberg` integration.
+
+This is an unofficial community library. It is not affiliated with or endorsed by Atomberg.
+
+## Installation
+
+```bash
+pip install ha-atomberg-integration
+```
+
+## Usage
+
+You need an API key and a refresh token, which you generate in the Atomberg Home app
+(see the [developer portal](https://developer.atomberg-iot.com/)). Keep them private.
+
+```python
+import asyncio
+
+import aiohttp
+
+from atomberg_api import AtombergClient
+
+
+async def main() -> None:
+    async with aiohttp.ClientSession() as session:
+        client = AtombergClient(session, "YOUR_API_KEY", "YOUR_REFRESH_TOKEN")
+        for device in await client.async_get_devices():
+            print(device.name, device.device_type)
+        states = await client.async_get_states()
+        print(states)
+
+
+asyncio.run(main())
+```
+
+## API limits
+
+The public API allows 100 requests per day per account, shared with the Atomberg Home app
+and any other tools using the same account. One call to `async_get_states()` returns the
+state of every device.
+
+## Errors
+
+- `AtombergAuthError`: the API key or refresh token was rejected.
+- `AtombergConnectionError`: a network or server error.
+- `AtombergError`: any other unexpected response. The two errors above inherit from it.
+
+## Development
+
+```bash
+pip install -e ".[test]"
+pytest
+```
+
+## License
+
+MIT

ha_atomberg_integration-1.1.1/pyproject.toml

@@ -0,0 +1,39 @@
+[build-system]
+requires = ["setuptools>=68"]
+build-backend = "setuptools.build_meta"
+
+[project]
+name = "ha-atomberg-integration"
+version = "1.1.1"
+description = "Async Python client for the Atomberg public cloud API (fans, water purifiers, smart locks)"
+readme = "README.md"
+license = "MIT"
+license-files = ["LICENSE"]
+requires-python = ">=3.12"
+authors = [{ name = "Ratul BMG" }]
+keywords = ["atomberg", "home-assistant", "smart-home", "iot"]
+classifiers = [
+  "Development Status :: 4 - Beta",
+  "Framework :: AsyncIO",
+  "Intended Audience :: Developers",
+  "Programming Language :: Python :: 3",
+  "Typing :: Typed",
+]
+dependencies = ["aiohttp>=3.9"]
+
+[project.optional-dependencies]
+test = ["pytest>=8", "pytest-asyncio>=0.23"]
+
+[project.urls]
+Homepage = "https://github.com/ratulbmg/ha-atomberg-integration"
+Issues = "https://github.com/ratulbmg/ha-atomberg-integration/issues"
+
+[tool.setuptools.packages.find]
+where = ["src"]
+
+[tool.setuptools.package-data]
+atomberg_api = ["py.typed"]
+
+[tool.pytest.ini_options]
+asyncio_mode = "auto"
+testpaths = ["tests"]

ha_atomberg_integration-1.1.1/setup.cfg

@@ -0,0 +1,4 @@
+[egg_info]
+tag_build = 
+tag_date = 0
+

ha_atomberg_integration-1.1.1/src/atomberg_api/__init__.py

@@ -0,0 +1,57 @@
+"""Async client for the Atomberg public cloud API."""
+
+from .client import (
+    API_URL,
+    COMMAND_BRIGHTNESS,
+    COMMAND_FALLBACK_MODE,
+    COMMAND_IDLE_SLEEP,
+    COMMAND_LED,
+    COMMAND_LIGHT_MODE,
+    COMMAND_MODE,
+    COMMAND_POWER,
+    COMMAND_SLEEP,
+    COMMAND_SLEEP_WINDOW,
+    COMMAND_SPEED,
+    COMMAND_START_UTC,
+    COMMAND_STOP_UTC,
+    COMMAND_TDS_THRESHOLD,
+    COMMAND_TIMER,
+    DEFAULT_FAN_COMMANDS,
+    DEVICE_TYPE_FAN,
+    DEVICE_TYPE_LOCK,
+    DEVICE_TYPE_WATER_PURIFIER,
+    AtombergAuthError,
+    AtombergClient,
+    AtombergConnectionError,
+    AtombergDevice,
+    AtombergError,
+    parse_state,
+)
+
+__all__ = [
+    "API_URL",
+    "COMMAND_BRIGHTNESS",
+    "COMMAND_FALLBACK_MODE",
+    "COMMAND_IDLE_SLEEP",
+    "COMMAND_LED",
+    "COMMAND_LIGHT_MODE",
+    "COMMAND_MODE",
+    "COMMAND_POWER",
+    "COMMAND_SLEEP",
+    "COMMAND_SLEEP_WINDOW",
+    "COMMAND_SPEED",
+    "COMMAND_START_UTC",
+    "COMMAND_STOP_UTC",
+    "COMMAND_TDS_THRESHOLD",
+    "COMMAND_TIMER",
+    "DEFAULT_FAN_COMMANDS",
+    "DEVICE_TYPE_FAN",
+    "DEVICE_TYPE_LOCK",
+    "DEVICE_TYPE_WATER_PURIFIER",
+    "AtombergAuthError",
+    "AtombergClient",
+    "AtombergConnectionError",
+    "AtombergDevice",
+    "AtombergError",
+    "parse_state",
+]

ha_atomberg_integration-1.1.1/src/atomberg_api/client.py

@@ -0,0 +1,235 @@
+"""Async client for the Atomberg public cloud API."""
+
+import asyncio
+from dataclasses import dataclass
+import json
+import time
+from typing import Any, Self
+
+from aiohttp import ClientError, ClientSession, ClientTimeout
+
+API_URL = "https://api.developer.atomberg-iot.com"
+REQUEST_TIMEOUT = ClientTimeout(total=30)
+ACCESS_TOKEN_LIFETIME = 23 * 60 * 60
+
+DEVICE_TYPE_FAN = "fan"
+DEVICE_TYPE_WATER_PURIFIER = "water_purifier"
+DEVICE_TYPE_LOCK = "lock"
+
+COMMAND_POWER = "power"
+COMMAND_SPEED = "speed"
+COMMAND_SLEEP = "sleep"
+COMMAND_TIMER = "timer"
+COMMAND_LED = "led"
+COMMAND_BRIGHTNESS = "brightness"
+COMMAND_LIGHT_MODE = "light_mode"
+
+COMMAND_MODE = "mode"
+COMMAND_FALLBACK_MODE = "fallback_mode"
+COMMAND_TDS_THRESHOLD = "tds_threshold"
+COMMAND_IDLE_SLEEP = "idle_sleep"
+COMMAND_SLEEP_WINDOW = "sleep_mode"
+COMMAND_START_UTC = "start_utc"
+COMMAND_STOP_UTC = "stop_utc"
+
+DEFAULT_FAN_COMMANDS = frozenset({COMMAND_POWER, COMMAND_SPEED})
+
+
+class AtombergError(Exception):
+    """Base error for the Atomberg API."""
+
+
+class AtombergConnectionError(AtombergError):
+    """Error talking to the Atomberg API."""
+
+
+class AtombergAuthError(AtombergError):
+    """The API key or refresh token was rejected."""
+
+
+@dataclass(frozen=True, slots=True)
+class AtombergDevice:
+    """A device registered to the Atomberg account."""
+
+    device_id: str
+    name: str
+    model: str | None
+    series: str | None
+    color: str | None
+    device_type: str
+    supported_commands: frozenset[str]
+
+    @classmethod
+    def from_api(cls, data: dict[str, Any]) -> Self:
+        """Build a device from an entry of ``get_list_of_devices``."""
+        commands = frozenset(
+            command["command"] for command in data.get("supported_commands") or []
+        )
+        device_type = data.get("device_type", DEVICE_TYPE_FAN)
+        return cls(
+            device_id=data["device_id"],
+            name=data["name"],
+            model=data.get("model"),
+            series=data.get("series"),
+            color=data.get("color"),
+            device_type=device_type,
+            supported_commands=commands
+            or (
+                DEFAULT_FAN_COMMANDS if device_type == DEVICE_TYPE_FAN else frozenset()
+            ),
+        )
+
+
+def parse_state(device: AtombergDevice, data: dict[str, Any]) -> dict[str, Any]:
+    """Map an entry of ``get_device_state`` to the keys used by the integration."""
+    if device.device_type in (DEVICE_TYPE_WATER_PURIFIER, DEVICE_TYPE_LOCK):
+        return {**data, "is_online": data.get("is_online", True)}
+    return {
+        "is_online": data.get("is_online", True),
+        COMMAND_POWER: data.get("power", False),
+        COMMAND_SPEED: data.get("last_recorded_speed"),
+        COMMAND_SLEEP: data.get("sleep_mode", False),
+        COMMAND_LED: data.get("led", False),
+        COMMAND_BRIGHTNESS: data.get("last_recorded_brightness"),
+        COMMAND_LIGHT_MODE: data.get("last_recorded_color"),
+        "timer_hours": data.get("timer_hours"),
+    }
+
+
+class AtombergClient:
+    """Async client for the Atomberg Home public API."""
+
+    def __init__(
+        self, session: ClientSession, api_key: str, refresh_token: str
+    ) -> None:
+        """Initialize the client."""
+        self._session = session
+        self._api_key = api_key
+        self._refresh_token = refresh_token
+        self._access_token: str | None = None
+        self._access_token_expires = 0.0
+        self._token_lock = asyncio.Lock()
+
+    async def async_get_devices(self) -> list[AtombergDevice]:
+        """Return all devices on the account."""
+        message = await self._async_authorized_request("GET", "/v1/get_list_of_devices")
+        try:
+            return [
+                AtombergDevice.from_api(device) for device in message["devices_list"]
+            ]
+        except (KeyError, TypeError) as err:
+            raise AtombergError("Unexpected device list response") from err
+
+    async def async_get_states(self) -> dict[str, dict[str, Any]]:
+        """Return the raw state of every device, keyed by device ID, in one call."""
+        message = await self._async_authorized_request(
+            "GET", "/v1/get_device_state", params={"device_id": "all"}
+        )
+        try:
+            return {state["device_id"]: state for state in message["device_state"]}
+        except (KeyError, TypeError) as err:
+            raise AtombergError("Unexpected device state response") from err
+
+    async def async_send_command(self, device_id: str, command: dict[str, Any]) -> None:
+        """Send a command to a device."""
+        await self._async_authorized_request(
+            "POST",
+            "/v1/send_command",
+            json_body={"device_id": device_id, "command": command},
+        )
+
+    async def async_get_guest_pin(
+        self, device_id: str, start_time: int | None = None, end_time: int | None = None
+    ) -> dict[str, Any]:
+        """Generate a temporary guest PIN for a lock."""
+        body: dict[str, Any] = {"device_id": device_id}
+        if start_time is not None and end_time is not None:
+            body["start_time"] = start_time
+            body["end_time"] = end_time
+        message = await self._async_authorized_request(
+            "POST", "/v1/get_lock_dynamic_pin", json_body=body
+        )
+        try:
+            data: dict[str, Any] = message["data"]
+        except (KeyError, TypeError) as err:
+            raise AtombergError("Unexpected guest PIN response") from err
+        return data
+
+    async def _async_get_access_token(self) -> str:
+        async with self._token_lock:
+            if self._access_token and time.monotonic() < self._access_token_expires:
+                return self._access_token
+            message = await self._async_request(
+                "GET",
+                "/v1/get_access_token",
+                bearer=self._refresh_token,
+                auth_error_statuses=(401, 403),
+            )
+            try:
+                self._access_token = message["access_token"]
+            except (KeyError, TypeError) as err:
+                raise AtombergError("Unexpected access token response") from err
+            self._access_token_expires = time.monotonic() + ACCESS_TOKEN_LIFETIME
+            return self._access_token
+
+    async def _async_authorized_request(
+        self,
+        method: str,
+        path: str,
+        *,
+        params: dict[str, str] | None = None,
+        json_body: dict[str, Any] | None = None,
+    ) -> Any:
+        token = await self._async_get_access_token()
+        try:
+            return await self._async_request(
+                method, path, bearer=token, params=params, json_body=json_body
+            )
+        except AtombergAuthError:
+            self._access_token = None
+        token = await self._async_get_access_token()
+        return await self._async_request(
+            method, path, bearer=token, params=params, json_body=json_body
+        )
+
+    async def _async_request(
+        self,
+        method: str,
+        path: str,
+        *,
+        bearer: str,
+        params: dict[str, str] | None = None,
+        json_body: dict[str, Any] | None = None,
+        auth_error_statuses: tuple[int, ...] = (401,),
+    ) -> Any:
+        headers = {
+            "Accept": "application/json",
+            "Authorization": f"Bearer {bearer}",
+            "x-api-key": self._api_key,
+        }
+        try:
+            async with self._session.request(
+                method,
+                f"{API_URL}{path}",
+                headers=headers,
+                params=params,
+                json=json_body,
+                timeout=REQUEST_TIMEOUT,
+            ) as response:
+                status = response.status
+                body = await response.text()
+        except (ClientError, TimeoutError) as err:
+            raise AtombergConnectionError(str(err)) from err
+
+        if status in auth_error_statuses:
+            raise AtombergAuthError(f"HTTP {status}")
+        if status >= 400:
+            raise AtombergConnectionError(f"HTTP {status}: {body[:200]}")
+
+        try:
+            data = json.loads(body)
+        except ValueError as err:
+            raise AtombergError("Response was not valid JSON") from err
+        if not isinstance(data, dict) or data.get("status") != "Success":
+            raise AtombergError(f"API reported failure: {body[:200]}")
+        return data.get("message")

ha_atomberg_integration-1.1.1/src/ha_atomberg_integration.egg-info/PKG-INFO

@@ -0,0 +1,86 @@
+Metadata-Version: 2.4
+Name: ha-atomberg-integration
+Version: 1.1.1
+Summary: Async Python client for the Atomberg public cloud API (fans, water purifiers, smart locks)
+Author: Ratul BMG
+License-Expression: MIT
+Project-URL: Homepage, https://github.com/ratulbmg/ha-atomberg-integration
+Project-URL: Issues, https://github.com/ratulbmg/ha-atomberg-integration/issues
+Keywords: atomberg,home-assistant,smart-home,iot
+Classifier: Development Status :: 4 - Beta
+Classifier: Framework :: AsyncIO
+Classifier: Intended Audience :: Developers
+Classifier: Programming Language :: Python :: 3
+Classifier: Typing :: Typed
+Requires-Python: >=3.12
+Description-Content-Type: text/markdown
+License-File: LICENSE
+Requires-Dist: aiohttp>=3.9
+Provides-Extra: test
+Requires-Dist: pytest>=8; extra == "test"
+Requires-Dist: pytest-asyncio>=0.23; extra == "test"
+Dynamic: license-file
+
+<p align="center"><img src="https://raw.githubusercontent.com/ratulbmg/ha-atomberg-integration/main/docs/logo.png" width="128" alt="ha-atomberg-integration logo"></p>
+
+# ha-atomberg-integration
+
+Async Python client for the [Atomberg public cloud API](https://developer.atomberg-iot.com/).
+It supports Atomberg fans, water purifiers, and smart locks, and is used by the
+Home Assistant `atomberg` integration.
+
+This is an unofficial community library. It is not affiliated with or endorsed by Atomberg.
+
+## Installation
+
+```bash
+pip install ha-atomberg-integration
+```
+
+## Usage
+
+You need an API key and a refresh token, which you generate in the Atomberg Home app
+(see the [developer portal](https://developer.atomberg-iot.com/)). Keep them private.
+
+```python
+import asyncio
+
+import aiohttp
+
+from atomberg_api import AtombergClient
+
+
+async def main() -> None:
+    async with aiohttp.ClientSession() as session:
+        client = AtombergClient(session, "YOUR_API_KEY", "YOUR_REFRESH_TOKEN")
+        for device in await client.async_get_devices():
+            print(device.name, device.device_type)
+        states = await client.async_get_states()
+        print(states)
+
+
+asyncio.run(main())
+```
+
+## API limits
+
+The public API allows 100 requests per day per account, shared with the Atomberg Home app
+and any other tools using the same account. One call to `async_get_states()` returns the
+state of every device.
+
+## Errors
+
+- `AtombergAuthError`: the API key or refresh token was rejected.
+- `AtombergConnectionError`: a network or server error.
+- `AtombergError`: any other unexpected response. The two errors above inherit from it.
+
+## Development
+
+```bash
+pip install -e ".[test]"
+pytest
+```
+
+## License
+
+MIT

ha_atomberg_integration-1.1.1/src/ha_atomberg_integration.egg-info/SOURCES.txt

@@ -0,0 +1,12 @@
+LICENSE
+README.md
+pyproject.toml
+src/atomberg_api/__init__.py
+src/atomberg_api/client.py
+src/atomberg_api/py.typed
+src/ha_atomberg_integration.egg-info/PKG-INFO
+src/ha_atomberg_integration.egg-info/SOURCES.txt
+src/ha_atomberg_integration.egg-info/dependency_links.txt
+src/ha_atomberg_integration.egg-info/requires.txt
+src/ha_atomberg_integration.egg-info/top_level.txt
+tests/test_client.py

ha_atomberg_integration-1.1.1/src/ha_atomberg_integration.egg-info/dependency_links.txt

@@ -0,0 +1 @@
+

ha_atomberg_integration-1.1.1/src/ha_atomberg_integration.egg-info/requires.txt

@@ -0,0 +1,5 @@
+aiohttp>=3.9
+
+[test]
+pytest>=8
+pytest-asyncio>=0.23

ha_atomberg_integration-1.1.1/src/ha_atomberg_integration.egg-info/top_level.txt

@@ -0,0 +1 @@
+atomberg_api

ha_atomberg_integration-1.1.1/tests/test_client.py

@@ -0,0 +1,294 @@
+"""Tests for the Atomberg API client."""
+
+from __future__ import annotations
+
+from collections import defaultdict
+import json
+from typing import Any
+
+import aiohttp
+import pytest
+
+from atomberg_api import (
+    API_URL,
+    AtombergAuthError,
+    AtombergClient,
+    AtombergConnectionError,
+    AtombergDevice,
+    AtombergError,
+)
+
+TOKEN_URL = f"{API_URL}/v1/get_access_token"
+DEVICES_URL = f"{API_URL}/v1/get_list_of_devices"
+STATE_URL = f"{API_URL}/v1/get_device_state"
+COMMAND_URL = f"{API_URL}/v1/send_command"
+PIN_URL = f"{API_URL}/v1/get_lock_dynamic_pin"
+TOKEN_RESPONSE = {"status": "Success", "message": {"access_token": "access-1"}}
+
+
+class _Response:
+    def __init__(self, status: int, body: str) -> None:
+        self.status = status
+        self._body = body
+
+    async def __aenter__(self) -> _Response:
+        return self
+
+    async def __aexit__(self, *args: object) -> None:
+        return None
+
+    async def text(self) -> str:
+        return self._body
+
+
+class FakeSession:
+    """Minimal stand-in for aiohttp.ClientSession."""
+
+    def __init__(self) -> None:
+        self._responses: dict[str, list[Any]] = defaultdict(list)
+        self.calls: dict[str, list[dict[str, Any]]] = defaultdict(list)
+
+    def add(
+        self,
+        url: str,
+        *,
+        payload: Any = None,
+        body: str | None = None,
+        status: int = 200,
+        exception: Exception | None = None,
+        repeat: bool = False,
+    ) -> None:
+        """Queue a response for a URL."""
+        text = body if body is not None else json.dumps(payload)
+        self._responses[url].append((status, text, exception, repeat))
+
+    def request(self, method: str, url: str, **kwargs: Any) -> _Response:
+        """Return the next queued response for the URL."""
+        self.calls[url].append({"method": method, **kwargs})
+        queue = self._responses[url]
+        status, text, exception, repeat = queue[0]
+        if not repeat and len(queue) > 1:
+            queue.pop(0)
+        if exception is not None:
+            raise exception
+        return _Response(status, text)
+
+
+@pytest.fixture
+def mocked() -> FakeSession:
+    """Return a fake session."""
+    return FakeSession()
+
+
+@pytest.fixture
+def client(mocked: FakeSession) -> AtombergClient:
+    """Return a client using the fake session."""
+    return AtombergClient(mocked, "key", "refresh")  # type: ignore[arg-type]
+
+
+def _calls(mocked: FakeSession, url: str) -> list[dict[str, Any]]:
+    return mocked.calls[url]
+
+
+async def test_token_is_reused(mocked: FakeSession, client: AtombergClient) -> None:
+    """One token request serves several calls and headers are set."""
+    mocked.add(TOKEN_URL, payload=TOKEN_RESPONSE)
+    mocked.add(
+        DEVICES_URL,
+        payload={"status": "Success", "message": {"devices_list": []}},
+        repeat=True,
+    )
+
+    assert await client.async_get_devices() == []
+    await client.async_get_devices()
+
+    token_calls = _calls(mocked, TOKEN_URL)
+    assert len(token_calls) == 1
+    assert token_calls[0]["headers"]["Authorization"] == "Bearer refresh"
+    assert token_calls[0]["headers"]["x-api-key"] == "key"
+    device_calls = _calls(mocked, DEVICES_URL)
+    assert device_calls[0]["headers"]["Authorization"] == "Bearer access-1"
+
+
+async def test_rejected_access_token_refreshes_once(
+    mocked: FakeSession, client: AtombergClient
+) -> None:
+    """A 401 triggers exactly one token refresh and retry before failing."""
+    mocked.add(TOKEN_URL, payload=TOKEN_RESPONSE, repeat=True)
+    mocked.add(DEVICES_URL, status=401, repeat=True)
+
+    with pytest.raises(AtombergAuthError):
+        await client.async_get_devices()
+
+    assert len(_calls(mocked, TOKEN_URL)) == 2
+    assert len(_calls(mocked, DEVICES_URL)) == 2
+
+
+@pytest.mark.parametrize(
+    ("status", "exception"),
+    [
+        (401, AtombergAuthError),
+        (403, AtombergAuthError),
+        (500, AtombergConnectionError),
+    ],
+)
+async def test_token_request_errors(
+    mocked: FakeSession,
+    client: AtombergClient,
+    status: int,
+    exception: type[AtombergError],
+) -> None:
+    """Token endpoint errors map to the right exception."""
+    mocked.add(TOKEN_URL, status=status)
+
+    with pytest.raises(exception):
+        await client.async_get_devices()
+
+
+async def test_connection_error(mocked: FakeSession, client: AtombergClient) -> None:
+    """A network failure maps to a connection error."""
+    mocked.add(TOKEN_URL, exception=TimeoutError())
+
+    with pytest.raises(AtombergConnectionError):
+        await client.async_get_devices()
+
+
+async def test_bad_token_response(mocked: FakeSession, client: AtombergClient) -> None:
+    """A token response without a token raises."""
+    mocked.add(TOKEN_URL, payload={"status": "Success", "message": {}})
+
+    with pytest.raises(AtombergError):
+        await client.async_get_devices()
+
+
+async def test_invalid_json(mocked: FakeSession, client: AtombergClient) -> None:
+    """A non-JSON body raises."""
+    mocked.add(TOKEN_URL, body="not json")
+
+    with pytest.raises(AtombergError):
+        await client.async_get_devices()
+
+
+@pytest.mark.parametrize(
+    "body",
+    [
+        {"status": "Failure", "message": "nope"},
+        {"status": "Success", "message": {"unexpected": []}},
+    ],
+)
+async def test_bad_device_list(
+    mocked: FakeSession, client: AtombergClient, body: dict[str, Any]
+) -> None:
+    """Unexpected device list responses raise."""
+    mocked.add(TOKEN_URL, payload=TOKEN_RESPONSE)
+    mocked.add(DEVICES_URL, payload=body)
+
+    with pytest.raises(AtombergError):
+        await client.async_get_devices()
+
+
+async def test_get_devices_parses(mocked: FakeSession, client: AtombergClient) -> None:
+    """Devices are parsed into AtombergDevice objects."""
+    mocked.add(TOKEN_URL, payload=TOKEN_RESPONSE)
+    mocked.add(
+        DEVICES_URL,
+        payload={
+            "status": "Success",
+            "message": {
+                "devices_list": [
+                    {"device_id": "abc", "name": "Fan", "series": "I1", "model": "M"}
+                ]
+            },
+        },
+    )
+
+    devices = await client.async_get_devices()
+
+    assert len(devices) == 1
+    assert isinstance(devices[0], AtombergDevice)
+    assert devices[0].device_id == "abc"
+
+
+async def test_get_states(mocked: FakeSession, client: AtombergClient) -> None:
+    """States are keyed by device ID."""
+    mocked.add(TOKEN_URL, payload=TOKEN_RESPONSE)
+    mocked.add(
+        STATE_URL,
+        payload={
+            "status": "Success",
+            "message": {"device_state": [{"device_id": "abc", "power": True}]},
+        },
+    )
+
+    assert await client.async_get_states() == {
+        "abc": {"device_id": "abc", "power": True}
+    }
+
+
+async def test_bad_states(mocked: FakeSession, client: AtombergClient) -> None:
+    """Unexpected state responses raise."""
+    mocked.add(TOKEN_URL, payload=TOKEN_RESPONSE)
+    mocked.add(STATE_URL, payload={"status": "Success", "message": {}})
+
+    with pytest.raises(AtombergError):
+        await client.async_get_states()
+
+
+async def test_send_command_payload(
+    mocked: FakeSession, client: AtombergClient
+) -> None:
+    """The command request body is correct."""
+    mocked.add(TOKEN_URL, payload=TOKEN_RESPONSE)
+    mocked.add(COMMAND_URL, payload={"status": "Success", "message": {}})
+
+    await client.async_send_command("abc", {"speed": 4})
+
+    assert _calls(mocked, COMMAND_URL)[0]["json"] == {
+        "device_id": "abc",
+        "command": {"speed": 4},
+    }
+
+
+@pytest.mark.parametrize(
+    ("window", "expected_body"),
+    [
+        ((None, None), {"device_id": "abc"}),
+        ((10, 20), {"device_id": "abc", "start_time": 10, "end_time": 20}),
+    ],
+)
+async def test_guest_pin_request(
+    mocked: FakeSession,
+    client: AtombergClient,
+    window: tuple[int | None, int | None],
+    expected_body: dict[str, Any],
+) -> None:
+    """The guest PIN request body and response parsing."""
+    mocked.add(TOKEN_URL, payload=TOKEN_RESPONSE)
+    mocked.add(
+        PIN_URL,
+        payload={"status": "Success", "message": {"data": {"otp": "123456"}}},
+    )
+
+    pin = await client.async_get_guest_pin("abc", *window)
+
+    assert pin == {"otp": "123456"}
+    assert _calls(mocked, PIN_URL)[0]["json"] == expected_body
+
+
+async def test_guest_pin_bad_response(
+    mocked: FakeSession, client: AtombergClient
+) -> None:
+    """A response without PIN data raises."""
+    mocked.add(TOKEN_URL, payload=TOKEN_RESPONSE)
+    mocked.add(PIN_URL, payload={"status": "Success", "message": {}})
+
+    with pytest.raises(AtombergError):
+        await client.async_get_guest_pin("abc")
+
+
+async def test_client_error_maps(mocked: FakeSession, client: AtombergClient) -> None:
+    """An aiohttp client error maps to a connection error."""
+    mocked.add(TOKEN_URL, exception=aiohttp.ClientError("boom"))
+
+    with pytest.raises(AtombergConnectionError):
+        await client.async_get_devices()