File: test_range_editor_spinner.py

package info (click to toggle)
python-traitsui 8.0.0-2
  • links: PTS, VCS
  • area: main
  • in suites: forky, sid, trixie
  • size: 18,232 kB
  • sloc: python: 58,982; makefile: 113
file content (116 lines) | stat: -rw-r--r-- 3,918 bytes parent folder | download
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
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
# (C) Copyright 2004-2023 Enthought, Inc., Austin, TX
# All rights reserved.
#
# This software is provided without warranty under the terms of the BSD
# license included in 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
#
# Thanks for using Enthought open source!

"""
Test case for bug (wx, Mac OS X)

Editing the text part of a spin control box and pressing the OK button
without de-focusing raises an AttributeError::

    Traceback (most recent call last):
    File "ETS/traitsui/traitsui/wx/range_editor.py", line 783, in update_object
        self.value = self.control.GetValue()
    AttributeError: 'NoneType' object has no attribute 'GetValue'
"""
import unittest

from traits.has_traits import HasTraits
from traits.trait_types import Int
from traitsui.item import Item
from traitsui.view import View
from traitsui.editors.range_editor import RangeEditor

from traitsui.tests._tools import (
    BaseTestMixin,
    create_ui,
    requires_toolkit,
    reraise_exceptions,
    ToolkitName,
)


class NumberWithSpinnerEditor(HasTraits):
    """Dialog containing a RangeEditor in 'spinner' mode for an Int."""

    number = Int()

    traits_view = View(
        Item(label="Enter 4, then press OK without defocusing"),
        Item("number", editor=RangeEditor(low=3, high=8, mode="spinner")),
        buttons=["OK"],
    )


class TestRangeEditorSpinner(BaseTestMixin, unittest.TestCase):
    def setUp(self):
        BaseTestMixin.setUp(self)

    def tearDown(self):
        BaseTestMixin.tearDown(self)

    @requires_toolkit([ToolkitName.wx])
    def test_wx_spin_control_editing_should_not_crash(self):
        # Bug: when editing the text part of a spin control box, pressing
        # the OK button raises an AttributeError on Mac OS X

        num = NumberWithSpinnerEditor()
        try:
            with reraise_exceptions(), create_ui(num) as ui:

                # the following is equivalent to clicking in the text control
                # of the range editor, enter a number, and clicking ok without
                # defocusing

                # SpinCtrl object
                spin = ui.control.FindWindowByName("wxSpinCtrl", ui.control)
                spin.SetFocusFromKbd()

                # on Windows, a wxSpinCtrl does not have children, and we
                # cannot do the more fine-grained testing below
                if len(spin.GetChildren()) == 0:
                    spin.SetValue("4")
                else:
                    # TextCtrl object of the spin control
                    spintxt = spin.FindWindowByName("text", spin)
                    spintxt.SetValue("4")

        except AttributeError:
            # if all went well, we should not be here
            self.fail("AttributeError raised")

    @requires_toolkit([ToolkitName.qt])
    def test_qt_spin_control_editing(self):
        # Behavior: when editing the text part of a spin control box, pressing
        # the OK button updates the value of the HasTraits class

        from pyface import qt

        num = NumberWithSpinnerEditor()
        with reraise_exceptions(), create_ui(num) as ui:

            # the following is equivalent to clicking in the text control of
            # the range editor, enter a number, and clicking ok without
            # defocusing

            # text element inside the spin control
            lineedit = ui.control.findChild(qt.QtGui.QLineEdit)
            lineedit.setFocus()
            lineedit.setText("4")

        # if all went well, the number traits has been updated and its value is
        # 4
        self.assertEqual(num.number, 4)


if __name__ == "__main__":
    # Executing the file opens the dialog for manual testing
    num = NumberWithSpinnerEditor()
    num.configure_traits()
    print(num.number)