]> git.saurik.com Git - wxWidgets.git/blob - tests/controls/dataviewctrltest.cpp
adding missing include
[wxWidgets.git] / tests / controls / dataviewctrltest.cpp
1 ///////////////////////////////////////////////////////////////////////////////
2 // Name: tests/controls/dataviewctrltest.cpp
3 // Purpose: wxDataViewCtrl unit test
4 // Author: Vaclav Slavik
5 // Created: 2011-08-08
6 // RCS-ID: $Id$
7 // Copyright: (c) 2011 Vaclav Slavik <vslavik@gmail.com>
8 ///////////////////////////////////////////////////////////////////////////////
9
10 // ----------------------------------------------------------------------------
11 // headers
12 // ----------------------------------------------------------------------------
13
14 #include "testprec.h"
15
16 #if wxUSE_DATAVIEWCTRL
17
18 #ifdef __BORLANDC__
19 #pragma hdrstop
20 #endif
21
22 #include "wx/app.h"
23 #include "wx/dataview.h"
24
25 #include "testableframe.h"
26
27 // ----------------------------------------------------------------------------
28 // test class
29 // ----------------------------------------------------------------------------
30
31 class DataViewCtrlTestCase : public CppUnit::TestCase
32 {
33 public:
34 DataViewCtrlTestCase() { }
35
36 virtual void setUp();
37 virtual void tearDown();
38
39 private:
40 CPPUNIT_TEST_SUITE( DataViewCtrlTestCase );
41 CPPUNIT_TEST( DeleteSelected );
42 CPPUNIT_TEST( DeleteNotSelected );
43 CPPUNIT_TEST( GetSelectionForMulti );
44 CPPUNIT_TEST( GetSelectionForSingle );
45 CPPUNIT_TEST_SUITE_END();
46
47 // Create wxDataViewTreeCtrl with the given style.
48 void Create(long style);
49
50 void DeleteSelected();
51 void DeleteNotSelected();
52 void GetSelectionForMulti();
53 void GetSelectionForSingle();
54
55 void TestSelectionFor0and1();
56
57 // the dataview control itself
58 wxDataViewTreeCtrl *m_dvc;
59
60 // and some of its items
61 wxDataViewItem m_root,
62 m_child1,
63 m_child2,
64 m_grandchild;
65
66 DECLARE_NO_COPY_CLASS(DataViewCtrlTestCase)
67 };
68
69 // register in the unnamed registry so that these tests are run by default
70 CPPUNIT_TEST_SUITE_REGISTRATION( DataViewCtrlTestCase );
71
72 // also include in its own registry so that these tests can be run alone
73 CPPUNIT_TEST_SUITE_NAMED_REGISTRATION( DataViewCtrlTestCase, "DataViewCtrlTestCase" );
74
75 // ----------------------------------------------------------------------------
76 // test initialization
77 // ----------------------------------------------------------------------------
78
79 void DataViewCtrlTestCase::Create(long style)
80 {
81 m_dvc = new wxDataViewTreeCtrl(wxTheApp->GetTopWindow(),
82 wxID_ANY,
83 wxDefaultPosition,
84 wxSize(400, 200),
85 style);
86
87 m_root = m_dvc->AppendContainer(wxDataViewItem(), "The root");
88 m_child1 = m_dvc->AppendContainer(m_root, "child1");
89 m_grandchild = m_dvc->AppendItem(m_child1, "grandchild");
90 m_child2 = m_dvc->AppendItem(m_root, "child2");
91
92 m_dvc->SetSize(400, 200);
93 m_dvc->ExpandAncestors(m_root);
94 m_dvc->Refresh();
95 m_dvc->Update();
96 }
97
98 void DataViewCtrlTestCase::setUp()
99 {
100 Create(wxDV_MULTIPLE);
101 }
102
103 void DataViewCtrlTestCase::tearDown()
104 {
105 delete m_dvc;
106 m_dvc = NULL;
107
108 m_root =
109 m_child1 =
110 m_child2 =
111 m_grandchild = wxDataViewItem();
112 }
113
114 // ----------------------------------------------------------------------------
115 // the tests themselves
116 // ----------------------------------------------------------------------------
117
118 void DataViewCtrlTestCase::DeleteSelected()
119 {
120 wxDataViewItemArray sel;
121 sel.push_back(m_child1);
122 sel.push_back(m_grandchild);
123 sel.push_back(m_child2);
124 m_dvc->SetSelections(sel);
125
126 // delete a selected item
127 m_dvc->DeleteItem(m_child1);
128
129 m_dvc->GetSelections(sel);
130
131 // m_child1 and its children should be removed from the selection now
132 CPPUNIT_ASSERT_EQUAL( 1, sel.size() );
133 CPPUNIT_ASSERT( sel[0] == m_child2 );
134 }
135
136 void DataViewCtrlTestCase::DeleteNotSelected()
137 {
138 wxDataViewItemArray sel;
139 sel.push_back(m_child1);
140 sel.push_back(m_grandchild);
141 m_dvc->SetSelections(sel);
142
143 // delete unselected item
144 m_dvc->DeleteItem(m_child2);
145
146 m_dvc->GetSelections(sel);
147
148 // m_child1 and its children should be unaffected
149 CPPUNIT_ASSERT_EQUAL( 2, sel.size() );
150 CPPUNIT_ASSERT( sel[0] == m_child1 );
151 CPPUNIT_ASSERT( sel[1] == m_grandchild );
152 }
153
154 void DataViewCtrlTestCase::TestSelectionFor0and1()
155 {
156 wxDataViewItemArray selections;
157
158 // Initially there is no selection.
159 CPPUNIT_ASSERT_EQUAL( 0, m_dvc->GetSelectedItemsCount() );
160 CPPUNIT_ASSERT( !m_dvc->HasSelection() );
161 CPPUNIT_ASSERT( !m_dvc->GetSelection().IsOk() );
162
163 CPPUNIT_ASSERT( !m_dvc->GetSelections(selections) );
164 CPPUNIT_ASSERT( selections.empty() );
165
166 // Select one item.
167 m_dvc->Select(m_child1);
168 CPPUNIT_ASSERT_EQUAL( 1, m_dvc->GetSelectedItemsCount() );
169 CPPUNIT_ASSERT( m_dvc->HasSelection() );
170 CPPUNIT_ASSERT( m_dvc->GetSelection().IsOk() );
171 CPPUNIT_ASSERT_EQUAL( 1, m_dvc->GetSelections(selections) );
172 CPPUNIT_ASSERT( selections[0] == m_child1 );
173 }
174
175 void DataViewCtrlTestCase::GetSelectionForMulti()
176 {
177 wxDataViewItemArray selections;
178
179 TestSelectionFor0and1();
180
181 // Also test with more than one selected item.
182 m_dvc->Select(m_child2);
183
184 CPPUNIT_ASSERT_EQUAL( 2, m_dvc->GetSelectedItemsCount() );
185 CPPUNIT_ASSERT( m_dvc->HasSelection() );
186 CPPUNIT_ASSERT( !m_dvc->GetSelection().IsOk() );
187 CPPUNIT_ASSERT_EQUAL( 2, m_dvc->GetSelections(selections) );
188 CPPUNIT_ASSERT( selections[1] == m_child2 );
189 }
190
191 void DataViewCtrlTestCase::GetSelectionForSingle()
192 {
193 delete m_dvc;
194 Create(0);
195
196 TestSelectionFor0and1();
197 }
198
199 #endif //wxUSE_DATAVIEWCTRL