Coverage for mindsdb / integrations / handlers / tdengine_handler / tests / test_tdengine_handler.py: 0%

1import unittest 

2from mindsdb.integrations.handlers.tdengine_handler.tdengine_handler import TDEngineHandler 

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

4 

5 

6class TDEngineHandlerTest(unittest.TestCase): 

7 @classmethod 

8 def setUpClass(cls): 

9 cls.kwargs = { 

10 "token": '<token_for_cloud>', 

11 "url": "********.cloud.tdengine.com", 

12 "database": "temp" 

13 } 

14 cls.handler = TDEngineHandler('test_tdengine_handler', connection_data=cls.kwargs) 

15 

16 def test_0_check_connection(self): 

17 assert self.handler.check_connection() 

18 

19 def test_1_connect(self): 

20 assert self.handler.connect() 

21 

22 def test_2_create_table(self): 

23 query = "CREATE Table `hari` USING `temp` (`id`) TAGS (0);" 

24 result = self.handler.query(query) 

25 assert result.type is not RESPONSE_TYPE.ERROR 

26 pass 

27 

28 def test_3_insert(self): 

29 query = "INSERT INTO hari VALUES (NOW, 12);" 

30 result = self.handler.query(query) 

31 assert result.type is not RESPONSE_TYPE.ERROR 

32 

33 def test_4_native_query_select(self): 

34 query = "SELECT * FROM hari;" 

35 result = self.handler.query(query) 

36 assert result.type is RESPONSE_TYPE.TABLE 

37 

38 def test_5_get_tables(self): 

39 tables = self.handler.get_tables() 

40 assert tables.type is RESPONSE_TYPE.TABLE 

41 

42 def test_6_get_columns(self): 

43 columns = self.handler.get_columns('hari') 

44 assert columns.type is not RESPONSE_TYPE.ERROR 

45 

46 

47if __name__ == '__main__': 

48 unittest.main()