Skip to content
This repository has been archived by the owner on Apr 26, 2024. It is now read-only.

Commit

Permalink
Rename get_http_client to get_federation_http_client.
Browse files Browse the repository at this point in the history
  • Loading branch information
clokep committed Nov 25, 2020
1 parent f559354 commit b22a244
Show file tree
Hide file tree
Showing 31 changed files with 50 additions and 45 deletions.
4 changes: 2 additions & 2 deletions synapse/crypto/keyring.py
Original file line number Diff line number Diff line change
Expand Up @@ -578,7 +578,7 @@ class PerspectivesKeyFetcher(BaseV2KeyFetcher):
def __init__(self, hs):
super().__init__(hs)
self.clock = hs.get_clock()
self.client = hs.get_http_client()
self.client = hs.get_federation_http_client()
self.key_servers = self.config.key_servers

async def get_keys(self, keys_to_fetch):
Expand Down Expand Up @@ -748,7 +748,7 @@ class ServerKeyFetcher(BaseV2KeyFetcher):
def __init__(self, hs):
super().__init__(hs)
self.clock = hs.get_clock()
self.client = hs.get_http_client()
self.client = hs.get_federation_http_client()

async def get_keys(self, keys_to_fetch):
"""
Expand Down
2 changes: 1 addition & 1 deletion synapse/federation/transport/client.py
Original file line number Diff line number Diff line change
Expand Up @@ -35,7 +35,7 @@ class TransportLayerClient:

def __init__(self, hs):
self.server_name = hs.hostname
self.client = hs.get_http_client()
self.client = hs.get_federation_http_client()

@log_function
def get_room_state_ids(self, destination, room_id, event_id):
Expand Down
2 changes: 1 addition & 1 deletion synapse/handlers/identity.py
Original file line number Diff line number Diff line change
Expand Up @@ -50,7 +50,7 @@ def __init__(self, hs):
self.http_client = SimpleHttpClient(hs)
# An HTTP client for contacting identity servers specified by clients.
self.blacklisting_http_client = hs.get_proxied_blacklisted_http_client()
self.federation_http_client = hs.get_http_client()
self.federation_http_client = hs.get_federation_http_client()
self.hs = hs

async def threepid_from_creds(
Expand Down
2 changes: 1 addition & 1 deletion synapse/rest/media/v1/media_repository.py
Original file line number Diff line number Diff line change
Expand Up @@ -66,7 +66,7 @@ class MediaRepository:
def __init__(self, hs):
self.hs = hs
self.auth = hs.get_auth()
self.client = hs.get_http_client()
self.client = hs.get_federation_http_client()
self.clock = hs.get_clock()
self.server_name = hs.hostname
self.store = hs.get_datastore()
Expand Down
2 changes: 1 addition & 1 deletion synapse/server.py
Original file line number Diff line number Diff line change
Expand Up @@ -381,7 +381,7 @@ def get_proxied_blacklisted_http_client(self) -> SimpleHttpClient:
)

@cache_in_self
def get_http_client(self) -> MatrixFederationHttpClient:
def get_federation_http_client(self) -> MatrixFederationHttpClient:
"""
An HTTP client for federation.
"""
Expand Down
4 changes: 3 additions & 1 deletion tests/api/test_filtering.py
Original file line number Diff line number Diff line change
Expand Up @@ -50,7 +50,9 @@ def setUp(self):
self.mock_http_client.put_json = DeferredMockCallable()

hs = yield setup_test_homeserver(
self.addCleanup, http_client=self.mock_http_client, keyring=Mock(),
self.addCleanup,
federation_http_client=self.mock_http_client,
keyring=Mock(),
)

self.filtering = hs.get_filtering()
Expand Down
2 changes: 1 addition & 1 deletion tests/app/test_frontend_proxy.py
Original file line number Diff line number Diff line change
Expand Up @@ -23,7 +23,7 @@ class FrontendProxyTests(HomeserverTestCase):
def make_homeserver(self, reactor, clock):

hs = self.setup_test_homeserver(
http_client=None, homeserver_to_use=GenericWorkerServer
federation_http_client=None, homeserver_to_use=GenericWorkerServer
)

return hs
Expand Down
4 changes: 2 additions & 2 deletions tests/app/test_openid_listener.py
Original file line number Diff line number Diff line change
Expand Up @@ -27,7 +27,7 @@
class FederationReaderOpenIDListenerTests(HomeserverTestCase):
def make_homeserver(self, reactor, clock):
hs = self.setup_test_homeserver(
http_client=None, homeserver_to_use=GenericWorkerServer
federation_http_client=None, homeserver_to_use=GenericWorkerServer
)
return hs

Expand Down Expand Up @@ -84,7 +84,7 @@ def test_openid_listener(self, names, expectation):
class SynapseHomeserverOpenIDListenerTests(HomeserverTestCase):
def make_homeserver(self, reactor, clock):
hs = self.setup_test_homeserver(
http_client=None, homeserver_to_use=SynapseHomeServer
federation_http_client=None, homeserver_to_use=SynapseHomeServer
)
return hs

Expand Down
6 changes: 4 additions & 2 deletions tests/crypto/test_keyring.py
Original file line number Diff line number Diff line change
Expand Up @@ -315,7 +315,7 @@ async def get_keys2(keys_to_fetch):
class ServerKeyFetcherTestCase(unittest.HomeserverTestCase):
def make_homeserver(self, reactor, clock):
self.http_client = Mock()
hs = self.setup_test_homeserver(http_client=self.http_client)
hs = self.setup_test_homeserver(federation_http_client=self.http_client)
return hs

def test_get_keys_from_server(self):
Expand Down Expand Up @@ -395,7 +395,9 @@ def make_homeserver(self, reactor, clock):
}
]

return self.setup_test_homeserver(http_client=self.http_client, config=config)
return self.setup_test_homeserver(
federation_http_client=self.http_client, config=config
)

def build_perspectives_response(
self, server_name: str, signing_key: SigningKey, valid_until_ts: int,
Expand Down
4 changes: 2 additions & 2 deletions tests/handlers/test_device.py
Original file line number Diff line number Diff line change
Expand Up @@ -27,7 +27,7 @@

class DeviceTestCase(unittest.HomeserverTestCase):
def make_homeserver(self, reactor, clock):
hs = self.setup_test_homeserver("server", http_client=None)
hs = self.setup_test_homeserver("server", federation_http_client=None)
self.handler = hs.get_device_handler()
self.store = hs.get_datastore()
return hs
Expand Down Expand Up @@ -229,7 +229,7 @@ def _record_user(

class DehydrationTestCase(unittest.HomeserverTestCase):
def make_homeserver(self, reactor, clock):
hs = self.setup_test_homeserver("server", http_client=None)
hs = self.setup_test_homeserver("server", federation_http_client=None)
self.handler = hs.get_device_handler()
self.registration = hs.get_registration_handler()
self.auth = hs.get_auth()
Expand Down
2 changes: 1 addition & 1 deletion tests/handlers/test_directory.py
Original file line number Diff line number Diff line change
Expand Up @@ -42,7 +42,7 @@ def register_query_handler(query_type, handler):
self.mock_registry.register_query_handler = register_query_handler

hs = self.setup_test_homeserver(
http_client=None,
federation_http_client=None,
resource_for_federation=Mock(),
federation_client=self.mock_federation,
federation_registry=self.mock_registry,
Expand Down
2 changes: 1 addition & 1 deletion tests/handlers/test_federation.py
Original file line number Diff line number Diff line change
Expand Up @@ -37,7 +37,7 @@ class FederationTestCase(unittest.HomeserverTestCase):
]

def make_homeserver(self, reactor, clock):
hs = self.setup_test_homeserver(http_client=None)
hs = self.setup_test_homeserver(federation_http_client=None)
self.handler = hs.get_federation_handler()
self.store = hs.get_datastore()
return hs
Expand Down
4 changes: 1 addition & 3 deletions tests/handlers/test_oidc.py
Original file line number Diff line number Diff line change
Expand Up @@ -150,9 +150,7 @@ def make_homeserver(self, reactor, clock):
config["oidc_config"] = oidc_config

hs = self.setup_test_homeserver(
http_client=self.http_client,
proxied_http_client=self.http_client,
config=config,
proxied_http_client=self.http_client, config=config,
)

self.handler = OidcHandler(hs)
Expand Down
2 changes: 1 addition & 1 deletion tests/handlers/test_presence.py
Original file line number Diff line number Diff line change
Expand Up @@ -463,7 +463,7 @@ class PresenceJoinTestCase(unittest.HomeserverTestCase):

def make_homeserver(self, reactor, clock):
hs = self.setup_test_homeserver(
"server", http_client=None, federation_sender=Mock()
"server", federation_http_client=None, federation_sender=Mock()
)
return hs

Expand Down
2 changes: 1 addition & 1 deletion tests/handlers/test_profile.py
Original file line number Diff line number Diff line change
Expand Up @@ -44,7 +44,7 @@ def register_query_handler(query_type, handler):

hs = yield setup_test_homeserver(
self.addCleanup,
http_client=None,
federation_http_client=None,
resource_for_federation=Mock(),
federation_client=self.mock_federation,
federation_server=Mock(),
Expand Down
6 changes: 3 additions & 3 deletions tests/handlers/test_typing.py
Original file line number Diff line number Diff line change
Expand Up @@ -70,7 +70,7 @@ def make_homeserver(self, reactor, clock):

hs = self.setup_test_homeserver(
notifier=Mock(),
http_client=mock_federation_client,
federation_http_client=mock_federation_client,
keyring=mock_keyring,
replication_streams={},
)
Expand Down Expand Up @@ -192,7 +192,7 @@ def test_started_typing_remote_send(self):
)
)

put_json = self.hs.get_http_client().put_json
put_json = self.hs.get_federation_http_client().put_json
put_json.assert_called_once_with(
"farm",
path="/_matrix/federation/v1/send/1000000",
Expand Down Expand Up @@ -270,7 +270,7 @@ def test_stopped_typing(self):

self.on_new_event.assert_has_calls([call("typing_key", 1, rooms=[ROOM_ID])])

put_json = self.hs.get_http_client().put_json
put_json = self.hs.get_federation_http_client().put_json
put_json.assert_called_once_with(
"farm",
path="/_matrix/federation/v1/send/1000000",
Expand Down
4 changes: 2 additions & 2 deletions tests/replication/_base.py
Original file line number Diff line number Diff line change
Expand Up @@ -67,7 +67,7 @@ def prepare(self, reactor, clock, hs):
# Make a new HomeServer object for the worker
self.reactor.lookups["testserv"] = "1.2.3.4"
self.worker_hs = self.setup_test_homeserver(
http_client=None,
federation_http_client=None,
homeserver_to_use=GenericWorkerServer,
config=self._get_worker_hs_config(),
reactor=self.reactor,
Expand Down Expand Up @@ -264,7 +264,7 @@ def make_worker_hs(
worker_app: Type of worker, e.g. `synapse.app.federation_sender`.
extra_config: Any extra config to use for this instances.
**kwargs: Options that get passed to `self.setup_test_homeserver`,
useful to e.g. pass some mocks for things like `http_client`
useful to e.g. pass some mocks for things like `federation_http_client`
Returns:
The new worker HomeServer instance.
Expand Down
10 changes: 5 additions & 5 deletions tests/replication/test_federation_sender_shard.py
Original file line number Diff line number Diff line change
Expand Up @@ -50,7 +50,7 @@ def test_send_event_single_sender(self):
self.make_worker_hs(
"synapse.app.federation_sender",
{"send_federation": True},
http_client=mock_client,
federation_http_client=mock_client,
)

user = self.register_user("user", "pass")
Expand Down Expand Up @@ -81,7 +81,7 @@ def test_send_event_sharded(self):
"worker_name": "sender1",
"federation_sender_instances": ["sender1", "sender2"],
},
http_client=mock_client1,
federation_http_client=mock_client1,
)

mock_client2 = Mock(spec=["put_json"])
Expand All @@ -93,7 +93,7 @@ def test_send_event_sharded(self):
"worker_name": "sender2",
"federation_sender_instances": ["sender1", "sender2"],
},
http_client=mock_client2,
federation_http_client=mock_client2,
)

user = self.register_user("user2", "pass")
Expand Down Expand Up @@ -144,7 +144,7 @@ def test_send_typing_sharded(self):
"worker_name": "sender1",
"federation_sender_instances": ["sender1", "sender2"],
},
http_client=mock_client1,
federation_http_client=mock_client1,
)

mock_client2 = Mock(spec=["put_json"])
Expand All @@ -156,7 +156,7 @@ def test_send_typing_sharded(self):
"worker_name": "sender2",
"federation_sender_instances": ["sender1", "sender2"],
},
http_client=mock_client2,
federation_http_client=mock_client2,
)

user = self.register_user("user3", "pass")
Expand Down
2 changes: 1 addition & 1 deletion tests/rest/admin/test_admin.py
Original file line number Diff line number Diff line change
Expand Up @@ -210,7 +210,7 @@ def write_to(r):
}
config["media_storage_providers"] = [provider_config]

hs = self.setup_test_homeserver(config=config, http_client=client)
hs = self.setup_test_homeserver(config=config, federation_http_client=client)

return hs

Expand Down
2 changes: 1 addition & 1 deletion tests/rest/client/v1/test_presence.py
Original file line number Diff line number Diff line change
Expand Up @@ -34,7 +34,7 @@ class PresenceTestCase(unittest.HomeserverTestCase):
def make_homeserver(self, reactor, clock):

hs = self.setup_test_homeserver(
"red", http_client=None, federation_client=Mock()
"red", federation_http_client=None, federation_client=Mock()
)

hs.presence_handler = Mock()
Expand Down
2 changes: 1 addition & 1 deletion tests/rest/client/v1/test_profile.py
Original file line number Diff line number Diff line change
Expand Up @@ -63,7 +63,7 @@ def setUp(self):
hs = yield setup_test_homeserver(
self.addCleanup,
"test",
http_client=None,
federation_http_client=None,
resource_for_client=self.mock_resource,
federation=Mock(),
federation_client=Mock(),
Expand Down
2 changes: 1 addition & 1 deletion tests/rest/client/v1/test_rooms.py
Original file line number Diff line number Diff line change
Expand Up @@ -45,7 +45,7 @@ class RoomBase(unittest.HomeserverTestCase):
def make_homeserver(self, reactor, clock):

self.hs = self.setup_test_homeserver(
"red", http_client=None, federation_client=Mock(),
"red", federation_http_client=None, federation_client=Mock(),
)

self.hs.get_federation_handler = Mock()
Expand Down
2 changes: 1 addition & 1 deletion tests/rest/client/v1/test_typing.py
Original file line number Diff line number Diff line change
Expand Up @@ -39,7 +39,7 @@ class RoomTypingTestCase(unittest.HomeserverTestCase):
def make_homeserver(self, reactor, clock):

hs = self.setup_test_homeserver(
"red", http_client=None, federation_client=Mock(),
"red", federation_http_client=None, federation_client=Mock(),
)

self.event_source = hs.get_event_sources().sources["typing"]
Expand Down
4 changes: 2 additions & 2 deletions tests/rest/key/v2/test_remote_key_resource.py
Original file line number Diff line number Diff line change
Expand Up @@ -39,7 +39,7 @@
class BaseRemoteKeyResourceTestCase(unittest.HomeserverTestCase):
def make_homeserver(self, reactor, clock):
self.http_client = Mock()
return self.setup_test_homeserver(http_client=self.http_client)
return self.setup_test_homeserver(federation_http_client=self.http_client)

def create_test_resource(self):
return create_resource_tree(
Expand Down Expand Up @@ -172,7 +172,7 @@ def prepare(self, reactor, clock, homeserver):
}
]
self.hs2 = self.setup_test_homeserver(
http_client=self.http_client2, config=config
federation_http_client=self.http_client2, config=config
)

# wire up outbound POST /key/v2/query requests from hs2 so that they
Expand Down
2 changes: 1 addition & 1 deletion tests/rest/media/v1/test_media_storage.py
Original file line number Diff line number Diff line change
Expand Up @@ -214,7 +214,7 @@ def write_to(r):
}
config["media_storage_providers"] = [provider_config]

hs = self.setup_test_homeserver(config=config, http_client=client)
hs = self.setup_test_homeserver(config=config, federation_http_client=client)

return hs

Expand Down
2 changes: 1 addition & 1 deletion tests/storage/test_e2e_room_keys.py
Original file line number Diff line number Diff line change
Expand Up @@ -26,7 +26,7 @@

class E2eRoomKeysHandlerTestCase(unittest.HomeserverTestCase):
def make_homeserver(self, reactor, clock):
hs = self.setup_test_homeserver("server", http_client=None)
hs = self.setup_test_homeserver("server", federation_http_client=None)
self.store = hs.get_datastore()
return hs

Expand Down
2 changes: 1 addition & 1 deletion tests/storage/test_purge.py
Original file line number Diff line number Diff line change
Expand Up @@ -27,7 +27,7 @@ class PurgeTests(HomeserverTestCase):
servlets = [room.register_servlets]

def make_homeserver(self, reactor, clock):
hs = self.setup_test_homeserver("server", http_client=None)
hs = self.setup_test_homeserver("server", federation_http_client=None)
return hs

def prepare(self, reactor, clock, hs):
Expand Down
2 changes: 1 addition & 1 deletion tests/storage/test_redaction.py
Original file line number Diff line number Diff line change
Expand Up @@ -34,7 +34,7 @@ def make_homeserver(self, reactor, clock):
config = self.default_config()
config["redaction_retention_period"] = "30d"
return self.setup_test_homeserver(
resource_for_federation=Mock(), http_client=None, config=config
resource_for_federation=Mock(), federation_http_client=None, config=config
)

def prepare(self, reactor, clock, hs):
Expand Down
2 changes: 1 addition & 1 deletion tests/storage/test_roommember.py
Original file line number Diff line number Diff line change
Expand Up @@ -36,7 +36,7 @@ class RoomMemberStoreTestCase(unittest.HomeserverTestCase):

def make_homeserver(self, reactor, clock):
hs = self.setup_test_homeserver(
resource_for_federation=Mock(), http_client=None
resource_for_federation=Mock(), federation_http_client=None
)
return hs

Expand Down
2 changes: 1 addition & 1 deletion tests/test_federation.py
Original file line number Diff line number Diff line change
Expand Up @@ -37,7 +37,7 @@ def setUp(self):
self.hs_clock = Clock(self.reactor)
self.homeserver = setup_test_homeserver(
self.addCleanup,
http_client=self.http_client,
federation_http_client=self.http_client,
clock=self.hs_clock,
reactor=self.reactor,
)
Expand Down
5 changes: 4 additions & 1 deletion tests/test_server.py
Original file line number Diff line number Diff line change
Expand Up @@ -38,7 +38,10 @@ def setUp(self):
self.reactor = ThreadedMemoryReactorClock()
self.hs_clock = Clock(self.reactor)
self.homeserver = setup_test_homeserver(
self.addCleanup, http_client=None, clock=self.hs_clock, reactor=self.reactor
self.addCleanup,
federation_http_client=None,
clock=self.hs_clock,
reactor=self.reactor,
)

def test_handler_for_request(self):
Expand Down

0 comments on commit b22a244

Please sign in to comment.