Coverage for mindsdb / integrations / handlers / solr_handler / tests / test_solr_handler.py: 0%

1import unittest 

2 

3from mindsdb.integrations.handlers.solr_handler.solr_handler import SolrHandler 

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

5 

6 

7class SolrHandlerTest(unittest.TestCase): 

8 @classmethod 

9 def setUpClass(cls): 

10 cls.kwargs = { 

11 "username": "demo_user", 

12 "password": "demo_password", 

13 "host": "172.22.0.4", 

14 "port": 8983, 

15 "server_path": "solr", 

16 "collection": "gettingstarted", 

17 "use_ssl": False 

18 } 

19 cls.handler = SolrHandler('test_solr_handler', **cls.kwargs) 

20 

21 def test_0_connect(self): 

22 self.handler.check_connection() 

23 

24 def test_2_get_tables(self): 

25 tbls = self.handler.get_tables() 

26 assert tbls['type'] is not RESPONSE_TYPE.ERROR 

27 

28 def test_6_describe_table(self): 

29 described = self.handler.get_columns("gettingstarted") 

30 assert described['type'] is RESPONSE_TYPE.TABLE 

31 

32 def test_7_select_query(self): 

33 query = "SELECT * FROM gettingstarted WHERE id='apple' limit 1000" 

34 result = self.handler.query(query) 

35 assert result['type'] is RESPONSE_TYPE.TABLE 

36 

37 

38if __name__ == '__main__': 

39 unittest.main()