From 1fca64ab72b53979909782e78421f20043523bb3 Mon Sep 17 00:00:00 2001
From: Ilya Shakhat <ishakhat@mirantis.com>
Date: Fri, 6 Feb 2015 12:55:38 +0300
Subject: [PATCH] Fix LP profile validation function

Change-Id: I28f3412882126a5c2edec44cd1a680ab6534aaa0
---
 stackalytics/processor/utils.py |  5 +++--
 tests/unit/test_utils.py        | 13 +++++++++++++
 2 files changed, 16 insertions(+), 2 deletions(-)

diff --git a/stackalytics/processor/utils.py b/stackalytics/processor/utils.py
index e05e05e02..bc3ac7ac2 100644
--- a/stackalytics/processor/utils.py
+++ b/stackalytics/processor/utils.py
@@ -269,5 +269,6 @@ def normalize_company_draft(name):
 
 
 def validate_lp_display_name(lp_profile):
-    if "<email address hidden>" == lp_profile['display_name']:
-        lp_profile['display_name'] = lp_profile['name']
+    if lp_profile:
+        if "<email address hidden>" == lp_profile['display_name']:
+            lp_profile['display_name'] = lp_profile['name']
diff --git a/tests/unit/test_utils.py b/tests/unit/test_utils.py
index 99945fbf7..8fbae147b 100644
--- a/tests/unit/test_utils.py
+++ b/tests/unit/test_utils.py
@@ -110,3 +110,16 @@ class TestUtils(testtools.TestCase):
 
         self.assertEqual(normalized_company_names,
                          correct_normalized_company_names)
+
+    def test_validate_lp_display_name(self):
+        profile = dict(name='johnny', display_name='John Smith')
+        utils.validate_lp_display_name(profile)
+        self.assertEqual('John Smith', profile['display_name'])
+
+        profile = dict(name='johnny', display_name='<email address hidden>')
+        utils.validate_lp_display_name(profile)
+        self.assertEqual('johnny', profile['display_name'])
+
+        profile = None
+        utils.validate_lp_display_name(profile)
+        self.assertEqual(None, profile)