Skip to content

Commit 8eb26a7

Browse files
committed
fix issue with tests
1 parent 8a781b4 commit 8eb26a7

File tree

1 file changed

+44
-2
lines changed

1 file changed

+44
-2
lines changed

tembo-operator/tests/integration_tests.rs

Lines changed: 44 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -6060,7 +6060,28 @@ CREATE EVENT TRIGGER pgrst_watch
60606060
// Assert there are 4 pods running: postgres, pooler, postgrest and ferretdb
60616061
let pods: Api<Pod> = Api::namespaced(test.client.clone(), &namespace);
60626062
let pods_list = pods.list(&Default::default()).await.unwrap();
6063-
assert_eq!(pods_list.items.len(), 4);
6063+
let required_pods = ["postgres", "pooler", "postgrest", "fdb-api"];
6064+
6065+
// Check each required pod exists
6066+
for required_pod in required_pods {
6067+
let pod_exists = pods_list.items.iter().any(|pod| {
6068+
pod.metadata
6069+
.name
6070+
.as_ref()
6071+
.map_or(false, |name| name.contains(required_pod))
6072+
});
6073+
6074+
assert!(
6075+
pod_exists,
6076+
"Required pod '{}' was not found in namespace. Found pods: {:?}",
6077+
required_pod,
6078+
pods_list
6079+
.items
6080+
.iter()
6081+
.filter_map(|pod| pod.metadata.name.as_ref())
6082+
.collect::<Vec<_>>()
6083+
);
6084+
}
60646085

60656086
// Stop the cluster and check to make sure it's not running to ensure
60666087
// hibernate is doing its job.
@@ -6113,7 +6134,28 @@ CREATE EVENT TRIGGER pgrst_watch
61136134

61146135
// Assert there are 4 pods running: postgres, pooler, postgrest and ferretdb
61156136
let pods_list = pods.list(&Default::default()).await.unwrap();
6116-
assert_eq!(pods_list.items.len(), 4);
6137+
let required_pods = ["postgres", "pooler", "postgrest", "fdb-api"];
6138+
6139+
// Check each required pod exists
6140+
for required_pod in required_pods {
6141+
let pod_exists = pods_list.items.iter().any(|pod| {
6142+
pod.metadata
6143+
.name
6144+
.as_ref()
6145+
.map_or(false, |name| name.contains(required_pod))
6146+
});
6147+
6148+
assert!(
6149+
pod_exists,
6150+
"Required pod '{}' was not found in namespace. Found pods: {:?}",
6151+
required_pod,
6152+
pods_list
6153+
.items
6154+
.iter()
6155+
.filter_map(|pod| pod.metadata.name.as_ref())
6156+
.collect::<Vec<_>>()
6157+
);
6158+
}
61176159

61186160
// Assert there are 4 IngressRouteTCPs created after starting. One for postgres, pooler,
61196161
// ferretdb and postgrest

0 commit comments

Comments
 (0)