progress

Author: lgarber-akamai

test/unit/linode_client_test.py

@@ -1185,7 +1185,12 @@ def test_firewall_create(self):
             }
 
             f = self.client.networking.firewall_create(
-                "test-firewall-1", rules, status="enabled"
+                "test-firewall-1",
+                rules,
+                devices=FirewallDevicesOptions(
+                    linodes=[123], nodebalancers=[456], interfaces=[789]
+                ),
+                status="enabled",
             )
 
             self.assertIsNotNone(f)
@@ -1200,6 +1205,11 @@ def test_firewall_create(self):
                     "label": "test-firewall-1",
                     "status": "enabled",
                     "rules": rules,
+                    "devices": {
+                        "linodes": [123],
+                        "nodebalancers": [456],
+                        "interface": [789],
+                    },
                 },
             )
 
@@ -1214,6 +1224,47 @@ def test_get_firewalls(self):
 
         self.assertEqual(firewall.id, 123)
 
+    def test_get_firewall_settings(self):
+        """
+        Tests that firewall settings can be retrieved
+        """
+        settings = self.client.networking.firewall_settings()
+
+        assert settings.default_firewall_ids.vpc_interface == 123
+        assert settings.default_firewall_ids.public_interface == 456
+        assert settings.default_firewall_ids.linode == 789
+        assert settings.default_firewall_ids.nodebalancer == 321
+
+        settings.invalidate()
+
+        assert settings.default_firewall_ids.vpc_interface == 123
+        assert settings.default_firewall_ids.public_interface == 456
+        assert settings.default_firewall_ids.linode == 789
+        assert settings.default_firewall_ids.nodebalancer == 321
+
+    def test_update_firewall_settings(self):
+        """
+        Tests that firewall settings can be updated
+        """
+        settings = self.client.networking.firewall_settings()
+
+        settings.default_firewall_ids.vpc_interface = 321
+        settings.default_firewall_ids.public_interface = 654
+        settings.default_firewall_ids.linode = 987
+        settings.default_firewall_ids.nodebalancer = 123
+
+        with self.mock_put("networking/firewalls/settings") as m:
+            settings.save()
+
+            assert m.call_data == {
+                "default_firewall_ids": {
+                    "vpc_interface": 321,
+                    "public_interface": 654,
+                    "linode": 987,
+                    "nodebalancer": 123,
+                }
+            }
+
     def test_ip_addresses_share(self):
         """
         Tests that you can submit a correct ip addresses share api request.