fixed #265

iwarapter · May 20, 2016 · 3a4589e · 3a4589e
1 parent f21615c
commit 3a4589e
Show file tree

Hide file tree

Showing 3 changed files with 9 additions and 3 deletions.
diff --git a/puppet-checks/src/main/java/com/iadams/sonarqube/puppet/checks/OneIncludePerLineCheck.java b/puppet-checks/src/main/java/com/iadams/sonarqube/puppet/checks/OneIncludePerLineCheck.java
@@ -52,7 +52,7 @@ public void init() {
 
   @Override
   public void visitNode(AstNode node) {
-    if ("include".equals(node.getTokenValue()) && !node.getFirstChild(PuppetGrammar.FUNCVALUES).getChildren(PuppetPunctuator.COMMA).isEmpty()) {
+    if ("include".equals(node.getTokenValue()) && node.hasDescendant(PuppetPunctuator.COMMA)) {
       addIssue(node, this, "Split this include statement into multiple lines.");
     }
   }

diff --git a/...ecks/src/test/groovy/com/iadams/sonarqube/puppet/checks/OneIncludePerLineCheckSpec.groovy b/...ecks/src/test/groovy/com/iadams/sonarqube/puppet/checks/OneIncludePerLineCheckSpec.groovy
@@ -44,6 +44,7 @@ class OneIncludePerLineCheckSpec extends Specification {
       .next().atLine(2).withMessage(MESSAGE)
       .next().atLine(3).withMessage(MESSAGE)
       .next().atLine(6).withMessage(MESSAGE)
+      .next().atLine(8).withMessage(MESSAGE)
       .noMore();
   }
-}
+}
diff --git a/.../src/test/groovy/com/iadams/sonarqube/puppet/parser/simple_statements/FunctionSpec.groovy b/.../src/test/groovy/com/iadams/sonarqube/puppet/parser/simple_statements/FunctionSpec.groovy
@@ -91,8 +91,13 @@ class FunctionSpec extends GrammarSpec {
     assertThat(p).matches('include $my_classes')
   }
 
+  def 'include with brackets'() {
+    expect:
+    assertThat(p).matches('include( "stdlib", "apache" )')
+  }
+
   def "function using hash array accessor"() {
     expect:
     assertThat(p).matches('dirname($logfiles[1][1])')
   }
-}
+}