def test_flags_no_flags_for_invalid_index():
""" Check that no flags are returned for invalid index """
# Given
model = TreeModel()
# project = ResearchProject("")
# project.add_plan().add_task()
# When
plan_index = model.index(0, 0, model.plans_index)
task_index = model.index(0, 0, plan_index)
# Then
assert model.flags(plan_index) == Qt.NoItemFlags
assert model.flags(task_index) == Qt.NoItemFlags
def fixture_link_updater():
""" Fixture to create an updater with uncommitted updates """
tree_model = TreeModel()
individuals_model = IndividualsModel([])
sources_model = SourcesModel([])
old_value = "Altered Ancestor"
new_value = "Foo"
project = ResearchProject("")
plan_altered = ResearchPlan()
plan_altered.ancestor = old_value
plan_altered.ancestor_link = "I001"
project.plans.append(plan_altered)
tree_model.set_project(project)
context = DataContext(
data_model=tree_model,
individuals_model=individuals_model,
sources_model=sources_model,
)
updater = LinkUpdater(context)
plan_index = tree_model.index(0, TreeModelCols.TEXT, QModelIndex())
updater.ancestor_updates = [{"index": plan_index, "value": new_value}]
return (updater, old_value, new_value, tree_model, plan_index)
def test_setdata_updates_link():
""" setData() works on the link """
# Given
model = TreeModel()
project = ResearchProject("")
plan = ResearchPlan()
project.plans.append(plan)
project.plans[0].add_task()
model.set_project(project)
plan_index = model.index(0, 0, model.plans_index)
task_index = model.index(0, TreeModelCols.LINK, plan_index)
# When
retval = model.setData(task_index, 1234)
assert retval is True
assert model.data(task_index, Qt.DisplayRole) == 1234
def test_has_children_handles_plan_index():
""" When no project is set, hasChildren should return False """
# Given
model = TreeModel()
project = ResearchProject("")
project.add_plan().add_task()
model.set_project(project)
# Then
assert model.hasChildren(model.index(0, 0, model.plans_index)) is True
def test_delete_removes_task():
""" Delete removes task from underlying data structure """
# Given
model = TreeModel()
project = ResearchProject("")
plan = ResearchPlan()
task = ResearchTask()
plan.tasks.append(task)
project.plans.append(plan)
model.set_project(project)
plan_index = model.index(0, 0, model.plans_index)
task_index = model.index(0, 0, plan_index)
# When
model.delete_node(task_index)
# Then
assert len(project.plans) == 1
assert len(plan.tasks) == 0
def test_setdata_updates_result():
""" setData() works on the result """
# Given
model = TreeModel()
project = ResearchProject("")
plan = ResearchPlan()
project.plans.append(plan)
project.plans[0].add_task()
model.set_project(project)
plan_index = model.index(0, 0, model.plans_index)
task_index = model.index(0, 2, plan_index)
result = ResearchResult(True)
result.description = "Test"
# When
retval = model.setData(task_index, result)
assert retval is True
assert model.data(task_index, Qt.DisplayRole) == result
def test_get_font_returns_strikeout_for_task_with_result():
""" A task with a nil result should show up as strike-out """
# Given
model = TreeModel()
project = ResearchProject("")
plan = ResearchPlan()
task = ResearchTask()
task.result = ResearchResult(False)
plan.tasks.append(task)
project.plans.append(plan)
model.set_project(project)
plan_index = model.index(0, 0, model.plans_index)
task_index = model.index(0, 0, plan_index)
# When
font = model.data(task_index, Qt.FontRole)
# Then
assert font.strikeOut() is True
def test_flags_set_for_line_items():
""" Check flags are set correctly for plans and tasks """
# Given
model = TreeModel()
project = ResearchProject("")
project.add_plan().add_task()
model.set_project(project)
# When
plan_index = model.index(0, 0, model.plans_index)
task_index = model.index(0, 0, plan_index)
# Then
plan_flags = int(model.flags(plan_index))
task_flags = int(model.flags(task_index))
assert plan_flags & Qt.ItemIsSelectable == Qt.ItemIsSelectable
assert plan_flags & Qt.ItemIsEditable == Qt.ItemIsEditable
assert plan_flags & Qt.ItemIsEnabled == Qt.ItemIsEnabled
assert task_flags & Qt.ItemIsSelectable == Qt.ItemIsSelectable
assert task_flags & Qt.ItemIsEditable == Qt.ItemIsEditable
assert task_flags & Qt.ItemIsEnabled == Qt.ItemIsEnabled
def test_delete_calls_signals(qtbot):
""" The start and end row removal signals should be emitted """
# Given
model = TreeModel()
project = ResearchProject("")
plan = ResearchPlan()
project.plans.append(plan)
model.set_project(project)
plan_index = model.index(0, 0, model.plans_index)
# When
with qtbot.waitSignals([model.rowsAboutToBeRemoved, model.rowsRemoved]):
model.delete_node(plan_index)
def test_setdata_returns_false_on_invalid_column():
""" When setting data, false is returned for invalid column """
# Given
model = TreeModel()
project = ResearchProject("")
plan = ResearchPlan()
project.plans.append(plan)
model.set_project(project)
plan_index = model.index(0, model.columnCount(None) + 1, model.plans_index)
# When
retval = model.setData(plan_index, "Foo")
assert retval is False
def test_setdata_updates_description():
""" setData() works on the description """
# Given
model = TreeModel()
project = ResearchProject("")
plan = ResearchPlan()
project.plans.append(plan)
model.set_project(project)
plan_index = model.index(0, 1, model.plans_index)
# When
retval = model.setData(plan_index, "Foo")
assert retval is True
assert model.data(plan_index, Qt.DisplayRole) == "Foo"
def test_setdata_fires_signal_on_change(qtbot):
""" When setting data, the dataChanged signal is emitted on changes """
# Given
model = TreeModel()
project = ResearchProject("")
plan = ResearchPlan()
project.plans.append(plan)
model.set_project(project)
plan_index = model.index(0, 0, model.plans_index)
# When
with qtbot.waitSignal(model.dataChanged):
retval = model.setData(plan_index, "Foo")
assert retval is True
def test_setdata_doesnot_fire_signal_if_no_change(qtbot):
""" When setting data, the dataChanged signal should only be emitted on actual changes """
# Given
model = TreeModel()
project = ResearchProject("")
plan = ResearchPlan()
project.plans.append(plan)
model.set_project(project)
plan_index = model.index(0, 0, model.plans_index)
# When
with qtbot.assertNotEmitted(model.dataChanged):
retval = model.setData(plan_index, plan.ancestor)
assert retval is True
def test_data_returns_none_for_invalid_role():
""" When getting data, invalid role returns None """
# Given
model = TreeModel()
project = ResearchProject("")
plan = ResearchPlan()
plan.ancestor = "Foo"
project.plans.append(plan)
model.set_project(project)
plan_index = model.index(0, 0, model.plans_index)
# When
retval = model.data(plan_index, Qt.UserRole)
# Then
assert retval is None
def test_data_returns_qicon_for_decoration_role():
""" When getting data, decoration role returns an icon """
# Given
model = TreeModel()
project = ResearchProject("")
plan = ResearchPlan()
plan.ancestor = "Foo"
project.plans.append(plan)
model.set_project(project)
plan_index = model.index(0, 0, model.plans_index)
# When
retval = model.data(plan_index, Qt.DecorationRole)
# Then
assert isinstance(retval, QIcon)
def test_data_returns_string_for_valid_index():
""" When getting data, ancestor is returned for valid index """
# Given
model = TreeModel()
project = ResearchProject("")
plan = ResearchPlan()
plan.ancestor = "Foo"
project.plans.append(plan)
model.set_project(project)
plan_index = model.index(0, 0, model.plans_index)
# When
retval = model.data(plan_index, Qt.DisplayRole)
# Then
assert retval is plan.ancestor
def test_add_creates_task():
""" Add should create a new task in the underlying data structure """
# Given
model = TreeModel()
project = ResearchProject("")
plan = ResearchPlan()
project.plans.append(plan)
model.set_project(project)
plan_index = model.index(0, 0, model.plans_index)
# When
model.add_node(plan_index)
# Then
assert len(project.plans) == 1
assert len(plan.tasks) == 1
def test_data_returns_none_for_invalid_column():
""" When getting data, invalid column returns None """
# Given
model = TreeModel()
project = ResearchProject("")
plan = ResearchPlan()
plan.ancestor = "Foo"
project.plans.append(plan)
model.set_project(project)
plan_index = model.index(0,
len(list(TreeModelCols)) + 1, model.plans_index)
# When
retval = model.data(plan_index, Qt.DisplayRole)
# Then
assert retval is None
