From 1d2cb3968fd9e0a830f7606a7dbbc86fa23d6c5a Mon Sep 17 00:00:00 2001
From: Matt Martz <matt@sivel.net>
Date: Tue, 24 Nov 2015 11:56:50 -0600
Subject: [PATCH] Still look for EXAMPLES and RETURN if DOCUMENTATION is
 invalid. Fixes #7

---
 ansible_testing/modules.py | 19 ++++++++++++++++++-
 1 file changed, 18 insertions(+), 1 deletion(-)

diff --git a/ansible_testing/modules.py b/ansible_testing/modules.py
index f0acaa380b8..29e078a3301 100644
--- a/ansible_testing/modules.py
+++ b/ansible_testing/modules.py
@@ -293,6 +293,22 @@ class ModuleValidator(Validator):
         if not os.path.isfile(py_path):
             self.errors.append('Missing python documentation file')
 
+    def _get_docs(self):
+        docs = None
+        examples = None
+        ret = None
+        for child in self.ast.body:
+            if isinstance(child, ast.Assign):
+                for grandchild in child.targets:
+                    if grandchild.id == 'DOCUMENTATION':
+                        docs = child.value.s
+                    elif grandchild.id == 'EXAMPLES':
+                        examples = child.value.s[1:]
+                    elif grandchild.id == 'RETURN':
+                        ret = child.value.s
+
+        return docs, examples, ret
+
     def _find_redeclarations(self):
         g = set()
         find_globals(g, self.ast.body)
@@ -352,7 +368,8 @@ class ModuleValidator(Validator):
                 doc, examples, ret = get_docstring(self.path, verbose=True)
                 trace = None
             except:
-                doc, examples, ret = get_docstring(self.path)
+                doc = None
+                _, examples, ret = self._get_docs()
                 trace = traceback.format_exc()
             finally:
                 sys.stdout = sys_stdout