Issue ansible error

app.py

@@ -92,6 +92,14 @@ def run_ansible_playbook(ansible_hosts, playbook):
                                    stats=stats, inventory=inventory)
     pb.run()
 
+
+    hosts = sorted(pb.stats.processed.keys())
+    for h in hosts:
+       t = pb.stats.summarize(h)
+       if t['failures'] > 0:
+           msg = "[%s]:[%s] playbook failed!" % (h, playbook)
+           raise AnsibleError(msg)
+
 def process_deployment(deployment):
     """Process deployment."""
     try:

test/test_app.py

@@ -34,6 +34,7 @@
     deployment_ansible
 from subprocess import CalledProcessError
 from urllib import quote
+from ansible.errors import AnsibleError
 
 
 class TestApp(Test):
@@ -419,6 +420,36 @@ def test_run_ansible_playbook(self, callbacks, ansible):
                                                      inventory=inventory)
         pb.run.assert_called_with()
 
+
+    @patch('app.ansible', autospec=True)
+    @patch('app.callbacks', autospec=True)
+    def test_run_ansible_playbook_raises_error(self, callbacks, ansible):
+        """Test run ansible playbook raises error works."""
+        ansible_hosts = 'ansible_hosts'
+        playbook = 'playbook.yml'
+
+        stats = MagicMock()
+
+        callbacks.AggregateStats.return_value = stats
+
+        playbook_cb = MagicMock()
+        callbacks.PlaybookCallbacks.return_value = playbook_cb
+
+        inventory = MagicMock()
+        ansible.inventory.Inventory.return_value = inventory
+
+        runner_cb = MagicMock()
+        callbacks.PlaybookRunnerCallbacks.return_value = runner_cb
+
+        pb = MagicMock()
+        ansible.playbook.PlayBook.return_value = pb
+
+        pb.stats.processed.keys.return_value = ['127.0.0.1']
+        pb.stats.summarize.return_value = {'failures': 1}
+
+        assert_raises(AnsibleError, run_ansible_playbook, ansible_hosts, playbook)
+
+
     @patch('app.requests')
     def test_get_status_non_url(self, rq_mock):
         """Test get_status non URL works."""