Coverage for mindsdb / integrations / utilities / test_utils.py: 0%

1from mindsdb.integrations.handlers.postgres_handler import Handler as PGHandler 

2 

3 

4PG_HANDLER_NAME = 'test_handler' 

5PG_CONNECTION_DATA = { 

6 "user": "demo_user", 

7 "password": "demo_password", 

8 "host": "samples.mindsdb.com", 

9 "port": "5432", 

10 "database": "demo" 

11} 

12 

13 

14class HandlerControllerMock: 

15 def __init__(self): 

16 self.handlers = { 

17 PG_HANDLER_NAME: PGHandler( 

18 PG_HANDLER_NAME, 

19 **{"connection_data": PG_CONNECTION_DATA} 

20 ) 

21 } 

22 

23 def get_handler(self, name): 

24 return self.handlers[name] 

25 

26 def get(self, name): 

27 return { 

28 'id': 0, 

29 'name': PG_HANDLER_NAME 

30 }