src/tests/kitemlistselectionmanagertest.cpp

   1 /***************************************************************************
   2  *   Copyright (C) 2011 by Peter Penz <peter.penz19@gmail.com>             *
   3  *                                                                         *
   4  *   This program is free software; you can redistribute it and/or modify  *
   5  *   it under the terms of the GNU General Public License as published by  *
   6  *   the Free Software Foundation; either version 2 of the License, or     *
   7  *   (at your option) any later version.                                   *
   8  *                                                                         *
   9  *   This program is distributed in the hope that it will be useful,       *
  10  *   but WITHOUT ANY WARRANTY; without even the implied warranty of        *
  11  *   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the         *
  12  *   GNU General Public License for more details.                          *
  13  *                                                                         *
  14  *   You should have received a copy of the GNU General Public License     *
  15  *   along with this program; if not, write to the                         *
  16  *   Free Software Foundation, Inc.,                                       *
  17  *   51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA            *
  18  ***************************************************************************/
  19
  20 #include <qtest_kde.h>
  21
  22 #include "kitemviews/kitemmodelbase.h"
  23 #include "kitemviews/kitemlistselectionmanager.h"
  24
  25 class DummyModel : public KItemModelBase
  26 {
  27 public:
  28     DummyModel();
  29     virtual int count() const;
  30     virtual QHash<QByteArray, QVariant> data(int index) const;
  31 };
  32
  33 DummyModel::DummyModel() :
  34     KItemModelBase()
  35 {
  36 }
  37
  38 int DummyModel::count() const
  39 {
  40     return 100;
  41 }
  42
  43 QHash<QByteArray, QVariant> DummyModel::data(int index) const
  44 {
  45     Q_UNUSED(index);
  46     return QHash<QByteArray, QVariant>();
  47 }
  48
  49
  50
  51 class KItemListSelectionManagerTest : public QObject
  52 {
  53     Q_OBJECT
  54
  55 private slots:
  56     void init();
  57     void cleanup();
  58
  59     void testConstructor();
  60
  61     void testSetSelected_data();
  62     void testSetSelected();
  63     void testItemsInserted();
  64     void testItemsRemoved();
  65
  66 private:
  67     KItemListSelectionManager* m_selectionManager;
  68 };
  69
  70 void KItemListSelectionManagerTest::init()
  71 {
  72     m_selectionManager = new KItemListSelectionManager();
  73     m_selectionManager->setModel(new DummyModel());
  74 }
  75
  76 void KItemListSelectionManagerTest::cleanup()
  77 {
  78     delete m_selectionManager->model();
  79     delete m_selectionManager;
  80     m_selectionManager = 0;
  81 }
  82
  83 void KItemListSelectionManagerTest::testConstructor()
  84 {
  85     QVERIFY(!m_selectionManager->hasSelection());
  86     QCOMPARE(m_selectionManager->selectedItems().count(), 0);
  87     QCOMPARE(m_selectionManager->currentItem(), 0);
  88     QCOMPARE(m_selectionManager->anchorItem(), -1);
  89 }
  90
  91 void KItemListSelectionManagerTest::testSetSelected_data()
  92 {
  93     QTest::addColumn<int>("index");
  94     QTest::addColumn<int>("count");
  95     QTest::addColumn<int>("expectedSelectionCount");
  96
  97     QTest::newRow("Select all") << 0 << 100 << 100;
  98     QTest::newRow("Sub selection 15 items") << 20 << 15 << 15;
  99     QTest::newRow("Sub selection 1 item") << 20 << 1 << 1;
 100     QTest::newRow("Too small index") << -1 << 100 << 0;
 101     QTest::newRow("Too large index") << 100 << 100 << 0;
 102     QTest::newRow("Too large count") << 0 << 100000 << 100;
 103     QTest::newRow("Too small count") << 0 << 0 << 0;
 104 }
 105
 106 void KItemListSelectionManagerTest::testSetSelected()
 107 {
 108     QFETCH(int, index);
 109     QFETCH(int, count);
 110     QFETCH(int, expectedSelectionCount);
 111     m_selectionManager->setSelected(index, count);
 112     QCOMPARE(m_selectionManager->selectedItems().count(), expectedSelectionCount);
 113 }
 114
 115 void KItemListSelectionManagerTest::testItemsInserted()
 116 {
 117     // Select items 10 to 12
 118     m_selectionManager->setSelected(10, 3);
 119     QSet<int> selectedItems = m_selectionManager->selectedItems();
 120     QVERIFY(selectedItems.contains(10));
 121     QVERIFY(selectedItems.contains(11));
 122     QVERIFY(selectedItems.contains(12));
 123
 124     // Insert items 0 to 4 -> selection must be 15 to 17
 125     m_selectionManager->itemsInserted(KItemRangeList() << KItemRange(0, 5));
 126     selectedItems = m_selectionManager->selectedItems();
 127     QVERIFY(selectedItems.contains(15));
 128     QVERIFY(selectedItems.contains(16));
 129     QVERIFY(selectedItems.contains(17));
 130
 131     // Insert 3 items between the selections
 132     m_selectionManager->itemsInserted(KItemRangeList() <<
 133                                       KItemRange(15, 1) <<
 134                                       KItemRange(16, 1) <<
 135                                       KItemRange(17, 1));
 136     selectedItems = m_selectionManager->selectedItems();
 137     QVERIFY(selectedItems.contains(16));
 138     QVERIFY(selectedItems.contains(18));
 139     QVERIFY(selectedItems.contains(20));
 140 }
 141
 142 void KItemListSelectionManagerTest::testItemsRemoved()
 143 {
 144     // Select items 10 to 15
 145     m_selectionManager->setSelected(10, 6);
 146     QSet<int> selectedItems = m_selectionManager->selectedItems();
 147     for (int i = 10; i <= 15; ++i) {
 148         QVERIFY(selectedItems.contains(i));
 149     }
 150
 151     // Remove items 0 to 4 -> selection must be 5 to 10
 152     m_selectionManager->itemsRemoved(KItemRangeList() << KItemRange(0, 5));
 153     selectedItems = m_selectionManager->selectedItems();
 154     for (int i = 5; i <= 10; ++i) {
 155         QVERIFY(selectedItems.contains(i));
 156     }
 157
 158     // Remove the items 6 , 8 and 10
 159     m_selectionManager->itemsRemoved(KItemRangeList() <<
 160                                      KItemRange(6, 1) <<
 161                                      KItemRange(8, 1) <<
 162                                      KItemRange(10, 1));
 163     selectedItems = m_selectionManager->selectedItems();
 164     QCOMPARE(selectedItems.count(), 3);
 165     QVERIFY(selectedItems.contains(5));
 166     QVERIFY(selectedItems.contains(6));
 167     QVERIFY(selectedItems.contains(7));
 168 }
 169
 170 QTEST_KDEMAIN(KItemListSelectionManagerTest, NoGUI)
 171
 172 #include "kitemlistselectionmanagertest.moc"