1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
|
#------------------------------------------------------------------------------
#
# Copyright (c) 2012, Enthought, Inc.
# All rights reserved.
#
# This software is provided without warranty under the terms of the BSD
# license included in enthought/LICENSE.txt and may be redistributed only
# under the conditions described in the aforementioned license. The license
# is also available online at http://www.enthought.com/licenses/BSD.txt
#
# Author: Pietro Berkes
# Date: Dec 2012
#
#------------------------------------------------------------------------------
from traits.api import Bool, HasTraits, Instance, Int, List
from traitsui.api import Item, TreeEditor, TreeNode, View
from traitsui.tests._tools import *
class Bogus(HasTraits):
""" A bogus class representing a bogus tree. """
bogus_list = List
class BogusTreeView(HasTraits):
""" A traitsui view visualizing Bogus objects as trees. """
bogus = Instance(Bogus)
hide_root = Bool
def default_traits_view(self):
nodes = [
TreeNode(node_for=[Bogus], children='bogus_list', label='=Bogus'),
]
tree_editor = TreeEditor(
nodes=nodes, hide_root=self.hide_root, editable=False
)
traits_view = View(
Item(name='bogus', id='engine', editor=tree_editor),
buttons = ['OK'],
)
return traits_view
def _test_tree_editor_releases_listeners(hide_root):
""" The TreeEditor should release the listener to the root node's children
when it's disposed of.
"""
with store_exceptions_on_all_threads():
bogus = Bogus(bogus_list=[Bogus()])
tree_editor_view = BogusTreeView(bogus=bogus, hide_root=hide_root)
ui = tree_editor_view.edit_traits()
# The TreeEditor sets a listener on the bogus object's children list
notifiers_list = bogus.trait('bogus_list')._notifiers(False)
nose.tools.assert_equal(1, len(notifiers_list))
# Manually close the UI
press_ok_button(ui)
# The listener should be removed after the UI has been closed
notifiers_list = bogus.trait('bogus_list')._notifiers(False)
nose.tools.assert_equal(0, len(notifiers_list))
def test_tree_editor_listeners_with_shown_root():
_test_tree_editor_releases_listeners(hide_root=False)
def test_tree_editor_listeners_with_hidden_root():
_test_tree_editor_releases_listeners(hide_root=True)
|
