Add experimental callback for mucking about with results after recording

Author: geeksam

app/models/lab_tech/experiment.rb

@@ -56,6 +56,11 @@ def self.named(experiment_name_or_id)
 
     ##### INSTANCE METHODS #####
 
+    # This method is experimental and should probably not be depended on
+    def __after_recording__(&block)
+      @after_recording = block
+    end
+
     def comparator
       @_scientist_comparator
     end
@@ -96,7 +101,10 @@ def name=(value) ; write_attribute :name, value ; end
 
     def publish(scientist_result)
       return if Rails.env.test? && !LabTech.publish_results_in_test_mode?
-      LabTech::Result.record_a_science( self, scientist_result, diff_with: @diff_with )
+
+      LabTech::Result.record_a_science( self, scientist_result, diff_with: @diff_with ).tap do |result|
+        @after_recording&.call result
+      end
     end
 
     # I don't encourage the willy-nilly destruction of experimental results...

spec/models/lab_tech/experiment_spec.rb

@@ -92,6 +92,21 @@ def wtf
         end
       end
 
+      specify "clients can intercept the LabTech::Result before it gets recorded (EXPERIMENTAL)" do
+        LabTech.science "wibble" do |e|
+          e.context( { a: 1 } )
+          e.use { :hello_world }
+          e.try { :howdy_earth }
+
+          e.__after_recording__ do |result|
+            result.update context: { b: 2 }
+          end
+        end
+
+        result = experiment.results.first
+        expect( result.context ).to eq( { b: 2 } )
+      end
+
       describe "diff-generating behavior" do
         let(:result) { experiment.results.first }