1
0
Fork 0
icann-rdap/icann-rdap-srv/tests/integration/storage/data.rs

586 lines
17 KiB
Rust
Raw Normal View History

#![allow(non_snake_case)]
use {
icann_rdap_common::{
prelude::Numberish,
response::{
Autnum, Domain, Entity, Help, Nameserver, Network, Notice, NoticeOrRemark, RdapResponse,
},
},
icann_rdap_srv::{
config::{ServiceConfig, StorageType},
storage::{
data::{
load_data, AutnumId, AutnumOrError::AutnumObject, DomainId, DomainOrError,
EntityId, EntityOrError::EntityObject, NameserverId,
NameserverOrError::NameserverObject, NetworkId, NetworkIdType,
NetworkOrError::NetworkObject, Template,
},
mem::{config::MemConfig, ops::Mem},
CommonConfig, StoreOps,
},
},
test_dir::{DirBuilder, TestDir},
};
async fn new_and_init_mem(data_dir: String) -> Mem {
let mem_config = MemConfig::builder()
.common_config(CommonConfig::default())
.build();
let mem = Mem::new(mem_config.clone());
mem.init().await.expect("initialzing memeory");
load_data(
&ServiceConfig::non_server()
.data_dir(data_dir)
.storage_type(StorageType::Memory(mem_config))
.build()
.expect("building service config"),
&mem,
false,
)
.await
.expect("loading data");
mem
}
#[tokio::test]
async fn GIVEN_data_dir_with_domain_WHEN_mem_init_THEN_domain_is_loaded() {
// GIVEN
let ldh_name = "foo.example";
let temp = TestDir::temp();
let domain = Domain::builder().ldh_name(ldh_name).build();
let domain_file = temp.path("foo_example.json");
std::fs::write(
domain_file,
serde_json::to_string(&domain).expect("serializing domain"),
)
.expect("writing file");
// WHEN
let mem = new_and_init_mem(temp.root().to_string_lossy().to_string()).await;
// THEN
let actual = mem
.get_domain_by_ldh(ldh_name)
.await
.expect("getting domain by ldh");
assert!(matches!(actual, RdapResponse::Domain(_)));
let RdapResponse::Domain(domain) = actual else {
panic!()
};
assert_eq!(domain.ldh_name.as_ref().expect("ldhName is none"), ldh_name)
}
#[tokio::test]
async fn GIVEN_data_dir_with_domain_template_WHEN_mem_init_THEN_domains_are_loaded() {
// GIVEN
let ldh1 = "foo.example";
let ldh2 = "bar.example";
let temp = TestDir::temp();
let template = Template::Domain {
domain: DomainOrError::DomainObject(Box::new(
Domain::builder().ldh_name("example").build(),
)),
ids: vec![
DomainId::builder().ldh_name(ldh1).build(),
DomainId::builder().ldh_name(ldh2).build(),
],
};
let template_file = temp.path("example.template");
std::fs::write(
template_file,
serde_json::to_string(&template).expect("serializing template"),
)
.expect("writing file");
// WHEN
let mem = new_and_init_mem(temp.root().to_string_lossy().to_string()).await;
// THEN
for ldh in [ldh1, ldh2] {
let actual = mem
.get_domain_by_ldh(ldh)
.await
.expect("getting domain by ldh");
assert!(matches!(actual, RdapResponse::Domain(_)));
let RdapResponse::Domain(domain) = actual else {
panic!()
};
assert_eq!(domain.ldh_name.as_ref().expect("ldhName is none"), ldh)
}
}
#[tokio::test]
async fn GIVEN_data_dir_with_entity_WHEN_mem_init_THEN_entity_is_loaded() {
// GIVEN
let handle = "foo.example";
let temp = TestDir::temp();
let entity = Entity::builder().handle(handle).build();
let domain_file = temp.path("foo_example.json");
std::fs::write(
domain_file,
serde_json::to_string(&entity).expect("serializing entity"),
)
.expect("writing file");
// WHEN
let mem = new_and_init_mem(temp.root().to_string_lossy().to_string()).await;
// THEN
let actual = mem
.get_entity_by_handle(handle)
.await
.expect("getting entity by ldh");
assert!(matches!(actual, RdapResponse::Entity(_)));
let RdapResponse::Entity(entity) = actual else {
panic!()
};
assert_eq!(
entity
.object_common
.handle
.as_ref()
.expect("handle is none"),
handle
)
}
#[tokio::test]
async fn GIVEN_data_dir_with_entity_template_WHEN_mem_init_THEN_entities_are_loaded() {
// GIVEN
let handle1 = "foo";
let handle2 = "bar";
let temp = TestDir::temp();
let template = Template::Entity {
entity: EntityObject(Box::new(Entity::builder().handle("example").build())),
ids: vec![
EntityId::builder().handle(handle1).build(),
EntityId::builder().handle(handle2).build(),
],
};
let template_file = temp.path("example.template");
std::fs::write(
template_file,
serde_json::to_string(&template).expect("serializing template"),
)
.expect("writing file");
// WHEN
let mem = new_and_init_mem(temp.root().to_string_lossy().to_string()).await;
// THEN
for handle in [handle1, handle2] {
let actual = mem
.get_entity_by_handle(handle)
.await
.expect("getting entity by handle");
assert!(matches!(actual, RdapResponse::Entity(_)));
let RdapResponse::Entity(entity) = actual else {
panic!()
};
assert_eq!(
entity
.object_common
.handle
.as_ref()
.expect("handle is none"),
handle
)
}
}
#[tokio::test]
async fn GIVEN_data_dir_with_nameserver_WHEN_mem_init_THEN_nameserver_is_loaded() {
// GIVEN
let ldh_name = "ns.foo.example";
let temp = TestDir::temp();
let nameserver = Nameserver::builder().ldh_name(ldh_name).build().unwrap();
let nameserver_file = temp.path("ns_foo_example.json");
std::fs::write(
nameserver_file,
serde_json::to_string(&nameserver).expect("serializing nameserver"),
)
.expect("writing file");
// WHEN
let mem = new_and_init_mem(temp.root().to_string_lossy().to_string()).await;
// THEN
let actual = mem
.get_nameserver_by_ldh(ldh_name)
.await
.expect("getting nameserver by ldh");
assert!(matches!(actual, RdapResponse::Nameserver(_)));
let RdapResponse::Nameserver(nameserver) = actual else {
panic!()
};
assert_eq!(
nameserver.ldh_name.as_ref().expect("ldhName is none"),
ldh_name
)
}
#[tokio::test]
async fn GIVEN_data_dir_with_nameserver_template_WHEN_mem_init_THEN_nameservers_are_loaded() {
// GIVEN
let ldh1 = "ns.foo.example";
let ldh2 = "ns.bar.example";
let temp = TestDir::temp();
let template = Template::Nameserver {
nameserver: NameserverObject(Box::new(
Nameserver::builder().ldh_name("example").build().unwrap(),
)),
ids: vec![
NameserverId::builder().ldh_name(ldh1).build(),
NameserverId::builder().ldh_name(ldh2).build(),
],
};
let template_file = temp.path("example.template");
std::fs::write(
template_file,
serde_json::to_string(&template).expect("serializing template"),
)
.expect("writing file");
// WHEN
let mem = new_and_init_mem(temp.root().to_string_lossy().to_string()).await;
// THEN
for ldh in [ldh1, ldh2] {
let actual = mem
.get_nameserver_by_ldh(ldh)
.await
.expect("getting nameserver by ldh");
assert!(matches!(actual, RdapResponse::Nameserver(_)));
let RdapResponse::Nameserver(nameserver) = actual else {
panic!()
};
assert_eq!(nameserver.ldh_name.as_ref().expect("ldhName is none"), ldh)
}
}
#[tokio::test]
async fn GIVEN_data_dir_with_autnum_WHEN_mem_init_THEN_autnum_is_loaded() {
// GIVEN
let num = 700u32;
let temp = TestDir::temp();
let autnum = Autnum::builder().autnum_range(num..num).build();
let autnum_file = temp.path("700.json");
std::fs::write(
autnum_file,
serde_json::to_string(&autnum).expect("serializing autnum"),
)
.expect("writing file");
// WHEN
let mem = new_and_init_mem(temp.root().to_string_lossy().to_string()).await;
// THEN
let actual = mem
.get_autnum_by_num(num)
.await
.expect("getting autnum by num");
assert!(matches!(actual, RdapResponse::Autnum(_)));
let RdapResponse::Autnum(autnum) = actual else {
panic!()
};
assert_eq!(
*autnum.start_autnum.as_ref().expect("startAutnum is none"),
Numberish::<u32>::from(num)
)
}
#[tokio::test]
async fn GIVEN_data_dir_with_autnum_template_WHEN_mem_init_THEN_autnums_are_loaded() {
// GIVEN
let num1 = 700u32;
let num2 = 800u32;
let temp = TestDir::temp();
let template = Template::Autnum {
autnum: AutnumObject(Box::new(Autnum::builder().autnum_range(0..0).build())),
ids: vec![
AutnumId::builder()
.start_autnum(num1)
.end_autnum(num1)
.build(),
AutnumId::builder()
.start_autnum(num2)
.end_autnum(num2)
.build(),
],
};
let template_file = temp.path("example.template");
std::fs::write(
template_file,
serde_json::to_string(&template).expect("serializing template"),
)
.expect("writing file");
// WHEN
let mem = new_and_init_mem(temp.root().to_string_lossy().to_string()).await;
// THEN
for num in [num1, num2] {
let actual = mem
.get_autnum_by_num(num)
.await
.expect("getting autnum by num");
assert!(matches!(actual, RdapResponse::Autnum(_)));
let RdapResponse::Autnum(autnum) = actual else {
panic!()
};
assert_eq!(
*autnum.start_autnum.as_ref().expect("startAutnum is none"),
Numberish::<u32>::from(num)
)
}
}
#[tokio::test]
async fn GIVEN_data_dir_with_network_WHEN_mem_init_THEN_network_is_loaded() {
// GIVEN
let temp = TestDir::temp();
let network = Network::builder()
.cidr("10.0.0.0/24")
.build()
.expect("cidr parsing");
let net_file = temp.path("ten_net.json");
std::fs::write(
net_file,
serde_json::to_string(&network).expect("serializing network"),
)
.expect("writing file");
// WHEN
let mem = new_and_init_mem(temp.root().to_string_lossy().to_string()).await;
// THEN
let actual = mem
.get_network_by_ipaddr("10.0.0.0")
.await
.expect("getting autnum by num");
assert!(matches!(actual, RdapResponse::Network(_)));
let RdapResponse::Network(network) = actual else {
panic!()
};
assert_eq!(
*network
.start_address
.as_ref()
.expect("startAddress is none"),
"10.0.0.0"
)
}
#[tokio::test]
async fn GIVEN_data_dir_with_network_template_with_cidr_WHEN_mem_init_THEN_networks_are_loaded() {
// GIVEN
let cidr1 = "10.0.0.0/24";
let cidr2 = "10.0.1.0/24";
let start1 = "10.0.0.0";
let start2 = "10.0.1.0";
let temp = TestDir::temp();
let template = Template::Network {
network: NetworkObject(Box::new(
Network::builder()
.cidr("1.1.1.1/32")
.build()
.expect("parsing cidr"),
)),
ids: vec![
NetworkId::builder()
.network_id(NetworkIdType::Cidr(cidr1.parse().expect("parsing cidr")))
.build(),
NetworkId::builder()
.network_id(NetworkIdType::Cidr(cidr2.parse().expect("parsing cidr")))
.build(),
],
};
let template_file = temp.path("example.template");
std::fs::write(
template_file,
serde_json::to_string(&template).expect("serializing template"),
)
.expect("writing file");
// WHEN
let mem = new_and_init_mem(temp.root().to_string_lossy().to_string()).await;
// THEN
for (cidr, start) in [(cidr1, start1), (cidr2, start2)] {
let actual = mem
.get_network_by_cidr(cidr)
.await
.expect("getting cidr by num");
assert!(matches!(actual, RdapResponse::Network(_)));
let RdapResponse::Network(network) = actual else {
panic!()
};
assert_eq!(
*network
.start_address
.as_ref()
.expect("startAddress is none"),
start
)
}
}
#[tokio::test]
async fn GIVEN_data_dir_with_network_template_with_range_WHEN_mem_init_THEN_networks_are_loaded() {
// GIVEN
let start1 = "10.0.0.0";
let start2 = "10.0.1.0";
let end1 = "10.0.0.255";
let end2 = "10.0.1.255";
let temp = TestDir::temp();
let template = Template::Network {
network: NetworkObject(Box::new(
Network::builder()
.cidr("1.1.1.1/32")
.build()
.expect("parsing cidr"),
)),
ids: vec![
NetworkId::builder()
.network_id(NetworkIdType::Range {
start_address: start1.to_string(),
end_address: end1.to_string(),
})
.build(),
NetworkId::builder()
.network_id(NetworkIdType::Range {
start_address: start2.to_string(),
end_address: end2.to_string(),
})
.build(),
],
};
let template_file = temp.path("example.template");
std::fs::write(
template_file,
serde_json::to_string(&template).expect("serializing template"),
)
.expect("writing file");
// WHEN
let mem = new_and_init_mem(temp.root().to_string_lossy().to_string()).await;
// THEN
for (start, end) in [(start1, end1), (start2, end2)] {
let actual = mem
.get_network_by_ipaddr(end)
.await
.expect("getting cidr by addr");
assert!(matches!(actual, RdapResponse::Network(_)));
let RdapResponse::Network(network) = actual else {
panic!()
};
assert_eq!(
*network
.start_address
.as_ref()
.expect("startAddress is none"),
start
)
}
}
#[tokio::test]
async fn GIVEN_data_dir_with_default_help_WHEN_mem_init_THEN_default_help_is_loaded() {
// GIVEN
let temp = TestDir::temp();
let srvhelp = Help::builder()
.notice(Notice(
NoticeOrRemark::builder()
.description_entry("foo".to_string())
.build(),
))
.build();
let srvhelp_file = temp.path("__default.help");
std::fs::write(
srvhelp_file,
serde_json::to_string(&srvhelp).expect("serializing srvhelp"),
)
.expect("writing file");
// WHEN
let mem = new_and_init_mem(temp.root().to_string_lossy().to_string()).await;
// THEN
let actual = mem
.get_srv_help(None)
.await
.expect("getting default srvhelp");
assert!(matches!(actual, RdapResponse::Help(_)));
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_data_dir_with_host_help_WHEN_mem_init_THEN_host_help_is_loaded() {
// GIVEN
let temp = TestDir::temp();
let srvhelp = Help::builder()
.notice(Notice(
NoticeOrRemark::builder()
.description_entry("bar".to_string())
.build(),
))
.build();
let srvhelp_file = temp.path("foo_example_com.help");
std::fs::write(
srvhelp_file,
serde_json::to_string(&srvhelp).expect("serializing srvhelp"),
)
.expect("writing file");
// WHEN
let mem = new_and_init_mem(temp.root().to_string_lossy().to_string()).await;
// THEN
let actual = mem
.get_srv_help(Some("foo.example.com"))
.await
.expect("getting default srvhelp");
assert!(matches!(actual, RdapResponse::Help(_)));
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"
);
}