1 files changed, 107 insertions, 0 deletions
diff --git a/test/unit/nickserver_test.rb b/test/unit/nickserver_test.rb
new file mode 100644
index 0000000..5753ec3
--- /dev/null
+++ b/test/unit/nickserver_test.rb
@@ -0,0 +1,107 @@
+require File.expand_path('test_helper', File.dirname(__FILE__))
+require 'json'
+
+class NickserverTest < MiniTest::Unit::TestCase
+
+  def test_GET_served_via_SKS
+    uid    = 'cloudadmin@leap.se'
+    key_id = 'E818C478D3141282F7590D29D041EB11B1647490'
+    stub_sks_vindex_reponse(uid, :body => file_content(:leap_vindex_result))
+    stub_sks_get_reponse(key_id, :body => file_content(:leap_public_key))
+
+    start do
+      params = {:query => {"address" => uid}}
+      get(params) do |http|
+        assert_equal file_content(:leap_public_key), JSON.parse(http.response)["openpgp"]
+        stop
+      end
+    end
+  end
+
+  def test_POST_served_via_SKS
+    uid    = 'cloudadmin@leap.se'
+    key_id = 'E818C478D3141282F7590D29D041EB11B1647490'
+    stub_sks_vindex_reponse(uid, :body => file_content(:leap_vindex_result))
+    stub_sks_get_reponse(key_id, :body => file_content(:leap_public_key))
+
+    start do
+      params = {:body => {"address" => uid}}
+      post(params) do |http|
+        assert_equal file_content(:leap_public_key), JSON.parse(http.response)["openpgp"]
+        stop
+      end
+    end
+  end
+
+  def test_GET_served_via_couch_not_found
+    uid    = 'bananas@localhost'
+    key_id = 'E818C478D3141282F7590D29D041EB11B1647490'
+    stub_couch_response(uid, :body => file_content(uid))
+
+    start do
+      params = {:query => {"address" => uid}}
+      get(params) do |http|
+        assert_equal 404, http.response_header.status
+        stop
+      end
+    end
+  end
+
+  protected
+
+  #
+  # start nickserver
+  #
+  def start(timeout = 1)
+    Timeout::timeout(timeout) do
+      EM.run do
+        Nickserver::Server.start
+        EM.epoll
+        yield
+      end
+    end
+  rescue Timeout::Error
+    flunk 'Eventmachine was not stopped before the timeout expired'
+  end
+
+  #
+  # http GET requests to nickserver
+  #
+  def get(params, &block)
+    request(:get, params, &block)
+  end
+
+  #
+  # http POST requests to nickserver
+  #
+  def post(params, &block)
+    request(:post, params, &block)
+  end
+
+  #
+  # http request to nickserver
+  #
+  # this works because http requests to localhost are not stubbed, but requests to other domains are.
+  #
+  def request(method, params)
+    http = EventMachine::HttpRequest.new("http://localhost:#{Nickserver::Config.port}/").send(method,params)
+
+    http.callback {
+      # p http.response_header.status
+      # p http.response_header
+      # p http.response
+      yield http
+    }.errback {
+      flunk(http.error) if http.error
+      EM.stop
+    }
+  end
+
+  #
+  # stop nickserver
+  #
+  def stop
+    EM.stop
+  end
+
+end