summaryrefslogtreecommitdiff
path: root/test/unit/hkp_test.rb
blob: c211690d6692172bb08e6060f3bbcd01b55b4576 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
require File.expand_path('test_helper', File.dirname(__FILE__))

class HkpTest < MiniTest::Unit::TestCase

  def test_key_info_expired
    fetch_key_info(:hkp_vindex_result, 'lemur@leap.se') do |keys|
      assert_equal 1, keys.length, 'should find a single key'
      assert_equal ['lemur@example.org', 'lemur@leap.se'].sort, keys.first.uids.sort, 'should find both uids'
      assert_equal '0EE5BE979282D80B9F7540F1CCD2ED94D21739E9', keys.first.keyid
    end
  end

  def test_key_info_multiple_valid_results
    fetch_key_info :hkp_vindex_result, 'gazelle@leap.se' do |keys|
      assert_equal 2, keys.length, 'should find two keys'
      assert_equal ['gazelle@leap.se'], keys.first.uids
      assert_equal '3790027A', keys.first.keyid
      assert keys.last.uids.include? 'gazelle@leap.se'
    end
  end

  def test_key_info_reject_keysize
    fetch_key_info :hkp_vindex_result, 'frog@leap.se' do |keys|
      assert_equal 1, keys.length, 'should find one key' # because short key gets ignored
      assert_equal '00440025', keys.first.keyid
    end
  end

  def test_key_info_not_found
    uid = 'leaping_lemur@leap.se'
    stub_sks_vindex_reponse(uid, :status => 404)
    test_em_errback "Nickserver::HKP::FetchKeyInfo.new.search '#{uid}'" do |error|
      assert_equal 404, error
    end
  end

  def test_key_info_real_network
    real_network do
      uid = 'elijah@riseup.net'
      test_em_callback "Nickserver::HKP::FetchKeyInfo.new.search '#{uid}'" do |keys|
        assert_equal 1, keys.size
        assert keys.first.keyid =~ /00440025$/
      end
    end
  end

  def test_fetch_key
    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))

    test_em_callback "Nickserver::HKP::FetchKey.new.get '#{uid}'" do |key_text|
      assert_equal file_content(:leap_public_key), key_text
    end
  end

  def test_fetch_key_not_found
    uid    = 'cloudadmin@leap.se'
    key_id = 'E818C478D3141282F7590D29D041EB11B1647490'

    stub_sks_vindex_reponse(uid, :body => file_content(:leap_vindex_result))
    stub_sks_get_reponse(key_id, :status => 404)

    test_em_errback "Nickserver::HKP::FetchKey.new.get '#{uid}'" do |error|
      assert_equal 404, error
    end
  end

  def test_fetch_key_too_short
    uid    = 'chiiph@leap.se'
    key_id = '9A753A6B'

    stub_sks_vindex_reponse(uid, :body => file_content(:short_key_vindex_result))
    test_em_errback "Nickserver::HKP::FetchKey.new.get '#{uid}'" do |error|
      assert_equal 500, error
    end
  end


  protected

  #
  # Takes a code snippet that returns a Deferrable, and yields the callback result.
  # Assertion fails if errback is called instead of callback.
  #
  # This method takes care of the calls to EM.run and EM.stop. It works kind of like EM.run_block,
  # except I couldn't get run_block to work with multiple nested HTTP requests.
  #
  def test_em_callback(code, &block)
    EM.run do
      deferrable = instance_eval(code)
      deferrable.callback {|response|
        EM.stop
        yield response
        return
      }
      deferrable.errback {|response|
        EM.stop
        flunk "Expecting callback, but errback invoked with response: #{response}"
      }
    end
    assert false, 'should not get here'
  end

  #
  # like test_em_callback, except value yielded is the result of errback, and
  # we raise an exception if errback was not called.
  #
  def test_em_errback(code, &block)
    EM.run do
      deferrable = instance_eval(code)
      deferrable.callback {|response|
        EM.stop
        flunk "Expecting errback, but callback invoked with response: #{response}"
      }
      deferrable.errback {|response|
        EM.stop
        yield response
        return
      }
    end
    assert false, 'should not get here'
  end

  def fetch_key_info(body_source, uid, &block)
    stub_sks_vindex_reponse(uid, :body => file_content(body_source))
    test_em_callback "Nickserver::HKP::FetchKeyInfo.new.search '#{uid}'", &block
  end

end