progress

Author: lgarber-akamai

test/unit/objects/firewall_test.py

@@ -65,7 +65,28 @@ def test_get_devices(self):
         Tests that devices can be pulled from a firewall
         """
         firewall = Firewall(self.client, 123)
-        self.assertEqual(len(firewall.devices), 1)
+        assert len(firewall.devices) == 2
+
+        assert firewall.devices[0].created is not None
+        assert firewall.devices[0].id == 123
+        assert firewall.devices[0].updated is not None
+
+        assert firewall.devices[0].entity.id == 123
+        assert firewall.devices[0].entity.label == "my-linode"
+        assert firewall.devices[0].entity.type == "linode"
+        assert firewall.devices[0].entity.url == "/v4/linode/instances/123"
+
+        assert firewall.devices[1].created is not None
+        assert firewall.devices[1].id == 456
+        assert firewall.devices[1].updated is not None
+
+        assert firewall.devices[1].entity.id == 123
+        assert firewall.devices[1].entity.label is None
+        assert firewall.devices[1].entity.type == "interface"
+        assert (
+            firewall.devices[1].entity.url
+            == "/v4/linode/instances/123/interfaces/123"
+        )
 
     def test_get_device(self):
         """