Fix issue loading Type using LinodeClient.load(...) (#365)

Author: lgarber-akamai

linode_api4/objects/linode.py

@@ -256,9 +256,10 @@ def _populate(self, json):
         """
         Allows changing the name "class" in JSON to "type_class" in python
         """
+
         super()._populate(json)
 
-        if "class" in json:
+        if json is not None and "class" in json:
             setattr(self, "type_class", json["class"])
         else:
             setattr(self, "type_class", None)

test/fixtures/linode_types_g6-nanode-1.json

@@ -0,0 +1,48 @@
+{
+  "disk": 20480,
+  "memory": 1024,
+  "transfer": 1000,
+  "addons": {
+    "backups": {
+      "price": {
+        "hourly": 0.003,
+        "monthly": 2
+      },
+      "region_prices": [
+        {
+          "id": "ap-west",
+          "hourly": 0.02,
+          "monthly": 20
+        },
+        {
+          "id": "ap-northeast",
+          "hourly": 0.02,
+          "monthly": 20
+        }
+      ]
+    }
+  },
+  "class": "nanode",
+  "network_out": 1000,
+  "vcpus": 1,
+  "gpus": 0,
+  "id": "g5-nanode-1",
+  "label": "Linode 1024",
+  "price": {
+    "hourly": 0.0075,
+    "monthly": 5
+  },
+  "region_prices": [
+    {
+      "id": "us-east",
+      "hourly": 0.02,
+      "monthly": 20
+    },
+    {
+      "id": "ap-northeast",
+      "hourly": 0.02,
+      "monthly": 20
+    }
+  ],
+  "successor": null
+}

test/unit/objects/linode_test.py

@@ -580,6 +580,15 @@ def test_get_type_gpu(self):
         self.assertEqual(t.gpus, 1)
         self.assertEqual(t._populated, True)
 
+    def test_load_type(self):
+        """
+        Tests that a type can be loaded using LinodeClient.load(...)
+        """
+
+        t = self.client.load(Type, "g6-nanode-1")
+        self.assertEqual(t._populated, True)
+        self.assertEqual(t.type_class, "nanode")
+
     def test_save_noforce(self):
         """
         Tests that a client will only save if changes are detected