Simplify type resolution code

This commit is contained in:
Michael Miller 2019-11-12 21:46:33 -07:00
parent cb466b4ff5
commit 38ec44c9ac

View file

@ -15,10 +15,7 @@ module Spectator::DSL
end
macro create_double(type_name, name, **stubs)
{%
type = type_name.resolve?
raise "Could not find a double labeled #{name}" unless type
%}
{% type_name.resolve? || raise("Could not find a double labeled #{name}") %}
{{type_name}}.new.tap do |%double|
{% for name, value in stubs %}
@ -57,10 +54,7 @@ module Spectator::DSL
end
macro create_null_double(type_name, name, **stubs)
{%
type = type_name.resolve?
raise "Could not find a double labeled #{name}" unless type
%}
{% type_name.resolve? || raise("Could not find a double labeled #{name}") %}
{{type_name}}.new(true).tap do |%double|
{% for name, value in stubs %}