Add simple specs for lexing line numbers

opal · Nov 23, 2013 · ad317e2 · ad317e2
1 parent 6e108ec
commit ad317e2
Showing 1 changed file with 27 additions and 0 deletions.
diff --git a/spec/cli/lexer_spec.rb b/spec/cli/lexer_spec.rb
@@ -0,0 +1,27 @@
+require File.expand_path('../spec_helper', __FILE__)
+require 'opal/parser'
+
+describe Opal::Lexer do
+  it "sets correct line information for each token" do
+    expect_lines("42").to eq([1])
+    expect_lines("\n3.142").to eq([2])
+    expect_lines("3.142\n42\n57").to eq([1, 2, 3])
+  end
+
+  it "increments the line count over multiple new lines" do
+    expect_lines("1\n\n\n2").to eq([1, 4])
+    expect_lines("\n\n\n3\n\n5\n\n").to eq([4, 6])
+  end
+
+  it "increments line numbers over =begin...=end blocks" do
+    expect_lines("=begin\n=end\n1").to eq([3])
+    expect_lines("=begin\nfoo\nbar\n=end\n42\n43").to eq([5, 6])
+  end
+
+  def expect_lines(source)
+    parsed = Opal::Parser.new.parse(source)
+    nodes = parsed.type == :block ? parsed.children : [parsed]
+
+    expect(nodes.map { |sexp| sexp.line })
+  end
+end