3 files changed, 30 insertions, 3 deletions
diff --git a/test/unit/error_response_test.rb b/test/unit/error_response_test.rb
new file mode 100644
index 0000000..7242b38
--- /dev/null
+++ b/test/unit/error_response_test.rb
@@ -0,0 +1,12 @@
+require 'test_helper'
+require 'nickserver/error_response'
+
+class ErrorResponseTest < Minitest::Test
+
+  def test_content
+    response = Nickserver::ErrorResponse.new "Not a valid address"
+    assert_equal "500 Not a valid address\n", response.content
+    assert_equal 500, response.status
+  end
+
+end
diff --git a/test/unit/nicknym/source_test.rb b/test/unit/nicknym/source_test.rb
index 586dc0e..41f5ac0 100644
--- a/test/unit/nicknym/source_test.rb
+++ b/test/unit/nicknym/source_test.rb
@@ -1,5 +1,6 @@
 require 'test_helper'
 require 'nickserver/nicknym/source'
+require 'nickserver/email_address'
 
 class NicknymSourceTest < Minitest::Test
 
@@ -21,6 +22,15 @@ class NicknymSourceTest < Minitest::Test
     adapter.verify
   end
 
+  def test_successful_query
+    adapter.expect :get, [200, 'dummy body'],
+      ['https://nicknym.leap_powered.tld', address: email_stub.to_s]
+    response = source.query(email_stub)
+    assert_equal 200, response.status
+    assert_equal 'dummy body', response.content
+    adapter.verify
+  end
+
   protected
 
   def source
@@ -30,4 +40,8 @@ class NicknymSourceTest < Minitest::Test
   def adapter
     @adapter ||= Minitest::Mock.new
   end
+
+  def email_stub
+    @email_stub ||= Nickserver::EmailAddress.new 'test@leap_powered.tld'
+  end
 end
diff --git a/test/unit/response_test.rb b/test/unit/response_test.rb
index 8a53066..ac7a3a8 100644
--- a/test/unit/response_test.rb
+++ b/test/unit/response_test.rb
@@ -3,9 +3,10 @@ require 'nickserver/response'
 
 class ResponseTest < Minitest::Test
 
-  def test_content
-    response = Nickserver::Response.new 500, "Not a valid address"
-    assert_equal "500 Not a valid address", response.content
+  def test_ok_response
+    response = Nickserver::Response.new 200, "content"
+    assert_equal "content", response.content
+    assert_equal 200, response.status
   end
 
 end