Add status method for basic state (#8)

* add status method for basic state.

* Update test_interface.py

Author: ctalkington

directv/directv.py

@@ -196,6 +196,16 @@ async def state(self, client: str = "0") -> State:
             program=program,
         )
 
+    async def status(self, client: str = "0") -> str:
+        """Get basic status of receiver client."""
+        try:
+            mode = await self._request("info/mode", params={"clientAddr": client})
+            return "standby" if mode["mode"] == 1 else "active"
+        except DIRECTVAccessRestricted:
+            return "unauthorized"
+        except DIRECTVError:
+            return "unavailable"
+
     async def tune(self, channel: str, client: str = "0") -> None:
         """Change the channel on the receiver."""
         major, minor = parse_channel_number(channel)

tests/test_interface.py

@@ -274,6 +274,90 @@ async def test_state_standby(aresponses):
         assert response.program is None
 
 
+@pytest.mark.asyncio
+async def test_status(aresponses):
+    """Test active state is handled correctly."""
+    aresponses.add(
+        MATCH_HOST,
+        "/info/mode",
+        "GET",
+        aresponses.Response(
+            status=200,
+            headers={"Content-Type": "application/json"},
+            text=load_fixture("info-mode.json"),
+        ),
+    )
+
+    async with ClientSession() as session:
+        dtv = DIRECTV(HOST, session=session)
+        response = await dtv.status()
+
+        assert response == "active"
+
+
+@pytest.mark.asyncio
+async def test_status_access_restricted(aresponses):
+    """Test unauthorized state is handled correctly."""
+    aresponses.add(
+        MATCH_HOST,
+        "/info/mode",
+        "GET",
+        aresponses.Response(
+            status=403,
+            headers={"Content-Type": "application/json"},
+            text=load_fixture("info-mode-restricted.json"),
+        ),
+    )
+
+    async with ClientSession() as session:
+        dtv = DIRECTV(HOST, session=session)
+        response = await dtv.status()
+
+        assert response == "unauthorized"
+
+
+@pytest.mark.asyncio
+async def test_status_standby(aresponses):
+    """Test standby status is handled correctly."""
+    aresponses.add(
+        MATCH_HOST,
+        "/info/mode",
+        "GET",
+        aresponses.Response(
+            status=200,
+            headers={"Content-Type": "application/json"},
+            text=load_fixture("info-mode-standby.json"),
+        ),
+    )
+
+    async with ClientSession() as session:
+        dtv = DIRECTV(HOST, session=session)
+        response = await dtv.status()
+
+        assert response == "standby"
+
+
+@pytest.mark.asyncio
+async def test_status_unavailable(aresponses):
+    """Test unavailable status is handled correctly."""
+    aresponses.add(
+        MATCH_HOST,
+        "/info/mode",
+        "GET",
+        aresponses.Response(
+            status=500,
+            headers={"Content-Type": "application/json"},
+            text=load_fixture("info-mode-error.json"),
+        ),
+    )
+
+    async with ClientSession() as session:
+        dtv = DIRECTV(HOST, session=session)
+        response = await dtv.status()
+
+        assert response == "unavailable"
+
+
 @pytest.mark.asyncio
 async def test_tune(aresponses):
     """Test tune is handled correctly."""