From e9a35d0b69b737fb018207d9daf1684e24a62f57 Mon Sep 17 00:00:00 2001
From: Sebastian Wilzbach <seb@wilzba.ch>
Date: Fri, 17 Feb 2017 05:49:55 +0100
Subject: [PATCH] Test extractor: visit the entire AST tree

---
 styles/tests_extractor.d | 9 ++++-----
 1 file changed, 4 insertions(+), 5 deletions(-)

diff --git a/styles/tests_extractor.d b/styles/tests_extractor.d
index 7213c32..0bcfdc4 100644
--- a/styles/tests_extractor.d
+++ b/styles/tests_extractor.d
@@ -54,11 +54,10 @@ class TestVisitor : ASTVisitor
 
     override void visit(const Declaration decl)
     {
-        if (decl.unittest_ !is null)
-        {
-           if (hasDdocHeader(sourceCode, decl))
-                print(decl.unittest_);
-        }
+        if (decl.unittest_ !is null && hasDdocHeader(sourceCode, decl))
+            print(decl.unittest_);
+
+        decl.accept(this);
     }
 
 private: