Add before and after as shorter variants

icy-arctic-fox · Jun 24, 2024 · 2917ebc · 2917ebc
1 parent 4031c6b
commit 2917ebc
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/src/spectator/core/hooks.cr b/src/spectator/core/hooks.cr
@@ -7,11 +7,19 @@ module Spectator::Core
       hooks << ExampleHook(Example).new(&block)
     end
 
+    def before(&block : Example ->) : Nil
+      before_each(&block)
+    end
+
     def after_each(&block : Example ->)
       hooks = @after_each ||= [] of ExampleHook(Example)
       hooks << ExampleHook(Example).new(&block)
     end
 
+    def after(&block : Example ->) : Nil
+      after_each(&block)
+    end
+
     def before_all(&block : ->)
       hooks = @before_all ||= [] of ContextHook
       hooks << ContextHook.new(&block)