Skip to content

Commit 5953314

Browse files
committed
adjust conductor tests so they use the correct spec values when updating
1 parent 4728615 commit 5953314

File tree

3 files changed

+20
-18
lines changed

3 files changed

+20
-18
lines changed

conductor/Cargo.lock

Lines changed: 1 addition & 1 deletion
Some generated files are not rendered by default. Learn more about customizing how changed files appear on GitHub.

conductor/testdata/operator-values.yaml

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -2,12 +2,14 @@ controller:
22
image:
33
tag: latest
44
extraEnv:
5-
- name: ENABLE_INITIAL_BACKUP
5+
- name: ENABLE_BACKUP
66
value: "false"
77
- name: RUST_LOG
88
value: info,kube=info,controller=info
99
- name: OPENTELEMETRY_ENDPOINT_URL
1010
value: http://tempo.monitoring.svc.cluster.local:4317
11+
- name: DATA_PLANE_BASEDOMAIN
12+
value: coredb-development.com
1113

1214
pod-init:
1315
image:

conductor/tests/integration_tests.rs

Lines changed: 16 additions & 16 deletions
Original file line numberDiff line numberDiff line change
@@ -154,7 +154,7 @@ mod test {
154154
image: "default-image-value".to_string()
155155
})
156156
});
157-
let spec: CoreDBSpec = serde_json::from_value(spec_js).unwrap();
157+
let mut spec: CoreDBSpec = serde_json::from_value(spec_js).unwrap();
158158

159159
let msg = types::CRUDevent {
160160
organization_name: org_name.clone(),
@@ -163,9 +163,11 @@ mod test {
163163
inst_id: "inst_02s4UKVbRy34SAYVSwZq2H".to_owned(),
164164
event_type: types::Event::Create,
165165
dbname: dbname.clone(),
166-
spec: Some(spec),
166+
spec: Some(spec.clone()),
167167
};
168168

169+
// println!("Message: {:?}", msg);
170+
169171
let msg_id = queue.send(&myqueue, &msg).await;
170172
println!("Create msg_id: {msg_id:?}");
171173

@@ -201,11 +203,11 @@ mod test {
201203
.await
202204
.expect("error deleting message");
203205

204-
let spec = msg.message.spec.expect("No spec found in message");
206+
let passed_spec = msg.message.spec.expect("No spec found in message");
205207

206208
// assert that the message returned by Conductor includes the new metrics values in the spec
207-
//println!("spec: {:?}", spec);
208-
assert!(spec
209+
println!("spec: {:?}", passed_spec);
210+
assert!(passed_spec
209211
.metrics
210212
.expect("no metrics in data-plane-event message")
211213
.queries
@@ -214,10 +216,10 @@ mod test {
214216
.contains_key("pg_postmaster"));
215217

216218
assert!(
217-
!spec.extensions.is_empty(),
219+
!passed_spec.extensions.is_empty(),
218220
"Extension object missing from spec"
219221
);
220-
let extensions = spec.extensions;
222+
let extensions = passed_spec.extensions.clone();
221223
assert!(
222224
!extensions.is_empty(),
223225
"Expected at least one extension: {:?}",
@@ -234,7 +236,7 @@ mod test {
234236
// ADD AN EXTENSION - ASSERT IT MAKES IT TO STATUS.EXTENSIONS
235237
// conductor receives a CRUDevent from control plane
236238
// take note of number of extensions at this point in time
237-
let mut extensions_add = extensions.clone();
239+
// let mut extensions_add = extensions.clone();
238240
let _install_location = ExtensionInstallLocation::default();
239241
let install_location = ExtensionInstallLocation {
240242
enabled: true,
@@ -243,24 +245,21 @@ mod test {
243245
..ExtensionInstallLocation::default()
244246
};
245247
let install_location = install_location.clone();
246-
extensions_add.push(Extension {
248+
spec.extensions.push(Extension {
247249
name: "pg_jsonschema".to_owned(),
248250
description: Some("fake description".to_string()),
249251
locations: vec![install_location],
250252
});
251-
let num_expected_extensions = extensions_add.len();
252-
let spec_js = serde_json::json!({
253-
"extensions": extensions_add,
254-
});
255-
let spec: CoreDBSpec = serde_json::from_value(spec_js).unwrap();
253+
let num_expected_extensions = spec.extensions.len();
254+
// println!("Updated spec: {:?}", spec.clone());
256255
let msg = types::CRUDevent {
257256
organization_name: org_name.clone(),
258257
data_plane_id: "org_02s3owPQskuGXHE8vYsGSY".to_owned(),
259258
org_id: "org_02s3owPQskuGXHE8vYsGSY".to_owned(),
260259
inst_id: "inst_02s4UKVbRy34SAYVSwZq2H".to_owned(),
261260
event_type: types::Event::Update,
262261
dbname: dbname.clone(),
263-
spec: Some(spec),
262+
spec: Some(spec.clone()),
264263
};
265264
let msg_id = queue.send(&myqueue, &msg).await;
266265
println!("Update msg_id: {msg_id:?}");
@@ -269,7 +268,7 @@ mod test {
269268
let mut extensions: Vec<Extension> = vec![];
270269
while num_expected_extensions != extensions.len() {
271270
let msg = get_dataplane_message(retries, retry_delay, &queue).await;
272-
//println!("msg: {:?}", msg);
271+
// println!("Update msg: {:?}", msg);
273272
queue
274273
.archive("myqueue_data_plane", msg.msg_id)
275274
.await
@@ -365,6 +364,7 @@ mod test {
365364
dbname: dbname.clone(),
366365
spec: None,
367366
};
367+
// println!("DELETE msg: {:?}", msg);
368368
let msg_id = queue.send(&myqueue, &msg).await;
369369
println!("Delete msg_id: {msg_id:?}");
370370

0 commit comments

Comments
 (0)