Poll in deferred image tests

Author: josefarias

spec/ruby_llm/image_generation_spec.rb

@@ -22,8 +22,17 @@ def save_and_verify_image(image)
   end
 end
 
-def wait_for_deferred_image
-  sleep(15) if VCR.current_cassette.recording?
+def save_and_verify_deferred_image(image)
+  tries = 0
+
+  begin
+    save_and_verify_image image
+  rescue RSpec::Expectations::ExpectationNotMetError => e
+    tries += 1
+    sleep 1
+    retry if tries < 20
+    raise e
+  end
 end
 
 RSpec.describe RubyLLM::Image do
@@ -68,8 +77,7 @@ def wait_for_deferred_image
 
       expect(image.deferred?).to be(true)
 
-      wait_for_deferred_image
-      save_and_verify_image image
+      save_and_verify_deferred_image image
     end
 
     it 'google/nano-banana, an official replicate model, can paint images with custom parameters' do
@@ -81,17 +89,15 @@ def wait_for_deferred_image
 
       expect(image.deferred?).to be(true)
 
-      wait_for_deferred_image
-      save_and_verify_image image
+      save_and_verify_deferred_image image
     end
 
     it 'prunaai/hidream-l1-fast, an unofficial replicate model, can paint images' do
       image = RubyLLM.paint('a siamese cat', model: 'prunaai/hidream-l1-fast', output_format: 'png')
 
       expect(image.deferred?).to be(true)
 
-      wait_for_deferred_image
-      save_and_verify_image image
+      save_and_verify_deferred_image image
     end
 
     it 'validates model existence' do

spec/support/rspec_configuration.rb

@@ -13,7 +13,7 @@
 
   config.around do |example|
     cassette_name = example.full_description.parameterize(separator: '_').delete_prefix('rubyllm_')
-    VCR.use_cassette(cassette_name) do
+    VCR.use_cassette(cassette_name, record: :new_episodes) do
       example.run
     end
   end