From a756a74b49ee8e75a5306e0147b718be95171b06 Mon Sep 17 00:00:00 2001
From: kevgliss <kevgliss@gmail.com>
Date: Mon, 4 Dec 2017 13:13:02 -0800
Subject: [PATCH] Ensures we can get multiple endpoints with the same name but
 different ports. (#1011)

---
 lemur/endpoints/service.py        | 10 ++++++++++
 lemur/plugins/lemur_aws/plugin.py |  2 ++
 lemur/plugins/lemur_aws/sts.py    |  2 --
 lemur/sources/service.py          |  4 ++--
 4 files changed, 14 insertions(+), 4 deletions(-)

diff --git a/lemur/endpoints/service.py b/lemur/endpoints/service.py
index 8feb8fbe..6334dcac 100644
--- a/lemur/endpoints/service.py
+++ b/lemur/endpoints/service.py
@@ -60,6 +60,16 @@ def get_by_dnsname(dnsname):
     return database.get(Endpoint, dnsname, field='dnsname')
 
 
+def get_by_dnsname_and_port(dnsname, port):
+    """
+    Retrieves and endpoint by it's dnsname and port.
+    :param dnsname:
+    :param port:
+    :return:
+    """
+    return Endpoint.query.filter(Endpoint.dnsname == dnsname).filter(Endpoint.port == port).scalar()
+
+
 def get_by_source(source_label):
     """
     Retrieves all endpoints for a given source.
diff --git a/lemur/plugins/lemur_aws/plugin.py b/lemur/plugins/lemur_aws/plugin.py
index c9da599b..c15e30ca 100644
--- a/lemur/plugins/lemur_aws/plugin.py
+++ b/lemur/plugins/lemur_aws/plugin.py
@@ -109,6 +109,8 @@ def get_elb_endpoints(account_number, region, elb_dict):
                                                          account_number=account_number, region=region)
             endpoint['policy'] = format_elb_cipher_policy(policy)
 
+        current_app.logger.debug("Found new endpoint. Endpoint: {}".format(endpoint))
+
         endpoints.append(endpoint)
 
     return endpoints
diff --git a/lemur/plugins/lemur_aws/sts.py b/lemur/plugins/lemur_aws/sts.py
index d82297e5..0ef1c3f8 100644
--- a/lemur/plugins/lemur_aws/sts.py
+++ b/lemur/plugins/lemur_aws/sts.py
@@ -22,8 +22,6 @@ def sts_client(service, service_type='client'):
                 current_app.config.get('LEMUR_INSTANCE_PROFILE', 'Lemur')
             )
 
-            current_app.logger.debug('Assuming Role. Role: {0}'.format(arn))
-
             # TODO add user specific information to RoleSessionName
             role = sts.assume_role(RoleArn=arn, RoleSessionName='lemur')
 
diff --git a/lemur/sources/service.py b/lemur/sources/service.py
index dcb126fd..a1c19e4b 100644
--- a/lemur/sources/service.py
+++ b/lemur/sources/service.py
@@ -72,7 +72,7 @@ def sync_endpoints(source):
         return new, updated
 
     for endpoint in endpoints:
-        exists = endpoint_service.get_by_dnsname(endpoint['dnsname'])
+        exists = endpoint_service.get_by_dnsname_and_port(endpoint['dnsname'], endpoint['port'])
 
         certificate_name = endpoint.pop('certificate_name')
 
@@ -99,7 +99,7 @@ def sync_endpoints(source):
             new += 1
 
         else:
-            current_app.logger.debug("Endpoint Updated: Name: {name}".format(name=endpoint['name']))
+            current_app.logger.debug("Endpoint Updated: {}".format(endpoint))
             endpoint_service.update(exists.id, **endpoint)
             updated += 1