iv-org-mirror
/
shard-spectator
mirror of https://gitea.invidious.io/iv-org/shard-spectator.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Make derived double names safer

This commit is contained in:
Michael Miller 2019-11-12 21:13:44 -07:00
parent 94e210c306
commit f50e71606e
1 changed files with 8 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
src/spectator/dsl/mocks.cr
View File

@ -2,16 +2,17 @@ require "../mocks"
module Spectator::DSL
macro double(name, **stubs, &block)
{% safe_name = name.id.symbolize.gsub(/\W/, "_").id %}
{% if block.is_a?(Nop) %}
Double{{name.id}}.new.tap do |%double|
Double{{safe_name}}.new.tap do |%double|
{% for name, value in stubs %}
allow(%double).to receive({{name.id}}).and_return({{value}})
{% end %}
end
{% else %}
class Double{{name.id}} < ::Spectator::Mocks::Double
class Double{{safe_name}} < ::Spectator::Mocks::Double
def initialize(null = false)
super({{name.id.symbolize}}, null)
super({{name.id.stringify}}, null)
end
def as_null_object
@ -24,16 +25,17 @@ module Spectator::DSL
end
macro null_double(name, **stubs, &block)
{% safe_name = name.id.symbolize.gsub(/\W/, "_").id %}
{% if block.is_a?(Nop) %}
Double{{name.id}}.new(true).tap do |%double|
Double{{safe_name}}.new(true).tap do |%double|
{% for name, value in stubs %}
allow(%double).to receive({{name.id}}).and_return({{value}})
{% end %}
end
{% else %}
class Double{{name.id}} < ::Spectator::Mocks::Double
class Double{{safe_name}} < ::Spectator::Mocks::Double
def initialize(null = true)
super({{name.id.symbolize}}, null)
super({{name.id.stringify}}, null)
end
def as_null_object