bitfab 0.20.0 → 0.20.1

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 9624757032370e1e318121cabcf2bfa984c7bcdcd07c9ecd5dcee449fef57a7c
-  data.tar.gz: c3925bbae804c16d21c3f373ec6cde4de340ee645fc6423505fc0b5d5f5e638e
+  metadata.gz: cd571926bd40dc39f337e87f9977bfb0fe3219c21532dc1cf7be382c175a4559
+  data.tar.gz: 17ff724ae1a42f97fa6eae82c97404de8dc9cef738a18674d76da48ba9ea762d
 SHA512:
-  metadata.gz: d2c33cd4a3432d79703685f00bb8c8278e2bebbd694b3a695f1dee646680a5af11370f233635f04590eeb0d5cedee850c496b40bee2fc176c34cf22fb721cde8
-  data.tar.gz: c64be06a07ca8ea807bca23e0f96acb15e1647945b33c3ea731f10c0b52759870b88637d86b680aee07caac7836e3818946f2fb85f0979d83b3ea89ef8ea1704
+  metadata.gz: 012d3433d2d59200c812557a95b4f2b7cb8d93d7885d9495f59b7e21d89a892ffcc3c0bb83200f4c9f324320d5a1153f4c59bd2facac2dc1924174a7f184d81f
+  data.tar.gz: 69aaffc897238560ebecf6fedbd81509eda58fd082126107585a913cc74a025456a32111356902251a0c5b3439ff2b84ef7ea5aac4325a10e531c93a5d460334

data/lib/bitfab/replay.rb

@@ -2,6 +2,7 @@
 
 require_relative "constants"
 require_relative "serialize"
+require_relative "traceable"
 
 module Bitfab
   # Replay mock strategies. Mirrors the Python and TypeScript SDKs.
@@ -117,6 +118,21 @@ module Bitfab
           "determines how many traces replay."
       end
 
+      # Reject a trace_function_key that contradicts the method's declared key.
+      # replay() fetches historical traces by trace_function_key but records the
+      # replayed spans under the method's own bitfab_span key (via send below),
+      # so a mismatch produces an incoherent test run (traces fetched for one
+      # function, recorded under another). Only fires when the method's key is
+      # introspectable; an untraced method falls through to the persistence
+      # check in complete_replay. Mirrors the TypeScript/Python SDKs.
+      declared_key = Traceable.trace_function_key_for(receiver, method_name)
+      if declared_key && declared_key != trace_function_key
+        raise ArgumentError,
+          "Method #{method_name} is traced under trace function key '#{declared_key}' but replay was " \
+          "called with '#{trace_function_key}'. Pass trace_function_key: '#{declared_key}' to replay it, " \
+          "or point at the method traced under '#{trace_function_key}'."
+      end
+
       http_client = client.instance_variable_get(:@http_client)
 
       # limit is meaningless with explicit trace_ids (the ID list determines

data/lib/bitfab/traceable.rb

@@ -69,9 +69,43 @@ module Bitfab
         end
       end
 
+      wrapper.instance_variable_set(:@bitfab_span_method, method_name_str)
+      wrapper.instance_variable_set(:@bitfab_span_key, trace_function_key)
       klass.prepend(wrapper)
     end
 
+    # Resolve the trace function key a traced method records spans under, or
+    # nil if the method carries no Bitfab span wrapper. Walks the receiver's
+    # ancestor chain for the wrapper module that bitfab_span / Traceable.wrap
+    # prepended. Instances resolve via the class. A Class/Module receiver
+    # resolves either way: singleton-class ancestors first (class/module
+    # methods wrapped on the singleton class), then the class's own ancestors
+    # (methods whose wrapper was prepended onto the class itself), so a
+    # class-method replay is guarded regardless of which surface carries the
+    # wrapper. Singleton-first keeps the method actually dispatched by
+    # `receiver.send` (a class method shadows a same-named instance wrapper).
+    #
+    # replay() uses this to reject a trace_function_key that contradicts the
+    # method's declared key: a mismatch fetches one function's historical
+    # traces but re-records the replay under the method's own key, producing
+    # an incoherent test run. Mirrors the TypeScript and Python SDKs, which
+    # throw on the same key mismatch.
+    def self.trace_function_key_for(receiver, method_name)
+      name = method_name.to_s
+      ancestors = if receiver.is_a?(Module)
+        receiver.singleton_class.ancestors + receiver.ancestors
+      else
+        receiver.class.ancestors
+      end
+      ancestors.each do |mod|
+        next unless mod.instance_variable_defined?(:@bitfab_span_method)
+        next unless mod.instance_variable_get(:@bitfab_span_method) == name
+
+        return mod.instance_variable_get(:@bitfab_span_key)
+      end
+      nil
+    end
+
     module ClassMethods
       # Set the trace function key for this class.
       # All spans declared in this class will be grouped under this key.
@@ -152,6 +186,8 @@ module Bitfab
           end
         end
 
+        wrapper.instance_variable_set(:@bitfab_span_method, method_name_str)
+        wrapper.instance_variable_set(:@bitfab_span_key, trace_function_key)
         prepend(wrapper)
       end
     end

data/lib/bitfab/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module Bitfab
-  VERSION = "0.20.0"
+  VERSION = "0.20.1"
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: bitfab
 version: !ruby/object:Gem::Version
-  version: 0.20.0
+  version: 0.20.1
 platform: ruby
 authors:
 - Harvest Team