Refactor AzureLoadBalancer to simplify private IP address retrieval; update test case for frontend_ip_configurations

Author: devanshjainms

src/modules/get_azure_lb.py

@@ -246,8 +246,6 @@ def get_private_ip_from_config(config):
             private_ip = (
                 config.get("private_ip_address")
                 or config.get("privateIpAddress")
-                or config.get("properties", {}).get("private_ip_address")
-                or config.get("properties", {}).get("privateIpAddress")
             )
             return private_ip
 

tests/modules/get_azure_lb_test.py

@@ -247,7 +247,7 @@ def __init__(self):
                 self.name = "nested-lb"
                 self.location = "test"
                 self.frontend_ip_configurations = [
-                    {"properties": {"private_ip_address": "10.0.0.5"}}
+                     {"private_ip_address": "10.0.0.5"}
                 ]
                 self.load_balancing_rules = []
                 self.probes = []