From fc0b46caca11afe98fea25b03ca01f475f54231f Mon Sep 17 00:00:00 2001
From: Michael Miller <icy.arctic.fox@gmail.com>
Date: Sat, 9 Nov 2019 09:08:09 -0700
Subject: [PATCH] Use Registry for storing stubs

---
 src/spectator/mocks/allow.cr | 8 +++++---
 1 file changed, 5 insertions(+), 3 deletions(-)

diff --git a/src/spectator/mocks/allow.cr b/src/spectator/mocks/allow.cr
index f6a70fc..00ecc99 100644
--- a/src/spectator/mocks/allow.cr
+++ b/src/spectator/mocks/allow.cr
@@ -1,10 +1,12 @@
+require "./registry"
+
 module Spectator::Mocks
-  struct Allow
-    def initialize(@mock : Double)
+  struct Allow(T)
+    def initialize(@mock : T)
     end
 
     def to(stub : MethodStub) : Nil
-      @mock.spectator_define_stub(stub)
+      Registry.add_stub(@mock, stub)
     end
   end
 end