add unit test for column reordering functions
[wxWidgets.git] / tests / controls / headerctrltest.cpp
1 ///////////////////////////////////////////////////////////////////////////////
2 // Name: tests/controls/headerctrltest.cpp
3 // Purpose: wxHeaderCtrl unit test
4 // Author: Vadim Zeitlin
5 // Created: 2008-11-26
6 // RCS-ID: $Id$
7 // Copyright: (c) 2008 Vadim Zeitlin <vadim@wxwidgets.org>
8 ///////////////////////////////////////////////////////////////////////////////
9
10 // ----------------------------------------------------------------------------
11 // headers
12 // ----------------------------------------------------------------------------
13
14 #include "testprec.h"
15
16 #ifdef __BORLANDC__
17 #pragma hdrstop
18 #endif
19
20 #ifndef WX_PRECOMP
21 #include "wx/app.h"
22 #include "wx/headerctrl.h"
23 #endif // WX_PRECOMP
24
25 // ----------------------------------------------------------------------------
26 // test class
27 // ----------------------------------------------------------------------------
28
29 class HeaderCtrlTestCase : public CppUnit::TestCase
30 {
31 public:
32 HeaderCtrlTestCase() { }
33
34 virtual void setUp();
35 virtual void tearDown();
36
37 private:
38 CPPUNIT_TEST_SUITE( HeaderCtrlTestCase );
39 CPPUNIT_TEST( AddDelete );
40 CPPUNIT_TEST( BestSize );
41 CPPUNIT_TEST( Reorder );
42 CPPUNIT_TEST_SUITE_END();
43
44 void AddDelete();
45 void BestSize();
46 void Reorder();
47
48 wxHeaderCtrlSimple *m_header;
49
50 DECLARE_NO_COPY_CLASS(HeaderCtrlTestCase)
51 };
52
53 // register in the unnamed registry so that these tests are run by default
54 CPPUNIT_TEST_SUITE_REGISTRATION( HeaderCtrlTestCase );
55
56 // also include in it's own registry so that these tests can be run alone
57 CPPUNIT_TEST_SUITE_NAMED_REGISTRATION( HeaderCtrlTestCase, "HeaderCtrlTestCase" );
58
59 // ----------------------------------------------------------------------------
60 // test initialization
61 // ----------------------------------------------------------------------------
62
63 void HeaderCtrlTestCase::setUp()
64 {
65 m_header = new wxHeaderCtrlSimple(wxTheApp->GetTopWindow());
66 }
67
68 void HeaderCtrlTestCase::tearDown()
69 {
70 delete m_header;
71 m_header = NULL;
72 }
73
74 // ----------------------------------------------------------------------------
75 // the tests themselves
76 // ----------------------------------------------------------------------------
77
78 void HeaderCtrlTestCase::AddDelete()
79 {
80 CPPUNIT_ASSERT_EQUAL( 0, m_header->GetColumnCount() );
81
82 m_header->AppendColumn(wxHeaderColumnSimple("Column 1"));
83 CPPUNIT_ASSERT_EQUAL( 1, m_header->GetColumnCount() );
84
85 m_header->AppendColumn(wxHeaderColumnSimple("Column 2"));
86 CPPUNIT_ASSERT_EQUAL( 2, m_header->GetColumnCount() );
87
88 m_header->InsertColumn(wxHeaderColumnSimple("Column 0"), 0);
89 CPPUNIT_ASSERT_EQUAL( 3, m_header->GetColumnCount() );
90
91 m_header->DeleteColumn(2);
92 CPPUNIT_ASSERT_EQUAL( 2, m_header->GetColumnCount() );
93 }
94
95 void HeaderCtrlTestCase::BestSize()
96 {
97 const wxSize sizeEmpty = m_header->GetBestSize();
98 CPPUNIT_ASSERT( sizeEmpty.x > 0 );
99 CPPUNIT_ASSERT( sizeEmpty.y > 0 );
100
101 m_header->AppendColumn(wxHeaderColumnSimple("Foo"));
102 m_header->AppendColumn(wxHeaderColumnSimple("Bar"));
103 const wxSize size = m_header->GetBestSize();
104 CPPUNIT_ASSERT_EQUAL( sizeEmpty.y, size.y );
105 }
106
107 void HeaderCtrlTestCase::Reorder()
108 {
109 static const int COL_COUNT = 4;
110
111 int n;
112
113 for ( n = 0; n < COL_COUNT; n++ )
114 m_header->AppendColumn(wxHeaderColumnSimple(wxString::Format("%d", n)));
115
116 wxArrayInt order = m_header->GetColumnsOrder(); // initial order: [0 1 2 3]
117 for ( n = 0; n < COL_COUNT; n++ )
118 CPPUNIT_ASSERT_EQUAL( n, order[n] );
119
120 wxHeaderCtrl::MoveColumnInOrderArray(order, 0, 2);
121 m_header->SetColumnsOrder(order); // change order to [1 2 0 3]
122
123 order = m_header->GetColumnsOrder();
124 CPPUNIT_ASSERT_EQUAL( 1, order[0] );
125 CPPUNIT_ASSERT_EQUAL( 2, order[1] );
126 CPPUNIT_ASSERT_EQUAL( 0, order[2] );
127 CPPUNIT_ASSERT_EQUAL( 3, order[3] );
128
129 order[2] = 3;
130 order[3] = 0;
131 m_header->SetColumnsOrder(order); // and now [1 2 3 0]
132 order = m_header->GetColumnsOrder();
133 CPPUNIT_ASSERT_EQUAL( 1, order[0] );
134 CPPUNIT_ASSERT_EQUAL( 2, order[1] );
135 CPPUNIT_ASSERT_EQUAL( 3, order[2] );
136 CPPUNIT_ASSERT_EQUAL( 0, order[3] );
137
138 wxHeaderCtrl::MoveColumnInOrderArray(order, 1, 3);
139 m_header->SetColumnsOrder(order); // finally [2 3 0 1]
140 order = m_header->GetColumnsOrder();
141 CPPUNIT_ASSERT_EQUAL( 2, order[0] );
142 CPPUNIT_ASSERT_EQUAL( 3, order[1] );
143 CPPUNIT_ASSERT_EQUAL( 0, order[2] );
144 CPPUNIT_ASSERT_EQUAL( 1, order[3] );
145 }
146