22# pylint: disable=no-value-for-parameter
33# pylint: disable=super-init-not-called
44# pylint: disable=unused-argument
5+ # pylint: disable=no-self-use
56
67from uuid import uuid4
78
@@ -47,7 +48,7 @@ class DummyRpcClient(RabbitMQRPCClient):
4748
4849 def __init__ (self ):
4950 self .client_name = "dummy_client"
50- self .settings = {} # Add a settings attribute to avoid AttributeError
51+ self .settings = {} # type: ignore # Add a settings attribute to avoid AttributeError
5152
5253 async def request (self , namespace : str , method_name : str , ** kwargs ):
5354 # Mock implementation of the request method
@@ -103,6 +104,7 @@ async def get_function(
103104 async def run_function (
104105 self , rabbitmq_rpc_client : RabbitMQRPCClient , function_id : str , inputs : dict
105106 ) -> dict :
107+ assert isinstance (rabbitmq_rpc_client , RabbitMQRPCClient )
106108 # Mimic running a function and returning a success status
107109 if function_id not in self ._functions :
108110 raise FunctionIDNotFoundError (function_id = function_id )
@@ -114,6 +116,7 @@ async def list_functions(
114116 pagination_offset : int = 0 ,
115117 pagination_limit : int = 10 ,
116118 ) -> tuple [list [RegisteredFunction ], PageMetaInfoLimitOffset ]:
119+ assert isinstance (rabbitmq_rpc_client , RabbitMQRPCClient )
117120 # Mimic listing all functions
118121 functions_list = list (self ._functions .values ())[
119122 pagination_offset : pagination_offset + pagination_limit
@@ -130,6 +133,7 @@ async def list_functions(
130133 async def delete_function (
131134 self , rabbitmq_rpc_client : RabbitMQRPCClient , function_id : str
132135 ) -> None :
136+ assert isinstance (rabbitmq_rpc_client , RabbitMQRPCClient )
133137 # Mimic deleting a function
134138 if function_id in self ._functions :
135139 del self ._functions [function_id ]
@@ -139,6 +143,7 @@ async def delete_function(
139143 async def register_function_job (
140144 self , rabbitmq_rpc_client : RabbitMQRPCClient , function_job : FunctionJob
141145 ) -> RegisteredFunctionJob :
146+ assert isinstance (rabbitmq_rpc_client , RabbitMQRPCClient )
142147 # Mimic registering a function job
143148 uid = uuid4 ()
144149 self ._function_jobs [uid ] = TypeAdapter (RegisteredFunctionJob ).validate_python (
@@ -158,6 +163,7 @@ async def register_function_job(
158163 async def get_function_job (
159164 self , rabbitmq_rpc_client : RabbitMQRPCClient , function_job_id : str
160165 ) -> RegisteredFunctionJob :
166+ assert isinstance (rabbitmq_rpc_client , RabbitMQRPCClient )
161167 # Mimic retrieval of a function job based on function_job_id and raise 404 if not found
162168 if function_job_id not in self ._function_jobs :
163169 raise FunctionJobIDNotFoundError (function_id = function_job_id )
@@ -169,6 +175,9 @@ async def list_function_jobs(
169175 pagination_offset : int ,
170176 pagination_limit : int ,
171177 ) -> tuple [list [RegisteredFunctionJob ], PageMetaInfoLimitOffset ]:
178+
179+ assert isinstance (rabbitmq_rpc_client , RabbitMQRPCClient )
180+
172181 # Mimic listing all function jobs
173182 function_jobs_list = list (self ._function_jobs .values ())[
174183 pagination_offset : pagination_offset + pagination_limit
@@ -185,6 +194,7 @@ async def list_function_jobs(
185194 async def delete_function_job (
186195 self , rabbitmq_rpc_client : RabbitMQRPCClient , function_job_id : str
187196 ) -> None :
197+ assert isinstance (rabbitmq_rpc_client , RabbitMQRPCClient )
188198 # Mimic deleting a function job
189199 if function_job_id in self ._function_jobs :
190200 del self ._function_jobs [function_job_id ]
@@ -196,6 +206,7 @@ async def register_function_job_collection(
196206 rabbitmq_rpc_client : RabbitMQRPCClient ,
197207 function_job_collection : FunctionJobCollection ,
198208 ) -> RegisteredFunctionJobCollection :
209+ assert isinstance (rabbitmq_rpc_client , RabbitMQRPCClient )
199210 # Mimic registering a function job collection
200211 uid = uuid4 ()
201212 self ._function_job_collections [uid ] = TypeAdapter (
@@ -213,6 +224,7 @@ async def register_function_job_collection(
213224 async def get_function_job_collection (
214225 self , rabbitmq_rpc_client : RabbitMQRPCClient , function_job_collection_id : str
215226 ) -> RegisteredFunctionJobCollection :
227+ assert isinstance (rabbitmq_rpc_client , RabbitMQRPCClient )
216228 # Mimic retrieval of a function job collection based on collection_id and raise 404 if not found
217229 if function_job_collection_id not in self ._function_job_collections :
218230 raise FunctionJobCollectionIDNotFoundError (
@@ -226,6 +238,7 @@ async def list_function_job_collections(
226238 pagination_offset : int ,
227239 pagination_limit : int ,
228240 ) -> tuple [list [RegisteredFunctionJobCollection ], PageMetaInfoLimitOffset ]:
241+ assert isinstance (rabbitmq_rpc_client , RabbitMQRPCClient )
229242 # Mimic listing all function job collections
230243 function_job_collections_list = list (self ._function_job_collections .values ())[
231244 pagination_offset : pagination_offset + pagination_limit
@@ -242,6 +255,7 @@ async def list_function_job_collections(
242255 async def delete_function_job_collection (
243256 self , rabbitmq_rpc_client : RabbitMQRPCClient , function_job_collection_id : str
244257 ) -> None :
258+ assert isinstance (rabbitmq_rpc_client , RabbitMQRPCClient )
245259 # Mimic deleting a function job collection
246260 if function_job_collection_id in self ._function_job_collections :
247261 del self ._function_job_collections [function_job_collection_id ]
@@ -253,6 +267,7 @@ async def delete_function_job_collection(
253267 async def update_function_title (
254268 self , rabbitmq_rpc_client : RabbitMQRPCClient , function_id : str , title : str
255269 ) -> RegisteredFunction :
270+ assert isinstance (rabbitmq_rpc_client , RabbitMQRPCClient )
256271 # Mimic updating the title of a function
257272 if function_id not in self ._functions :
258273 raise FunctionIDNotFoundError (function_id = function_id )
@@ -262,6 +277,7 @@ async def update_function_title(
262277 async def update_function_description (
263278 self , rabbitmq_rpc_client : RabbitMQRPCClient , function_id : str , description : str
264279 ) -> RegisteredFunction :
280+ assert isinstance (rabbitmq_rpc_client , RabbitMQRPCClient )
265281 # Mimic updating the description of a function
266282 if function_id not in self ._functions :
267283 raise FunctionIDNotFoundError (function_id = function_id )
0 commit comments