text: ///////////////////////////////////////////////////////////////////////////////
  // Name:        tests/controls/treelistctrltest.cpp
  // Purpose:     wxTreeListCtrl unit test.
  // Author:      Vadim Zeitlin
  // Created:     2011-08-27
  // 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
  