pyrox.dev
1import ./make-test-python.nix ({pkgs, lib, ...}:
2
3let
4 cfg = {
5 clusterId = "066ae264-2a5d-4729-8001-6ad265f50b03";
6 monA = {
7 name = "a";
8 ip = "192.168.1.1";
9 };
10 osd0 = {
11 name = "0";
12 ip = "192.168.1.2";
13 key = "AQBCEJNa3s8nHRAANvdsr93KqzBznuIWm2gOGg==";
14 uuid = "55ba2294-3e24-478f-bee0-9dca4c231dd9";
15 };
16 osd1 = {
17 name = "1";
18 ip = "192.168.1.3";
19 key = "AQBEEJNac00kExAAXEgy943BGyOpVH1LLlHafQ==";
20 uuid = "5e97a838-85b6-43b0-8950-cb56d554d1e5";
21 };
22 osd2 = {
23 name = "2";
24 ip = "192.168.1.4";
25 key = "AQAdyhZeIaUlARAAGRoidDAmS6Vkp546UFEf5w==";
26 uuid = "ea999274-13d0-4dd5-9af9-ad25a324f72f";
27 };
28 };
29 generateCephConfig = { daemonConfig }: {
30 enable = true;
31 global = {
32 fsid = cfg.clusterId;
33 monHost = cfg.monA.ip;
34 monInitialMembers = cfg.monA.name;
35 };
36 } // daemonConfig;
37
38 generateHost = { pkgs, cephConfig, networkConfig, ... }: {
39 virtualisation = {
40 memorySize = 1024;
41 emptyDiskImages = [ 20480 ];
42 vlans = [ 1 ];
43 };
44
45 networking = networkConfig;
46
47 environment.systemPackages = with pkgs; [
48 bash
49 sudo
50 ceph
51 xfsprogs
52 netcat-openbsd
53 ];
54
55 boot.kernelModules = [ "xfs" ];
56
57 services.ceph = cephConfig;
58 };
59
60 networkMonA = {
61 dhcpcd.enable = false;
62 interfaces.eth1.ipv4.addresses = pkgs.lib.mkOverride 0 [
63 { address = cfg.monA.ip; prefixLength = 24; }
64 ];
65 firewall = {
66 allowedTCPPorts = [ 6789 3300 ];
67 allowedTCPPortRanges = [ { from = 6800; to = 7300; } ];
68 };
69 };
70 cephConfigMonA = generateCephConfig { daemonConfig = {
71 mon = {
72 enable = true;
73 daemons = [ cfg.monA.name ];
74 };
75 mgr = {
76 enable = true;
77 daemons = [ cfg.monA.name ];
78 };
79 }; };
80
81 networkOsd = osd: {
82 dhcpcd.enable = false;
83 interfaces.eth1.ipv4.addresses = pkgs.lib.mkOverride 0 [
84 { address = osd.ip; prefixLength = 24; }
85 ];
86 firewall = {
87 allowedTCPPortRanges = [ { from = 6800; to = 7300; } ];
88 };
89 };
90
91 cephConfigOsd = osd: generateCephConfig { daemonConfig = {
92 osd = {
93 enable = true;
94 daemons = [ osd.name ];
95 };
96 }; };
97
98 # Following deployment is based on the manual deployment described here:
99 # https://docs.ceph.com/docs/master/install/manual-deployment/
100 # For other ways to deploy a ceph cluster, look at the documentation at
101 # https://docs.ceph.com/docs/master/
102 testscript = { ... }: ''
103 start_all()
104
105 monA.wait_for_unit("network.target")
106 osd0.wait_for_unit("network.target")
107 osd1.wait_for_unit("network.target")
108 osd2.wait_for_unit("network.target")
109
110 # Bootstrap ceph-mon daemon
111 monA.succeed(
112 "sudo -u ceph ceph-authtool --create-keyring /tmp/ceph.mon.keyring --gen-key -n mon. --cap mon 'allow *'",
113 "sudo -u ceph ceph-authtool --create-keyring /etc/ceph/ceph.client.admin.keyring --gen-key -n client.admin --cap mon 'allow *' --cap osd 'allow *' --cap mds 'allow *' --cap mgr 'allow *'",
114 "sudo -u ceph ceph-authtool /tmp/ceph.mon.keyring --import-keyring /etc/ceph/ceph.client.admin.keyring",
115 "monmaptool --create --add ${cfg.monA.name} ${cfg.monA.ip} --fsid ${cfg.clusterId} /tmp/monmap",
116 "sudo -u ceph ceph-mon --mkfs -i ${cfg.monA.name} --monmap /tmp/monmap --keyring /tmp/ceph.mon.keyring",
117 "sudo -u ceph mkdir -p /var/lib/ceph/mgr/ceph-${cfg.monA.name}/",
118 "sudo -u ceph touch /var/lib/ceph/mon/ceph-${cfg.monA.name}/done",
119 "systemctl start ceph-mon-${cfg.monA.name}",
120 )
121 monA.wait_for_unit("ceph-mon-${cfg.monA.name}")
122 monA.succeed("ceph mon enable-msgr2")
123 monA.succeed("ceph config set mon auth_allow_insecure_global_id_reclaim false")
124
125 # Can't check ceph status until a mon is up
126 monA.succeed("ceph -s | grep 'mon: 1 daemons'")
127
128 # Start the ceph-mgr daemon, it has no deps and hardly any setup
129 monA.succeed(
130 "ceph auth get-or-create mgr.${cfg.monA.name} mon 'allow profile mgr' osd 'allow *' mds 'allow *' > /var/lib/ceph/mgr/ceph-${cfg.monA.name}/keyring",
131 "systemctl start ceph-mgr-${cfg.monA.name}",
132 )
133 monA.wait_for_unit("ceph-mgr-a")
134 monA.wait_until_succeeds("ceph -s | grep 'quorum ${cfg.monA.name}'")
135 monA.wait_until_succeeds("ceph -s | grep 'mgr: ${cfg.monA.name}(active,'")
136
137 # Send the admin keyring to the OSD machines
138 monA.succeed("cp /etc/ceph/ceph.client.admin.keyring /tmp/shared")
139 osd0.succeed("cp /tmp/shared/ceph.client.admin.keyring /etc/ceph")
140 osd1.succeed("cp /tmp/shared/ceph.client.admin.keyring /etc/ceph")
141 osd2.succeed("cp /tmp/shared/ceph.client.admin.keyring /etc/ceph")
142
143 # Bootstrap OSDs
144 osd0.succeed(
145 "mkfs.xfs /dev/vdb",
146 "mkdir -p /var/lib/ceph/osd/ceph-${cfg.osd0.name}",
147 "mount /dev/vdb /var/lib/ceph/osd/ceph-${cfg.osd0.name}",
148 "ceph-authtool --create-keyring /var/lib/ceph/osd/ceph-${cfg.osd0.name}/keyring --name osd.${cfg.osd0.name} --add-key ${cfg.osd0.key}",
149 'echo \'{"cephx_secret": "${cfg.osd0.key}"}\' | ceph osd new ${cfg.osd0.uuid} -i -',
150 )
151 osd1.succeed(
152 "mkfs.xfs /dev/vdb",
153 "mkdir -p /var/lib/ceph/osd/ceph-${cfg.osd1.name}",
154 "mount /dev/vdb /var/lib/ceph/osd/ceph-${cfg.osd1.name}",
155 "ceph-authtool --create-keyring /var/lib/ceph/osd/ceph-${cfg.osd1.name}/keyring --name osd.${cfg.osd1.name} --add-key ${cfg.osd1.key}",
156 'echo \'{"cephx_secret": "${cfg.osd1.key}"}\' | ceph osd new ${cfg.osd1.uuid} -i -',
157 )
158 osd2.succeed(
159 "mkfs.xfs /dev/vdb",
160 "mkdir -p /var/lib/ceph/osd/ceph-${cfg.osd2.name}",
161 "mount /dev/vdb /var/lib/ceph/osd/ceph-${cfg.osd2.name}",
162 "ceph-authtool --create-keyring /var/lib/ceph/osd/ceph-${cfg.osd2.name}/keyring --name osd.${cfg.osd2.name} --add-key ${cfg.osd2.key}",
163 'echo \'{"cephx_secret": "${cfg.osd2.key}"}\' | ceph osd new ${cfg.osd2.uuid} -i -',
164 )
165
166 # Initialize the OSDs with regular filestore
167 osd0.succeed(
168 "ceph-osd -i ${cfg.osd0.name} --mkfs --osd-uuid ${cfg.osd0.uuid}",
169 "chown -R ceph:ceph /var/lib/ceph/osd",
170 "systemctl start ceph-osd-${cfg.osd0.name}",
171 )
172 osd1.succeed(
173 "ceph-osd -i ${cfg.osd1.name} --mkfs --osd-uuid ${cfg.osd1.uuid}",
174 "chown -R ceph:ceph /var/lib/ceph/osd",
175 "systemctl start ceph-osd-${cfg.osd1.name}",
176 )
177 osd2.succeed(
178 "ceph-osd -i ${cfg.osd2.name} --mkfs --osd-uuid ${cfg.osd2.uuid}",
179 "chown -R ceph:ceph /var/lib/ceph/osd",
180 "systemctl start ceph-osd-${cfg.osd2.name}",
181 )
182 monA.wait_until_succeeds("ceph osd stat | grep -e '3 osds: 3 up[^,]*, 3 in'")
183 monA.wait_until_succeeds("ceph -s | grep 'mgr: ${cfg.monA.name}(active,'")
184 monA.wait_until_succeeds("ceph -s | grep 'HEALTH_OK'")
185
186 monA.succeed(
187 "ceph osd pool create multi-node-test 32 32",
188 "ceph osd pool ls | grep 'multi-node-test'",
189 "ceph osd pool rename multi-node-test multi-node-other-test",
190 "ceph osd pool ls | grep 'multi-node-other-test'",
191 )
192 monA.wait_until_succeeds("ceph -s | grep '2 pools, 33 pgs'")
193 monA.succeed("ceph osd pool set multi-node-other-test size 2")
194 monA.wait_until_succeeds("ceph -s | grep 'HEALTH_OK'")
195 monA.wait_until_succeeds("ceph -s | grep '33 active+clean'")
196 monA.fail(
197 "ceph osd pool ls | grep 'multi-node-test'",
198 "ceph osd pool delete multi-node-other-test multi-node-other-test --yes-i-really-really-mean-it",
199 )
200
201 # Shut down ceph on all machines in a very unpolite way
202 monA.crash()
203 osd0.crash()
204 osd1.crash()
205 osd2.crash()
206
207 # Start it up
208 osd0.start()
209 osd1.start()
210 osd2.start()
211 monA.start()
212
213 # Ensure the cluster comes back up again
214 monA.succeed("ceph -s | grep 'mon: 1 daemons'")
215 monA.wait_until_succeeds("ceph -s | grep 'quorum ${cfg.monA.name}'")
216 monA.wait_until_succeeds("ceph osd stat | grep -e '3 osds: 3 up[^,]*, 3 in'")
217 monA.wait_until_succeeds("ceph -s | grep 'mgr: ${cfg.monA.name}(active,'")
218 monA.wait_until_succeeds("ceph -s | grep 'HEALTH_OK'")
219 '';
220in {
221 name = "basic-multi-node-ceph-cluster";
222 meta = with pkgs.lib.maintainers; {
223 maintainers = [ lejonet ];
224 };
225
226 nodes = {
227 monA = generateHost { pkgs = pkgs; cephConfig = cephConfigMonA; networkConfig = networkMonA; };
228 osd0 = generateHost { pkgs = pkgs; cephConfig = cephConfigOsd cfg.osd0; networkConfig = networkOsd cfg.osd0; };
229 osd1 = generateHost { pkgs = pkgs; cephConfig = cephConfigOsd cfg.osd1; networkConfig = networkOsd cfg.osd1; };
230 osd2 = generateHost { pkgs = pkgs; cephConfig = cephConfigOsd cfg.osd2; networkConfig = networkOsd cfg.osd2; };
231 };
232
233 testScript = testscript;
234})