void tst_TreeModel::testIteration()
{
TreeModel<> m;
TreeItem *r = m.rootItem();
TreeItem *group0 = createItem("group0");
TreeItem *group1 = createItem("group1");
TreeItem *item10 = createItem("item10");
TreeItem *item11 = createItem("item11");
TreeItem *item12 = createItem("item12");
group1->appendChild(item10);
group1->appendChild(item11);
TreeItem *group2 = createItem("group2");
TreeItem *item20 = createItem("item20");
TreeItem *item21 = createItem("item21");
TreeItem *item22 = createItem("item22");
r->appendChild(group0);
r->appendChild(group1);
r->appendChild(group2);
group1->appendChild(item12);
group2->appendChild(item20);
group2->appendChild(item21);
group2->appendChild(item22);
QCOMPARE(r->childCount(), 3);
QCOMPARE(countLevelItems(r, 1), 3);
QCOMPARE(countLevelItems(r, 2), 6);
QCOMPARE(countLevelItems(r, 3), 0);
QCOMPARE(countLevelItems(group0, 1), 0);
QCOMPARE(countLevelItems(group1, 1), 3);
QCOMPARE(countLevelItems(group1, 2), 0);
QCOMPARE(countLevelItems(group2, 1), 3);
QCOMPARE(countLevelItems(group2, 2), 0);
}
void tst_TreeModel::testMixed()
{
TreeModel<TreeItem, ItemA, ItemB> m;
TreeItem *r = m.rootItem();
TreeItem *ra;
r->appendChild(new ItemA);
r->appendChild(ra = new ItemA);
ra->appendChild(new ItemB);
ra->appendChild(new ItemB);
int n = 0;
m.forItemsAtLevel<1>([&n](ItemA *) { ++n; });
QCOMPARE(n, 2);
n = 0;
m.forItemsAtLevel<2>([&n](ItemB *) { ++n; });
QCOMPARE(n, 2);
}
