def test_given_no_callable_included__export_returns_empty_diagram():
manager = CallTreeManager()
with generate_file('file.cpp',
'void f();\nvoid g() { f(); }') as file_name:
manager.open(file_name)
manager.select_tu(file_name)
manager.select_root('g()')
expected = '@startuml\n\n\n@enduml'
actual = manager.export()
assert actual == expected
def test_given_two_functions_included__export_returns_diagram_with_call():
manager = CallTreeManager()
with generate_file('file.cpp',
'void f();\nvoid g() { f(); }') as file_name:
manager.open(file_name)
manager.select_tu(file_name)
manager.select_root('g()')
manager.include('g()')
manager.include('f()')
expected = '@startuml\n\n -> "file.cpp": g()\nactivate "file.cpp"\n"file.cpp" -> "file.cpp": f()\nactivate "file.cpp"\ndeactivate "file.cpp"\ndeactivate "file.cpp"\n\n@enduml'
actual = manager.export()
assert actual == expected
def test_given_methods_of_same_class__use_class_as_participant():
manager = CallTreeManager()
with generate_file('file.cpp',
'class Foo {\nvoid bar();\nvoid baz() { bar(); }\n};'
) as file_name:
manager.open(file_name)
manager.select_tu(file_name)
manager.select_root('Foo::baz()')
manager.include('Foo::baz()')
manager.include('Foo::bar()')
expected = '@startuml\n\n -> "Foo": baz()\nactivate "Foo"\n"Foo" -> "Foo": bar()\nactivate "Foo"\ndeactivate "Foo"\ndeactivate "Foo"\n\n@enduml'
actual = manager.export()
assert actual == expected
def test_given_call_graph_of_depth_two__functions_exported_depth_first():
manager = CallTreeManager()
with generate_file(
'file.cpp',
'void i();\nvoid h();\nvoid g() { h(); }\nvoid f() { g();\ni(); }'
) as file_name:
manager.open(file_name)
manager.select_tu(file_name)
manager.select_root('f()')
manager.include('f()')
manager.include('g()')
manager.include('h()')
manager.include('i()')
expected = '@startuml\n\n -> "file.cpp": f()\nactivate "file.cpp"\n"file.cpp" -> "file.cpp": g()\nactivate "file.cpp"\n"file.cpp" -> "file.cpp": h()\nactivate "file.cpp"\ndeactivate "file.cpp"\ndeactivate "file.cpp"\n"file.cpp" -> "file.cpp": i()\nactivate "file.cpp"\ndeactivate "file.cpp"\ndeactivate "file.cpp"\n\n@enduml'
actual = manager.export()
assert actual == expected
def test_given_tu_with_tree_three_functions_deep__tree_of_select_root_on_middle_function_has_depth_two(
):
manager = CallTreeManager()
content = 'void f();\nvoid g() { f(); }\nvoid h() { g(); }\n'
with generate_file('file.cpp', content) as file_name:
manager.open(file_name)
manager.select_tu(file_name)
root = manager.select_root('g()')
children = manager.get_calls_of(root.name)
assert len(children) == 1
grand_children = manager.get_calls_of(children[0].name)
assert len(grand_children) == 0
def test_given_tu_referencing_function_in_other_tu__load_definition_grows_call_tree(
):
manager = CallTreeManager()
with generate_file('g.cpp',
'extern void f();\nvoid g() { f(); }') as file_name:
manager.open(file_name)
manager.select_tu(file_name)
root = manager.select_root('g()')
with generate_file('f.cpp', 'void f() { f(); }\n') as file_name:
manager.state_ = CallTreeManagerState.READY_TO_SELECT_TU
manager.open(
file_name) # TODO(KNR): a hack, use a compilation DB instead
manager.load_definition('f()')
children = manager.get_calls_of(root.name)
grand_children = manager.get_calls_of(children[0].name)
assert len(grand_children) == 1
class EntryDialog(QDialog, Ui_EntryDialog):
def __init__(self, parent=None):
super(EntryDialog, self).__init__(parent)
self.setupUi(self)
# TODO(KNR): prevent editing the entry file and entry point lists
self.manager_ = CallTreeManager()
self.browse_compilation_database_button_.clicked.connect(
self.on_browse)
self.compilation_database_path_.editingFinished.connect(
self.on_edit_db_path)
self.extra_arguments_.editingFinished.connect(self.on_edit_extra_args)
self.entry_files_ = QStandardItemModel(self.entry_file_list_)
self.entry_file_list_.setModel(self.entry_files_)
self.entry_file_selection_ = self.entry_file_list_.selectionModel()
self.entry_file_selection_.currentChanged.connect(
self.on_select_entry_file)
self.entry_points_ = QStandardItemModel(self.entry_point_list_)
def on_edit_extra_args(self):
args = self.extra_arguments_.text()
self.manager_.state_ = CallTreeManagerState.INITIALIZED # TODO(KNR): yuck
self.manager_.set_extra_arguments(args)
if self.compilation_database_path_.text():
self.on_edit_db_path()
def on_browse(self):
path = QFileDialog.getOpenFileName(self, 'Open compilation database',
'', '*.json')
if path and len(path) > 0:
path = path[0]
if not path:
return
self.compilation_database_path_.setText(
path) # TODO(KNR): prevent double-event
self.set_db_path(path)
def on_edit_db_path(self):
path = self.compilation_database_path_.text()
if not path:
return
self.set_db_path(path)
def set_db_path(self, db_path):
compilation_database_directory = os.path.dirname(db_path)
os.chdir(compilation_database_directory)
self.entry_points_.clear()
self.entry_files_.clear()
tu_list = self.manager_.open(db_path)
common_path = os.path.commonprefix(list(tu_list))
for tu in tu_list:
item = QStandardItem(tu.replace(common_path, ''))
item.setData(tu)
self.entry_files_.appendRow(item)
def on_select_entry_file(self, current, previous):
if not current:
return
entry_file_path = self.entry_files_.item(current.row(), 0).data()
callable_list = self.manager_.select_tu(entry_file_path)
self.entry_points_.clear()
for callable in callable_list:
item = CallableItem(callable)
self.entry_points_.appendRow(item)
self.entry_point_list_.setModel(self.entry_points_)
def reject(self):
QApplication.instance().quit()
def accept(self):
if not self.entry_point_list_.selectionModel():
return
current = self.entry_point_list_.selectionModel().selectedIndexes()
if current and len(current) > 0:
entry_point = self.entry_points_.item(current[0].row(), 0)
self.manager_.select_root(entry_point.callable.name)
self.hide()
self.window_ = DiagramConfiguration(self.manager_, entry_point)
self.window_.show()