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
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
|
# (C) Copyright 2005-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!
import os
import unittest
from ..python_editor import PythonEditor
from ..toolkit import toolkit_object
from ..window import Window
GuiTestAssistant = toolkit_object("util.gui_test_assistant:GuiTestAssistant")
no_gui_test_assistant = GuiTestAssistant.__name__ == "Unimplemented"
PYTHON_SCRIPT = os.path.join(
os.path.dirname(__file__), "python_shell_script.py"
)
@unittest.skipIf(no_gui_test_assistant, "No GuiTestAssistant")
class TestPythonEditor(unittest.TestCase, GuiTestAssistant):
def setUp(self):
GuiTestAssistant.setUp(self)
self.window = Window()
self.window.create()
def tearDown(self):
if self.widget.control is not None:
with self.delete_widget(self.window.control):
self.widget.destroy()
if self.window.control is not None:
with self.delete_widget(self.window.control):
self.window.destroy()
del self.widget
del self.window
GuiTestAssistant.tearDown(self)
def test_lifecycle(self):
# test that create/destroy works
self.widget = PythonEditor(self.window.control)
self.assertIsNone(self.widget.control)
with self.event_loop():
self.widget.create(parent=self.window.control)
self.assertIsNotNone(self.widget.control)
with self.event_loop():
self.widget.destroy()
def test_one_stage_create(self):
# test that automatic creation works
with self.event_loop():
with self.assertWarns(DeprecationWarning):
self.widget = PythonEditor(self.window.control, create=True)
self.assertIsNotNone(self.widget.control)
with self.event_loop():
self.widget.destroy()
def test_two_stage_create(self):
# test that create=False works
with self.assertWarns(DeprecationWarning):
self.widget = PythonEditor(self.window.control, create=False)
self.assertIsNone(self.widget.control)
with self.event_loop():
self.widget.create(parent=self.window.control)
self.assertIsNotNone(self.widget.control)
with self.event_loop():
self.widget.destroy()
def test_show_line_numbers(self):
# test that destroy works
with self.event_loop():
self.widget = PythonEditor(
parent=self.window.control,
show_line_numbers=False,
)
self.widget.create()
with self.event_loop():
self.widget.show_line_numbers = True
with self.event_loop():
self.widget.show_line_numbers = False
with self.event_loop():
self.widget.destroy()
def test_load(self):
# test that destroy works
with self.event_loop():
self.widget = PythonEditor(
parent=self.window.control,
)
self.widget.create()
with self.assertTraitChanges(self.widget, "changed", count=1):
with self.event_loop():
self.widget.path = PYTHON_SCRIPT
self.assertFalse(self.widget.dirty)
with self.event_loop():
self.widget.destroy()
def test_select_line(self):
# test that destroy works
with self.event_loop():
self.widget = PythonEditor(
parent=self.window.control,
path=PYTHON_SCRIPT,
)
self.widget.create()
with self.event_loop():
self.widget.select_line(3)
with self.event_loop():
self.widget.destroy()
|
