Coverage for mindsdb / integrations / handlers / faunadb_handler / tests / test_faunadb_handler.py: 0%

1import unittest 

2from mindsdb_sql_parser import parse_sql 

3 

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

5 RESPONSE_TYPE, 

6) 

7from mindsdb.integrations.handlers.faunadb_handler.faunadb_handler import ( 

8 FaunaDBHandler, 

9) 

10 

11 

12class FaunadbHandlerTest(unittest.TestCase): 

13 @classmethod 

14 def setUpClass(cls): 

15 cls.kwargs = { 

16 "connection_data": { 

17 "fauna_secret": "fnAFQFQPZNAAUYkCYkdvozJsm9tH2VbX55AULhsH", 

18 "fauna_endpoint": "https://db.fauna.com:443/", 

19 } 

20 } 

21 cls.handler = FaunaDBHandler("test_faunadb_handler", **cls.kwargs) 

22 

23 def test_0_connect(self): 

24 self.handler.connect() 

25 

26 def test_1_check_connection(self): 

27 self.handler.check_connection() 

28 self.assertTrue(self.handler.is_connected) 

29 

30 def test_2_select(self): 

31 query = parse_sql("SELECT * FROM books;") 

32 res = self.handler.query(query) 

33 assert res.type is RESPONSE_TYPE.TABLE 

34 

35 def test_3_describe_db(self): 

36 res = self.handler.get_tables() 

37 assert res.type is RESPONSE_TYPE.TABLE 

38 

39 

40if __name__ == "__main__": 

41 unittest.main()