use rabbitmq_http_client::{
blocking_api::Client,
commons::VirtualHostLimitTarget,
requests::{EnforcedLimitParams, VirtualHostParams},
};
use crate::test_helpers::{PASSWORD, USERNAME, endpoint, rabbitmq_version_is_at_least};
#[test]
fn test_blocking_list_all_vhost_limits() {
if !rabbitmq_version_is_at_least(3, 13, 0) {
return;
}
let endpoint = endpoint();
let rc = Client::new(&endpoint, USERNAME, PASSWORD);
let vh_params = VirtualHostParams::named("test_list_all_vhost_limits");
let result1 = rc.create_vhost(&vh_params);
assert!(result1.is_ok());
let limit = EnforcedLimitParams::new(VirtualHostLimitTarget::MaxQueues, 500);
let result2 = rc.set_vhost_limit(vh_params.name, limit);
assert!(result2.is_ok());
let result3 = rc.list_all_vhost_limits();
assert!(result3.is_ok());
let vec = result3.unwrap();
assert!(vec.iter().any(|it| it.vhost == vh_params.name));
let key1 = VirtualHostLimitTarget::MaxConnections;
assert!(
!vec.iter()
.any(|it| it.vhost == vh_params.name && it.limits.get(key1.as_ref()).is_some())
);
let key2 = VirtualHostLimitTarget::MaxQueues;
assert!(
vec.iter()
.any(|it| it.vhost == vh_params.name && it.limits.get(key2.as_ref()).is_some())
);
rc.delete_vhost(vh_params.name, false).unwrap();
}
#[test]
fn test_blocking_list_vhost_limits() {
if !rabbitmq_version_is_at_least(3, 13, 0) {
return;
}
let endpoint = endpoint();
let rc = Client::new(&endpoint, USERNAME, PASSWORD);
let vh_params = VirtualHostParams::named("test_list_vhost_limits");
let result1 = rc.create_vhost(&vh_params);
assert!(result1.is_ok());
let limit = EnforcedLimitParams::new(VirtualHostLimitTarget::MaxConnections, 500);
let result2 = rc.set_vhost_limit(vh_params.name, limit);
assert!(result2.is_ok());
let result3 = rc.list_vhost_limits(vh_params.name);
assert!(result3.is_ok());
let vec = result3.unwrap();
let key1 = VirtualHostLimitTarget::MaxConnections;
assert!(
vec.iter()
.any(|it| it.vhost == vh_params.name && it.limits.get(key1.as_ref()).is_some())
);
let key2 = VirtualHostLimitTarget::MaxQueues;
assert!(
!vec.iter()
.any(|it| it.vhost == vh_params.name && it.limits.get(key2.as_ref()).is_some())
);
rc.delete_vhost(vh_params.name, false).unwrap();
}