Fixing some small bugs related to integration tests (v2)

lib/ansible/executor/play_iterator.py

@@ -239,7 +239,7 @@ class PlayIterator:
         self._host_states[host.name] = s
 
     def get_failed_hosts(self):
-        return dict((host, True) for (host, state) in self._host_states.iteritems() if state.run_state == self.ITERATING_COMPLETE and state.fail_state != self.FAILED_NONE)
+        return dict((host, True) for (host, state) in self._host_states.iteritems() if state.fail_state != self.FAILED_NONE)
 
     def get_original_task(self, host, task):
         '''

lib/ansible/inventory/group.py

@@ -59,11 +59,9 @@ class Group:
             depth=self.depth,
         )
 
-        debug("serializing group, result is: %s" % result)
         return result
 
     def deserialize(self, data):
-        debug("deserializing group, data is: %s" % data)
         self.__init__()
         self.name = data.get('name')
         self.vars = data.get('vars', dict())

lib/ansible/module_utils/basic.py

@@ -588,8 +588,8 @@ class AnsibleModule(object):
                     return True
                 rc = selinux.lsetfilecon(self._to_filesystem_str(path),
                                          str(':'.join(new_context)))
-            except OSError:
-                self.fail_json(path=path, msg='invalid selinux context', new_context=new_context, cur_context=cur_context, input_was=context)
+            except OSError, e:
+                self.fail_json(path=path, msg='invalid selinux context: %s' % str(e), new_context=new_context, cur_context=cur_context, input_was=context)
             if rc != 0:
                 self.fail_json(path=path, msg='set selinux context failed')
             changed = True

lib/ansible/parsing/yaml/dumper.py

@@ -0,0 +1,37 @@
+# (c) 2012-2014, Michael DeHaan <michael.dehaan@gmail.com>
+#
+# This file is part of Ansible
+#
+# Ansible is free software: you can redistribute it and/or modify
+# it under the terms of the GNU General Public License as published by
+# the Free Software Foundation, either version 3 of the License, or
+# (at your option) any later version.
+#
+# Ansible is distributed in the hope that it will be useful,
+# but WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+# GNU General Public License for more details.
+#
+# You should have received a copy of the GNU General Public License
+# along with Ansible.  If not, see <http://www.gnu.org/licenses/>.
+
+# Make coding more python3-ish
+from __future__ import (absolute_import, division, print_function)
+__metaclass__ = type
+
+import yaml
+
+from ansible.parsing.yaml.objects import AnsibleUnicode
+
+class AnsibleDumper(yaml.SafeDumper):
+    '''
+    A simple stub class that allows us to add representers
+    for our overridden object types.
+    '''
+    pass
+
+AnsibleDumper.add_representer(
+    AnsibleUnicode,
+    yaml.representer.SafeRepresenter.represent_unicode
+)
+

lib/ansible/plugins/filter/core.py

@@ -38,16 +38,21 @@ from jinja2.filters import environmentfilter
 from distutils.version import LooseVersion, StrictVersion
 
 from ansible import errors
+from ansible.parsing.yaml.dumper import AnsibleDumper
 from ansible.utils.hashing import md5s, checksum_s
 from ansible.utils.unicode import unicode_wrap, to_unicode
 
 
 UUID_NAMESPACE_ANSIBLE = uuid.UUID('361E6D51-FAEC-444A-9079-341386DA8E2E')
 
-
-def to_nice_yaml(*a, **kw):
+def to_yaml(a, *args, **kw):
     '''Make verbose, human readable yaml'''
-    transformed = yaml.safe_dump(*a, indent=4, allow_unicode=True, default_flow_style=False, **kw)
+    transformed = yaml.dump(a, Dumper=AnsibleDumper, allow_unicode=True, **kw)
+    return to_unicode(transformed)
+
+def to_nice_yaml(a, *args, **kw):
+    '''Make verbose, human readable yaml'''
+    transformed = yaml.dump(a, Dumper=AnsibleDumper, indent=4, allow_unicode=True, default_flow_style=False, **kw)
     return to_unicode(transformed)
 
 def to_json(a, *args, **kw):
@@ -288,7 +293,7 @@ class FilterModule(object):
             'from_json': json.loads,
 
             # yaml
-            'to_yaml': yaml.safe_dump,
+            'to_yaml': to_yaml,
             'to_nice_yaml': to_nice_yaml,
             'from_yaml': yaml.safe_load,
 

lib/ansible/plugins/strategies/__init__.py

@@ -73,24 +73,28 @@ class StrategyBase:
         self._blocked_hosts     = dict()
 
     def run(self, iterator, connection_info, result=True):
-        # save the counts on failed/unreachable hosts, as the cleanup/handler
-        # methods will clear that information during their runs
-        num_failed      = len(self._tqm._failed_hosts)
-        num_unreachable = len(self._tqm._unreachable_hosts)
+        # save the failed/unreachable hosts, as the run_handlers()
+        # method will clear that information during its execution
+        failed_hosts      = self._tqm._failed_hosts.keys()
+        unreachable_hosts = self._tqm._unreachable_hosts.keys()
 
         debug("running handlers")
         result &= self.run_handlers(iterator, connection_info)
 
+        # now update with the hosts (if any) that failed or were
+        # unreachable during the handler execution phase
+        failed_hosts      = set(failed_hosts).union(self._tqm._failed_hosts.keys())
+        unreachable_hosts = set(unreachable_hosts).union(self._tqm._unreachable_hosts.keys())
+
         # send the stats callback
         self._tqm.send_callback('v2_playbook_on_stats', self._tqm._stats)
 
-        if not result:
-            if num_unreachable > 0:
-                return 3
-            elif num_failed > 0:
-                return 2
-            else:
-                return 1
+        if len(unreachable_hosts) > 0:
+            return 3
+        elif len(failed_hosts) > 0:
+            return 2
+        elif not result:
+            return 1
         else:
             return 0
 
@@ -145,7 +149,7 @@ class StrategyBase:
                     task_result = result[1]
                     host = task_result._host
                     task = task_result._task
-                    if result[0] == 'host_task_failed':
+                    if result[0] == 'host_task_failed' or 'failed' in task_result._result:
                         if not task.ignore_errors:
                             debug("marking %s as failed" % host.name)
                             iterator.mark_host_failed(host)

lib/ansible/plugins/strategies/linear.py

@@ -211,7 +211,7 @@ class StrategyModule(StrategyBase):
                 try:
                     included_files = IncludedFile.process_include_results(host_results, self._tqm, iterator=iterator, loader=self._loader)
                 except AnsibleError, e:
-                    return 1
+                    return False
 
                 if len(included_files) > 0:
                     noop_task = Task()
@@ -252,7 +252,7 @@ class StrategyModule(StrategyBase):
             except (IOError, EOFError), e:
                 debug("got IOError/EOFError in task loop: %s" % e)
                 # most likely an abort, return failed
-                return 1
+                return False
 
         # run the base class run() method, which executes the cleanup function
         # and runs any outstanding handlers which have been triggered

lib/ansible/template/__init__.py

@@ -238,14 +238,6 @@ class Templar:
             environment.filters.update(self._get_filters())
             environment.template_class = AnsibleJ2Template
 
-            # FIXME: may not be required anymore, as the basedir stuff will
-            #        be handled by the loader?
-            #if '_original_file' in vars:
-            #    basedir = os.path.dirname(vars['_original_file'])
-            #    filesdir = os.path.abspath(os.path.join(basedir, '..', 'files'))
-            #    if os.path.exists(filesdir):
-            #        basedir = filesdir
-
             try:
                 t = environment.from_string(data)
             except TemplateSyntaxError, e: