aboutsummaryrefslogtreecommitdiff
path: root/test/tc_mauve_udp_server.rb
blob: 9a82de82a5c8dfdaabc2e09c5192d95c65159222 (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
# encoding: utf-8

$:.unshift "../lib"

require 'th_mauve'
require 'th_mauve_resolv'
require 'mauve/udp_server'
require 'mauve/server'

class TcMauveUdpServer < Mauve::UnitTest
  # Note: this test fires up the UDP socket.  It *will* fail if more
  # than one instance runs at the same time on the same machine.
  include Mauve

  def setup
    super
    Server.instance.packet_buffer.clear # blarg!
    @server = UDPServer.instance # blarg!
  end

  def test_listens
    update = generic_update()
    t = Thread.new do @server.__send__(:main_loop) end
    sleep(0.2)
    sender.send(update)
    Timeout.timeout(2) do t.join end
    assert_equal update, Proto::AlertUpdate.new.parse_from_string(Server.packet_pop[0])
  end

  def test_closes
    update = generic_update()
    t = @server.run
    sleep(0.2)
    @server.stop
    sleep(0.2)
    sender.send(update)
    Timeout.timeout(2) do t.join end
    assert_nil Server.packet_pop
  end


  def sender
    Sender.new(["#{@server.ip}:#{@server.port}"])
  end

  def generic_update
    update = Proto::AlertUpdate.new
    alert = Proto::Alert.new
    alert.id = "alertid"
    update.alert << alert
    update
  end

end