2017-10-26 16:46:58 +00:00
|
|
|
require "spec"
|
|
|
|
require "../src/ameba"
|
2017-10-30 20:00:01 +00:00
|
|
|
|
2017-11-01 20:05:41 +00:00
|
|
|
module Ameba
|
2017-11-07 21:50:25 +00:00
|
|
|
struct DummyRule < Rule::Base
|
2017-11-22 06:44:29 +00:00
|
|
|
properties do
|
|
|
|
description : String = "Dummy rule that does nothing."
|
|
|
|
end
|
|
|
|
|
2017-11-01 20:05:41 +00:00
|
|
|
def test(source)
|
|
|
|
end
|
2017-10-30 20:00:01 +00:00
|
|
|
end
|
2017-11-30 21:50:07 +00:00
|
|
|
|
2018-01-29 22:25:36 +00:00
|
|
|
struct NamedRule < Rule::Base
|
|
|
|
properties do
|
|
|
|
description : String = "A rule with a custom name."
|
|
|
|
end
|
|
|
|
|
|
|
|
def test(source)
|
|
|
|
end
|
|
|
|
|
|
|
|
def self.name
|
|
|
|
"BreakingRule"
|
|
|
|
end
|
|
|
|
end
|
|
|
|
|
2017-11-30 21:50:07 +00:00
|
|
|
struct ErrorRule < Rule::Base
|
|
|
|
def test(source)
|
2018-01-28 18:06:04 +00:00
|
|
|
source.error self, 1, 1, "This rule always adds an error"
|
2017-11-30 21:50:07 +00:00
|
|
|
end
|
|
|
|
end
|
2017-11-01 20:05:41 +00:00
|
|
|
|
2018-05-03 15:57:47 +00:00
|
|
|
struct ScopeRule < Rule::Base
|
|
|
|
getter scopes = [] of AST::Scope
|
|
|
|
|
|
|
|
def test(source)
|
|
|
|
end
|
|
|
|
|
|
|
|
def test(source, node : Crystal::ASTNode, scope : AST::Scope)
|
|
|
|
@scopes << scope
|
|
|
|
end
|
|
|
|
end
|
|
|
|
|
2017-11-14 07:24:36 +00:00
|
|
|
class DummyFormatter < Formatter::BaseFormatter
|
|
|
|
property started_sources : Array(Source)?
|
|
|
|
property finished_sources : Array(Source)?
|
|
|
|
property started_source : Source?
|
|
|
|
property finished_source : Source?
|
|
|
|
|
|
|
|
def started(sources)
|
|
|
|
@started_sources = sources
|
|
|
|
end
|
|
|
|
|
|
|
|
def source_finished(source : Source)
|
|
|
|
@started_source = source
|
|
|
|
end
|
|
|
|
|
|
|
|
def source_started(source : Source)
|
|
|
|
@finished_source = source
|
|
|
|
end
|
|
|
|
|
|
|
|
def finished(sources)
|
|
|
|
@finished_sources = sources
|
|
|
|
end
|
|
|
|
end
|
|
|
|
|
2017-11-01 20:05:41 +00:00
|
|
|
struct BeValidExpectation
|
|
|
|
def match(source)
|
|
|
|
source.valid?
|
|
|
|
end
|
|
|
|
|
|
|
|
def failure_message(source)
|
|
|
|
String.build do |str|
|
|
|
|
str << "Source expected to be valid, but there are errors:\n\n"
|
2018-05-08 21:14:46 +00:00
|
|
|
source.errors.reject(&.disabled?).each do |e|
|
2017-11-07 20:02:51 +00:00
|
|
|
str << " * #{e.rule.name}: #{e.message}\n"
|
2017-11-01 20:05:41 +00:00
|
|
|
end
|
|
|
|
end
|
|
|
|
end
|
|
|
|
|
|
|
|
def negative_failure_message(source)
|
|
|
|
"Source expected to be invalid, but it is valid."
|
|
|
|
end
|
|
|
|
end
|
2018-05-03 15:57:47 +00:00
|
|
|
|
|
|
|
class TestNodeVisitor < Crystal::Visitor
|
|
|
|
NODES = [
|
|
|
|
Crystal::Var,
|
|
|
|
Crystal::Assign,
|
|
|
|
Crystal::OpAssign,
|
|
|
|
Crystal::MultiAssign,
|
|
|
|
Crystal::Block,
|
|
|
|
Crystal::Def,
|
|
|
|
Crystal::If,
|
|
|
|
Crystal::While,
|
|
|
|
Crystal::MacroLiteral,
|
|
|
|
]
|
|
|
|
|
|
|
|
def initialize(node)
|
|
|
|
node.accept self
|
|
|
|
end
|
|
|
|
|
|
|
|
def visit(node : Crystal::ASTNode)
|
|
|
|
true
|
|
|
|
end
|
|
|
|
|
|
|
|
{% for node in NODES %}
|
|
|
|
{{getter_name = node.stringify.split("::").last.underscore + "_nodes"}}
|
|
|
|
|
|
|
|
getter {{getter_name.id}} = [] of {{node}}
|
|
|
|
|
|
|
|
def visit(node : {{node}})
|
|
|
|
{{getter_name.id}} << node
|
|
|
|
true
|
|
|
|
end
|
|
|
|
{% end %}
|
|
|
|
end
|
2017-11-01 20:05:41 +00:00
|
|
|
end
|
|
|
|
|
|
|
|
def be_valid
|
|
|
|
Ameba::BeValidExpectation.new
|
2017-11-01 10:49:03 +00:00
|
|
|
end
|
2018-05-03 15:57:47 +00:00
|
|
|
|
|
|
|
def as_node(source)
|
|
|
|
Crystal::Parser.new(source).parse
|
|
|
|
end
|
|
|
|
|
|
|
|
def as_nodes(source)
|
|
|
|
Ameba::TestNodeVisitor.new(as_node source)
|
|
|
|
end
|