wxTreebook is supposed to allow not specifying any valid window for the
top-level pages, but this didn't work any longer, probably since the
changes of 02a92e23f3 (see #4379), as a
possibly null page was dereferenced without checking, resulting in a
crash.
Fix this by adding a missing check.
Also rename DoGetNonNullPage() to TryGetNonNullPage() to make it more
clear that this function can return null and add a unit test checking
that calling AddPage(NULL) really works (or at least doesn't crash).
See https://github.com/wxWidgets/wxWidgets/pull/921
		
	
		
			
				
	
	
		
			163 lines
		
	
	
		
			4.4 KiB
		
	
	
	
		
			C++
		
	
	
	
	
	
			
		
		
	
	
			163 lines
		
	
	
		
			4.4 KiB
		
	
	
	
		
			C++
		
	
	
	
	
	
| ///////////////////////////////////////////////////////////////////////////////
 | |
| // Name:        tests/controls/treebooktest.cpp
 | |
| // Purpose:     wxtreebook unit test
 | |
| // Author:      Steven Lamerton
 | |
| // Created:     2010-07-02
 | |
| // Copyright:   (c) 2010 Steven Lamerton
 | |
| ///////////////////////////////////////////////////////////////////////////////
 | |
| 
 | |
| #include "testprec.h"
 | |
| 
 | |
| #if wxUSE_TREEBOOK
 | |
| 
 | |
| #ifdef __BORLANDC__
 | |
|     #pragma hdrstop
 | |
| #endif
 | |
| 
 | |
| #ifndef WX_PRECOMP
 | |
|     #include "wx/app.h"
 | |
|     #include "wx/panel.h"
 | |
| #endif // WX_PRECOMP
 | |
| 
 | |
| #include "wx/treebook.h"
 | |
| #include "bookctrlbasetest.h"
 | |
| 
 | |
| class TreebookTestCase : public BookCtrlBaseTestCase, public CppUnit::TestCase
 | |
| {
 | |
| public:
 | |
|     TreebookTestCase() { }
 | |
| 
 | |
|     virtual void setUp() wxOVERRIDE;
 | |
|     virtual void tearDown() wxOVERRIDE;
 | |
| 
 | |
| private:
 | |
|     virtual wxBookCtrlBase *GetBase() const wxOVERRIDE { return m_treebook; }
 | |
| 
 | |
|     virtual wxEventType GetChangedEvent() const wxOVERRIDE
 | |
|         { return wxEVT_TREEBOOK_PAGE_CHANGED; }
 | |
| 
 | |
|     virtual wxEventType GetChangingEvent() const wxOVERRIDE
 | |
|         { return wxEVT_TREEBOOK_PAGE_CHANGING; }
 | |
| 
 | |
|     CPPUNIT_TEST_SUITE( TreebookTestCase );
 | |
|         wxBOOK_CTRL_BASE_TESTS();
 | |
|         CPPUNIT_TEST( Image );
 | |
|         CPPUNIT_TEST( SubPages );
 | |
|         CPPUNIT_TEST( ContainerPage );
 | |
|         CPPUNIT_TEST( Expand );
 | |
|         CPPUNIT_TEST( Delete );
 | |
|     CPPUNIT_TEST_SUITE_END();
 | |
| 
 | |
|     void SubPages();
 | |
|     void ContainerPage();
 | |
|     void Expand();
 | |
|     void Delete();
 | |
| 
 | |
|     wxTreebook *m_treebook;
 | |
| 
 | |
|     wxDECLARE_NO_COPY_CLASS(TreebookTestCase);
 | |
| };
 | |
| 
 | |
| // register in the unnamed registry so that these tests are run by default
 | |
| CPPUNIT_TEST_SUITE_REGISTRATION( TreebookTestCase );
 | |
| 
 | |
| // also include in its own registry so that these tests can be run alone
 | |
| CPPUNIT_TEST_SUITE_NAMED_REGISTRATION( TreebookTestCase, "TreebookTestCase" );
 | |
| 
 | |
| void TreebookTestCase::setUp()
 | |
| {
 | |
|     m_treebook = new wxTreebook(wxTheApp->GetTopWindow(), wxID_ANY);
 | |
|     AddPanels();
 | |
| }
 | |
| 
 | |
| void TreebookTestCase::tearDown()
 | |
| {
 | |
|     wxDELETE(m_treebook);
 | |
| }
 | |
| 
 | |
| void TreebookTestCase::SubPages()
 | |
| {
 | |
|     wxPanel* subpanel1 = new wxPanel(m_treebook);
 | |
|     wxPanel* subpanel2 = new wxPanel(m_treebook);
 | |
|     wxPanel* subpanel3 = new wxPanel(m_treebook);
 | |
| 
 | |
|     m_treebook->AddSubPage(subpanel1, "Subpanel 1", false, 0);
 | |
| 
 | |
|     CPPUNIT_ASSERT_EQUAL(2, m_treebook->GetPageParent(3));
 | |
| 
 | |
|     m_treebook->InsertSubPage(1, subpanel2, "Subpanel 2", false, 1);
 | |
| 
 | |
|     CPPUNIT_ASSERT_EQUAL(1, m_treebook->GetPageParent(2));
 | |
| 
 | |
|     m_treebook->AddSubPage(subpanel3, "Subpanel 3", false, 2);
 | |
| 
 | |
|     CPPUNIT_ASSERT_EQUAL(3, m_treebook->GetPageParent(5));
 | |
| }
 | |
| 
 | |
| void TreebookTestCase::ContainerPage()
 | |
| {
 | |
|     // Get rid of the pages added in setUp().
 | |
|     m_treebook->DeleteAllPages();
 | |
|     CHECK( m_treebook->GetPageCount() == 0 );
 | |
| 
 | |
|     // Adding a page without the associated window should be allowed.
 | |
|     REQUIRE_NOTHROW( m_treebook->AddPage(NULL, "Container page") );
 | |
|     CHECK( m_treebook->GetPageParent(0) == -1 );
 | |
| 
 | |
|     m_treebook->AddSubPage(new wxPanel(m_treebook), "Child page");
 | |
|     CHECK( m_treebook->GetPageParent(1) == 0 );
 | |
| }
 | |
| 
 | |
| void TreebookTestCase::Expand()
 | |
| {
 | |
|     wxPanel* subpanel1 = new wxPanel(m_treebook);
 | |
|     wxPanel* subpanel2 = new wxPanel(m_treebook);
 | |
|     wxPanel* subpanel3 = new wxPanel(m_treebook);
 | |
| 
 | |
|     m_treebook->AddSubPage(subpanel1, "Subpanel 1", false, 0);
 | |
|     m_treebook->InsertSubPage(1, subpanel2, "Subpanel 2", false, 1);
 | |
|     m_treebook->AddSubPage(subpanel3, "Subpanel 3", false, 2);
 | |
| 
 | |
|     CPPUNIT_ASSERT(!m_treebook->IsNodeExpanded(1));
 | |
|     CPPUNIT_ASSERT(!m_treebook->IsNodeExpanded(3));
 | |
| 
 | |
|     m_treebook->CollapseNode(1);
 | |
| 
 | |
|     CPPUNIT_ASSERT(!m_treebook->IsNodeExpanded(1));
 | |
| 
 | |
|     m_treebook->ExpandNode(3, false);
 | |
| 
 | |
|     CPPUNIT_ASSERT(!m_treebook->IsNodeExpanded(3));
 | |
| 
 | |
|     m_treebook->ExpandNode(1);
 | |
| 
 | |
|     CPPUNIT_ASSERT(m_treebook->IsNodeExpanded(1));
 | |
| }
 | |
| 
 | |
| void TreebookTestCase::Delete()
 | |
| {
 | |
|     wxPanel* subpanel1 = new wxPanel(m_treebook);
 | |
|     wxPanel* subpanel2 = new wxPanel(m_treebook);
 | |
|     wxPanel* subpanel3 = new wxPanel(m_treebook);
 | |
| 
 | |
|     m_treebook->AddSubPage(subpanel1, "Subpanel 1", false, 0);
 | |
|     m_treebook->InsertSubPage(1, subpanel2, "Subpanel 2", false, 1);
 | |
|     m_treebook->AddSubPage(subpanel3, "Subpanel 3", false, 2);
 | |
| 
 | |
|     CPPUNIT_ASSERT_EQUAL(6, m_treebook->GetPageCount());
 | |
| 
 | |
|     m_treebook->DeletePage(3);
 | |
| 
 | |
|     CPPUNIT_ASSERT_EQUAL(3, m_treebook->GetPageCount());
 | |
| 
 | |
|     m_treebook->DeletePage(1);
 | |
| 
 | |
|     CPPUNIT_ASSERT_EQUAL(1, m_treebook->GetPageCount());
 | |
| 
 | |
|     m_treebook->DeletePage(0);
 | |
| 
 | |
|     CPPUNIT_ASSERT_EQUAL(0, m_treebook->GetPageCount());
 | |
| }
 | |
| 
 | |
| #endif // wxUSE_TREEBOOK
 |