Skip to content

Commit daf120a

Browse files
committed
more tests
1 parent 6d5ab79 commit daf120a

File tree

2 files changed

+137
-4
lines changed

2 files changed

+137
-4
lines changed

src/client.rs

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -180,6 +180,7 @@ impl Client {
180180
// - if in transaction mode, this lives for the duration of one transaction.
181181
let mut shard: Option<usize> = None;
182182

183+
// Active database role we want to talk to, e.g. primary or replica.
183184
let mut role: Option<Role> = None;
184185

185186
loop {
Lines changed: 136 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -1,13 +1,145 @@
11
SET SERVER ROLE TO 'primary';
2-
SELECT 1;
2+
SET SHARDING KEY TO '1';
3+
INSERT INTO data (id, value) VALUES (1, 'value_1');
34

45
SET SERVER ROLE TO 'replica';
5-
SELECT 1;
6+
SET SHARDING KEY TO '1';
7+
SELECT * FROM data WHERE id = 1;
8+
9+
---
10+
11+
SET SERVER ROLE TO 'primary';
12+
SET SHARDING KEY TO '2';
13+
INSERT INTO data (id, value) VALUES (2, 'value_1');
14+
15+
SET SERVER ROLE TO 'replica';
16+
SET SHARDING KEY TO '2';
17+
SELECT * FROM data WHERE id = 2;
18+
19+
---
20+
21+
SET SERVER ROLE TO 'primary';
22+
SET SHARDING KEY TO '3';
23+
INSERT INTO data (id, value) VALUES (3, 'value_1');
24+
25+
SET SERVER ROLE TO 'replica';
26+
SET SHARDING KEY TO '3';
27+
SELECT * FROM data WHERE id = 3;
28+
29+
---
30+
31+
SET SERVER ROLE TO 'primary';
32+
SET SHARDING KEY TO '4';
33+
INSERT INTO data (id, value) VALUES (4, 'value_1');
34+
35+
SET SERVER ROLE TO 'replica';
36+
SET SHARDING KEY TO '4';
37+
SELECT * FROM data WHERE id = 4;
38+
39+
---
40+
41+
SET SERVER ROLE TO 'primary';
42+
SET SHARDING KEY TO '5';
43+
INSERT INTO data (id, value) VALUES (5, 'value_1');
44+
45+
SET SERVER ROLE TO 'replica';
46+
SET SHARDING KEY TO '5';
47+
SELECT * FROM data WHERE id = 5;
48+
49+
---
50+
51+
SET SERVER ROLE TO 'primary';
52+
SET SHARDING KEY TO '6';
53+
INSERT INTO data (id, value) VALUES (6, 'value_1');
54+
55+
SET SERVER ROLE TO 'replica';
56+
SET SHARDING KEY TO '6';
57+
SELECT * FROM data WHERE id = 6;
58+
59+
---
60+
61+
SET SERVER ROLE TO 'primary';
62+
SET SHARDING KEY TO '7';
63+
INSERT INTO data (id, value) VALUES (7, 'value_1');
64+
65+
SET SERVER ROLE TO 'replica';
66+
SET SHARDING KEY TO '7';
67+
SELECT * FROM data WHERE id = 7;
68+
69+
---
70+
71+
SET SERVER ROLE TO 'primary';
72+
SET SHARDING KEY TO '8';
73+
INSERT INTO data (id, value) VALUES (8, 'value_1');
74+
75+
SET SERVER ROLE TO 'replica';
76+
SET SHARDING KEY TO '8';
77+
SELECT * FROM data WHERE id = 8;
78+
79+
---
80+
81+
SET SERVER ROLE TO 'primary';
82+
SET SHARDING KEY TO '9';
83+
INSERT INTO data (id, value) VALUES (9, 'value_1');
84+
85+
SET SERVER ROLE TO 'replica';
86+
SET SHARDING KEY TO '9';
87+
SELECT * FROM data WHERE id = 9;
88+
89+
---
90+
91+
SET SERVER ROLE TO 'primary';
92+
SET SHARDING KEY TO '10';
93+
INSERT INTO data (id, value) VALUES (10, 'value_1');
94+
95+
SET SERVER ROLE TO 'replica';
96+
SET SHARDING KEY TO '10';
97+
SELECT * FROM data WHERE id = 10;
98+
99+
---
100+
101+
SET SERVER ROLE TO 'primary';
102+
SET SHARDING KEY TO '11';
103+
INSERT INTO data (id, value) VALUES (11, 'value_1');
104+
105+
SET SERVER ROLE TO 'replica';
106+
SET SHARDING KEY TO '11';
107+
SELECT * FROM data WHERE id = 11;
108+
109+
---
110+
111+
SET SERVER ROLE TO 'primary';
112+
SET SHARDING KEY TO '12';
113+
INSERT INTO data (id, value) VALUES (12, 'value_1');
114+
115+
SET SERVER ROLE TO 'replica';
116+
SET SHARDING KEY TO '12';
117+
SELECT * FROM data WHERE id = 12;
118+
119+
---
120+
121+
SET SERVER ROLE TO 'primary';
122+
SET SHARDING KEY TO '13';
123+
INSERT INTO data (id, value) VALUES (13, 'value_1');
124+
125+
SET SERVER ROLE TO 'replica';
126+
SET SHARDING KEY TO '13';
127+
SELECT * FROM data WHERE id = 13;
128+
129+
---
130+
131+
SET SERVER ROLE TO 'primary';
132+
SET SHARDING KEY TO '14';
133+
INSERT INTO data (id, value) VALUES (14, 'value_1');
134+
135+
SET SERVER ROLE TO 'replica';
136+
SET SHARDING KEY TO '14';
137+
SELECT * FROM data WHERE id = 14;
138+
139+
---
6140

7-
SET SHARDING KEY TO '1234';
8141
SET SERVER ROLE TO 'primary';
9142
SELECT 1;
10143

11144
SET SERVER ROLE TO 'replica';
12-
SET SHARDING KEY TO '4321';
13145
SELECT 1;

0 commit comments

Comments
 (0)