25
25
from StringIO import StringIO
26
26
27
27
try :
28
- import xml .etree .cElementTree as ET
28
+ import xml .etree .cElementTree as ET
29
29
except ImportError :
30
- import xml .etree .ElementTree as ET
30
+ import xml .etree .ElementTree as ET
31
31
32
32
class ScriptTest (unittest .TestCase ):
33
33
def test_error_on_script_with_null_scheme (self ):
@@ -45,17 +45,17 @@ def stream_events(self, inputs, ew):
45
45
46
46
script = NewScript ()
47
47
48
- _out = StringIO ()
49
- _err = StringIO ()
50
- ew = EventWriter (_out , _err )
48
+ out = StringIO ()
49
+ err = StringIO ()
50
+ ew = EventWriter (out , err )
51
51
52
- _in = StringIO ()
52
+ in_stream = StringIO ()
53
53
54
54
args = ["--scheme" ]
55
- return_value = script .run_script (args , ew , _in )
55
+ return_value = script .run_script (args , ew , in_stream )
56
56
57
- self .assertEqual ("" , _out .getvalue ())
58
- self .assertEqual ("FATAL Modular input script returned a null scheme.\n " , _err .getvalue ())
57
+ self .assertEqual ("" , out .getvalue ())
58
+ self .assertEqual ("FATAL Modular input script returned a null scheme.\n " , err .getvalue ())
59
59
self .assertNotEqual (0 , return_value )
60
60
61
61
def test_scheme_properly_generated_by_script (self ):
@@ -89,17 +89,17 @@ def stream_events(self, inputs, ew):
89
89
90
90
script = NewScript ()
91
91
92
- _out = StringIO ()
93
- _err = StringIO ()
94
- ew = EventWriter (_out , _err )
92
+ out = StringIO ()
93
+ err = StringIO ()
94
+ ew = EventWriter (out , err )
95
95
96
96
args = ["--scheme" ]
97
- return_value = script .run_script (args , ew , _err )
97
+ return_value = script .run_script (args , ew , err )
98
98
99
- self .assertEqual ("" , _err .getvalue ())
99
+ self .assertEqual ("" , err .getvalue ())
100
100
self .assertEqual (0 , return_value )
101
101
102
- found = ET .fromstring (_out .getvalue ())
102
+ found = ET .fromstring (out .getvalue ())
103
103
expected = ET .parse (open ("data/scheme_without_defaults.xml" )).getroot ()
104
104
105
105
self .assertTrue (xml_compare (expected , found ))
@@ -122,16 +122,16 @@ def stream_events(self, inputs, ew):
122
122
123
123
script = NewScript ()
124
124
125
- _out = StringIO ()
126
- _err = StringIO ()
127
- ew = EventWriter (_out , _err )
125
+ out = StringIO ()
126
+ err = StringIO ()
127
+ ew = EventWriter (out , err )
128
128
129
129
args = ["--validate-arguments" ]
130
130
131
131
return_value = script .run_script (args , ew , open ("data/validation.xml" ))
132
132
133
- self .assertEqual ("" , _err .getvalue ())
134
- self .assertEqual ("" , _out .getvalue ())
133
+ self .assertEqual ("" , err .getvalue ())
134
+ self .assertEqual ("" , out .getvalue ())
135
135
self .assertEqual (0 , return_value )
136
136
137
137
def test_failed_validation (self ):
@@ -151,18 +151,18 @@ def stream_events(self, inputs, ew):
151
151
152
152
script = NewScript ()
153
153
154
- _out = StringIO ()
155
- _err = StringIO ()
156
- ew = EventWriter (_out , _err )
154
+ out = StringIO ()
155
+ err = StringIO ()
156
+ ew = EventWriter (out , err )
157
157
158
158
args = ["--validate-arguments" ]
159
159
160
160
return_value = script .run_script (args , ew , open ("data/validation.xml" ))
161
161
162
162
expected = ET .parse (open ("data/validation_error.xml" )).getroot ()
163
- found = ET .fromstring (_out .getvalue ())
163
+ found = ET .fromstring (out .getvalue ())
164
164
165
- self .assertEqual ("" , _err .getvalue ())
165
+ self .assertEqual ("" , err .getvalue ())
166
166
self .assertTrue (xml_compare (expected , found ))
167
167
self .assertNotEqual (0 , return_value )
168
168
@@ -193,17 +193,17 @@ def stream_events(self, inputs, ew):
193
193
script = NewScript ()
194
194
input_configuration = open ("data/conf_with_2_inputs.xml" )
195
195
196
- _out = StringIO ()
197
- _err = StringIO ()
198
- ew = EventWriter (_out , _err )
196
+ out = StringIO ()
197
+ err = StringIO ()
198
+ ew = EventWriter (out , err )
199
199
200
200
return_value = script .run_script ([], ew , input_configuration )
201
201
202
202
self .assertEqual (0 , return_value )
203
- self .assertEqual ("" , _err .getvalue ())
203
+ self .assertEqual ("" , err .getvalue ())
204
204
205
205
expected = ET .parse (open ("data/stream_with_two_events.xml" )).getroot ()
206
- found = ET .fromstring (_out .getvalue ())
206
+ found = ET .fromstring (out .getvalue ())
207
207
208
208
self .assertTrue (xml_compare (expected , found ))
209
209
0 commit comments