def _leaf_item(toolbox=None):
if toolbox is None:
toolbox = create_toolboxui_with_project()
project_item = View(toolbox, toolbox.project(), toolbox, "View",
"A View item", 0.0, 0.0)
item = LeafProjectTreeItem(project_item, toolbox)
return toolbox, item
def test_category_of_item(self):
root = RootProjectTreeItem()
category = CategoryProjectTreeItem("category", "category description")
root.add_child(category)
model = ProjectItemModel(self.toolbox, root)
self.assertEqual(model.category_of_item("nonexistent item"), None)
project_item = ProjectItem("item", "item description", 0.0, 0.0,
self.toolbox.project(), self.toolbox)
item = LeafProjectTreeItem(project_item, self.toolbox)
category.add_child(item)
found_category = model.category_of_item("item")
self.assertEqual(found_category.name, category.name)
def test_set_leaf_item_name(self):
root = RootProjectTreeItem()
model = ProjectItemModel(root)
category = CategoryProjectTreeItem("category", "category description")
model.insert_item(category)
category_index = model.find_category("category")
project_item = ProjectItem("item", "item description", 0.0, 0.0,
self.toolbox.project())
leaf = LeafProjectTreeItem(project_item, self.toolbox)
model.insert_item(leaf, category_index)
leaf_index = model.find_item("item")
model.set_leaf_item_name(leaf_index, "new view item name")
leaf_item = model.get_item("new view item name")
self.assertIsNotNone(leaf_item)
def test_insert_item_leaf_item(self):
root = RootProjectTreeItem()
model = ProjectItemModel(self.toolbox, root)
category = CategoryProjectTreeItem("category", "category description")
model.insert_item(category)
category_index = model.find_category("category")
mock_project_item = NonCallableMagicMock()
mock_project_item.name = "project item"
mock_project_item.description = "project item description"
leaf = LeafProjectTreeItem(mock_project_item, self.toolbox)
model.insert_item(leaf, category_index)
self.assertEqual(model.rowCount(), 1)
self.assertEqual(model.rowCount(category_index), 1)
self.assertEqual(model.n_items(), 1)
self.assertEqual(model.items("category"), [leaf])
def _leaf_item(toolbox):
project_item = ProjectItem("PI", "A Project item", 0.0, 0.0,
toolbox.project())
item = LeafProjectTreeItem(project_item, toolbox)
return item
def _leaf_item(toolbox=None):
if toolbox is None:
toolbox = create_toolboxui_with_project()
project_item = ProjectItem("PI", "A Project item", 0.0, 0.0, toolbox.project(), toolbox)
item = LeafProjectTreeItem(project_item, toolbox)
return toolbox, item