iv-org-mirror/shard-spectator
1
0
Fork 0
mirror of https://gitea.invidious.io/iv-org/shard-spectator.git synced 2024-08-15 00:53:35 +00:00
Code Issues Projects Releases Packages Wiki Activity

Make derived double names safer

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