@@ -33,8 +33,9 @@ fn construct_jvm_args(
3333 zk : & ZookeeperCluster ,
3434 role : & Role < ZookeeperConfigFragment , ZookeeperServerRoleConfig , JavaCommonConfig > ,
3535 role_group : & str ,
36+ product_version : & str ,
3637) -> Result < Vec < String > , Error > {
37- let logging_framework = zk. logging_framework ( ) ;
38+ let logging_framework = zk. logging_framework ( product_version ) ;
3839
3940 let jvm_args = vec ! [
4041 format!( "-Djava.security.properties={STACKABLE_CONFIG_DIR}/{JVM_SECURITY_PROPERTIES_FILE}" ) ,
@@ -67,8 +68,9 @@ pub fn construct_non_heap_jvm_args(
6768 zk : & ZookeeperCluster ,
6869 role : & Role < ZookeeperConfigFragment , ZookeeperServerRoleConfig , JavaCommonConfig > ,
6970 role_group : & str ,
71+ product_version : & str ,
7072) -> Result < String , Error > {
71- let mut jvm_args = construct_jvm_args ( zk, role, role_group) ?;
73+ let mut jvm_args = construct_jvm_args ( zk, role, role_group, product_version ) ?;
7274 jvm_args. retain ( |arg| !is_heap_jvm_argument ( arg) ) ;
7375
7476 Ok ( jvm_args. join ( " " ) )
@@ -122,8 +124,15 @@ mod tests {
122124 replicas: 1
123125 "# ;
124126 let ( zookeeper, merged_config, role, rolegroup) = construct_boilerplate ( input) ;
125- let non_heap_jvm_args = construct_non_heap_jvm_args ( & zookeeper, & role, & rolegroup) . unwrap ( ) ;
126- let zk_server_heap_env = construct_zk_server_heap_env ( & merged_config) . unwrap ( ) ;
127+ let non_heap_jvm_args = construct_non_heap_jvm_args (
128+ & zookeeper,
129+ & role,
130+ & rolegroup,
131+ zookeeper. spec . image . product_version ( ) ,
132+ )
133+ . expect ( "test: function must pass" ) ;
134+ let zk_server_heap_env =
135+ construct_zk_server_heap_env ( & merged_config) . expect ( "test: function must pass" ) ;
127136
128137 assert_eq ! (
129138 non_heap_jvm_args,
@@ -167,8 +176,15 @@ mod tests {
167176 - -Dhttps.proxyPort=1234
168177 "# ;
169178 let ( zookeeper, merged_config, role, rolegroup) = construct_boilerplate ( input) ;
170- let non_heap_jvm_args = construct_non_heap_jvm_args ( & zookeeper, & role, & rolegroup) . unwrap ( ) ;
171- let zk_server_heap_env = construct_zk_server_heap_env ( & merged_config) . unwrap ( ) ;
179+ let non_heap_jvm_args = construct_non_heap_jvm_args (
180+ & zookeeper,
181+ & role,
182+ & rolegroup,
183+ zookeeper. spec . image . product_version ( ) ,
184+ )
185+ . expect ( "test: function must pass" ) ;
186+ let zk_server_heap_env =
187+ construct_zk_server_heap_env ( & merged_config) . expect ( "test: function must pass" ) ;
172188
173189 assert_eq ! (
174190 non_heap_jvm_args,
@@ -197,8 +213,12 @@ mod tests {
197213 let rolegroup_ref = zookeeper. server_rolegroup_ref ( "default" ) ;
198214 let merged_config = zookeeper
199215 . merged_config ( & zookeeper_role, & rolegroup_ref)
200- . unwrap ( ) ;
201- let role = zookeeper. spec . servers . clone ( ) . unwrap ( ) ;
216+ . expect ( "test: merged config can be created" ) ;
217+ let role = zookeeper
218+ . spec
219+ . servers
220+ . clone ( )
221+ . expect ( "test: server role is defined in the ZookeeperCluster" ) ;
202222
203223 ( zookeeper, merged_config, role, "default" . to_owned ( ) )
204224 }
0 commit comments