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
|
require 'helper'
describe Statsd::Admin do
before do
class Statsd::Admin
o, $VERBOSE = $VERBOSE, nil
alias connect_old connect
def connect
$connect_count ||= 0
$connect_count += 1
end
$VERBOSE = o
end
@admin = Statsd::Admin.new('localhost', 1234)
@socket = @admin.instance_variable_set(:@socket, FakeTCPSocket.new)
end
after do
class Statsd::Admin
o, $VERBOSE = $VERBOSE, nil
alias connect connect_old
$VERBOSE = o
end
end
describe "#initialize" do
it "should set the host and port" do
@admin.host.must_equal 'localhost'
@admin.port.must_equal 1234
end
it "should default the host to 127.0.0.1 and port to 8126" do
statsd = Statsd::Admin.new
statsd.host.must_equal '127.0.0.1'
statsd.port.must_equal 8126
end
end
describe "#host and #port" do
it "should set host and port" do
@admin.host = '1.2.3.4'
@admin.port = 5678
@admin.host.must_equal '1.2.3.4'
@admin.port.must_equal 5678
end
it "should not resolve hostnames to IPs" do
@admin.host = 'localhost'
@admin.host.must_equal 'localhost'
end
it "should set nil host to default" do
@admin.host = nil
@admin.host.must_equal '127.0.0.1'
end
it "should set nil port to default" do
@admin.port = nil
@admin.port.must_equal 8126
end
end
%w(gauges counters timers).each do |action|
describe "##{action}" do
it "should send a command and return a Hash" do
["{'foo.bar': 0,\n",
"'foo.baz': 1,\n",
"'foo.quux': 2 }\n",
"END\n","\n"].each do |line|
@socket.write line
end
result = @admin.send action.to_sym
result.must_be_kind_of Hash
result.size.must_equal 3
@socket.readline.must_equal "#{action}\n"
end
end
describe "#del#{action}" do
it "should send a command and return an Array" do
["deleted: foo.bar\n",
"deleted: foo.baz\n",
"deleted: foo.quux\n",
"END\n", "\n"].each do |line|
@socket.write line
end
result = @admin.send "del#{action}", "foo.*"
result.must_be_kind_of Array
result.size.must_equal 3
@socket.readline.must_equal "del#{action} foo.*\n"
end
end
end
describe "#stats" do
it "should send a command and return a Hash" do
["whatever: 0\n", "END\n", "\n"].each do |line|
@socket.write line
end
result = @admin.stats
result.must_be_kind_of Hash
result["whatever"].must_equal 0
@socket.readline.must_equal "stats\n"
end
end
describe "#connect" do
it "should reconnect" do
c = $connect_count
@admin.connect
($connect_count - c).must_equal 1
end
end
end
|
