AST Util

spec/ameba/ast/traverse_spec.cr

@@ -1,10 +1,10 @@
-require "../spec_helper"
+require "../../spec_helper"
 
-module Ameba
+module Ameba::AST
   rule = DummyRule.new
   source = Source.new ""
 
-  describe "AST Traverse" do
+  describe "Traverse" do
     {% for name in NODE_VISITORS %}
       describe "{{name}}" do
         it "allow to visit {{name}} node" do

spec/ameba/ast/util_spec.cr

@@ -0,0 +1,39 @@
+require "../../spec_helper"
+
+module Ameba::AST
+  struct Test
+    include Util
+  end
+
+  subject = Test.new
+
+  describe Util do
+    describe "#literal?" do
+      [
+        Crystal::ArrayLiteral.new,
+        Crystal::BoolLiteral.new(false),
+        Crystal::CharLiteral.new('a'),
+        Crystal::HashLiteral.new,
+        Crystal::NamedTupleLiteral.new,
+        Crystal::NilLiteral.new,
+        Crystal::NumberLiteral.new(42),
+        Crystal::RegexLiteral.new(Crystal::NilLiteral.new),
+        Crystal::StringLiteral.new(""),
+        Crystal::SymbolLiteral.new(""),
+        Crystal::TupleLiteral.new([] of Crystal::ASTNode),
+        Crystal::RangeLiteral.new(
+          Crystal::NilLiteral.new,
+          Crystal::NilLiteral.new,
+          true),
+      ].each do |literal|
+        it "returns true if node is #{literal}" do
+          subject.literal?(literal).should be_true
+        end
+      end
+
+      it "returns false if node is not a literal" do
+        subject.literal?(Crystal::Nop).should be_false
+      end
+    end
+  end
+end