pass tenant context to discovery session

asab/api/discovery.py

@@ -19,6 +19,7 @@
 	jwcrypto = None
 
 from .. import Service
+from ..contextvars import Tenant
 
 
 L = logging.getLogger(__name__)
@@ -404,6 +405,12 @@ def session(self, base_url=None, auth=None, headers=None, **kwargs) -> aiohttp.C
 				"Only instances of aiohttp.ClientRequest or the literal string 'internal' are allowed. "
 				"Found {}.".format(type(auth))
 			)
+
+		# Set tenant header if we are in tenant context
+		tenant = Tenant.get()
+		if tenant is not None:
+			headers["X-Tenant"] = tenant
+
 		return aiohttp.ClientSession(
 			base_url,
 			connector=aiohttp.TCPConnector(resolver=DiscoveryResolver(self)),