1 files changed, 20 insertions, 0 deletions
diff --git a/test/integration/dispatcher_test.rb b/test/integration/dispatcher_test.rb
index b3a50a0..4f13e6b 100644
--- a/test/integration/dispatcher_test.rb
+++ b/test/integration/dispatcher_test.rb
@@ -48,6 +48,20 @@ class Nickserver::DispatcherTest < Minitest::Test
     assert_response success
   end
 
+  def test_email_via_hkp_nicknym_unreachable
+    handle address: ['valid@email.tld'], headers: { "Host" => "http://nickserver.me" }
+    stub_nicknym_raises
+    hkp_source.expect :query, success, [Nickserver::EmailAddress]
+    assert_response success
+  end
+
+  def test_email_via_hkp_nicknym_unreachable
+    handle address: ['valid@email.tld'], headers: { "Host" => "http://nickserver.me" }
+    stub_nicknym_raises
+    hkp_source.expect :query, nil, [Nickserver::EmailAddress]
+    assert_response response(status: 502, content: "HTTP::ConnectionError")
+  end
+
   def test_email_via_nicknym
     handle address: ['valid@email.tld'], headers: { "Host" => "http://nickserver.me" }
     nicknym_source.expect :available_for?, true, [String]
@@ -90,6 +104,12 @@ class Nickserver::DispatcherTest < Minitest::Test
     end
   end
 
+  def stub_nicknym_raises
+    def nicknym_source.available_for?(*_args)
+      raise HTTP::ConnectionError
+    end
+  end
+
   def nicknym_source
     @nicknym_source ||= Minitest::Mock.new
   end