2121from mock import Mock
2222
2323from twisted .internet import defer
24-
2524from synapse .api .constants import EventTypes , JoinRules , RoomCreationPreset
2625from synapse .rest import admin
2726from synapse .rest .client .v1 import login , room
@@ -62,7 +61,7 @@ def get_json(uri, args={}, headers=None):
6261 address_domain = args ["address" ].split ("@" )[1 ]
6362 return defer .succeed ({"hs" : address_domain })
6463
65- def post_urlencoded_get_json (uri , args = {}, headers = None ):
64+ def post_json_get_json (uri , post_json , args = {}, headers = None ):
6665 token = "" .join (random .choice (string .ascii_letters ) for _ in range (10 ))
6766 return defer .succeed (
6867 {
@@ -84,11 +83,13 @@ def post_urlencoded_get_json(uri, args={}, headers=None):
8483 mock_federation_client = Mock (spec = ["send_invite" ])
8584 mock_federation_client .send_invite .side_effect = send_invite
8685
87- mock_http_client = Mock (spec = ["get_json" , "post_urlencoded_get_json" ])
86+ mock_http_client = Mock (
87+ spec = ["get_json" , "post_json_get_json" ],
88+ )
8889 # Mocking the response for /info on the IS API.
8990 mock_http_client .get_json .side_effect = get_json
9091 # Mocking the response for /store-invite on the IS API.
91- mock_http_client .post_urlencoded_get_json .side_effect = post_urlencoded_get_json
92+ mock_http_client .post_json_get_json .side_effect = post_json_get_json
9293 self .hs = self .setup_test_homeserver (
9394 config = config ,
9495 federation_client = mock_federation_client ,
0 commit comments