--- /dev/null
+
+
+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()
+
projects / wxWidgets.git / blobdiff