Fix tests

ezh · Feb 23, 2020 · a2cea97 · a2cea97
1 parent bbaff04
commit a2cea97
Show file tree

Hide file tree

Showing 3 changed files with 7 additions and 3 deletions.
diff --git a/cloudselect/selector.py b/cloudselect/selector.py
@@ -109,7 +109,7 @@ def adjust(representation, representation_maximum_field_length):
             raise AssertionError(
                 "There should be at least 2 fields in instance representation: id and something meaningful",
             )
-        return [find(i.split("\t", 1)[0]) for i in selected.split("\n")]
+        return [find(i.split("\t", 1)[0].strip()) for i in selected.split("\n")]
 
     def get_editor(self):
         """Get editor path."""

diff --git a/test/test_cloudselect.py b/test/test_cloudselect.py
@@ -94,6 +94,7 @@ def test_cli_configuration_read(cfgdir):
             "discovery": {
                 "aws": "cloudselect.discovery.aws",
                 "hetzner": "cloudselect.discovery.hetzner",
+                "kubernetes": "cloudselect.discovery.kubernetes",
                 "local": "cloudselect.discovery.local",
             },
             "group": {"simple": "cloudselect.group.simple"},

diff --git a/test/test_selector.py b/test/test_selector.py
@@ -23,7 +23,7 @@ def test_completer(capsys):
     selector = cloud.fabric(configuration, args)
     selector.complete("", 0)
     captured = capsys.readouterr()
-    assert captured.out == "a\nb\nc\nempty\nsingle\n"
+    assert captured.out == "a\nb\nc\nempty\nk8s\nsingle\n"
     assert captured.err == ""
 
 
@@ -35,7 +35,10 @@ def test_profile_list(capsys):
     selector = cloud.fabric(configuration, args)
     selector.profile_list()
     captured = capsys.readouterr()
-    assert captured.out == "CloudSelect profiles:\n- a\n- b\n- c\n- empty\n- single\n"
+    assert (
+        captured.out
+        == "CloudSelect profiles:\n- a\n- b\n- c\n- empty\n- k8s\n- single\n"
+    )
     assert captured.err == ""