diff options
Diffstat (limited to '')
| -rw-r--r-- | third_party/aom/tools/auto_refactor/test_auto_refactor.py | 675 |
1 files changed, 675 insertions, 0 deletions
diff --git a/third_party/aom/tools/auto_refactor/test_auto_refactor.py b/third_party/aom/tools/auto_refactor/test_auto_refactor.py new file mode 100644 index 0000000000..6b1e269efa --- /dev/null +++ b/third_party/aom/tools/auto_refactor/test_auto_refactor.py @@ -0,0 +1,675 @@ +#!/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() |