mindsdb

1
import logging
2
from locust import between, HttpUser
3
from load.test_postgresql import PostgreSQLConnectionBehavior
4
from utils.config import get_value_from_json_env_var
5

6

7
class DBConnectionUser(HttpUser):
8
    tasks = [PostgreSQLConnectionBehavior]
9
    wait_time = between(5, 15)
10
    config = get_value_from_json_env_var("INTEGRATIONS_CONFIG", "mindsdb_cloud")
11
    host = config['host']
12

13
    def __init__(self, *args, **kwargs):
14
        super().__init__(*args, **kwargs)
15
        try:
16
            response = self.client.post('/cloud/login', json={
17
                'email': self.config['user'],
18
                'password': self.config['password']
19
            })
20
            response.raise_for_status()
21
        except Exception as e:
22
            logging.error('Logging to MindsDB failed: ', e)
23