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

Get stubs on class methods (self.) working

Browse source
This commit is contained in:
Michael Miller 2019-11-11 23:30:30 -07:00
parent 233e9573de
commit c3cba6962c
1 changed files with 10 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

12
src/spectator/mocks/stubs.cr
View file

@ -2,11 +2,13 @@ module Spectator::Mocks
module Stubs module Stubs
private macro stub(definition, _file = __FILE__, _line = __LINE__, &block) private macro stub(definition, _file = __FILE__, _line = __LINE__, &block)
{% {%
receiver = nil
name = nil name = nil
params = nil params = nil
args = nil args = nil
body = nil body = nil
if definition.is_a?(Call) # stub foo { :bar } if definition.is_a?(Call) # stub foo { :bar }
receiver = definition.receiver.id
named = false named = false
name = definition.name.id name = definition.name.id
params = definition.args params = definition.args
@ -31,6 +33,12 @@ module Spectator::Mocks
else else
:super :super
end.id end.id
receiver = if receiver == :self.id
original = :previous_def.id
"self."
else
""
end.id
%} %}
{% if body && !body.is_a?(Nop) %} {% if body && !body.is_a?(Nop) %}
@ -42,7 +50,7 @@ module Spectator::Mocks
::Spectator::SpecBuilder.add_default_stub({{@type.name}}, %ds) ::Spectator::SpecBuilder.add_default_stub({{@type.name}}, %ds)
{% end %} {% end %}
def {{name}}({{params.splat}}){% if definition.is_a?(TypeDeclaration) %} : {{definition.type}}{% end %} def {{receiver}}{{name}}({{params.splat}}){% if definition.is_a?(TypeDeclaration) %} : {{definition.type}}{% end %}
if (%harness = ::Spectator::Harness.current?) if (%harness = ::Spectator::Harness.current?)
%args = ::Spectator::Mocks::GenericArguments.create({{args.splat}}) %args = ::Spectator::Mocks::GenericArguments.create({{args.splat}})
%call = ::Spectator::Mocks::GenericMethodCall.new({{name.symbolize}}, %args) %call = ::Spectator::Mocks::GenericMethodCall.new({{name.symbolize}}, %args)
@ -54,7 +62,7 @@ module Spectator::Mocks
{{original}}({{args.splat}}) {{original}}({{args.splat}})
end end
def {{name}}({{params.splat}}){% if definition.is_a?(TypeDeclaration) %} : {{definition.type}}{% end %} def {{receiver}}{{name}}({{params.splat}}){% if definition.is_a?(TypeDeclaration) %} : {{definition.type}}{% end %}
if (%harness = ::Spectator::Harness.current?) if (%harness = ::Spectator::Harness.current?)
%args = ::Spectator::Mocks::GenericArguments.create({{args.splat}}) %args = ::Spectator::Mocks::GenericArguments.create({{args.splat}})
%call = ::Spectator::Mocks::GenericMethodCall.new({{name.symbolize}}, %args) %call = ::Spectator::Mocks::GenericMethodCall.new({{name.symbolize}}, %args)