1
0
Fork 0
icann-rdap/icann-rdap-srv/tests/integration/storage/mem/mod.rs
Daniel Baumann b06d3acde8
Adding upstream version 0.0.22.
Signed-off-by: Daniel Baumann <daniel@debian.org>
2025-05-08 18:41:54 +02:00

743 lines
19 KiB
Rust

#![allow(non_snake_case)]
use {
icann_rdap_common::{
prelude::Numberish,
response::{
Autnum, Common, Domain, Entity, Help, Nameserver, Network, Notice, NoticeOrRemark,
ObjectCommon, RdapResponse,
},
},
icann_rdap_srv::storage::{
mem::{config::MemConfig, ops::Mem},
CommonConfig, StoreOps,
},
rstest::rstest,
};
#[tokio::test]
async fn GIVEN_domain_in_mem_WHEN_new_truncate_tx_THEN_no_domain_in_mem() {
// GIVEN
let mem = Mem::default();
let mut tx = mem.new_tx().await.expect("new transaction");
tx.add_domain(&Domain::builder().ldh_name("foo.example").build())
.await
.expect("add domain in tx");
tx.commit().await.expect("tx commit");
// WHEN
let tx = mem.new_truncate_tx().await.expect("new truncate tx");
tx.commit().await.expect("tx commit");
// THEN
let actual = mem
.get_domain_by_ldh("foo.example")
.await
.expect("getting domain by ldh");
let RdapResponse::ErrorResponse(error) = actual else {
panic!()
};
assert_eq!(error.error_code, 404)
}
#[tokio::test]
async fn GIVEN_domain_in_mem_WHEN_lookup_domain_by_ldh_THEN_domain_returned() {
// GIVEN
let mem = Mem::default();
let mut tx = mem.new_tx().await.expect("new transaction");
tx.add_domain(&Domain::builder().ldh_name("foo.example").build())
.await
.expect("add domain in tx");
tx.commit().await.expect("tx commit");
// WHEN
let actual = mem
.get_domain_by_ldh("foo.example")
.await
.expect("getting domain by ldh");
// THEN
let RdapResponse::Domain(domain) = actual else {
panic!()
};
assert_eq!(
domain.ldh_name.as_ref().expect("ldhName is none"),
"foo.example"
)
}
#[tokio::test]
async fn GIVEN_domain_in_mem_WHEN_lookup_domain_by_unicode_THEN_domain_returned() {
// GIVEN
let mem = Mem::default();
let mut tx = mem.new_tx().await.expect("new transaction");
tx.add_domain(
&Domain::idn()
.unicode_name("foo.example")
.ldh_name("foo.example")
.build(),
)
.await
.expect("add domain in tx");
tx.commit().await.expect("tx commit");
// WHEN
let actual = mem
.get_domain_by_unicode("foo.example")
.await
.expect("getting domain by unicode");
// THEN
let RdapResponse::Domain(domain) = actual else {
panic!()
};
assert_eq!(
domain.unicode_name.as_ref().expect("unicodeName is none"),
"foo.example"
)
}
#[tokio::test]
async fn GIVEN_domain_in_mem_WHEN_search_domain_by_name_THEN_domain_returned() {
// GIVEN
let mem = Mem::default();
let mut tx = mem.new_tx().await.expect("new transaction");
tx.add_domain(
&Domain::idn()
.unicode_name("foo.example.com")
.ldh_name("foo.example.com")
.build(),
)
.await
.expect("add domain in tx");
tx.commit().await.expect("tx commit");
// WHEN
let actual = mem
.search_domains_by_name("foo.example.*")
.await
.expect("getting domain by unicode");
// THEN
let RdapResponse::DomainSearchResults(domains) = actual else {
panic!()
};
assert_eq!(domains.clone().results.len(), 1);
assert_eq!(
domains
.results
.first()
.expect("at least one")
.unicode_name
.as_ref()
.expect("unicodeName is none"),
"foo.example.com"
)
}
#[tokio::test]
async fn GIVEN_domain_in_mem_but_search_not_enabled_WHEN_search_domain_by_name_THEN_not_implemented(
) {
// GIVEN
let mem_config = MemConfig::builder()
.common_config(
CommonConfig::builder()
.domain_search_by_name_enable(false)
.build(),
)
.build();
let mem = Mem::new(mem_config);
let mut tx = mem.new_tx().await.expect("new transaction");
tx.add_domain(
&Domain::idn()
.unicode_name("foo.example.com")
.ldh_name("foo.example.com")
.build(),
)
.await
.expect("add domain in tx");
tx.commit().await.expect("tx commit");
// WHEN
let actual = mem
.search_domains_by_name("foo.example.*")
.await
.expect("getting domain by unicode");
// THEN
let RdapResponse::ErrorResponse(_e) = actual else {
panic!()
};
}
#[tokio::test]
async fn GIVEN_no_domain_in_mem_WHEN_lookup_domain_by_ldh_THEN_404_returned() {
// GIVEN
let mem = Mem::default();
// WHEN
let actual = mem
.get_domain_by_ldh("foo.example")
.await
.expect("getting domain by ldh");
// THEN
let RdapResponse::ErrorResponse(error) = actual else {
panic!()
};
assert_eq!(error.error_code, 404)
}
#[tokio::test]
async fn GIVEN_entity_in_mem_WHEN_lookup_entity_by_handle_THEN_entity_returned() {
// GIVEN
let mem = Mem::default();
let mut tx = mem.new_tx().await.expect("new transaction");
tx.add_entity(&Entity::builder().handle("foo").build())
.await
.expect("add entity in tx");
tx.commit().await.expect("entity tx commit");
// WHEN
let actual = mem
.get_entity_by_handle("foo")
.await
.expect("getting entity by handle");
// THEN
let RdapResponse::Entity(entity) = actual else {
panic!()
};
assert_eq!(
entity
.object_common
.handle
.as_ref()
.expect("handle is none"),
"foo"
)
}
#[tokio::test]
async fn GIVEN_no_entity_in_mem_WHEN_lookup_entity_by_handle_THEN_404_returned() {
// GIVEN
let mem = Mem::default();
// WHEN
let actual = mem
.get_entity_by_handle("foo")
.await
.expect("getting entity by handle");
// THEN
let RdapResponse::ErrorResponse(error) = actual else {
panic!()
};
assert_eq!(error.error_code, 404)
}
#[tokio::test]
async fn GIVEN_nameserver_in_mem_WHEN_lookup_nameserver_by_ldh_THEN_nameserver_returned() {
// GIVEN
let mem = Mem::default();
let mut tx = mem.new_tx().await.expect("new transaction");
tx.add_nameserver(
&Nameserver::builder()
.ldh_name("ns.foo.example")
.build()
.unwrap(),
)
.await
.expect("add nameserver in tx");
tx.commit().await.expect("tx commit");
// WHEN
let actual = mem
.get_nameserver_by_ldh("ns.foo.example")
.await
.expect("getting nameserver by ldh");
// THEN
let RdapResponse::Nameserver(nameserver) = actual else {
panic!()
};
assert_eq!(
nameserver.ldh_name.as_ref().expect("ldhName is none"),
"ns.foo.example"
)
}
#[tokio::test]
async fn GIVEN_no_nameserver_in_mem_WHEN_lookup_nameserver_by_ldh_THEN_404_returned() {
// GIVEN
let mem = Mem::default();
// WHEN
let actual = mem
.get_nameserver_by_ldh("ns.foo.example")
.await
.expect("getting nameserver by ldh");
// THEN
let RdapResponse::ErrorResponse(error) = actual else {
panic!()
};
assert_eq!(error.error_code, 404)
}
#[tokio::test]
async fn GIVEN_autnum_in_mem_WHEN_lookup_autnum_by_start_autnum_THEN_autnum_returned() {
// GIVEN
let mem = Mem::default();
let mut tx = mem.new_tx().await.expect("new transaction");
tx.add_autnum(&Autnum::builder().autnum_range(700..710).build())
.await
.expect("add autnum in tx");
tx.commit().await.expect("tx commit");
// WHEN
let actual = mem
.get_autnum_by_num(700)
.await
.expect("getting autnum by num");
// THEN
let RdapResponse::Autnum(autnum) = actual else {
panic!()
};
assert_eq!(
*autnum.start_autnum.as_ref().expect("startNum is none"),
Numberish::<u32>::from(700)
);
assert_eq!(
*autnum.end_autnum.as_ref().expect("startNum is none"),
Numberish::<u32>::from(710)
);
}
#[tokio::test]
async fn GIVEN_autnum_in_mem_WHEN_lookup_autnum_by_end_autnum_THEN_autnum_returned() {
// GIVEN
let mem = Mem::default();
let mut tx = mem.new_tx().await.expect("new transaction");
tx.add_autnum(&Autnum::builder().autnum_range(700..710).build())
.await
.expect("add autnum in tx");
tx.commit().await.expect("tx commit");
// WHEN
let actual = mem
.get_autnum_by_num(710)
.await
.expect("getting autnum by num");
// THEN
let RdapResponse::Autnum(autnum) = actual else {
panic!()
};
assert_eq!(
*autnum.start_autnum.as_ref().expect("startNum is none"),
Numberish::<u32>::from(700)
);
assert_eq!(
*autnum.end_autnum.as_ref().expect("startNum is none"),
Numberish::<u32>::from(710)
);
}
#[tokio::test]
async fn GIVEN_no_autnum_in_mem_WHEN_lookup_autnum_by_num_THEN_404_returned() {
// GIVEN
let mem = Mem::default();
// WHEN
let actual = mem
.get_autnum_by_num(700)
.await
.expect("getting autnum by num");
// THEN
let RdapResponse::ErrorResponse(error) = actual else {
panic!()
};
assert_eq!(error.error_code, 404)
}
#[rstest]
#[case("192.168.0.0/24", "192.168.0.1", "192.168.0.0", "192.168.0.255")]
#[case("192.168.0.0/24", "192.168.0.0", "192.168.0.0", "192.168.0.255")]
#[case("192.168.0.0/24", "192.168.0.254", "192.168.0.0", "192.168.0.255")]
#[case("192.168.0.0/24", "192.168.0.255", "192.168.0.0", "192.168.0.255")]
#[tokio::test]
async fn GIVEN_network_in_mem_WHEN_lookup_network_by_address_THEN_network_returned(
#[case] cidr: &str,
#[case] addr: &str,
#[case] start: &str,
#[case] end: &str,
) {
// GIVEN
let mem = Mem::default();
let mut tx = mem.new_tx().await.expect("new transaction");
tx.add_network(&Network::builder().cidr(cidr).build().expect("cidr parsing"))
.await
.expect("add network in tx");
tx.commit().await.expect("tx commit");
// WHEN
let actual = mem
.get_network_by_ipaddr(addr)
.await
.expect("getting network by num");
// THEN
let RdapResponse::Network(network) = actual else {
panic!()
};
assert_eq!(
*network
.start_address
.as_ref()
.expect("startAddress is none"),
start
);
assert_eq!(
*network.end_address.as_ref().expect("endAddress is none"),
end
);
}
#[tokio::test]
async fn GIVEN_no_network_in_mem_WHEN_lookup_network_by_address_THEN_404_returned() {
// GIVEN
let mem = Mem::default();
// WHEN
let actual = mem
.get_network_by_ipaddr("192.168.0.1")
.await
.expect("getting network by address");
// THEN
let RdapResponse::ErrorResponse(error) = actual else {
panic!()
};
assert_eq!(error.error_code, 404)
}
#[rstest]
#[case(&["192.168.0.0/16", "192.168.0.0/8", "192.168.0.0/24"], "192.168.0.1", "192.168.0.0", "192.168.0.255")]
#[case(&["2001::/64", "2001::/56", "2001::/20"], "2001::1", "2001::", "2001::ffff:ffff:ffff:ffff")]
#[tokio::test]
async fn GIVEN_contained_networks_in_mem_WHEN_lookup_network_by_address_THEN_most_specific_network_returned(
#[case] cidrs: &[&str],
#[case] addr: &str,
#[case] start: &str,
#[case] end: &str,
) {
// GIVEN
let mem = Mem::default();
let mut tx = mem.new_tx().await.expect("new transaction");
for cidr in cidrs {
tx.add_network(
&Network::builder()
.cidr(*cidr)
.build()
.expect("cidr parsing"),
)
.await
.expect("add network in tx");
}
tx.commit().await.expect("tx commit");
// WHEN
let actual = mem
.get_network_by_ipaddr(addr)
.await
.expect("getting network by num");
// THEN
let RdapResponse::Network(network) = actual else {
panic!()
};
assert_eq!(
*network
.start_address
.as_ref()
.expect("startAddress is none"),
start
);
assert_eq!(
*network.end_address.as_ref().expect("endAddress is none"),
end
);
}
#[tokio::test]
async fn GIVEN_offbit_network_in_mem_WHEN_lookup_network_by_first_address_THEN_network_returned() {
// GIVEN
let start = "10.0.0.0";
let end = "10.0.1.255";
let mem = Mem::default();
let mut tx = mem.new_tx().await.expect("new transaction");
tx.add_network(&Network {
common: Common {
rdap_conformance: None,
notices: None,
},
object_common: ObjectCommon {
object_class_name: "ip network".to_string(),
handle: None,
remarks: None,
links: None,
events: None,
status: None,
port_43: None,
entities: None,
redacted: None,
},
start_address: Some(start.to_string()),
end_address: Some(end.to_string()),
ip_version: Some("v4".to_string()),
name: None,
network_type: None,
parent_handle: None,
country: None,
cidr0_cidrs: None,
})
.await
.expect("add network in tx");
tx.commit().await.expect("tx commit");
// WHEN
let actual = mem
.get_network_by_ipaddr(start)
.await
.expect("getting network by num");
// THEN
let RdapResponse::Network(network) = actual else {
panic!()
};
assert_eq!(
*network
.start_address
.as_ref()
.expect("startAddress is none"),
start
);
assert_eq!(
*network.end_address.as_ref().expect("endAddress is none"),
end
);
}
#[tokio::test]
async fn GIVEN_offbit_network_in_mem_WHEN_lookup_network_by_last_address_THEN_network_returned() {
// GIVEN
let start = "10.0.0.0";
let end = "10.0.1.255";
let mem = Mem::default();
let mut tx = mem.new_tx().await.expect("new transaction");
tx.add_network(&Network {
common: Common {
rdap_conformance: None,
notices: None,
},
object_common: ObjectCommon {
object_class_name: "ip network".to_string(),
handle: None,
remarks: None,
links: None,
events: None,
status: None,
port_43: None,
entities: None,
redacted: None,
},
start_address: Some(start.to_string()),
end_address: Some(end.to_string()),
ip_version: Some("v4".to_string()),
name: None,
network_type: None,
parent_handle: None,
country: None,
cidr0_cidrs: None,
})
.await
.expect("add network in tx");
tx.commit().await.expect("tx commit");
// WHEN
let actual = mem
.get_network_by_ipaddr(end)
.await
.expect("getting network by num");
// THEN
let RdapResponse::Network(network) = actual else {
panic!()
};
assert_eq!(
*network
.start_address
.as_ref()
.expect("startAddress is none"),
start
);
assert_eq!(
*network.end_address.as_ref().expect("endAddress is none"),
end
);
}
#[rstest]
#[case("192.168.0.0/16", "192.168.0.0/24", "192.168.0.0", "192.168.255.255")]
#[case("192.168.0.0/16", "192.168.0.0/16", "192.168.0.0", "192.168.255.255")]
#[tokio::test]
async fn GIVEN_network_in_mem_WHEN_lookup_network_by_cidr_THEN_network_returned(
#[case] cidr: &str,
#[case] lookup: &str,
#[case] start: &str,
#[case] end: &str,
) {
// GIVEN
let mem = Mem::default();
let mut tx = mem.new_tx().await.expect("new transaction");
tx.add_network(&Network::builder().cidr(cidr).build().expect("cidr parsing"))
.await
.expect("add network in tx");
tx.commit().await.expect("tx commit");
// WHEN
let actual = mem
.get_network_by_cidr(lookup)
.await
.expect("getting network by cidr");
// THEN
let RdapResponse::Network(network) = actual else {
panic!()
};
assert_eq!(
*network
.start_address
.as_ref()
.expect("startAddress is none"),
start
);
assert_eq!(
*network.end_address.as_ref().expect("endAddress is none"),
end
);
}
#[tokio::test]
async fn GIVEN_no_network_in_mem_WHEN_lookup_network_by_cidr_THEN_404_returned() {
// GIVEN
let mem = Mem::default();
// WHEN
let actual = mem
.get_network_by_cidr("192.168.0.0/24")
.await
.expect("getting network by address");
// THEN
let RdapResponse::ErrorResponse(error) = actual else {
panic!()
};
assert_eq!(error.error_code, 404)
}
#[tokio::test]
async fn GIVEN_default_help_in_mem_WHEN_lookup_help_with_no_host_THEN_get_default_help() {
// GIVEN
let mem = Mem::default();
let mut tx = mem.new_tx().await.expect("new transaction");
tx.add_srv_help(
&Help::builder()
.notice(Notice(
NoticeOrRemark::builder()
.description_entry("foo".to_string())
.build(),
))
.build(),
None,
)
.await
.expect("adding srv help");
tx.commit().await.expect("tx commit");
// WHEN
let actual = mem.get_srv_help(None).await.expect("getting srv helf");
// THEN
let RdapResponse::Help(srvhelp) = actual else {
panic!()
};
let notice = srvhelp
.common
.notices
.expect("no notices in srvhelp")
.first()
.expect("notices empty")
.to_owned();
assert_eq!(
notice
.description
.as_ref()
.expect("no description!")
.vec()
.first()
.expect("no description in notice"),
"foo"
);
}
#[tokio::test]
async fn GIVEN_help_in_mem_WHEN_lookup_help_with_host_THEN_get_host_help() {
// GIVEN
let mem = Mem::default();
let mut tx = mem.new_tx().await.expect("new transaction");
tx.add_srv_help(
&Help::builder()
.notice(Notice(
NoticeOrRemark::builder()
.description_entry("bar".to_string())
.build(),
))
.build(),
Some("bar.example.com"),
)
.await
.expect("adding srv help");
tx.commit().await.expect("tx commit");
// WHEN
let actual = mem
.get_srv_help(Some("bar.example.com"))
.await
.expect("getting srv helf");
// THEN
let RdapResponse::Help(srvhelp) = actual else {
panic!()
};
let notice = srvhelp
.common
.notices
.expect("no notices in srvhelp")
.first()
.expect("notices empty")
.to_owned();
assert_eq!(
notice
.description
.as_ref()
.expect("no description")
.vec()
.first()
.expect("no description in notice"),
"bar"
);
}