add retry.

1 files changed, 14 insertions, 0 deletions

diff --git a/test/net/client_test.rb b/test/net/client_test.rb
index caec258..2037cfd 100644
--- a/test/net/client_test.rb
+++ b/test/net/client_test.rb
@@ -6,6 +6,7 @@ class ClientTest < Minitest::Test
   def initialize(*args)
     super
     @subject = Net::Hippie::Client.new
+    @subject.logger = Logger.new('/dev/null')
   end
 
   def test_get
@@ -17,6 +18,19 @@ class ClientTest < Minitest::Test
     end
   end
 
+  def test_get_with_retry
+    uri = URI.parse('https://www.example.org/api/scim/v2/schemas')
+    WebMock.stub_request(:get, uri.to_s)
+      .to_timeout.then
+      .to_timeout.then
+      .to_return(status: 200, body: { 'success' => 'true' }.to_json)
+    response = subject.with_retry(retries: 3) do |client|
+      client.get(uri)
+    end
+    refute_nil response
+    assert_equal('true', JSON.parse(response.body)['success'])
+  end
+
   def test_get_with_string_uri
     VCR.use_cassette('get_breaches') do
       response = subject.get('https://haveibeenpwned.com/api/breaches')