feat(shield): Service and checks (#1504)

providers/aws/services/elb/elb_service.py

@@ -12,6 +12,8 @@ class ELB:
     def __init__(self, audit_info):
         self.service = "elb"
         self.session = audit_info.audit_session
+        self.audited_partition = audit_info.audited_partition
+        self.audited_account = audit_info.audited_account
         self.regional_clients = generate_regional_clients(self.service, audit_info)
         self.loadbalancers = []
         self.__threading_call__(self.__describe_load_balancers__)
@@ -48,6 +50,7 @@ class ELB:
                     self.loadbalancers.append(
                         LoadBalancer(
                             name=elb["LoadBalancerName"],
+                            arn=f"arn:{self.audited_partition}:elasticloadbalancing:{regional_client.region}:{self.audited_account}:loadbalancer/{elb['LoadBalancerName']}",
                             dns=elb["DNSName"],
                             region=regional_client.region,
                             scheme=elb["Scheme"],
@@ -85,6 +88,7 @@ class Listener(BaseModel):
 class LoadBalancer(BaseModel):
     name: str
     dns: str
+    arn: str
     region: str
     scheme: str
     access_logs: Optional[bool]

providers/aws/services/elb/elb_service_test.py

@@ -83,6 +83,10 @@ class Test_ELB_Service:
         assert elb.loadbalancers[0].name == "my-lb"
         assert elb.loadbalancers[0].region == AWS_REGION
         assert elb.loadbalancers[0].scheme == "internal"
+        assert (
+            elb.loadbalancers[0].arn
+            == f"arn:aws:elasticloadbalancing:{AWS_REGION}:{AWS_ACCOUNT_NUMBER}:loadbalancer/my-lb"
+        )
 
     # Test ELB Describe Load Balancers Attributes
     @mock_ec2
@@ -124,3 +128,7 @@ class Test_ELB_Service:
         assert elb.loadbalancers[0].region == AWS_REGION
         assert elb.loadbalancers[0].scheme == "internal"
         assert elb.loadbalancers[0].access_logs
+        assert (
+            elb.loadbalancers[0].arn
+            == f"arn:aws:elasticloadbalancing:{AWS_REGION}:{AWS_ACCOUNT_NUMBER}:loadbalancer/my-lb"
+        )