diff --git a/lib/netsnmp/client.rb b/lib/netsnmp/client.rb index fdb3db9..4f484e3 100644 --- a/lib/netsnmp/client.rb +++ b/lib/netsnmp/client.rb @@ -140,7 +140,7 @@ def handle_retries retries = @retries begin yield - rescue Timeout::Error => e + rescue Timeout => e raise e if retries.zero? retries -= 1 retry diff --git a/lib/netsnmp/errors.rb b/lib/netsnmp/errors.rb index e85aafd..da55f16 100644 --- a/lib/netsnmp/errors.rb +++ b/lib/netsnmp/errors.rb @@ -4,4 +4,5 @@ module NETSNMP Error = Class.new(StandardError) ConnectionFailed = Class.new(Error) AuthenticationFailed = Class.new(Error) + Timeout = Class.new(Error) end