From 18c9de1bc73e403634711e7cc2201788b464acef Mon Sep 17 00:00:00 2001
From: Andrey Pavlov <apavlov@mirantis.com>
Date: Thu, 24 Mar 2016 14:45:49 +0300
Subject: [PATCH] Fix marathon framework autodetection

TrivialFix

Change-Id: I3f0406e6a634350a6dca2675037978b2bff73379
---
 kolla_mesos/common/mesos_utils.py | 4 ++--
 kolla_mesos/tests/fakes/mesos.py  | 6 ++++--
 2 files changed, 6 insertions(+), 4 deletions(-)

diff --git a/kolla_mesos/common/mesos_utils.py b/kolla_mesos/common/mesos_utils.py
index b0839ccc..eba32ce5 100644
--- a/kolla_mesos/common/mesos_utils.py
+++ b/kolla_mesos/common/mesos_utils.py
@@ -62,9 +62,9 @@ def get_number_of_nodes(mesos_client):
 def get_marathon(mesos_client):
     frameworks = mesos_client.get_frameworks()
     frameworks_names = map(operator.itemgetter('name'), frameworks)
+    marathon_framework = None
     for framework_name in frameworks_names:
         if "marathon" in framework_name:
             marathon_framework = framework_name
-        else:
-            marathon_framework = None
+            break
     return marathon_framework
diff --git a/kolla_mesos/tests/fakes/mesos.py b/kolla_mesos/tests/fakes/mesos.py
index c5af32fb..9620a211 100644
--- a/kolla_mesos/tests/fakes/mesos.py
+++ b/kolla_mesos/tests/fakes/mesos.py
@@ -38,7 +38,8 @@ MESOS_STATE_TAGGED_SLAVES = {
     ],
     'frameworks': [
         {'name': 'chronos_autodetect'},
-        {'name': 'marathon_autodetect'}
+        {'name': 'marathon_autodetect'},
+        {'name': 'another_framework'}
     ]
 }
 MESOS_STATE_UNTAGGED_SLAVES = {
@@ -55,7 +56,8 @@ MESOS_STATE_NO_SLAVES = {
 MESOS_STATE_FRAMEWORKS = {
     'frameworks': [
         {'name': 'chronos_autodetect'},
-        {'name': 'marathon_autodetect'}
+        {'name': 'marathon_autodetect'},
+        {'name': 'another_framework'}
     ]
 }