Skip to content

Commit 588c178

Browse files
committed
test: Use yaml_from_str_singleton_map helper instead of defining own
1 parent 04ded1e commit 588c178

3 files changed

Lines changed: 21 additions & 29 deletions

File tree

rust/operator-binary/src/crd/mod.rs

Lines changed: 0 additions & 11 deletions
Original file line numberDiff line numberDiff line change
@@ -1733,17 +1733,6 @@ mod tests {
17331733

17341734
use crate::crd::v1alpha1;
17351735

1736-
pub fn deserialize_yaml_str<'a, T: serde::de::Deserialize<'a>>(value: &'a str) -> T {
1737-
let deserializer = serde_yaml::Deserializer::from_str(value);
1738-
serde_yaml::with::singleton_map_recursive::deserialize(deserializer).unwrap()
1739-
}
1740-
1741-
pub fn deserialize_yaml_file<'a, T: serde::de::Deserialize<'a>>(path: &'a str) -> T {
1742-
let file = std::fs::File::open(path).unwrap();
1743-
let deserializer = serde_yaml::Deserializer::from_reader(file);
1744-
serde_yaml::with::singleton_map_recursive::deserialize(deserializer).unwrap()
1745-
}
1746-
17471736
impl RoundtripTestData for v1alpha1::DruidClusterSpec {
17481737
fn roundtrip_test_data() -> Vec<Self> {
17491738
vec![]

rust/operator-binary/src/crd/resource.rs

Lines changed: 9 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -252,13 +252,13 @@ mod test {
252252
},
253253
k8s_openapi::apimachinery::pkg::api::resource::Quantity,
254254
role_utils::{CommonConfiguration, RoleGroup},
255+
utils::yaml_from_str_singleton_map,
255256
};
256257

257258
use super::*;
258259
use crate::crd::{
259260
MiddleManagerConfig,
260261
storage::{HistoricalStorage, default_free_percentage_empty_dir},
261-
tests::deserialize_yaml_file,
262262
v1alpha1,
263263
};
264264

@@ -406,9 +406,10 @@ mod test {
406406

407407
#[test]
408408
fn test_resources() -> Result<(), Error> {
409-
let cluster = deserialize_yaml_file::<v1alpha1::DruidCluster>(
410-
"test/resources/crd/resource_merge/druid_cluster.yaml",
411-
);
409+
let cluster: v1alpha1::DruidCluster = yaml_from_str_singleton_map(include_str!(
410+
"../../test/resources/crd/resource_merge/druid_cluster.yaml"
411+
))
412+
.expect("failed to parse YAML");
412413

413414
let config = cluster.merged_config().unwrap();
414415
if let Some(RoleGroup {
@@ -482,9 +483,10 @@ mod test {
482483

483484
#[test]
484485
fn test_segment_cache() -> Result<(), Error> {
485-
let cluster = deserialize_yaml_file::<v1alpha1::DruidCluster>(
486-
"test/resources/crd/resource_merge/segment_cache.yaml",
487-
);
486+
let cluster: v1alpha1::DruidCluster = yaml_from_str_singleton_map(include_str!(
487+
"../../test/resources/crd/resource_merge/segment_cache.yaml"
488+
))
489+
.expect("failed to parse YAML");
488490

489491
// ---------- default role group
490492
let config = cluster.merged_config().unwrap();

rust/operator-binary/src/crd/tls.rs

Lines changed: 12 additions & 11 deletions
Original file line numberDiff line numberDiff line change
@@ -32,8 +32,9 @@ pub fn tls_default() -> Option<String> {
3232
#[cfg(test)]
3333
mod tests {
3434
use indoc::formatdoc;
35+
use stackable_operator::utils::yaml_from_str_singleton_map;
3536

36-
use crate::crd::{tests::deserialize_yaml_str, tls::DruidTls, v1alpha1::DruidClusterConfig};
37+
use crate::crd::{tls::DruidTls, v1alpha1::DruidClusterConfig};
3738

3839
const BASE_DRUID_CONFIGURATION: &str = r#"
3940
deepStorage:
@@ -50,8 +51,8 @@ zookeeperConfigMapName: zk-config-map
5051

5152
#[test]
5253
fn test_tls_default() {
53-
let druid_cluster_config =
54-
deserialize_yaml_str::<DruidClusterConfig>(BASE_DRUID_CONFIGURATION);
54+
let druid_cluster_config: DruidClusterConfig =
55+
yaml_from_str_singleton_map(BASE_DRUID_CONFIGURATION).expect("failed to parse YAML");
5556

5657
assert_eq!(
5758
druid_cluster_config.tls,
@@ -69,8 +70,8 @@ zookeeperConfigMapName: zk-config-map
6970
tls:
7071
serverAndInternalSecretClass: druid-secret-class
7172
"};
72-
dbg!(&input);
73-
let druid_cluster_config = deserialize_yaml_str::<DruidClusterConfig>(&input);
73+
let druid_cluster_config: DruidClusterConfig =
74+
yaml_from_str_singleton_map(&input).expect("failed to parse YAML");
7475

7576
assert_eq!(
7677
druid_cluster_config.tls,
@@ -87,8 +88,8 @@ zookeeperConfigMapName: zk-config-map
8788
{BASE_DRUID_CONFIGURATION}
8889
tls: null
8990
"};
90-
dbg!(&input);
91-
let druid_cluster_config = deserialize_yaml_str::<DruidClusterConfig>(&input);
91+
let druid_cluster_config: DruidClusterConfig =
92+
yaml_from_str_singleton_map(&input).expect("failed to parse YAML");
9293

9394
assert_eq!(druid_cluster_config.tls, None,);
9495
assert_eq!(druid_cluster_config.authentication, vec![]);
@@ -101,8 +102,8 @@ zookeeperConfigMapName: zk-config-map
101102
tls:
102103
serverAndInternalSecretClass: null
103104
"};
104-
dbg!(&input);
105-
let druid_cluster_config = deserialize_yaml_str::<DruidClusterConfig>(&input);
105+
let druid_cluster_config: DruidClusterConfig =
106+
yaml_from_str_singleton_map(&input).expect("failed to parse YAML");
106107

107108
assert_eq!(
108109
druid_cluster_config.tls,
@@ -122,8 +123,8 @@ zookeeperConfigMapName: zk-config-map
122123
authentication:
123124
- authenticationClass: druid-user-authentication-class
124125
"};
125-
dbg!(&input);
126-
let druid_cluster_config = deserialize_yaml_str::<DruidClusterConfig>(&input);
126+
let druid_cluster_config: DruidClusterConfig =
127+
yaml_from_str_singleton_map(&input).expect("failed to parse YAML");
127128

128129
assert_eq!(
129130
druid_cluster_config.tls,

0 commit comments

Comments
 (0)