372 lines
14 KiB
Python
372 lines
14 KiB
Python
|
|
||
|
|
|
||
|
|
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()
|
||
|
|
|