sus 0.26.0 → 0.27.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 731c63c1341c591c641c0c9a8d05d387ca0258adef3aeb002b4509e5d6672e69
-  data.tar.gz: 4f0b0fd74e104f5fb82f48df8e7c8011dea872b07f5ccd7b314908b401f68597
+  metadata.gz: 15dd552861dc2a448ac4717e95092d21b76e9f09e50492269a898c8d7db464e3
+  data.tar.gz: 47e84419dadb77bfa29e82b8383f7ae90b18d8a170cdd4136e2ef75f3f5afb38
 SHA512:
-  metadata.gz: 0fae67b800bfc9f5c05581c2e3a4e3c45d5fe13f9403acc12c771441a2f333d775e8305c20449d14669bce69cdc0b234325e9adacb33ea9847c117ba750fe22c
-  data.tar.gz: edb022b7eef312a19c03452fee531ace489038d9320b6e68a13146a482136fb70f00273f3f43206d8b5f78e752c52ec46645eeab407ed815169290189a28230b
+  metadata.gz: aa482cd7ea24f6b6f5b32f59582403171ecf8695f490edc0a1149366566d05424e8199d813f4ff0cbd5ef6b2abd327013f35b112a84c905bba62ded1bee9e478
+  data.tar.gz: 3bf33b5fdcf42584ba46e360b9a83367b2f09ebc15a6b23497f2485fcd105ea5c354e10e4a15b084624aff8af2a506305771e9edd4754cc7b7e700a99fd50683

data/lib/sus/mock.rb

@@ -62,6 +62,23 @@ module Sus
 
 			return self
 		end
+		
+		# Wrap a method, yielding the original method as the first argument, so you can call it from within the hook.
+		def wrap(method, &hook)
+			execution_context = Thread.current
+			
+			@interceptor.define_method(method) do |*arguments, **options, &block|
+				if execution_context == Thread.current
+					original = proc do |*arguments, **options|
+						super(*arguments, **options)
+					end
+					
+					hook.call(original, *arguments, **options, &block) 
+				else
+					super(*arguments, **options, &block)
+				end
+			end
+		end
 	end
 
 	module Mocks

data/lib/sus/version.rb

@@ -4,5 +4,5 @@
 # Copyright, 2021-2024, by Samuel Williams.
 
 module Sus
-	VERSION = "0.26.0"
+	VERSION = "0.27.0"
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: sus
 version: !ruby/object:Gem::Version
-  version: 0.26.0
+  version: 0.27.0
 platform: ruby
 authors:
 - Samuel Williams