Fix ansible-test target completion on Python 3.

carlcantprogram · Aug 15, 2017 · 0ff5815 · 0ff5815
1 parent ebbb536
commit 0ff5815
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/test/runner/lib/target.py b/test/runner/lib/target.py
@@ -7,6 +7,7 @@
 import errno
 import itertools
 import abc
+import sys
 
 from lib.util import ApplicationError
 
@@ -21,7 +22,8 @@ def find_target_completion(target_func, prefix):
     """
     try:
         targets = target_func()
-        prefix = prefix.encode()
+        if sys.version_info[0] == 2:
+            prefix = prefix.encode()
         short = os.environ.get('COMP_TYPE') == '63'  # double tab completion from bash
         matches = walk_completion_targets(targets, prefix, short)
         return matches