| #!/usr/bin/env python | |
| # Copyright (c) 2021, Alliance for Open Media. All rights reserved | |
| # | |
| # This source code is subject to the terms of the BSD 2 Clause License and | |
| # the Alliance for Open Media Patent License 1.0. If the BSD 2 Clause License | |
| # was not distributed with this source code in the LICENSE file, you can | |
| # obtain it at www.aomedia.org/license/software. If the Alliance for Open | |
| # Media Patent License 1.0 was not distributed with this source code in the | |
| # PATENTS file, you can obtain it at www.aomedia.org/license/patent. | |
| # | |
| import pprint | |
| import re | |
| import os, sys | |
| import io | |
| import unittest as googletest | |
| sys.path[0:0] = ['.', '..'] | |
| from pycparser import c_parser, parse_file | |
| from pycparser.c_ast import * | |
| from pycparser.c_parser import CParser, Coord, ParseError | |
| from auto_refactor import * | |
| def get_c_file_path(filename): | |
| return os.path.join('c_files', filename) | |
| class TestStructInfo(googletest.TestCase): | |
| def setUp(self): | |
| filename = get_c_file_path('struct_code.c') | |
| self.ast = parse_file(filename) | |
| def test_build_struct_info(self): | |
| struct_info = build_struct_info(self.ast) | |
| typedef_name_dic = struct_info.typedef_name_dic | |
| self.assertEqual('T1' in typedef_name_dic, True) | |
| self.assertEqual('T4' in typedef_name_dic, True) | |
| self.assertEqual('T5' in typedef_name_dic, True) | |
| struct_name_dic = struct_info.struct_name_dic | |
| struct_name = 'S1' | |
| self.assertEqual(struct_name in struct_name_dic, True) | |
| struct_item = struct_name_dic[struct_name] | |
| self.assertEqual(struct_item.is_union, False) | |
| struct_name = 'S3' | |
| self.assertEqual(struct_name in struct_name_dic, True) | |
| struct_item = struct_name_dic[struct_name] | |
| self.assertEqual(struct_item.is_union, False) | |
| struct_name = 'U5' | |
| self.assertEqual(struct_name in struct_name_dic, True) | |
| struct_item = struct_name_dic[struct_name] | |
| self.assertEqual(struct_item.is_union, True) | |
| self.assertEqual(len(struct_info.struct_item_list), 6) | |
| def test_get_child_decl_status(self): | |
| struct_info = build_struct_info(self.ast) | |
| struct_item = struct_info.typedef_name_dic['T4'] | |
| decl_status = struct_item.child_decl_map['x'] | |
| self.assertEqual(decl_status.struct_item, None) | |
| self.assertEqual(decl_status.is_ptr_decl, False) | |
| decl_status = struct_item.child_decl_map['s3'] | |
| self.assertEqual(decl_status.struct_item.struct_name, 'S3') | |
| self.assertEqual(decl_status.is_ptr_decl, False) | |
| struct_item = struct_info.typedef_name_dic['T6'] | |
| decl_status = struct_item.child_decl_map['x'] | |
| self.assertEqual(decl_status.struct_item, None) | |
| self.assertEqual(decl_status.is_ptr_decl, False) | |
| decl_status = struct_item.child_decl_map['y'] | |
| self.assertEqual(decl_status.struct_item, None) | |
| self.assertEqual(decl_status.is_ptr_decl, False) | |
| decl_status = struct_item.child_decl_map['z'] | |
| self.assertEqual(decl_status.struct_item, None) | |
| self.assertEqual(decl_status.is_ptr_decl, False) | |
| struct_item = struct_info.typedef_name_dic['T7'] | |
| decl_status = struct_item.child_decl_map['y'] | |
| self.assertEqual('x' in decl_status.struct_item.child_decl_map, True) | |
| struct_item = struct_info.typedef_name_dic['T7'] | |
| decl_status = struct_item.child_decl_map['z'] | |
| self.assertEqual('w' in decl_status.struct_item.child_decl_map, True) | |
| class TestParseLvalue(googletest.TestCase): | |
| def setUp(self): | |
| filename = get_c_file_path('parse_lvalue.c') | |
| self.ast = parse_file(filename) | |
| self.func_dictionary = build_func_dictionary(self.ast) | |
| def test_parse_lvalue(self): | |
| func_node = self.func_dictionary['func'] | |
| func_body_items = func_node.body.block_items | |
| id_list = parse_lvalue(func_body_items[0].lvalue) | |
| ref_id_list = ['cpi', 'rd', 'u'] | |
| self.assertEqual(id_list, ref_id_list) | |
| id_list = parse_lvalue(func_body_items[2].lvalue) | |
| ref_id_list = ['y'] | |
| self.assertEqual(id_list, ref_id_list) | |
| id_list = parse_lvalue(func_body_items[3].lvalue) | |
| ref_id_list = ['cpi', 'rd2', 'v'] | |
| self.assertEqual(id_list, ref_id_list) | |
| id_list = parse_lvalue(func_body_items[4].lvalue) | |
| ref_id_list = ['cpi', 'rd', 'arr'] | |
| self.assertEqual(id_list, ref_id_list) | |
| id_list = parse_lvalue(func_body_items[5].lvalue) | |
| ref_id_list = ['cpi', 'rd3', 'arr'] | |
| self.assertEqual(id_list, ref_id_list) | |
| def test_parse_lvalue_2(self): | |
| func_node = self.func_dictionary['parse_lvalue_2'] | |
| func_body_items = func_node.body.block_items | |
| id_list = parse_lvalue(func_body_items[0].init) | |
| ref_id_list = ['cpi', 'rd2'] | |
| self.assertEqual(id_list, ref_id_list) | |
| class TestIDStatusNode(googletest.TestCase): | |
| def test_add_descendant(self): | |
| root = IDStatusNode('root') | |
| id_chain1 = ['cpi', 'rd', 'u'] | |
| id_chain2 = ['cpi', 'rd', 'v'] | |
| root.add_descendant(id_chain1) | |
| root.add_descendant(id_chain2) | |
| ref_children_list1 = ['cpi'] | |
| children_list1 = list(root.children.keys()) | |
| self.assertEqual(children_list1, ref_children_list1) | |
| ref_children_list2 = ['rd'] | |
| children_list2 = list(root.children['cpi'].children.keys()) | |
| self.assertEqual(children_list2, ref_children_list2) | |
| ref_children_list3 = ['u', 'v'] | |
| children_list3 = list(root.children['cpi'].children['rd'].children.keys()) | |
| self.assertEqual(children_list3, ref_children_list3) | |
| def test_get_descendant(self): | |
| root = IDStatusNode('root') | |
| id_chain1 = ['cpi', 'rd', 'u'] | |
| id_chain2 = ['cpi', 'rd', 'v'] | |
| ref_descendant_1 = root.add_descendant(id_chain1) | |
| ref_descendant_2 = root.add_descendant(id_chain2) | |
| descendant_1 = root.get_descendant(id_chain1) | |
| self.assertEqual(descendant_1 is ref_descendant_1, True) | |
| descendant_2 = root.get_descendant(id_chain2) | |
| self.assertEqual(descendant_2 is ref_descendant_2, True) | |
| id_chain3 = ['cpi', 'rd', 'h'] | |
| descendant_3 = root.get_descendant(id_chain3) | |
| self.assertEqual(descendant_3, None) | |
| class TestFuncInOut(googletest.TestCase): | |
| def setUp(self): | |
| c_filename = get_c_file_path('func_in_out.c') | |
| self.ast = parse_file(c_filename) | |
| self.func_dictionary = build_func_dictionary(self.ast) | |
| self.struct_info = build_struct_info(self.ast) | |
| def test_get_func_param_id_map(self): | |
| func_def_node = self.func_dictionary['func'] | |
| param_id_map = get_func_param_id_map(func_def_node) | |
| ref_param_id_map_keys = ['cpi', 'x'] | |
| self.assertEqual(list(param_id_map.keys()), ref_param_id_map_keys) | |
| def test_assign_refer_status_1(self): | |
| func_def_node = self.func_dictionary['func_assign_refer_status_1'] | |
| visitor = FuncInOutVisitor(func_def_node, self.struct_info, | |
| self.func_dictionary) | |
| visitor.visit(func_def_node.body) | |
| root = visitor.get_id_tree_stack() | |
| body_id_tree = visitor.body_id_tree | |
| id_chain = ['rd'] | |
| descendant = body_id_tree.get_descendant(id_chain) | |
| self.assertEqual(descendant.get_assign(), False) | |
| self.assertEqual(descendant.get_refer(), False) | |
| ref_link_id_chain = ['cpi', 'rd'] | |
| self.assertEqual(ref_link_id_chain, descendant.get_link_id_chain()) | |
| id_chain = ['cpi', 'rd'] | |
| descendant = root.get_id_node(id_chain) | |
| self.assertEqual(descendant.get_assign(), False) | |
| self.assertEqual(descendant.get_refer(), False) | |
| self.assertEqual(None, descendant.get_link_id_chain()) | |
| def test_assign_refer_status_2(self): | |
| func_def_node = self.func_dictionary['func_assign_refer_status_2'] | |
| visitor = FuncInOutVisitor(func_def_node, self.struct_info, | |
| self.func_dictionary) | |
| visitor.visit(func_def_node.body) | |
| root = visitor.get_id_tree_stack() | |
| body_id_tree = visitor.body_id_tree | |
| id_chain = ['rd2'] | |
| descendant = body_id_tree.get_descendant(id_chain) | |
| self.assertEqual(descendant.get_assign(), False) | |
| self.assertEqual(descendant.get_refer(), False) | |
| ref_link_id_chain = ['cpi', 'rd'] | |
| self.assertEqual(ref_link_id_chain, descendant.get_link_id_chain()) | |
| id_chain = ['cpi', 'rd'] | |
| descendant = root.get_id_node(id_chain) | |
| self.assertEqual(descendant.get_assign(), False) | |
| self.assertEqual(descendant.get_refer(), False) | |
| self.assertEqual(None, descendant.get_link_id_chain()) | |
| def test_assign_refer_status_3(self): | |
| func_def_node = self.func_dictionary['func_assign_refer_status_3'] | |
| visitor = FuncInOutVisitor(func_def_node, self.struct_info, | |
| self.func_dictionary) | |
| visitor.visit(func_def_node.body) | |
| root = visitor.get_id_tree_stack() | |
| body_id_tree = visitor.body_id_tree | |
| id_chain = ['a'] | |
| descendant = body_id_tree.get_descendant(id_chain) | |
| self.assertEqual(descendant.get_assign(), True) | |
| self.assertEqual(descendant.get_refer(), False) | |
| self.assertEqual(None, descendant.get_link_id_chain()) | |
| id_chain = ['cpi', 'y'] | |
| descendant = root.get_id_node(id_chain) | |
| self.assertEqual(descendant.get_assign(), False) | |
| self.assertEqual(descendant.get_refer(), True) | |
| self.assertEqual(None, descendant.get_link_id_chain()) | |
| def test_assign_refer_status_4(self): | |
| func_def_node = self.func_dictionary['func_assign_refer_status_4'] | |
| visitor = FuncInOutVisitor(func_def_node, self.struct_info, | |
| self.func_dictionary) | |
| visitor.visit(func_def_node.body) | |
| root = visitor.get_id_tree_stack() | |
| body_id_tree = visitor.body_id_tree | |
| id_chain = ['b'] | |
| descendant = body_id_tree.get_descendant(id_chain) | |
| self.assertEqual(descendant.get_assign(), True) | |
| self.assertEqual(descendant.get_refer(), False) | |
| self.assertEqual(None, descendant.get_link_id_chain()) | |
| id_chain = ['cpi', 'y'] | |
| descendant = root.get_id_node(id_chain) | |
| self.assertEqual(descendant.get_assign(), True) | |
| self.assertEqual(descendant.get_refer(), True) | |
| self.assertEqual(None, descendant.get_link_id_chain()) | |
| def test_assign_refer_status_5(self): | |
| func_def_node = self.func_dictionary['func_assign_refer_status_5'] | |
| visitor = FuncInOutVisitor(func_def_node, self.struct_info, | |
| self.func_dictionary) | |
| visitor.visit(func_def_node.body) | |
| root = visitor.get_id_tree_stack() | |
| body_id_tree = visitor.body_id_tree | |
| id_chain = ['rd5'] | |
| descendant = body_id_tree.get_descendant(id_chain) | |
| self.assertEqual(descendant.get_assign(), False) | |
| self.assertEqual(descendant.get_refer(), False) | |
| id_chain = ['cpi', 'rd2'] | |
| descendant = root.get_id_node(id_chain) | |
| self.assertEqual(descendant.get_assign(), False) | |
| self.assertEqual(descendant.get_refer(), False) | |
| self.assertEqual(None, descendant.get_link_id_chain()) | |
| def test_assign_refer_status_6(self): | |
| func_def_node = self.func_dictionary['func_assign_refer_status_6'] | |
| visitor = FuncInOutVisitor(func_def_node, self.struct_info, | |
| self.func_dictionary) | |
| visitor.visit(func_def_node.body) | |
| root = visitor.get_id_tree_stack() | |
| id_chain = ['cpi', 'rd'] | |
| descendant = root.get_id_node(id_chain) | |
| self.assertEqual(descendant.get_assign(), True) | |
| self.assertEqual(descendant.get_refer(), False) | |
| self.assertEqual(None, descendant.get_link_id_chain()) | |
| id_chain = ['cpi2', 'rd'] | |
| descendant = root.get_id_node(id_chain) | |
| self.assertEqual(descendant.get_assign(), False) | |
| self.assertEqual(descendant.get_refer(), True) | |
| self.assertEqual(None, descendant.get_link_id_chain()) | |
| def test_assign_refer_status_7(self): | |
| func_def_node = self.func_dictionary['func_assign_refer_status_7'] | |
| visitor = FuncInOutVisitor(func_def_node, self.struct_info, | |
| self.func_dictionary) | |
| visitor.visit(func_def_node.body) | |
| root = visitor.get_id_tree_stack() | |
| id_chain = ['cpi', 'arr'] | |
| descendant = root.get_id_node(id_chain) | |
| self.assertEqual(descendant.get_assign(), True) | |
| self.assertEqual(descendant.get_refer(), False) | |
| def test_assign_refer_status_8(self): | |
| func_def_node = self.func_dictionary['func_assign_refer_status_8'] | |
| visitor = FuncInOutVisitor(func_def_node, self.struct_info, | |
| self.func_dictionary) | |
| visitor.visit(func_def_node.body) | |
| root = visitor.get_id_tree_stack() | |
| id_chain = ['cpi', 'arr'] | |
| descendant = root.get_id_node(id_chain) | |
| self.assertEqual(descendant.get_assign(), False) | |
| self.assertEqual(descendant.get_refer(), True) | |
| def test_assign_refer_status_9(self): | |
| func_def_node = self.func_dictionary['func_assign_refer_status_9'] | |
| visitor = FuncInOutVisitor(func_def_node, self.struct_info, | |
| self.func_dictionary) | |
| visitor.visit(func_def_node.body) | |
| root = visitor.get_id_tree_stack() | |
| id_chain = ['cpi', 'rd', 'u'] | |
| descendant = root.get_id_node(id_chain) | |
| self.assertEqual(descendant.get_assign(), True) | |
| self.assertEqual(descendant.get_refer(), False) | |
| def test_assign_refer_status_10(self): | |
| func_def_node = self.func_dictionary['func_assign_refer_status_10'] | |
| visitor = FuncInOutVisitor(func_def_node, self.struct_info, | |
| self.func_dictionary) | |
| visitor.visit(func_def_node.body) | |
| root = visitor.get_id_tree_stack() | |
| id_chain = ['cpi', 'rd', 'u'] | |
| descendant = root.get_id_node(id_chain) | |
| self.assertEqual(descendant.get_assign(), False) | |
| self.assertEqual(descendant.get_refer(), True) | |
| id_chain = ['cpi', 'arr'] | |
| descendant = root.get_id_node(id_chain) | |
| self.assertEqual(descendant.get_assign(), True) | |
| self.assertEqual(descendant.get_refer(), False) | |
| def test_assign_refer_status_11(self): | |
| func_def_node = self.func_dictionary['func_assign_refer_status_11'] | |
| visitor = FuncInOutVisitor(func_def_node, self.struct_info, | |
| self.func_dictionary) | |
| visitor.visit(func_def_node.body) | |
| root = visitor.get_id_tree_stack() | |
| id_chain = ['cpi', 'rd2', 'v'] | |
| descendant = root.get_id_node(id_chain) | |
| self.assertEqual(descendant.get_assign(), True) | |
| self.assertEqual(descendant.get_refer(), False) | |
| def test_assign_refer_status_12(self): | |
| func_def_node = self.func_dictionary['func_assign_refer_status_12'] | |
| visitor = FuncInOutVisitor(func_def_node, self.struct_info, | |
| self.func_dictionary) | |
| visitor.visit(func_def_node.body) | |
| root = visitor.get_id_tree_stack() | |
| id_chain = ['cpi', 'rd'] | |
| descendant = root.get_id_node(id_chain) | |
| self.assertEqual(descendant.get_assign(), True) | |
| self.assertEqual(descendant.get_refer(), False) | |
| id_chain = ['cpi2', 'rd'] | |
| descendant = root.get_id_node(id_chain) | |
| self.assertEqual(descendant.get_assign(), False) | |
| self.assertEqual(descendant.get_refer(), True) | |
| def test_assign_refer_status_13(self): | |
| func_def_node = self.func_dictionary['func_assign_refer_status_13'] | |
| visitor = FuncInOutVisitor(func_def_node, self.struct_info, | |
| self.func_dictionary) | |
| visitor.visit(func_def_node.body) | |
| root = visitor.get_id_tree_stack() | |
| id_chain = ['cpi', 'z'] | |
| descendant = root.get_id_node(id_chain) | |
| self.assertEqual(descendant.get_assign(), True) | |
| self.assertEqual(descendant.get_refer(), False) | |
| id_chain = ['cpi', 'w'] | |
| descendant = root.get_id_node(id_chain) | |
| self.assertEqual(descendant.get_assign(), True) | |
| self.assertEqual(descendant.get_refer(), False) | |
| def test_id_status_forrest_1(self): | |
| func_def_node = self.func_dictionary['func'] | |
| visitor = FuncInOutVisitor(func_def_node, self.struct_info, | |
| self.func_dictionary) | |
| visitor.visit(func_def_node.body) | |
| root = visitor.get_id_tree_stack().top() | |
| children_names = set(root.get_children().keys()) | |
| ref_children_names = set(['cpi', 'x']) | |
| self.assertEqual(children_names, ref_children_names) | |
| root = visitor.body_id_tree | |
| children_names = set(root.get_children().keys()) | |
| ref_children_names = set(['a', 'ref_rd', 'ref_rd2', 'ref_rd3', 'b']) | |
| self.assertEqual(children_names, ref_children_names) | |
| def test_id_status_forrest_show(self): | |
| func_def_node = self.func_dictionary['func_id_forrest_show'] | |
| visitor = FuncInOutVisitor(func_def_node, self.struct_info, | |
| self.func_dictionary) | |
| visitor.visit(func_def_node.body) | |
| visitor.get_id_tree_stack().top().show() | |
| def test_id_status_forrest_2(self): | |
| func_def_node = self.func_dictionary['func_id_forrest_show'] | |
| visitor = FuncInOutVisitor(func_def_node, self.struct_info, | |
| self.func_dictionary) | |
| visitor.visit(func_def_node.body) | |
| root = visitor.get_id_tree_stack().top() | |
| self.assertEqual(root, root.root) | |
| id_chain = ['cpi', 'rd'] | |
| descendant = root.get_descendant(id_chain) | |
| self.assertEqual(root, descendant.root) | |
| id_chain = ['b'] | |
| descendant = root.get_descendant(id_chain) | |
| self.assertEqual(root, descendant.root) | |
| def test_link_id_chain_1(self): | |
| func_def_node = self.func_dictionary['func_link_id_chain_1'] | |
| visitor = FuncInOutVisitor(func_def_node, self.struct_info, | |
| self.func_dictionary) | |
| visitor.visit(func_def_node.body) | |
| root = visitor.get_id_tree_stack() | |
| id_chain = ['cpi', 'rd', 'u'] | |
| descendant = root.get_id_node(id_chain) | |
| self.assertEqual(descendant.get_assign(), True) | |
| def test_link_id_chain_2(self): | |
| func_def_node = self.func_dictionary['func_link_id_chain_2'] | |
| visitor = FuncInOutVisitor(func_def_node, self.struct_info, | |
| self.func_dictionary) | |
| visitor.visit(func_def_node.body) | |
| root = visitor.get_id_tree_stack() | |
| id_chain = ['cpi', 'rd', 'xd', 'u'] | |
| descendant = root.get_id_node(id_chain) | |
| self.assertEqual(descendant.get_assign(), True) | |
| def test_func_call_1(self): | |
| func_def_node = self.func_dictionary['func_call_1'] | |
| visitor = FuncInOutVisitor(func_def_node, self.struct_info, | |
| self.func_dictionary) | |
| visitor.visit(func_def_node.body) | |
| root = visitor.get_id_tree_stack() | |
| id_chain = ['cpi', 'y'] | |
| descendant = root.get_id_node(id_chain) | |
| self.assertEqual(descendant.get_assign(), True) | |
| self.assertEqual(descendant.get_refer(), False) | |
| id_chain = ['y'] | |
| descendant = root.get_id_node(id_chain) | |
| self.assertEqual(descendant.get_assign(), False) | |
| self.assertEqual(descendant.get_refer(), True) | |
| def test_func_call_2(self): | |
| func_def_node = self.func_dictionary['func_call_2'] | |
| visitor = FuncInOutVisitor(func_def_node, self.struct_info, | |
| self.func_dictionary) | |
| visitor.visit(func_def_node.body) | |
| root = visitor.get_id_tree_stack() | |
| id_chain = ['cpi', 'rd', 'u'] | |
| descendant = root.get_id_node(id_chain) | |
| self.assertEqual(descendant.get_assign(), True) | |
| self.assertEqual(descendant.get_refer(), False) | |
| id_chain = ['cpi', 'rd'] | |
| descendant = root.get_id_node(id_chain) | |
| self.assertEqual(descendant.get_assign(), False) | |
| self.assertEqual(descendant.get_refer(), False) | |
| def test_func_call_3(self): | |
| func_def_node = self.func_dictionary['func_call_3'] | |
| visitor = FuncInOutVisitor(func_def_node, self.struct_info, | |
| self.func_dictionary) | |
| visitor.visit(func_def_node.body) | |
| root = visitor.get_id_tree_stack() | |
| id_chain = ['cpi', 'y'] | |
| descendant = root.get_id_node(id_chain) | |
| self.assertEqual(descendant.get_assign(), True) | |
| self.assertEqual(descendant.get_refer(), True) | |
| def test_func_call_4(self): | |
| func_def_node = self.func_dictionary['func_call_4'] | |
| visitor = FuncInOutVisitor(func_def_node, self.struct_info, | |
| self.func_dictionary) | |
| visitor.visit(func_def_node.body) | |
| root = visitor.get_id_tree_stack() | |
| id_chain = ['cpi', 'rd', 'u'] | |
| descendant = root.get_id_node(id_chain) | |
| self.assertEqual(descendant.get_assign(), True) | |
| self.assertEqual(descendant.get_refer(), False) | |
| id_chain = ['cpi', 'rd', 'xd', 'u'] | |
| descendant = root.get_id_node(id_chain) | |
| self.assertEqual(descendant.get_assign(), True) | |
| self.assertEqual(descendant.get_refer(), False) | |
| def test_func_call_5(self): | |
| func_def_node = self.func_dictionary['func_call_5'] | |
| visitor = FuncInOutVisitor(func_def_node, self.struct_info, | |
| self.func_dictionary) | |
| visitor.visit(func_def_node.body) | |
| root = visitor.get_id_tree_stack() | |
| id_chain = ['cpi', 'y'] | |
| descendant = root.get_id_node(id_chain) | |
| self.assertEqual(descendant.get_assign(), True) | |
| self.assertEqual(descendant.get_refer(), False) | |
| def test_func_compound_1(self): | |
| func_def_node = self.func_dictionary['func_compound_1'] | |
| visitor = FuncInOutVisitor(func_def_node, self.struct_info, | |
| self.func_dictionary) | |
| visitor.visit(func_def_node.body) | |
| root = visitor.get_id_tree_stack() | |
| id_chain = ['cpi', 'y'] | |
| descendant = root.get_id_node(id_chain) | |
| self.assertEqual(descendant.get_assign(), True) | |
| self.assertEqual(descendant.get_refer(), True) | |
| def test_func_compound_2(self): | |
| func_def_node = self.func_dictionary['func_compound_2'] | |
| visitor = FuncInOutVisitor(func_def_node, self.struct_info, | |
| self.func_dictionary) | |
| visitor.visit(func_def_node.body) | |
| root = visitor.get_id_tree_stack() | |
| id_chain = ['cpi', 'y'] | |
| descendant = root.get_id_node(id_chain) | |
| self.assertEqual(descendant.get_assign(), False) | |
| self.assertEqual(descendant.get_refer(), True) | |
| id_chain = ['cpi', 'rd', 'u'] | |
| descendant = root.get_id_node(id_chain) | |
| self.assertEqual(descendant.get_assign(), True) | |
| self.assertEqual(descendant.get_refer(), False) | |
| def test_func_compound_3(self): | |
| func_def_node = self.func_dictionary['func_compound_3'] | |
| visitor = FuncInOutVisitor(func_def_node, self.struct_info, | |
| self.func_dictionary) | |
| visitor.visit(func_def_node.body) | |
| root = visitor.get_id_tree_stack() | |
| id_chain = ['cpi', 'rd', 'u'] | |
| descendant = root.get_id_node(id_chain) | |
| self.assertEqual(descendant.get_assign(), True) | |
| self.assertEqual(descendant.get_refer(), False) | |
| def test_func_compound_4(self): | |
| func_def_node = self.func_dictionary['func_compound_4'] | |
| visitor = FuncInOutVisitor(func_def_node, self.struct_info, | |
| self.func_dictionary) | |
| visitor.visit(func_def_node.body) | |
| root = visitor.get_id_tree_stack() | |
| id_chain = ['cpi', 'y'] | |
| descendant = root.get_id_node(id_chain) | |
| self.assertEqual(descendant.get_assign(), True) | |
| self.assertEqual(descendant.get_refer(), True) | |
| def test_func_compound_5(self): | |
| func_def_node = self.func_dictionary['func_compound_5'] | |
| visitor = FuncInOutVisitor(func_def_node, self.struct_info, | |
| self.func_dictionary) | |
| visitor.visit(func_def_node.body) | |
| root = visitor.get_id_tree_stack() | |
| id_chain = ['cpi', 'y'] | |
| descendant = root.get_id_node(id_chain) | |
| self.assertEqual(descendant.get_assign(), True) | |
| self.assertEqual(descendant.get_refer(), True) | |
| def test_func_compound_6(self): | |
| func_def_node = self.func_dictionary['func_compound_6'] | |
| visitor = FuncInOutVisitor(func_def_node, self.struct_info, | |
| self.func_dictionary) | |
| visitor.visit(func_def_node.body) | |
| root = visitor.get_id_tree_stack() | |
| id_chain = ['cpi', 'y'] | |
| descendant = root.get_id_node(id_chain) | |
| self.assertEqual(descendant.get_assign(), True) | |
| self.assertEqual(descendant.get_refer(), True) | |
| class TestDeclStatus(googletest.TestCase): | |
| def setUp(self): | |
| filename = get_c_file_path('decl_status_code.c') | |
| self.ast = parse_file(filename) | |
| self.func_dictionary = build_func_dictionary(self.ast) | |
| self.struct_info = build_struct_info(self.ast) | |
| def test_parse_decl_node(self): | |
| func_def_node = self.func_dictionary['main'] | |
| decl_list = func_def_node.body.block_items | |
| decl_status = parse_decl_node(self.struct_info, decl_list[0]) | |
| self.assertEqual(decl_status.name, 'a') | |
| self.assertEqual(decl_status.is_ptr_decl, False) | |
| decl_status = parse_decl_node(self.struct_info, decl_list[1]) | |
| self.assertEqual(decl_status.name, 't1') | |
| self.assertEqual(decl_status.is_ptr_decl, False) | |
| decl_status = parse_decl_node(self.struct_info, decl_list[2]) | |
| self.assertEqual(decl_status.name, 's1') | |
| self.assertEqual(decl_status.is_ptr_decl, False) | |
| decl_status = parse_decl_node(self.struct_info, decl_list[3]) | |
| self.assertEqual(decl_status.name, 't2') | |
| self.assertEqual(decl_status.is_ptr_decl, True) | |
| def test_parse_decl_node_2(self): | |
| func_def_node = self.func_dictionary['parse_decl_node_2'] | |
| decl_list = func_def_node.body.block_items | |
| decl_status = parse_decl_node(self.struct_info, decl_list[0]) | |
| self.assertEqual(decl_status.name, 'arr') | |
| self.assertEqual(decl_status.is_ptr_decl, True) | |
| self.assertEqual(decl_status.struct_item, None) | |
| def test_parse_decl_node_3(self): | |
| func_def_node = self.func_dictionary['parse_decl_node_3'] | |
| decl_list = func_def_node.body.block_items | |
| decl_status = parse_decl_node(self.struct_info, decl_list[0]) | |
| self.assertEqual(decl_status.name, 'a') | |
| self.assertEqual(decl_status.is_ptr_decl, True) | |
| self.assertEqual(decl_status.struct_item, None) | |
| def test_parse_decl_node_4(self): | |
| func_def_node = self.func_dictionary['parse_decl_node_4'] | |
| decl_list = func_def_node.body.block_items | |
| decl_status = parse_decl_node(self.struct_info, decl_list[0]) | |
| self.assertEqual(decl_status.name, 't1') | |
| self.assertEqual(decl_status.is_ptr_decl, True) | |
| self.assertEqual(decl_status.struct_item.typedef_name, 'T1') | |
| self.assertEqual(decl_status.struct_item.struct_name, 'S1') | |
| def test_parse_decl_node_5(self): | |
| func_def_node = self.func_dictionary['parse_decl_node_5'] | |
| decl_list = func_def_node.body.block_items | |
| decl_status = parse_decl_node(self.struct_info, decl_list[0]) | |
| self.assertEqual(decl_status.name, 't2') | |
| self.assertEqual(decl_status.is_ptr_decl, True) | |
| self.assertEqual(decl_status.struct_item.typedef_name, 'T1') | |
| self.assertEqual(decl_status.struct_item.struct_name, 'S1') | |
| def test_parse_decl_node_6(self): | |
| func_def_node = self.func_dictionary['parse_decl_node_6'] | |
| decl_list = func_def_node.body.block_items | |
| decl_status = parse_decl_node(self.struct_info, decl_list[0]) | |
| self.assertEqual(decl_status.name, 't3') | |
| self.assertEqual(decl_status.is_ptr_decl, True) | |
| self.assertEqual(decl_status.struct_item.typedef_name, 'T1') | |
| self.assertEqual(decl_status.struct_item.struct_name, 'S1') | |
| if __name__ == '__main__': | |
| googletest.main() |