Fix tests

ezh · Feb 11, 2020 · 46d5c7d · 46d5c7d
1 parent 2672def
commit 46d5c7d
Show file tree

Hide file tree

Showing 3 changed files with 14 additions and 12 deletions.
diff --git a/cloudselect/cloud.json.dist b/cloudselect/cloud.json.dist
@@ -32,7 +32,8 @@
             "bastion": "cloudselect.pathfinder.bastion"
         },
         "report": {
-            "json": "cloudselect.report.json"
+            "json": "cloudselect.report.json",
+            "json_pp": "cloudselect.report.json_pp"
         }
     }
 }
diff --git a/cloudselect/selector.py b/cloudselect/selector.py
@@ -152,13 +152,11 @@ def select(self):
         if args.edit is None or args.edit:
             if args.edit is None:
                 configuration = os.path.join(configpath, "{}".format(extension))
-                self.edit(configuration)
-            else:
-                profile = os.path.join(configpath, "{}.{}".format(args.edit, extension))
-                self.edit(profile)
-        elif args.reporter is None:
-            self.reporter_list()
-        elif not args.profile:
-            self.profile_list()
-        else:
-            self.profile_process()
+                return self.edit(configuration)
+            profile = os.path.join(configpath, "{}.{}".format(args.edit, extension))
+            return self.edit(profile)
+        if args.reporter is None:
+            return self.reporter_list()
+        if not args.profile:
+            return self.profile_list()
+        return self.profile_process()
diff --git a/test/test_cloudselect.py b/test/test_cloudselect.py
@@ -111,7 +111,10 @@ def test_cli_configuration_read():
             },
             "group": {"simple": "cloudselect.group.simple"},
             "pathfinder": {"bastion": "cloudselect.pathfinder.bastion"},
-            "report": {"json": "cloudselect.report.json"},
+            "report": {
+                "json": "cloudselect.report.json",
+                "json_pp": "cloudselect.report.json_pp",
+            },
         },
     }
     cloud = CloudSelect()