Added check for datetime string format + tests

Author: Seluj78

shellhub/models/device.py

@@ -60,16 +60,23 @@ def __init__(self, api_object: shellhub.models.base.ShellHub, device_json):  # t
         self.info = ShellHubDeviceInfo(device_json["info"])
         self.public_key = device_json["public_key"]
         self.tenant_id = device_json["tenant_id"]
-        self.last_seen = datetime.fromisoformat(device_json["last_seen"])
+        self.last_seen = self._safe_isoformat_to_datetime(device_json["last_seen"])
         self.online = device_json["online"]
         self.namespace = device_json["namespace"]
         self.status = device_json["status"]
-        self.status_updated_at = datetime.fromisoformat(device_json["status_updated_at"])
-        self.created_at = datetime.fromisoformat(device_json["created_at"])
+        self.status_updated_at = self._safe_isoformat_to_datetime(device_json["status_updated_at"])
+        self.created_at = self._safe_isoformat_to_datetime(device_json["created_at"])
         self.remote_addr = device_json["remote_addr"]
         self.tags = device_json["tags"]
         self.acceptable = device_json["acceptable"]
 
+    @staticmethod
+    def _safe_isoformat_to_datetime(date_string: str) -> datetime:
+        try:
+            return datetime.fromisoformat(date_string)
+        except ValueError as e:
+            raise ShellHubApiError(f"Invalid date string: {date_string} (Couldn't convert to datetime)") from e
+
     def delete(self) -> bool:
         """
         Delete the device from the API

tests/test_devices.py

@@ -1,3 +1,5 @@
+from datetime import datetime
+
 import pytest
 
 from shellhub.exceptions import ShellHubApiError
@@ -118,16 +120,48 @@ def test_get_device(self, shellhub, requests_mock):
             == "-----BEGIN RSA PUBLIC KEY-----\nxxx\nxxx\nxxx\nxxx\nxxx\nxxx\n-----END RSA PUBLIC KEY-----\n"
         )
         assert device.tenant_id == "1"
-        assert device.last_seen == "1970-01-01T00:00:00Z"
+        assert device.last_seen == datetime.fromisoformat("1970-01-01T00:00:00Z")
         assert device.online
         assert device.namespace == "dev"
         assert device.status == "accepted"
-        assert device.status_updated_at == "1970-01-01T00:00:00Z"
-        assert device.created_at == "1970-01-01T00:00:00Z"
+        assert device.status_updated_at == datetime.fromisoformat("1970-01-01T00:00:00Z")
+        assert device.created_at == datetime.fromisoformat("1970-01-01T00:00:00Z")
         assert device.remote_addr == "0.0.0.0"
         assert device.tags == []
         assert not device.acceptable
 
+    def test_get_incorrect_datetime_format(self, shellhub, requests_mock):
+        mock_response = {
+            "uid": "1",
+            "name": "default",
+            "identity": {"mac": "06:04:ju:le:s7:08"},
+            "info": {
+                "id": "ubuntu",
+                "pretty_name": "Ubuntu 20.04.2 LTS",
+                "version": "v0.14.1",
+                "arch": "amd64",
+                "platform": "docker",
+            },
+            "public_key": "-----BEGIN RSA PUBLIC KEY-----\nxxx\nxxx\nxxx\n"
+            "xxx\nxxx\nxxx\n-----END RSA PUBLIC KEY-----\n",
+            "tenant_id": "1",
+            "last_seen": "-1",
+            "online": True,
+            "namespace": "dev",
+            "status": "accepted",
+            "status_updated_at": "-1",
+            "created_at": "-1",
+            "remote_addr": "0.0.0.0",
+            "position": {"latitude": 0, "longitude": 0},
+            "tags": [],
+            "public_url": False,
+            "public_url_address": "",
+            "acceptable": False,
+        }
+        requests_mock.get(f"{MOCKED_DOMAIN_URL}/api/devices/1", json=mock_response)
+        with pytest.raises(ShellHubApiError):
+            shellhub.get_device("1")
+
 
 class TestDeleteDevice:
     def test_delete_device(self, shellhub_device, requests_mock):