Class Test_JavaParser

tests/test_parser/test_java.py:10–82

class Test_JavaParser(unittest.TestCase):
    def setUp(self) -> None:        
        with open('tests/test_parser/test_sample/java_test_sample.java', 'r') as file:
            self.code_sample = file.read()
            
        tree = parse_code(self.code_sample, 'java')
        self.root_node = tree.root_node

        return super().setUp()

    def test_get_function_list(self):
        root = self.root_node
        
        function_list = JavaParser.get_function_list(root)
        
        self.assertEqual(len(function_list), 2)

    def test_get_class_list(self):
        root = self.root_node
        
        class_list = JavaParser.get_class_list(root)
        
        self.assertEqual(len(class_list), 1)

    def test_get_docstring(self):
        code_sample = """
        public class SaveFileController {
            /**
            * Adds new user and saves to file.
            *
            * @param context instance of Context
            * @param user instance of User
            * @see User
            */
            public void addNewUser(Context context, User user){
                    loadFromFile(context);
                this.allUsers.add(user);
                saveToFile(context);
            }
        }
        """
        tree = parse_code(code_sample, 'java', './')
        root = tree.root_node
        
        fn = list(JavaParser.get_function_list(root))[0]

        docs = JavaParser.get_docstring(fn)
        self.assertEqual(docs, '/**\n            * Adds new user and saves to file.\n            *\n            * @param context instance of Context\n            * @param user instance of User\n            * @see User\n            */')
        

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

        for key in ['identifier', 'parameters', 'return_type']:
            self.assertTrue(key in metadata.keys())