-
Notifications
You must be signed in to change notification settings - Fork 1
/
test_candidate.rb
144 lines (127 loc) · 4.3 KB
/
test_candidate.rb
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
132
133
134
135
136
137
138
139
140
141
142
143
144
require 'rubygems'
require 'bud'
require 'test/unit'
require 'candidate'
class TestCandidate < Test::Unit::TestCase
class C
include Bud
include Candidate
state do
table :see_server_type, [] => [:state]
table :see_next_current_term, [] => [:state]
table :see_output_req_vote, [:num, :candidate, :voter, :term, :last_index, :last_term]
table :see_reset, reset.schema
end
bootstrap do
current_term <= [[42]]
log <= [[0, 0, "dummy"]]
candidate <= [[2]]
end
bloom do
see_server_type <+- server_type
see_reset <= reset
log <+ log
candidate <+ candidate
current_term <+ current_term
see_next_current_term <+- next_current_term
member <= [ [1, 'localhost:23451'], [2, 'localhost:23452'],
[3, 'localhost:23453'], [4, 'localhost:23454'],
[5, 'localhost:12345']]
see_output_req_vote <= outputSndRequestVote do |o|
[budtime, o.candidate, o.voter, o.term, o.last_index, o.last_term]
end
ip_port_scratch <= [["somecandidatestring"]]
end
end
def pseudo_tick(tick)
@candidate.current_term <+ [[tick]]
@candidate.tick
end
def setup
@candidate = C.new(:port =>12345)
@candidate.run_bg
end
def teardown
@candidate.stop
end
def test_higher_opponent
4.times { pseudo_tick(42) }
@candidate.inputSndRequestVote <+ [['localhost:12347', 'localhost:12345', 43, 2, 3]]
4.times { pseudo_tick(42) }
@candidate.sync_do do
assert_equal(NodeProtocol::FOLLOWER, @candidate.see_server_type.first.state)
end
end
def test_lower_opponent
4.times { pseudo_tick(42) }
@candidate.inputSndRequestVote <+ [['localhost:12347', 'localhost:12345', 1, 2, 3]]
4.times { pseudo_tick(42) }
@candidate.sync_do do
assert_equal(0, @candidate.see_server_type.length)
end
end
def test_equal_opponent
4.times { pseudo_tick(42) }
@candidate.inputSndRequestVote <+ [['localhost:12347', 'localhost:12345', 1, 2, 3]]
4.times { pseudo_tick(42) }
@candidate.sync_do do
assert_equal(0, @candidate.see_server_type.length)
end
end
def test_new_election
4.times { pseudo_tick(42) }
@candidate.ring <+ [["RINGED", "OMG"]]
sleep(0.5) #ensure 300 ms has passed
4.times { pseudo_tick(42)}
@candidate.sync_do do
assert_equal(1, @candidate.see_next_current_term.length)
end
end
def test_elected_leader
4.times { pseudo_tick(42) }
@candidate.async_do {
@candidate.inputRspRequestVote <+ [ ['localhost:12345', 'localhost:23451', 42, 'true'],
['localhost:12345', 'localhost:23454', 42, 'true']]
}
4.times { pseudo_tick(42) }
@candidate.sync_do do
assert_equal(NodeProtocol::LEADER, @candidate.see_server_type.first.state)
end
end
def test_election_no_result_yet
4.times { pseudo_tick(42) }
@candidate.inputRspRequestVote <+ [ ['localhost:12345', 'localhost:23451', 42, 'true'] ]
sleep(0.5) #ensure 300 ms has passed
4.times { pseudo_tick(42) }
@candidate.sync_do do
assert_operator 5, :<, @candidate.see_output_req_vote.length
assert_equal(0, @candidate.see_server_type.length)
end
end
def test_append_entry_greater_term_leader
4.times { pseudo_tick(42) }
@candidate.inputSndAppendEntries <+ [['localhost:12345', 'localhoast:23451', 43, 3, 42, "entry", 3]]
4.times { pseudo_tick(42) }
@candidate.sync_do do
assert_equal(NodeProtocol::FOLLOWER, @candidate.see_server_type.first.state)
end
end
def test_append_entry_less_term_leader
4.times { pseudo_tick(42) }
@candidate.inputSndAppendEntries <+ [['localhost:12345', 'localhoast:23451', 41, 3, 42, "entry", 3]]
4.times { pseudo_tick(42) }
@candidate.sync_do do
assert_equal(0, @candidate.see_server_type.length)
assert_equal(0, @candidate.see_reset.length)
end
end
def test_append_entry_equal_term_leader
4.times { pseudo_tick(42) }
@candidate.inputSndAppendEntries <+ [['localhost:12345', 'localhoast:23451', 42, 3, 41, "entry", 3]]
4.times { pseudo_tick(42) }
@candidate.sync_do do
assert_equal(NodeProtocol::FOLLOWER, @candidate.see_server_type.length)
assert_equal(1, @candidate.see_reset.length)
end
end
end