2 files changed, 7 insertions, 1 deletions
diff --git a/lib/nickserver/wkd/source.rb b/lib/nickserver/wkd/source.rb
index bf30c94..da7abb8 100644
--- a/lib/nickserver/wkd/source.rb
+++ b/lib/nickserver/wkd/source.rb
@@ -8,7 +8,8 @@ module Nickserver::Wkd
   class Source < Nickserver::Source
     def query(email)
       url = Url.new(email)
-      status, blob = adapter.get url
+      status, blob = adapter.get url,
+                                 rescue: 'failed to connect: getaddrinfo'
       if status == 200
         Nickserver::KeyResponse.new(email.to_s, armor_key(blob))
       end
diff --git a/lib/nickserver/wkd/url.rb b/lib/nickserver/wkd/url.rb
index 0ccff38..759222d 100644
--- a/lib/nickserver/wkd/url.rb
+++ b/lib/nickserver/wkd/url.rb
@@ -13,6 +13,11 @@ module Nickserver::Wkd
       "https://#{domain}/.well-known/openpgpkey/hu/#{encoded_digest}"
     end
 
+    # needed to compare when used as an arg calling a test mock
+    def ==(other)
+      self.to_s == other.to_s
+    end
+
     protected
 
     attr_reader :domain, :local_part