Coverage for mindsdb / integrations / handlers / rockset_handler / tests / test_rockset_handler.py: 0%

1import unittest 

2from mindsdb.api.executor.data_types.response_type import RESPONSE_TYPE 

3from mindsdb.integrations.handlers.rockset_handler.rockset_handler import RocksetHandler 

4 

5 

6class RocksetHandlerTest(unittest.TestCase): 

7 @classmethod 

8 def setUpClass(cls): 

9 cls.kwargs = { 

10 "connection_args": { 

11 "host": '127.0.0.1', 

12 "port": 3306, 

13 "user": "rockset", 

14 "password": "rockset" 

15 } 

16 } 

17 cls.handler = RocksetHandler('test_rockset_handler', **cls.kwargs) 

18 

19 def test_0_connect(self): 

20 self.handler.check_connection() 

21 

22 def test_1_get_tables(self): 

23 tables = self.handler.get_tables() 

24 self.assertEqual(tables, []) 

25 

26 def test_2_get_columns(self): 

27 columns = self.handler.get_columns('test') 

28 self.assertEqual(columns, []) 

29 

30 def test_3_query(self): 

31 response = self.handler.query('SELECT 1') 

32 self.assertEqual(response['type'], RESPONSE_TYPE.QUERY) 

33 self.assertEqual(response['data'], [[1]])