Some random test apps that I've been playing with
git-svn-id: https://svn.wxwidgets.org/svn/wx/wxWidgets/trunk@42803 c3d73ce0-8a6f-49c7-b76d-6d57e0e08775
This commit is contained in:
Robin Dunn
371
wxPython/tests/test_comboTreeBox.py
Normal file
371
wxPython/tests/test_comboTreeBox.py
Normal file
@@ -0,0 +1,371 @@
|
||||
|
||||
|
||||
import wx, unittest
|
||||
from wx.lib.combotreebox import ComboTreeBox, IterableTreeCtrl
|
||||
|
||||
|
||||
class ComboTreeBoxTest(unittest.TestCase):
|
||||
def setUp(self):
|
||||
self.comboBoxEventReceived = False
|
||||
frame = wx.Frame(None)
|
||||
self.comboBox = ComboTreeBox(frame, platform=platform)
|
||||
self.tree = self.comboBox._popupFrame.GetTree()
|
||||
|
||||
def onComboBox(self, event):
|
||||
self.comboBoxEventReceived = True
|
||||
|
||||
def testComboBoxIsEmptyByDefault(self):
|
||||
self.assertEqual(0, self.comboBox.GetCount())
|
||||
|
||||
def testAddingOneItem(self):
|
||||
self.comboBox.Append('Item 1')
|
||||
self.assertEqual(1, self.comboBox.GetCount())
|
||||
|
||||
def testAddingTwoItems(self):
|
||||
self.comboBox.Append('Item 1')
|
||||
self.comboBox.Append('Item 2')
|
||||
self.assertEqual(2, self.comboBox.GetCount())
|
||||
|
||||
def testAddingTwoParentAndChild(self):
|
||||
item1 = self.comboBox.Append('Item 1')
|
||||
self.comboBox.Append('Item 2', item1)
|
||||
self.assertEqual(2, self.comboBox.GetCount())
|
||||
|
||||
def testSelectingAnItemPutsItInTheComboBox(self):
|
||||
self.comboBox.Append('Item 1')
|
||||
self.comboBox.Bind(wx.EVT_COMBOBOX, self.onComboBox)
|
||||
self.comboBox.NotifyItemSelected('Item 1')
|
||||
self.failUnless(self.comboBoxEventReceived)
|
||||
|
||||
def testClear(self):
|
||||
self.comboBox.Append('Item 1')
|
||||
self.comboBox.Clear()
|
||||
self.assertEqual(0, self.comboBox.GetCount())
|
||||
|
||||
def testDelete(self):
|
||||
self.comboBox.Append('Item 1')
|
||||
self.comboBox.Delete(self.tree.GetFirstItem())
|
||||
self.assertEqual(0, self.comboBox.GetCount())
|
||||
|
||||
def testGetSelection_NoItems(self):
|
||||
self.failIf(self.comboBox.GetSelection().IsOk())
|
||||
|
||||
def testGetSelection_NoSelection(self):
|
||||
self.comboBox.Append('Item 1')
|
||||
self.failIf(self.comboBox.GetSelection().IsOk())
|
||||
|
||||
def testGetSelection_WithSelection(self):
|
||||
item1 = self.comboBox.Append('Item 1')
|
||||
self.comboBox.SetValue('Item 1')
|
||||
self.assertEqual(item1, self.comboBox.GetSelection())
|
||||
|
||||
def testGetSelection_EquallyNamedNodes_SelectedInTree(self):
|
||||
item1 = self.comboBox.Append('Item')
|
||||
item2 = self.comboBox.Append('Item')
|
||||
self.tree.SelectItem(item2)
|
||||
self.assertEqual(self.tree.GetSelection(), self.comboBox.GetSelection())
|
||||
|
||||
def testGetSelection_EquallyNamedNodes_TypedInTextBox(self):
|
||||
item1 = self.comboBox.Append('Item')
|
||||
item2 = self.comboBox.Append('Item')
|
||||
self.comboBox.SetValue('Item')
|
||||
self.assertEqual(item1, self.comboBox.GetSelection())
|
||||
|
||||
def testFindString_NotPresent(self):
|
||||
self.comboBox.Append('Item 1')
|
||||
self.failIf(self.comboBox.FindString('Item 2').IsOk())
|
||||
|
||||
def testFindString_Present(self):
|
||||
self.comboBox.Append('Item 1')
|
||||
self.assertEqual(self.tree.GetFirstItem(),
|
||||
self.comboBox.FindString('Item 1'))
|
||||
|
||||
def testFindString_Child(self):
|
||||
parent = self.comboBox.Append('Parent')
|
||||
child = self.comboBox.Append('Child', parent=parent)
|
||||
self.assertEqual(child, self.comboBox.FindString('Child'))
|
||||
|
||||
def testGetString_NotPresent(self):
|
||||
self.assertEqual('', self.comboBox.GetString(self.tree.GetFirstItem()))
|
||||
|
||||
def testGetString_Present(self):
|
||||
self.comboBox.Append('Item 1')
|
||||
self.assertEqual('Item 1',
|
||||
self.comboBox.GetString(self.tree.GetFirstItem()))
|
||||
|
||||
def testGetStringSelection_NotPresent(self):
|
||||
self.assertEqual('', self.comboBox.GetStringSelection())
|
||||
|
||||
def testGetStringSelection_Present(self):
|
||||
self.comboBox.SetValue('Item 1')
|
||||
self.assertEqual('Item 1', self.comboBox.GetStringSelection())
|
||||
|
||||
def testInsertAsFirstItem(self):
|
||||
self.comboBox.Insert('Item 1')
|
||||
self.assertEqual('Item 1',
|
||||
self.comboBox.GetString(self.tree.GetFirstItem()))
|
||||
|
||||
def testInsertAsFirstItemBeforeExistingItem(self):
|
||||
item1 = self.comboBox.Append('Item 1')
|
||||
item2 = self.comboBox.Insert('Item 2')
|
||||
self.assertEqual(item2, self.tree.GetFirstItem())
|
||||
|
||||
def testInsertAsFirstChildBeforeExistingChild(self):
|
||||
parent = self.comboBox.Append('parent')
|
||||
child1 = self.comboBox.Append('child 1', parent)
|
||||
child2 = self.comboBox.Insert('child 2', parent=parent)
|
||||
self.assertEqual(child2, self.tree.GetFirstChild(parent)[0])
|
||||
|
||||
def testSelect(self):
|
||||
item1 = self.comboBox.Append('Item 1')
|
||||
self.comboBox.Select(item1)
|
||||
self.assertEqual('Item 1', self.comboBox.GetValue())
|
||||
|
||||
def testSetString(self):
|
||||
item1 = self.comboBox.Append('Item 1')
|
||||
self.comboBox.SetString(item1, 'Item 2')
|
||||
self.assertEqual('Item 2', self.comboBox.GetString(item1))
|
||||
|
||||
def testSetStringSelection_ExistingString(self):
|
||||
self.comboBox.Append('Hi')
|
||||
self.comboBox.SetStringSelection('Hi')
|
||||
self.assertEqual('Hi', self.comboBox.GetStringSelection())
|
||||
|
||||
def testSetStringSelection_NonExistingString(self):
|
||||
self.comboBox.SetStringSelection('Hi')
|
||||
self.assertEqual('', self.comboBox.GetStringSelection())
|
||||
|
||||
def testAppendWithClientData(self):
|
||||
item1 = self.comboBox.Append('Item 1', clientData=[1,2,3])
|
||||
self.assertEqual([1,2,3], self.comboBox.GetClientData(item1))
|
||||
|
||||
def testInsertWithClientData(self):
|
||||
item1 = self.comboBox.Append('Item 1')
|
||||
item2 = self.comboBox.Insert('Item 2', previous=item1,
|
||||
clientData=[1,2,3])
|
||||
self.assertEqual([1,2,3], self.comboBox.GetClientData(item2))
|
||||
|
||||
def testSetClientData(self):
|
||||
item1 = self.comboBox.Append('Item 1')
|
||||
self.comboBox.SetClientData(item1, [1,2,3])
|
||||
self.assertEqual([1,2,3], self.comboBox.GetClientData(item1))
|
||||
|
||||
def testFindClientData(self):
|
||||
item1 = self.comboBox.Append('Item 1', clientData='A')
|
||||
self.assertEqual(item1, self.comboBox.FindClientData('A'))
|
||||
|
||||
def testFindClientData_NoItems(self):
|
||||
self.failIf(self.comboBox.FindClientData('A'))
|
||||
|
||||
def testFindClientData_NoSuchData(self):
|
||||
item1 = self.comboBox.Append('Item 1', clientData='A')
|
||||
self.failIf(self.comboBox.FindClientData('B'))
|
||||
|
||||
def testSetClientDataSelection(self):
|
||||
item1 = self.comboBox.Append('Item 1', clientData='A')
|
||||
self.comboBox.SetClientDataSelection('A')
|
||||
self.assertEqual(item1, self.comboBox.GetSelection())
|
||||
|
||||
def testSetClientDataSelection_NoSuchData(self):
|
||||
item1 = self.comboBox.Append('Item 1', clientData='A')
|
||||
self.comboBox.SetClientDataSelection('B')
|
||||
self.failIf(self.comboBox.GetSelection())
|
||||
|
||||
|
||||
class SortedComboTreeBoxTest(unittest.TestCase):
|
||||
def setUp(self):
|
||||
frame = wx.Frame(None)
|
||||
self.comboBox = ComboTreeBox(frame, style=wx.CB_SORT, platform=platform)
|
||||
self.tree = self.comboBox._popupFrame.GetTree()
|
||||
|
||||
def testAppend(self):
|
||||
itemB = self.comboBox.Append('B')
|
||||
itemA = self.comboBox.Append('A')
|
||||
self.assertEqual(itemA, self.tree.GetFirstItem())
|
||||
|
||||
def testInsert(self):
|
||||
itemA = self.comboBox.Append('A')
|
||||
itemB = self.comboBox.Insert('B')
|
||||
self.assertEqual(itemA, self.tree.GetFirstItem())
|
||||
|
||||
def testAppend_Child(self):
|
||||
itemA = self.comboBox.Append('A')
|
||||
itemA2 = self.comboBox.Append('2', parent=itemA)
|
||||
itemA1 = self.comboBox.Append('1', parent=itemA)
|
||||
self.assertEqual(itemA1, self.tree.GetFirstChild(itemA)[0])
|
||||
|
||||
def testInsert_Child(self):
|
||||
itemA = self.comboBox.Append('A')
|
||||
itemA1 = self.comboBox.Append('1', parent=itemA)
|
||||
itemA2 = self.comboBox.Insert('2', parent=itemA)
|
||||
self.assertEqual(itemA1, self.tree.GetFirstChild(itemA)[0])
|
||||
|
||||
def testSetString(self):
|
||||
itemB = self.comboBox.Append('B')
|
||||
itemC = self.comboBox.Append('C')
|
||||
self.comboBox.SetString(itemC, 'A')
|
||||
self.assertEqual(itemC, self.tree.GetFirstItem())
|
||||
|
||||
|
||||
class ReadOnlyComboTreeBoxTest(unittest.TestCase):
|
||||
def setUp(self):
|
||||
frame = wx.Frame(None)
|
||||
self.comboBox = ComboTreeBox(frame, style=wx.CB_READONLY)
|
||||
self.tree = self.comboBox._popupFrame.GetTree()
|
||||
|
||||
def testSetValue_ToNonExistingValue(self):
|
||||
self.comboBox.SetValue('Ignored value')
|
||||
self.assertEqual('', self.comboBox.GetValue())
|
||||
|
||||
def testSetValue_ToExistingValue(self):
|
||||
self.comboBox.Append('This works')
|
||||
self.comboBox.SetValue('This works')
|
||||
self.assertEqual('This works', self.comboBox.GetValue())
|
||||
|
||||
|
||||
class IterableTreeCtrlTest(unittest.TestCase):
|
||||
def setUp(self):
|
||||
self.frame = wx.Frame(None)
|
||||
self.tree = IterableTreeCtrl(self.frame)
|
||||
self.root = self.tree.AddRoot('root')
|
||||
|
||||
def testPreviousOfRootIsInvalid(self):
|
||||
item = self.tree.GetPreviousItem(self.root)
|
||||
self.failIf(item.IsOk())
|
||||
|
||||
def testPreviousOfChildOfRootIsRoot(self):
|
||||
child = self.tree.AppendItem(self.root, 'child')
|
||||
self.assertEqual(self.root, self.tree.GetPreviousItem(child))
|
||||
|
||||
def testPreviousOfSecondChildOfRootIsFirstChild(self):
|
||||
child1 = self.tree.AppendItem(self.root, 'child1')
|
||||
child2 = self.tree.AppendItem(self.root, 'child2')
|
||||
self.assertEqual(child1, self.tree.GetPreviousItem(child2))
|
||||
|
||||
def testPreviousOfGrandChildIsChild(self):
|
||||
child = self.tree.AppendItem(self.root, 'child')
|
||||
grandchild = self.tree.AppendItem(child, 'grandchild')
|
||||
self.assertEqual(child, self.tree.GetPreviousItem(grandchild))
|
||||
|
||||
def testPreviousOfSecondChildWhenFirstChildHasChildIsThatChild(self):
|
||||
child1 = self.tree.AppendItem(self.root, 'child1')
|
||||
grandchild = self.tree.AppendItem(child1, 'child of child1')
|
||||
child2 = self.tree.AppendItem(self.root, 'child2')
|
||||
self.assertEqual(grandchild, self.tree.GetPreviousItem(child2))
|
||||
|
||||
def testPreviousOfSecondChildWhenFirstChildHasGrandChildIsThatGrandChild(self):
|
||||
child1 = self.tree.AppendItem(self.root, 'child1')
|
||||
grandchild = self.tree.AppendItem(child1, 'child of child1')
|
||||
greatgrandchild = self.tree.AppendItem(grandchild,
|
||||
'grandchild of child1')
|
||||
child2 = self.tree.AppendItem(self.root, 'child2')
|
||||
self.assertEqual(greatgrandchild, self.tree.GetPreviousItem(child2))
|
||||
|
||||
def testNextOfRootIsInvalidWhenRootHasNoChildren(self):
|
||||
item = self.tree.GetNextItem(self.root)
|
||||
self.failIf(item.IsOk())
|
||||
|
||||
def testNextOfRootIsItsChildWhenRootHasOneChild(self):
|
||||
child = self.tree.AppendItem(self.root, 'child')
|
||||
self.assertEqual(child, self.tree.GetNextItem(self.root))
|
||||
|
||||
def testNextOfLastChildIsInvalid(self):
|
||||
child = self.tree.AppendItem(self.root, 'child')
|
||||
self.failIf(self.tree.GetNextItem(child).IsOk())
|
||||
|
||||
def testNextOfFirstChildIsSecondChild(self):
|
||||
child1 = self.tree.AppendItem(self.root, 'child1')
|
||||
child2 = self.tree.AppendItem(self.root, 'child2')
|
||||
self.assertEqual(child2, self.tree.GetNextItem(child1))
|
||||
|
||||
def testNextOfGrandChildIsItsParentsSibling(self):
|
||||
child1 = self.tree.AppendItem(self.root, 'child1')
|
||||
grandchild = self.tree.AppendItem(child1, 'child of child1')
|
||||
child2 = self.tree.AppendItem(self.root, 'child2')
|
||||
self.assertEqual(child2, self.tree.GetNextItem(grandchild))
|
||||
|
||||
def testNextOfGreatGrandChildIsItsParentsSiblingRecursively(self):
|
||||
child1 = self.tree.AppendItem(self.root, 'child1')
|
||||
grandchild = self.tree.AppendItem(child1, 'child of child1')
|
||||
greatgrandchild = self.tree.AppendItem(grandchild,
|
||||
'grandchild of child1')
|
||||
child2 = self.tree.AppendItem(self.root, 'child2')
|
||||
self.assertEqual(child2, self.tree.GetNextItem(greatgrandchild))
|
||||
|
||||
def testNextOfGrandChildWhenItIsLastIsInvalid(self):
|
||||
child = self.tree.AppendItem(self.root, 'child')
|
||||
grandchild = self.tree.AppendItem(child, 'child of child')
|
||||
self.failIf(self.tree.GetNextItem(grandchild).IsOk())
|
||||
|
||||
def testFirstItemIsRoot(self):
|
||||
self.assertEqual(self.root, self.tree.GetFirstItem())
|
||||
|
||||
def testGetFirstItemWithoutRootIsInvalid(self):
|
||||
tree = IterableTreeCtrl(self.frame)
|
||||
self.failIf(tree.GetFirstItem().IsOk())
|
||||
|
||||
def testGetSelection_NoSelection(self):
|
||||
self.tree.Unselect()
|
||||
self.failIf(self.tree.GetSelection().IsOk())
|
||||
|
||||
def testGetSelection_RootItemSelected(self):
|
||||
self.tree.SelectItem(self.tree.GetRootItem())
|
||||
self.assertEqual(self.tree.GetRootItem(), self.tree.GetSelection())
|
||||
|
||||
def testGetSelection_OtherItem(self):
|
||||
child = self.tree.AppendItem(self.root, 'child')
|
||||
self.tree.SelectItem(child)
|
||||
self.assertEqual(child, self.tree.GetSelection())
|
||||
|
||||
|
||||
class IterableTreeCtrlWithHiddenRootTest(unittest.TestCase):
|
||||
def setUp(self):
|
||||
frame = wx.Frame(None)
|
||||
self.tree = IterableTreeCtrl(frame, style=wx.TR_HIDE_ROOT)
|
||||
self.root = self.tree.AddRoot('root')
|
||||
|
||||
def testPreviousOfChildOfRootIsInvalid(self):
|
||||
child = self.tree.AppendItem(self.root, 'child')
|
||||
self.failIf(self.tree.GetPreviousItem(child).IsOk())
|
||||
|
||||
def testNextOfGrandChildWhenItIsLastIsInvalid(self):
|
||||
child = self.tree.AppendItem(self.root, 'child')
|
||||
grandchild = self.tree.AppendItem(child, 'child of child')
|
||||
self.failIf(self.tree.GetNextItem(grandchild).IsOk())
|
||||
|
||||
def testRootIsNotTheFirstItem(self):
|
||||
self.failIf(self.tree.GetFirstItem().IsOk())
|
||||
|
||||
def testFirstChildOfRootIsTheFirstItem(self):
|
||||
child = self.tree.AppendItem(self.root, 'child')
|
||||
self.assertEqual(child, self.tree.GetFirstItem())
|
||||
|
||||
def testGetSelection_NoSelection(self):
|
||||
self.tree.Unselect()
|
||||
self.failIf(self.tree.GetSelection().IsOk())
|
||||
|
||||
def testGetSelection_RootItemSelected(self):
|
||||
# Apparently, selecting a hidden root item crashes wxPython on
|
||||
# Windows, so don't do that.
|
||||
if '__WXMSW__' not in wx.PlatformInfo:
|
||||
self.tree.SelectItem(self.tree.GetRootItem())
|
||||
self.failIf(self.tree.GetSelection().IsOk())
|
||||
|
||||
def testGetSelection_OtherItem(self):
|
||||
child = self.tree.AppendItem(self.root, 'child')
|
||||
self.tree.SelectItem(child)
|
||||
self.assertEqual(child, self.tree.GetSelection())
|
||||
|
||||
|
||||
|
||||
if __name__ == '__main__':
|
||||
import sys
|
||||
if len(sys.argv) > 1:
|
||||
platform = sys.argv[1].upper()
|
||||
del sys.argv[1]
|
||||
else:
|
||||
platform = None
|
||||
|
||||
app = wx.App(False)
|
||||
unittest.main()
|
||||
|
||||
Reference in New Issue
Block a user