aboutsummaryrefslogtreecommitdiff
path: root/test/tc_mauve_sender.rb
blob: 858a7ec1296d7a904f3ae180d886362750504afa (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
# encoding: utf-8
$:.unshift "../lib"

require 'th_mauve_resolv'
require 'test/unit'
require 'pp'
require 'timecop'
require 'mauve/sender'
require 'locale'


class TcMauveSender < Test::Unit::TestCase
  include Mauve

  def setup
    Timecop.freeze(Time.local(2011,8,1,0,0,0,0))
  end

  def teardown
    Timecop.return
  end

  def test_sanitise
    Locale.clear
    Locale.current = "en_GB.ISO-8859-1"

    #
    # Set up a couple of crazy sources.
    #
    utf8_source = "Å ðîßtáñt plàñët"
    iso88591_source = utf8_source.encode(Locale.current.charset)

    #
    # Make sure our two sources are distinct
    #
    assert(iso88591_source != utf8_source)

    sender = Sender.new("test-1.example.com")
    update = Mauve::Proto::AlertUpdate.new
    update.source = iso88591_source
    update.replace = false

    alert = Mauve::Proto::Alert.new
    update.alert << alert

    alert_cleared = Mauve::Proto::Alert.new
    update.alert << alert_cleared
    alert_cleared.clear_time = Time.now.to_i

    #
    # Make sure the update has the correct source
    #
    assert_equal(iso88591_source, update.source)

    #
    # Sanitize
    #
    update = sender.sanitize(update)

    #
    # Now make sure the sanitization has changed it back to UTF-8
    #
    assert_equal(utf8_source, update.source)

    #
    # Now make sure the transmission time + id have been set
    #
    assert_equal(Time.now.to_i, update.transmission_time)
    assert_kind_of(Integer, update.transmission_id)

    #
    # Make sure that the alert has its raise time set by default
    #
    assert_equal(Time.now.to_i, alert.raise_time)
    assert_equal(0, alert_cleared.raise_time)
  end


end