rspec-retryable 0.2.0 → 0.4.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 236247a97d1e78fbff8c640dfe567b9c47bca7b72dd05e54edc3d6456cf4e412
-  data.tar.gz: ee51a9030277de2b6cfb3aa7e6be7cd44bb5a2acafe1ed7d7bba6b9d61e0b85c
+  metadata.gz: d740994dab3e9b7a7aab752f4588db55caddeac9a51ed0508388ef20b09c341d
+  data.tar.gz: ad5919d6b7b34eab56768805e88df68d6be400ecf82c6ba11042bafabedd5ea2
 SHA512:
-  metadata.gz: ab64dc3ef170cf3dd8b2dccf555eb16156c5cf84dce5b269fd81ff9a55213985d3e2aba57e4f7819c2f3b1eead695544c7fb6c4a4c1d707c2d3aabf93c6573ff
-  data.tar.gz: e49515cbf957ed1ebdfcb0e7ba3b6f6415e01d70b0ca5a36801f55e7035ac9555395200af71f3bafe608b64cd9784bc7c20a8cb2408618b5ae32135e54338971
+  metadata.gz: c67f9c02d7be75d705b2b2b7fb851e55c9cd364b0071d769386dd63b9f2eb713f00135c10b47e1b4b81ca79dd6ee93a54204a957efa0b827ff3bda91cc5d1c2f
+  data.tar.gz: 9f21a5aa684eff320f9d3440f414f16b8c789f783925a69a2d6251130495046d6d3875c8498f97606ecac74354f8ce78f6758d616eda0f9fbbb78e0f77a40ab2

data/README.md

@@ -1,35 +1,125 @@
 # RSpec::Retryable
 
-TODO: Delete this and the text below, and describe your gem
-
-Welcome to your new gem! In this directory, you'll find the files you need to be able to package up your Ruby library into a gem. Put your Ruby code in the file `lib/rspec/retryable`. To experiment with that code, run `bin/console` for an interactive prompt.
+RSpec Retryable is a gem that allows you to retry RSpec examples based on custom handlers. This can be useful for handling flaky tests or tests that depend on external systems.
 
 ## Installation
 
-TODO: Replace `UPDATE_WITH_YOUR_GEM_NAME_IMMEDIATELY_AFTER_RELEASE_TO_RUBYGEMS_ORG` with your gem name right after releasing it to RubyGems.org. Please do not do it earlier due to security reasons. Alternatively, replace this section with instructions to install your gem from git if you don't plan to release to RubyGems.org.
-
-Install the gem and add to the application's Gemfile by executing:
+Install the gem and add it to the application's Gemfile by executing:
 
 ```bash
-bundle add UPDATE_WITH_YOUR_GEM_NAME_IMMEDIATELY_AFTER_RELEASE_TO_RUBYGEMS_ORG
+$ bundle add rspec-retryable
 ```
 
 If bundler is not being used to manage dependencies, install the gem by executing:
 
 ```bash
-gem install UPDATE_WITH_YOUR_GEM_NAME_IMMEDIATELY_AFTER_RELEASE_TO_RUBYGEMS_ORG
+$ gem install rspec-retryable
 ```
 
 ## Usage
 
-TODO: Write usage instructions here
+Simply calling `RSpec::Retryable.bind` enables the ability to retry on test examples. You can add your first handler by using `RSpec::Retryable.handlers.register(SomeHandler)`.
 
-## Development
+#### Example
+
+Create a file `retry_setup.rb` with the following content:
+
+```retry_setup.rb
+require 'rspec/retryable'
+
+class SimpleRetryHandler
+  MAX_RETRIES = 2
+
+  def initialize
+    # Initialization code here
+    @retries = Hash.new(0)
+  end
+
+  def call(payload)
+    # use payload to set retry or not based on RSpec example state
+
+    if @retries[payload.example.id] < MAX_RETRIES
+      # Set `payload.retry` to `true` to enable rspec retry
+      payload.retry = true if payload.state == :failed
+    else
+      # Pass down to next handler
+      yield
+    end
+  end
+end
+
+RSpec::Retryable.bind
+
+RSpec::Retryable.handlers.register(SimpleRetryHandler)
+```
+
+Then requires the setup in your `spec_helper.rb`
+
+```spec_helper.rb
+require 'retry_setup'
+
+RSpec.configure do |config|
+  # ... some rspec setup
+end
+```
+
+`payload` holds below information:
+
+- `example`: (read-only) RSpec example
+- `state`: Current state of the example, can be alterted by handlers
+- `notify`: default to `true`, if set to `false`, reporter will not be notified
+- `result`: this is the final result returned to RSpec runner
+- `retry`: default to `false`, if set to `true`, the example will be retried
 
