39
39
40
40
class ConfigLoadingFileTestCase (ConfigFileTestCase ):
41
41
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" ] )
43
43
with self .assertRaises (ConfigError ):
44
44
HomeServerConfig .load_config ("" , ["-c" , self .config_file ])
45
45
with self .assertRaises (ConfigError ):
@@ -76,7 +76,7 @@ def test_generates_and_loads_macaroon_secret_key(self) -> None:
76
76
)
77
77
78
78
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" ] )
80
80
config1 = HomeServerConfig .load_config ("" , ["-c" , self .config_file ])
81
81
config2 = HomeServerConfig .load_config ("" , ["-c" , self .config_file ])
82
82
config3 = HomeServerConfig .load_or_generate_config ("" , ["-c" , self .config_file ])
@@ -111,7 +111,7 @@ def test_disable_registration(self) -> None:
111
111
self .assertTrue (config3 .registration .enable_registration )
112
112
113
113
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" ] )
115
115
self .add_lines_to_config (["enable_metrics: true" ])
116
116
117
117
# The default Metrics Flags are off by default.
@@ -131,6 +131,7 @@ def test_depreciated_identity_server_flag_throws_error(self) -> None:
131
131
[
132
132
"turn_shared_secret_path: /does/not/exist" ,
133
133
"registration_shared_secret_path: /does/not/exist" ,
134
+ "macaroon_secret_key_path: /does/not/exist" ,
134
135
* ["redis:\n enabled: true\n password_path: /does/not/exist" ]
135
136
* (hiredis is not None ),
136
137
]
@@ -146,16 +147,20 @@ def test_secret_files_missing(self, config_str: str) -> None:
146
147
[
147
148
(
148
149
"turn_shared_secret_path: {}" ,
149
- lambda c : c .voip .turn_shared_secret ,
150
+ lambda c : c .voip .turn_shared_secret . encode ( "utf-8" ) ,
150
151
),
151
152
(
152
153
"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 ,
154
159
),
155
160
* [
156
161
(
157
162
"redis:\n enabled: true\n password_path: {}" ,
158
- lambda c : c .redis .redis_password ,
163
+ lambda c : c .redis .redis_password . encode ( "utf-8" ) ,
159
164
)
160
165
]
161
166
* (hiredis is not None ),
@@ -164,11 +169,13 @@ def test_secret_files_missing(self, config_str: str) -> None:
164
169
def test_secret_files_existing (
165
170
self , config_line : str , get_secret : Callable [[RootConfig ], str ]
166
171
) -> 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
+ )
168
175
with tempfile .NamedTemporaryFile (buffering = 0 ) as secret_file :
169
176
secret_file .write (b"53C237" )
170
177
171
178
self .add_lines_to_config (["" , config_line .format (secret_file .name )])
172
179
config = HomeServerConfig .load_config ("" , ["-c" , self .config_file ])
173
180
174
- self .assertEqual (get_secret (config ), "53C237" )
181
+ self .assertEqual (get_secret (config ), b "53C237" )
0 commit comments