implement wxTreeCtrl::ItemHasChildren() properly for virtual root item; added unit...

[wxWidgets.git] / tests / controls / treectrltest.cpp

diff --git a/tests/controls/treectrltest.cpp b/tests/controls/treectrltest.cpp
new file mode 100644 (file)
index 0000000..4d0bee2
--- /dev/null
+++ b/tests/controls/treectrltest.cpp
@@ -0,0 +1,96 @@
+///////////////////////////////////////////////////////////////////////////////
+// Name:        tests/controls/treectrltest.cpp
+// Purpose:     wxTreeCtrl unit test
+// Author:      Vadim Zeitlin
+// Created:     2008-11-26
+// RCS-ID:      $Id$
+// Copyright:   (c) 2008 Vadim Zeitlin <vadim@wxwidgets.org>
+///////////////////////////////////////////////////////////////////////////////
+
+// ----------------------------------------------------------------------------
+// headers
+// ----------------------------------------------------------------------------
+
+#include "testprec.h"
+
+#ifdef __BORLANDC__
+    #pragma hdrstop
+#endif
+
+#ifndef WX_PRECOMP
+    #include "wx/app.h"
+#endif // WX_PRECOMP
+
+#include "wx/treectrl.h"
+
+// ----------------------------------------------------------------------------
+// test class
+// ----------------------------------------------------------------------------
+
+class TreeCtrlTestCase : public CppUnit::TestCase
+{
+public:
+    TreeCtrlTestCase() { }
+
+    virtual void setUp();
+    virtual void tearDown();
+
+private:
+    CPPUNIT_TEST_SUITE( TreeCtrlTestCase );
+        CPPUNIT_TEST( HasChildren );
+        CPPUNIT_TEST( PseudoTest_SetHiddenRoot );
+        CPPUNIT_TEST( HasChildren );
+    CPPUNIT_TEST_SUITE_END();
+
+    void HasChildren();
+    void PseudoTest_SetHiddenRoot() { ms_hiddenRoot = true; }
+
+    static bool ms_hiddenRoot;
+
+    wxTreeCtrl *m_tree;
+
+    DECLARE_NO_COPY_CLASS(TreeCtrlTestCase)
+};
+
+// register in the unnamed registry so that these tests are run by default
+CPPUNIT_TEST_SUITE_REGISTRATION( TreeCtrlTestCase );
+
+// also include in it's own registry so that these tests can be run alone
+CPPUNIT_TEST_SUITE_NAMED_REGISTRATION( TreeCtrlTestCase, "TreeCtrlTestCase" );
+
+// ----------------------------------------------------------------------------
+// test initialization
+// ----------------------------------------------------------------------------
+
+bool TreeCtrlTestCase::ms_hiddenRoot = false;
+
+void TreeCtrlTestCase::setUp()
+{
+    m_tree = new wxTreeCtrl(wxTheApp->GetTopWindow());
+    if ( ms_hiddenRoot )
+        m_tree->ToggleWindowStyle(wxTR_HIDE_ROOT); // actually set it
+}
+
+void TreeCtrlTestCase::tearDown()
+{
+    delete m_tree;
+    m_tree = NULL;
+}
+
+// ----------------------------------------------------------------------------
+// the tests themselves
+// ----------------------------------------------------------------------------
+
+void TreeCtrlTestCase::HasChildren()
+{
+    const wxTreeItemId root = m_tree->AddRoot("root");
+    const wxTreeItemId child1 = m_tree->AppendItem(root, "child1");
+    const wxTreeItemId child2 = m_tree->AppendItem(root, "child2");
+    const wxTreeItemId grandchild = m_tree->AppendItem(child1, "grandchild");
+
+    CPPUNIT_ASSERT( m_tree->HasChildren(root) );
+    CPPUNIT_ASSERT( m_tree->HasChildren(child1) );
+    CPPUNIT_ASSERT( !m_tree->HasChildren(child2) );
+    CPPUNIT_ASSERT( !m_tree->HasChildren(grandchild) );
+}
+