67 lines
1.1 KiB
Ruby
67 lines
1.1 KiB
Ruby
|
#!/usr/bin/env ruby
|
||
|
|
|
||
|
|
TEST_REGEX = /TEST_F\([a-zA-Z0-9_]+,\s+([a-zA-Z0-9_]+)\)/
|
||
|
|
|
||
|
|
class Context
|
||
|
|
attr_accessor :name, :ev, :src
|
||
|
|
def initialize
|
||
|
|
@name = ""
|
||
|
|
@src = ""
|
||
|
|
@ev = []
|
||
|
|
end
|
||
|
|
end
|
||
|
|
|
||
|
|
class String
|
||
|
|
def snakecase
|
||
|
|
self
|
||
|
|
.gsub(/([A-Z]+)([A-Z][a-z])/, '\1_\2')
|
||
|
|
.gsub(/([a-z\d])([A-Z])/, '\1_\2')
|
||
|
|
.tr('-', '_')
|
||
|
|
.gsub(/\s/, '_')
|
||
|
|
.gsub(/__+/, '_')
|
||
|
|
.downcase
|
||
|
|
end
|
||
|
|
end
|
||
|
|
|
||
|
|
ctx = nil
|
||
|
|
|
||
|
|
tests = []
|
||
|
|
IO.foreach(ARGV[0]) do |line|
|
||
|
|
line.strip!
|
||
|
|
if ctx
|
||
|
|
fail "unexpected TEST_F" if line =~ TEST_REGEX
|
||
|
|
if line =~ /^}/
|
||
|
|
tests << ctx
|
||
|
|
ctx = nil
|
||
|
|
end
|
||
|
|
if line =~ /^EXPECT_CALL/
|
||
|
|
fail 'not end with ;' unless line[-1] == ';'
|
||
|
|
v = line.gsub('(', ' ').gsub(')', ' ').split
|
||
|
|
ctx.ev << v[2]
|
||
|
|
end
|
||
|
|
else
|
||
|
|
next unless line =~ TEST_REGEX
|
||
|
|
name = $1
|
||
|
|
next unless name =~ /^(Ex\d+_\d+)/
|
||
|
|
str = $1.upcase
|
||
|
|
$stderr.puts "found #{name}"
|
||
|
|
ctx = Context.new
|
||
|
|
ctx.name = "test_#{name.snakecase}"
|
||
|
|
ctx.src = str
|
||
|
|
end
|
||
|
|
end
|
||
|
|
|
||
|
|
# code gen
|
||
|
|
tests.each do |t|
|
||
|
|
next if t.ev.size == 0
|
||
|
|
puts "#[test]"
|
||
|
|
puts "fn #{t.name}() {"
|
||
|
|
puts " let mut v = str_to_test_events(#{t.src}).into_iter();"
|
||
|
|
t.ev.each do |e|
|
||
|
|
puts " assert_next!(v, TestEvent::#{e});"
|
||
|
|
end
|
||
|
|
puts "}"
|
||
|
|
puts
|
||
|
|
end
|
||
|
|
|