Skip to content

Commit 6233a63

Browse files
committed
update unit test
1 parent 215fbbc commit 6233a63

File tree

1 file changed

+24
-16
lines changed

1 file changed

+24
-16
lines changed

rust/operator-binary/src/hdfs_controller.rs

Lines changed: 24 additions & 16 deletions
Original file line numberDiff line numberDiff line change
@@ -847,11 +847,14 @@ spec:
847847
default:
848848
replicas: 1
849849
dataNodes:
850+
envOverrides:
851+
COMMON_VAR: role-value # overridden by role group below
852+
ROLE_VAR: role-value # only defined here at role level
850853
roleGroups:
851854
default:
852855
envOverrides:
853-
MY_ENV: my-value
854-
HADOOP_HOME: /not/the/default/path
856+
COMMON_VAR: group-value # overrides role value
857+
GROUP_VAR: group-value # only defined here at group level
855858
replicas: 1
856859
";
857860
let product_config = "
@@ -903,33 +906,38 @@ properties: []
903906
)
904907
.unwrap();
905908
let containers = pb.build().unwrap().spec.unwrap().containers;
906-
let main_container = containers
909+
let env_vars = containers
907910
.iter()
908911
.find(|c| c.name == role.to_string())
912+
.unwrap()
913+
.env
914+
.clone()
909915
.unwrap();
910916

911917
assert_eq!(
912-
main_container
913-
.env
914-
.clone()
915-
.unwrap()
916-
.into_iter()
917-
.find(|e| e.name == "MY_ENV")
918+
env_vars
919+
.iter()
920+
.find(|e| e.name == "COMMON_VAR")
918921
.unwrap()
919922
.value,
920-
Some("my-value".to_string())
923+
Some("group-value".to_string())
921924
);
922925

923926
assert_eq!(
924-
main_container
925-
.env
926-
.clone()
927+
env_vars
928+
.iter()
929+
.find(|e| e.name == "ROLE_VAR")
927930
.unwrap()
928-
.into_iter()
929-
.find(|e| e.name == "HADOOP_HOME")
931+
.value,
932+
Some("role-value".to_string())
933+
);
934+
assert_eq!(
935+
env_vars
936+
.iter()
937+
.find(|e| e.name == "GROUP_VAR")
930938
.unwrap()
931939
.value,
932-
Some("/not/the/default/path".to_string())
940+
Some("group-value".to_string())
933941
);
934942
}
935943
}

0 commit comments

Comments
 (0)