forked from pcmanus/cassandra-dtest-old
/
taketoken_test.py
83 lines (65 loc) · 2.6 KB
/
taketoken_test.py
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
import time
from dtest import Tester, debug
from assertions import assert_unavailable
from tools import (create_c1c2_table, insert_c1c2, query_c1c2, retry_till_success,
insert_columns, new_node, no_vnodes, since)
class TestTakeToken(Tester):
@since('2.0')
def taketoken_test(self):
debug("Creating a ring")
cluster = self.cluster
cluster.set_configuration_options(values={
'initial_token': None,
'num_tokens': 10,
'hinted_handoff_enabled' : False,
'write_request_timeout_in_ms' : 60000,
'read_request_timeout_in_ms' : 60000,
'dynamic_snitch_badness_threshold' : 0.0}, batch_commitlog=True)
cluster.populate(3).start()
[node1, node2, node3] = cluster.nodelist()
cluster.start()
debug("Set to talk to node 2")
n2cursor = self.patient_cql_connection(node2).cursor()
self.create_ks(n2cursor, 'ks', 2)
create_c1c2_table(self, n2cursor)
debug("Generating some data for all nodes")
for n in xrange(10,20):
insert_c1c2(n2cursor, n, 'ALL')
node1.flush()
debug("Writing data to node2")
for n in xrange(30,1000):
insert_c1c2(n2cursor, n, 'ONE')
node2.flush()
debug("Getting token from node 1")
n1cursor = self.patient_cql_connection(node1).cursor()
n1cursor.execute('SELECT tokens FROM system.local')
n1tokens = n1cursor.fetchone()
n3cursor = self.patient_cql_connection(node3).cursor()
n3cursor.execute('SELECT tokens FROM system.local')
n3tokens = n3cursor.fetchone()
debug("Relocate tokens from node1 to node3")
i = 0;
tl = "";
for t in n1tokens[0]:
if i == 8:
break
t = '\\%s' % t
tl = "%s %s" % (tl, t);
i += 1
cmd = "taketoken %s" % tl
debug(cmd)
node3.nodetool(cmd)
time.sleep(1)
debug("Check that the tokens were really moved")
n3cursor.execute('SELECT tokens FROM system.local')
n3tokens = n3cursor.fetchone()
n1cursor.execute('SELECT tokens FROM system.local')
n1tokens = n1cursor.fetchone()
debug("n1 %s n3 %s" % (n1tokens,n3tokens))
assert len(n3tokens[0]) == 18
assert len(n1tokens[0]) == 2
debug("Checking that no data was lost")
for n in xrange(10,20):
query_c1c2(n2cursor, n, 'ALL')
for n in xrange(30,1000):
query_c1c2(n2cursor, n, 'ALL')