--- /dev/null
+///////////////////////////////////////////////////////////////////////////////
+// Name: tests/controls/treelistctrltest.cpp
+// Purpose: wxTreeListCtrl unit test.
+// Author: Vadim Zeitlin
+// Created: 2011-08-27
+// RCS-ID: $Id$
+// Copyright: (c) 2011 Vadim Zeitlin <vadim@wxwidgets.org>
+///////////////////////////////////////////////////////////////////////////////
+
+// ----------------------------------------------------------------------------
+// headers
+// ----------------------------------------------------------------------------
+
+#include "testprec.h"
+
+#if wxUSE_TREELISTCTRL
+
+#ifdef __BORLANDC__
+ #pragma hdrstop
+#endif
+
+#include "wx/treelist.h"
+
+#include "wx/app.h"
+
+// ----------------------------------------------------------------------------
+// test class
+// ----------------------------------------------------------------------------
+
+class TreeListCtrlTestCase : public CppUnit::TestCase
+{
+public:
+ TreeListCtrlTestCase() { }
+
+ virtual void setUp();
+ virtual void tearDown();
+
+private:
+ CPPUNIT_TEST_SUITE( TreeListCtrlTestCase );
+ CPPUNIT_TEST( Traversal );
+ CPPUNIT_TEST( ItemText );
+ CPPUNIT_TEST( ItemCheck );
+ CPPUNIT_TEST_SUITE_END();
+
+ // Create the control with the given style.
+ void Create(long style);
+
+ // Add an item to the tree and increment m_numItems.
+ wxTreeListItem AddItem(const char *label,
+ wxTreeListItem parent = wxTreeListItem(),
+ const char *numFiles = "",
+ const char *size = "");
+
+
+ // Tests:
+ void Traversal();
+ void ItemText();
+ void ItemCheck();
+
+
+ // The control itself.
+ wxTreeListCtrl *m_treelist;
+
+ // And some of its items.
+ wxTreeListItem m_code,
+ m_code_osx,
+ m_code_osx_cocoa;
+
+ // Also the total number of items in it initially
+ unsigned m_numItems;
+
+ wxDECLARE_NO_COPY_CLASS(TreeListCtrlTestCase);
+};
+
+// register in the unnamed registry so that these tests are run by default
+CPPUNIT_TEST_SUITE_REGISTRATION( TreeListCtrlTestCase );
+
+// also include in its own registry so that these tests can be run alone
+CPPUNIT_TEST_SUITE_NAMED_REGISTRATION( TreeListCtrlTestCase, "TreeListCtrlTestCase" );
+
+// ----------------------------------------------------------------------------
+// test initialization
+// ----------------------------------------------------------------------------
+
+wxTreeListItem
+TreeListCtrlTestCase::AddItem(const char *label,
+ wxTreeListItem parent,
+ const char *numFiles,
+ const char *size)
+{
+ if ( !parent.IsOk() )
+ parent = m_treelist->GetRootItem();
+
+ wxTreeListItem item = m_treelist->AppendItem(parent, label);
+ m_treelist->SetItemText(item, 1, numFiles);
+ m_treelist->SetItemText(item, 2, size);
+
+ m_numItems++;
+
+ return item;
+}
+
+void TreeListCtrlTestCase::Create(long style)
+{
+ m_treelist = new wxTreeListCtrl(wxTheApp->GetTopWindow(),
+ wxID_ANY,
+ wxDefaultPosition,
+ wxSize(400, 200),
+ style);
+
+ m_treelist->AppendColumn("Component");
+ m_treelist->AppendColumn("# Files");
+ m_treelist->AppendColumn("Size");
+
+ // Fill the control with the same data as used in the treelist sample:
+ m_code = AddItem("Code");
+ AddItem("wxMSW", m_code, "313", "3.94 MiB");
+ AddItem("wxGTK", m_code, "180", "1.66 MiB");
+
+ m_code_osx = AddItem("wxOSX", m_code, "265", "2.36 MiB");
+ AddItem("Core", m_code_osx, "31", "347 KiB");
+ AddItem("Carbon", m_code_osx, "91", "1.34 MiB");
+ m_code_osx_cocoa = AddItem("Cocoa", m_code_osx, "46", "512 KiB");
+
+ wxTreeListItem Documentation = AddItem("Documentation");
+ AddItem("HTML", Documentation, "many");
+ AddItem("CHM", Documentation, "1");
+
+ wxTreeListItem Samples = AddItem("Samples");
+ AddItem("minimal", Samples, "1", "7 KiB");
+ AddItem("widgets", Samples, "28", "419 KiB");
+
+ m_treelist->Refresh();
+ m_treelist->Update();
+}
+
+void TreeListCtrlTestCase::setUp()
+{
+ m_numItems = 0;
+ Create(wxTL_MULTIPLE | wxTL_3STATE);
+}
+
+void TreeListCtrlTestCase::tearDown()
+{
+ delete m_treelist;
+ m_treelist = NULL;
+}
+
+// ----------------------------------------------------------------------------
+// the tests themselves
+// ----------------------------------------------------------------------------
+
+// Test various tree traversal methods.
+void TreeListCtrlTestCase::Traversal()
+{
+ // GetParent() tests:
+ wxTreeListItem root = m_treelist->GetRootItem();
+ CPPUNIT_ASSERT( !m_treelist->GetItemParent(root) );
+
+ CPPUNIT_ASSERT_EQUAL( root, m_treelist->GetItemParent(m_code) );
+ CPPUNIT_ASSERT_EQUAL( m_code, m_treelist->GetItemParent(m_code_osx) );
+
+
+ // GetFirstChild() and GetNextSibling() tests:
+ CPPUNIT_ASSERT_EQUAL( m_code, m_treelist->GetFirstChild(root) );
+ CPPUNIT_ASSERT_EQUAL
+ (
+ m_code_osx,
+ m_treelist->GetNextSibling
+ (
+ m_treelist->GetNextSibling
+ (
+ m_treelist->GetFirstChild(m_code)
+ )
+ )
+ );
+
+ // Get{First,Next}Item() test:
+ unsigned numItems = 0;
+ for ( wxTreeListItem item = m_treelist->GetFirstItem();
+ item.IsOk();
+ item = m_treelist->GetNextItem(item) )
+ {
+ numItems++;
+ }
+
+ CPPUNIT_ASSERT_EQUAL( m_numItems, numItems );
+}
+
+// Test accessing items text.
+void TreeListCtrlTestCase::ItemText()
+{
+ CPPUNIT_ASSERT_EQUAL( "Cocoa", m_treelist->GetItemText(m_code_osx_cocoa) );
+ CPPUNIT_ASSERT_EQUAL( "46", m_treelist->GetItemText(m_code_osx_cocoa, 1) );
+
+ m_treelist->SetItemText(m_code_osx_cocoa, "wxCocoa");
+ CPPUNIT_ASSERT_EQUAL( "wxCocoa", m_treelist->GetItemText(m_code_osx_cocoa) );
+
+ m_treelist->SetItemText(m_code_osx_cocoa, 1, "47");
+ CPPUNIT_ASSERT_EQUAL( "47", m_treelist->GetItemText(m_code_osx_cocoa, 1) );
+}
+
+// Test checking and unchecking items.
+void TreeListCtrlTestCase::ItemCheck()
+{
+ CPPUNIT_ASSERT_EQUAL( wxCHK_UNCHECKED,
+ m_treelist->GetCheckedState(m_code) );
+
+ m_treelist->CheckItemRecursively(m_code);
+ CPPUNIT_ASSERT_EQUAL( wxCHK_CHECKED,
+ m_treelist->GetCheckedState(m_code) );
+ CPPUNIT_ASSERT_EQUAL( wxCHK_CHECKED,
+ m_treelist->GetCheckedState(m_code_osx) );
+ CPPUNIT_ASSERT_EQUAL( wxCHK_CHECKED,
+ m_treelist->GetCheckedState(m_code_osx_cocoa) );
+
+ m_treelist->UncheckItem(m_code_osx_cocoa);
+ CPPUNIT_ASSERT_EQUAL( wxCHK_UNCHECKED,
+ m_treelist->GetCheckedState(m_code_osx_cocoa) );
+
+ m_treelist->UpdateItemParentStateRecursively(m_code_osx_cocoa);
+ CPPUNIT_ASSERT_EQUAL( wxCHK_UNDETERMINED,
+ m_treelist->GetCheckedState(m_code_osx) );
+ CPPUNIT_ASSERT_EQUAL( wxCHK_UNDETERMINED,
+ m_treelist->GetCheckedState(m_code) );
+
+ m_treelist->CheckItemRecursively(m_code_osx, wxCHK_UNCHECKED);
+ m_treelist->UpdateItemParentStateRecursively(m_code_osx_cocoa);
+ CPPUNIT_ASSERT_EQUAL( wxCHK_UNCHECKED,
+ m_treelist->GetCheckedState(m_code_osx) );
+ CPPUNIT_ASSERT_EQUAL( wxCHK_UNDETERMINED,
+ m_treelist->GetCheckedState(m_code) );
+}
+
+#endif // wxUSE_TREELISTCTRL
projects / wxWidgets.git / blobdiff