Playing around with mocking and type restrictions

src/spectator/dsl/mocks.cr

@@ -111,12 +111,29 @@ module Spectator::DSL
               else
                 :module
               end %}
-    {% begin %}
+    {% if block %}
-      {{type.id}} ::{{resolved.id}}
+      {% begin %}
-        include ::Spectator::Mocks::Stubs
+        {{type.id}} ::{{resolved.id}}
+          include ::Spectator::Mocks::Stubs
 
-        {{block.body}}
+          {{block.body}}
-      end
+        end
+
+        {% if type == :class %}
+          {{type.id}} Mock{{resolved.id}} < ::{{resolved.id}}
+          end
+        {% elsif type == :module %}
+          class Mock{{resolved.id}}
+            include ::{{resolved.id}}
+          end
+        {% end %}
+      {% end %}
+    {% else %}
+      {% if type == :struct %}
+        {% raise "Cannot use mock(#{resolved.id}) to create a struct" %}
+      {% else %}
+        Mock{{resolved.id}}.new
+      {% end %}
     {% end %}
   end