Coverage for mindsdb / integrations / handlers / aurora_handler / connection_args.py: 0%

1from collections import OrderedDict 

2 

3from mindsdb.integrations.libs.const import HANDLER_CONNECTION_ARG_TYPE as ARG_TYPE 

4 

5 

6connection_args = OrderedDict( 

7 user={ 

8 'type': ARG_TYPE.STR, 

9 'description': 'The user name used to authenticate with the Amazon Aurora DB cluster.' 

10 }, 

11 password={ 

12 'type': ARG_TYPE.PWD, 

13 'description': 'The password to authenticate the user with the Amazon Aurora DB cluster.', 

14 'secret': True 

15 }, 

16 database={ 

17 'type': ARG_TYPE.STR, 

18 'description': 'The database name to use when connecting with the Amazon Aurora DB cluster.' 

19 }, 

20 host={ 

21 'type': ARG_TYPE.STR, 

22 'description': 'The host name or IP address of the Amazon Aurora DB cluster.' 

23 }, 

24 port={ 

25 'type': ARG_TYPE.INT, 

26 'description': 'The TCP/IP port of the Amazon Aurora DB cluster. Must be an integer.' 

27 }, 

28 db_engine={ 

29 'type': ARG_TYPE.STR, 

30 'description': "The database engine of the Amazon Aurora DB cluster. This can take one of two values: 'mysql' or 'postgresql'. This parameter is optional, but if it is not provided, 'aws_access_key_id' and 'aws_secret_access_key' parameters must be provided" 

31 }, 

32 aws_access_key_id={ 

33 'type': ARG_TYPE.STR, 

34 'description': "The access key for the AWS account. This parameter is optional and is only required to be provided if the 'db_engine' parameter is not provided." 

35 }, 

36 aws_secret_access_key={ 

37 'type': ARG_TYPE.STR, 

38 'description': "The secret key for the AWS account. This parameter is optional and is only required to be provided if the 'db_engine' parameter is not provided.", 

39 'secret': True 

40 }, 

41) 

42 

43connection_args_example = OrderedDict( 

44 db_engine='mysql', 

45 host='mysqlcluster.cluster-123456789012.us-east-1.rds.amazonaws.com', 

46 port=3306, 

47 user='root', 

48 password='password', 

49 database='database' 

50)