-After checking out the repo, run `bin/setup` to install dependencies. Then, run `rake spec` to run the tests. You can also run `bin/console` for an interactive prompt that will allow you to experiment.
+#### Interact with multiple Handlers
+
+Since we use [Chain-of-responsibility pattern](http://en.wikipedia.org/wiki/Chain-of-responsibility_pattern) to define handlers, it's possible to chain handlers with a payload passing down as `Rack::Builder` or `ActionDispatch::MiddlewareStack` to manage a stack of handlers, for example:
+
+```retry_setup.rb
+RSpec::Retryable.bind
+
+class FirstRetryHandler
+  def call(payload)
+    puts "-> First handler in"
+
+    # ... do something based on payload state ...
+
+    yield
+
+    puts "<- First handler out"
+  end
+end
+
+# This handler stops retry when expected condition met
+class SecondRetryHandler
+  def call(payload)
+    puts "-> Second handler in"
+
+    # ... do something based on payload state ...
+
+    yield
+
+    puts "<- Second handler out"
+  end
+end
+
+RSpec::Retryable.handlers.register(FirstRetryHandler)
+RSpec::Retryable.handlers.register(SecondRetryHandler)
+```
+
+When execute tests, the output will look like:
+
+```
+-> First handler in
+-> Second handler in
+<- Second handler out
+<- First handler out
+```
+
+## Development
 
-To install this gem onto your local machine, run `bundle exec rake install`. To release a new version, update the version number in `version.rb`, and then run `bundle exec rake release`, which will create a git tag for the version, push git commits and the created tag, and push the `.gem` file to [rubygems.org](https://rubygems.org).
+After checking out the repo, run `bin/setup` to install dependencies. Then, run `bin/rspec` to run the tests.
 
 ## Contributing
 
-Bug reports and pull requests are welcome on GitHub at https://github.com/ngan/rspec-retryable.
+Bug reports and pull requests are welcome on GitHub at https://github.com/Gusto/rspec-retryable.

data/lib/rspec/retryable/example.rb

@@ -28,8 +28,6 @@ module RSpec
         if @payload.retry
           # Replaced the final result by the retry result
           @payload.result = retry_example
-          # Update the execution result status to the new state
-          execution_result.status = @payload.state
         end
 
         # Notify reporter only if it's not handled by the handlers
@@ -58,6 +56,10 @@ module RSpec
         @payload.notify = false
         # Duplicate the example for re-run
         new_example = duplicate_with(metadata)
+        # The shared_group_inclusion_backtrace was reserved from duplication, but we need to keep as-is for
+        # reporter/formatter to show the correct backtrace.
+        # https://github.com/rspec/rspec/blob/0ae9cc43163507458494e631ca8e473bf24cdb26/rspec-core/lib/rspec/core/metadata.rb#L341
+        new_example.metadata[:shared_group_inclusion_backtrace] = metadata[:shared_group_inclusion_backtrace]
         new_example.instance_variable_set(:@id, id)
         # Taken from https://github.com/rspec/rspec-core/blob/main/lib/rspec/core/example_group.rb#L644-L646
         instance = new_example.example_group.new(new_example.inspect_output)
@@ -65,7 +67,16 @@ module RSpec
         # Use same reporter from example instead of the one passing in to behave like
         # a fresh new example.
 
-        new_example.run(instance, reporter)
+        result = new_example.run(instance, reporter)
+        # Update the execution result status to the new state from retry
+        execution_result.status = new_example.execution_result.status
+
+        if execution_result.status == :failed
+          # Sets exception when retry failed
+          execution_result.exception = new_example.execution_result.exception
+        end
+
+        result
       end
 
       def notify_reporter

data/lib/rspec/retryable/handlers.rb

@@ -11,6 +11,10 @@ module RSpec
         @handlers << klass.new(*args, **kwargs)
       end
 
+      def reset!
+        @handlers.clear
+      end
+
       def invoke(payload)
         traverse(0, payload)
       end

data/lib/rspec/retryable/version.rb

@@ -2,6 +2,6 @@
 
 module RSpec
   module Retryable
-    VERSION = "0.2.0"
+    VERSION = "0.4.0"
   end
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: rspec-retryable
 version: !ruby/object:Gem::Version
-  version: 0.2.0
+  version: 0.4.0
 platform: ruby
 authors:
 - Ngan Pham
 autorequire:
 bindir: exe
 cert_chain: []
-date: 2024-09-03 00:00:00.000000000 Z
+date: 2025-05-22 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: rspec-core
@@ -62,7 +62,7 @@ required_rubygems_version: !ruby/object:Gem::Requirement
     - !ruby/object:Gem::Version
       version: '0'
 requirements: []
-rubygems_version: 3.5.9
+rubygems_version: 3.5.22
 signing_key:
 specification_version: 4
 summary: Adds ability to fully control RSpec retrying