From e4aae1f60aeef0a8c6b1d317cf35a77e34d10ce4 Mon Sep 17 00:00:00 2001
From: Michael Miller <icy.arctic.fox@gmail.com>
Date: Sat, 9 Nov 2019 23:05:22 -0700
Subject: [PATCH] Change Registry to a class

---
 src/spectator/mocks/registry.cr | 14 ++++++--------
 1 file changed, 6 insertions(+), 8 deletions(-)

diff --git a/src/spectator/mocks/registry.cr b/src/spectator/mocks/registry.cr
index dbfaf49..b0b1f75 100644
--- a/src/spectator/mocks/registry.cr
+++ b/src/spectator/mocks/registry.cr
@@ -1,7 +1,5 @@
 module Spectator::Mocks
-  module Registry
-    extend self
-
+  class Registry
     alias Key = Tuple(String, UInt64)
 
     private struct Entry
@@ -9,14 +7,14 @@ module Spectator::Mocks
       getter calls = Deque(MethodCall).new
     end
 
-    @@entries = {} of Key => Entry
+    @entries = {} of Key => Entry
 
     def prepare(context : TestContext) : Nil
       # TODO
     end
 
     def reset : Nil
-      @@entries.clear
+      @entries.clear
     end
 
     def add_stub(object, stub : MethodStub) : Nil
@@ -39,10 +37,10 @@ module Spectator::Mocks
 
     private def fetch(object)
       key = unique_key(object)
-      if @@entries.has_key?(key)
-        @@entries[key]
+      if @entries.has_key?(key)
+        @entries[key]
       else
-        @@entries[key] = Entry.new
+        @entries[key] = Entry.new
       end
     end