Merge pull request #33 from ikalnitsky/discovery-method

Fix /discovery-methods endpoint
Mirantis · Sep 12, 2016 · b123d7e · b123d7e
2 parents cf2c772 + 6315fb8
commit b123d7e
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 2 deletions.
diff --git a/kostyor/db/api.py b/kostyor/db/api.py
@@ -40,8 +40,8 @@ def get_upgrade(upgrade_id):
     return u_task.to_dict()
 
 
-def get_discovery_methods(db_session):
-    return {'discovery_methods': constants.DISCOVERY_METHODS}
+def get_discovery_methods():
+    return constants.DISCOVERY_METHODS
 
 
 def get_upgrade_versions(cluster_id):

diff --git a/kostyor/tests/unit/test_db_api.py b/kostyor/tests/unit/test_db_api.py
@@ -70,3 +70,8 @@ def test_update_cluster(self):
         result = db_api.get_cluster(cluster['id'])
 
         self.assertEqual(update['name'], result['name'])
+
+    def test_discovery_methods(self):
+        methods = db_api.get_discovery_methods()
+        # we don't care about ordering here, so let's compare sorted arrays
+        self.assertEqual(sorted(methods), sorted([constants.OPENSTACK]))