3939
4040class ConfigLoadingFileTestCase (ConfigFileTestCase ):
4141 def test_load_fails_if_server_name_missing (self ) -> None :
42- self .generate_config_and_remove_lines_containing ("server_name" )
42+ self .generate_config_and_remove_lines_containing ([ "server_name" ] )
4343 with self .assertRaises (ConfigError ):
4444 HomeServerConfig .load_config ("" , ["-c" , self .config_file ])
4545 with self .assertRaises (ConfigError ):
@@ -76,7 +76,7 @@ def test_generates_and_loads_macaroon_secret_key(self) -> None:
7676 )
7777
7878 def test_load_succeeds_if_macaroon_secret_key_missing (self ) -> None :
79- self .generate_config_and_remove_lines_containing ("macaroon" )
79+ self .generate_config_and_remove_lines_containing ([ "macaroon" ] )
8080 config1 = HomeServerConfig .load_config ("" , ["-c" , self .config_file ])
8181 config2 = HomeServerConfig .load_config ("" , ["-c" , self .config_file ])
8282 config3 = HomeServerConfig .load_or_generate_config ("" , ["-c" , self .config_file ])
@@ -111,7 +111,7 @@ def test_disable_registration(self) -> None:
111111 self .assertTrue (config3 .registration .enable_registration )
112112
113113 def test_stats_enabled (self ) -> None :
114- self .generate_config_and_remove_lines_containing ("enable_metrics" )
114+ self .generate_config_and_remove_lines_containing ([ "enable_metrics" ] )
115115 self .add_lines_to_config (["enable_metrics: true" ])
116116
117117 # The default Metrics Flags are off by default.
@@ -131,6 +131,7 @@ def test_depreciated_identity_server_flag_throws_error(self) -> None:
131131 [
132132 "turn_shared_secret_path: /does/not/exist" ,
133133 "registration_shared_secret_path: /does/not/exist" ,
134+ "macaroon_secret_key_path: /does/not/exist" ,
134135 * ["redis:\n enabled: true\n password_path: /does/not/exist" ]
135136 * (hiredis is not None ),
136137 ]
@@ -146,16 +147,20 @@ def test_secret_files_missing(self, config_str: str) -> None:
146147 [
147148 (
148149 "turn_shared_secret_path: {}" ,
149- lambda c : c .voip .turn_shared_secret ,
150+ lambda c : c .voip .turn_shared_secret . encode ( "utf-8" ) ,
150151 ),
151152 (
152153 "registration_shared_secret_path: {}" ,
153- lambda c : c .registration .registration_shared_secret ,
154+ lambda c : c .registration .registration_shared_secret .encode ("utf-8" ),
155+ ),
156+ (
157+ "macaroon_secret_key_path: {}" ,
158+ lambda c : c .key .macaroon_secret_key ,
154159 ),
155160 * [
156161 (
157162 "redis:\n enabled: true\n password_path: {}" ,
158- lambda c : c .redis .redis_password ,
163+ lambda c : c .redis .redis_password . encode ( "utf-8" ) ,
159164 )
160165 ]
161166 * (hiredis is not None ),
@@ -164,11 +169,13 @@ def test_secret_files_missing(self, config_str: str) -> None:
164169 def test_secret_files_existing (
165170 self , config_line : str , get_secret : Callable [[RootConfig ], str ]
166171 ) -> None :
167- self .generate_config_and_remove_lines_containing ("registration_shared_secret" )
172+ self .generate_config_and_remove_lines_containing (
173+ ["registration_shared_secret" , "macaroon_secret_key" ]
174+ )
168175 with tempfile .NamedTemporaryFile (buffering = 0 ) as secret_file :
169176 secret_file .write (b"53C237" )
170177
171178 self .add_lines_to_config (["" , config_line .format (secret_file .name )])
172179 config = HomeServerConfig .load_config ("" , ["-c" , self .config_file ])
173180
174- self .assertEqual (get_secret (config ), "53C237" )
181+ self .assertEqual (get_secret (config ), b "53C237" )
0 commit comments