Class Test_CppParser

tests/test_parser/test_cpp.py:10–100

class Test_CppParser(unittest.TestCase):
    def setUp(self) -> None:
        with open('tests/test_parser/test_sample/cpp_test_sample.cpp', 'r') as file:
            self.code_sample = file.read()
            
        tree = parse_code(self.code_sample, 'c++')
        self.root_node = tree.root_node

        return super().setUp()

    def test_get_function_list(self):
        root = self.root_node
        
        function_list = CppParser.get_function_list(root)
        
        self.assertEqual(len(function_list), 3)
        
    def test_get_class_list(self):
        root = self.root_node
        
        class_list = CppParser.get_class_list(root)
        
        self.assertEqual(len(class_list), 2)

    def test_get_function_metadata(self):
        root = self.root_node
        
        function = list(CppParser.get_function_list(root))[0]
        metadata = CppParser.get_function_metadata(function)

        for key in ['identifier', 'parameters', 'return_type']:
            self.assertTrue(key in metadata.keys(), "Missing {}".format(key))
        self.assertEqual(metadata['parameters'], {'a': 'int', 'b': 'int'})
        self.assertEqual(metadata['identifier'], 'sum2number')
        self.assertEqual(metadata['return_type'], 'int')
    
    def test_get_class_metadata(self):
        root = self.root_node
        
        classes = list(CppParser.get_class_list(root))[0]
        metadata = CppParser.get_class_metadata(classes)

        self.assertEqual(metadata['parameters'], {'Vehicle': None, 'B': None})
        self.assertEqual(metadata['identifier'], 'Car')

    def test_get_docstring(self):
        code_sample = """
        /**
        * Find 2 sum
        *
        * @param nums List number.
        * @param target Sum target.
        * @return postion of 2 number.
        */
        vector<int> twoSum(vector<int>& nums, int target) {
            map<int,int> m;
            vector<int> v;
            int n= nums.size();