blob: 5ae1ba6b9d9c32e81239f41269ac688ddf4ba2e6 [file] [log] [blame]
Guido van Rossumb5a755e2007-07-18 18:15:48 +00001import base64
Fred Drakeba613c32005-02-10 18:33:30 +00002import datetime
Skip Montanaro3e7bba92001-10-19 16:06:52 +00003import sys
Guido van Rossumb5a755e2007-07-18 18:15:48 +00004import time
Skip Montanaro419abda2001-10-01 17:47:44 +00005import unittest
Georg Brandl38eceaa2008-05-26 11:14:17 +00006import xmlrpc.client as xmlrpclib
7import xmlrpc.server
Guido van Rossumaf554a02007-08-16 23:48:43 +00008import threading
Georg Brandl24420152008-05-26 16:32:26 +00009import http.client
Christian Heimes255f53b2007-12-08 15:33:56 +000010import socket
11import os
Senthil Kumaranb3af08f2009-04-01 20:20:43 +000012import re
R. David Murray0548ce02009-10-26 08:24:14 +000013import io
14import contextlib
Benjamin Petersonee8712c2008-05-20 21:35:26 +000015from test import support
Skip Montanaro419abda2001-10-01 17:47:44 +000016
17alist = [{'astring': 'foo@bar.baz.spam',
18 'afloat': 7283.43,
Skip Montanaro3e7bba92001-10-19 16:06:52 +000019 'anint': 2**20,
Guido van Rossume2a383d2007-01-15 16:59:06 +000020 'ashortlong': 2,
Skip Montanaro419abda2001-10-01 17:47:44 +000021 'anotherlist': ['.zyx.41'],
Guido van Rossum54a40cb2007-08-27 22:27:41 +000022 'abase64': xmlrpclib.Binary(b"my dog has fleas"),
Guido van Rossume7ba4952007-06-06 23:52:48 +000023 'boolean': False,
Guido van Rossumef87d6e2007-05-02 19:09:54 +000024 'unicode': '\u4000\u6000\u8000',
25 'ukey\u4000': 'regular value',
Fred Drakeba613c32005-02-10 18:33:30 +000026 'datetime1': xmlrpclib.DateTime('20050210T11:41:23'),
27 'datetime2': xmlrpclib.DateTime(
Guido van Rossumcd16bf62007-06-13 18:07:49 +000028 (2005, 2, 10, 11, 41, 23, 0, 1, -1)),
Fred Drakeba613c32005-02-10 18:33:30 +000029 'datetime3': xmlrpclib.DateTime(
Guido van Rossumcd16bf62007-06-13 18:07:49 +000030 datetime.datetime(2005, 2, 10, 11, 41, 23)),
Skip Montanaro419abda2001-10-01 17:47:44 +000031 }]
32
33class XMLRPCTestCase(unittest.TestCase):
34
35 def test_dump_load(self):
Guido van Rossum54a40cb2007-08-27 22:27:41 +000036 dump = xmlrpclib.dumps((alist,))
37 load = xmlrpclib.loads(dump)
38 self.assertEquals(alist, load[0][0])
Skip Montanaro419abda2001-10-01 17:47:44 +000039
Fred Drakeba613c32005-02-10 18:33:30 +000040 def test_dump_bare_datetime(self):
Skip Montanaro174dd222005-05-14 20:54:16 +000041 # This checks that an unwrapped datetime.date object can be handled
42 # by the marshalling code. This can't be done via test_dump_load()
43 # since with use_datetime set to 1 the unmarshaller would create
44 # datetime objects for the 'datetime[123]' keys as well
Guido van Rossumcd16bf62007-06-13 18:07:49 +000045 dt = datetime.datetime(2005, 2, 10, 11, 41, 23)
Fred Drakeba613c32005-02-10 18:33:30 +000046 s = xmlrpclib.dumps((dt,))
Skip Montanaro174dd222005-05-14 20:54:16 +000047 (newdt,), m = xmlrpclib.loads(s, use_datetime=1)
48 self.assertEquals(newdt, dt)
Fred Drakeba613c32005-02-10 18:33:30 +000049 self.assertEquals(m, None)
50
Skip Montanaro174dd222005-05-14 20:54:16 +000051 (newdt,), m = xmlrpclib.loads(s, use_datetime=0)
52 self.assertEquals(newdt, xmlrpclib.DateTime('20050210T11:41:23'))
53
Christian Heimesdae2a892008-04-19 00:55:37 +000054 def test_datetime_before_1900(self):
Christian Heimes81ee3ef2008-05-04 22:42:01 +000055 # same as before but with a date before 1900
Christian Heimesdae2a892008-04-19 00:55:37 +000056 dt = datetime.datetime(1, 2, 10, 11, 41, 23)
57 s = xmlrpclib.dumps((dt,))
58 (newdt,), m = xmlrpclib.loads(s, use_datetime=1)
59 self.assertEquals(newdt, dt)
60 self.assertEquals(m, None)
61
62 (newdt,), m = xmlrpclib.loads(s, use_datetime=0)
63 self.assertEquals(newdt, xmlrpclib.DateTime('00010210T11:41:23'))
64
Christian Heimes05e8be12008-02-23 18:30:17 +000065 def test_cmp_datetime_DateTime(self):
66 now = datetime.datetime.now()
67 dt = xmlrpclib.DateTime(now.timetuple())
Benjamin Petersonc9c0f202009-06-30 23:06:06 +000068 self.assertTrue(dt == now)
69 self.assertTrue(now == dt)
Christian Heimes05e8be12008-02-23 18:30:17 +000070 then = now + datetime.timedelta(seconds=4)
Benjamin Petersonc9c0f202009-06-30 23:06:06 +000071 self.assertTrue(then >= dt)
72 self.assertTrue(dt < then)
Skip Montanaro174dd222005-05-14 20:54:16 +000073
Andrew M. Kuchlingbdb39012005-12-04 19:11:17 +000074 def test_bug_1164912 (self):
75 d = xmlrpclib.DateTime()
Tim Peters536cf992005-12-25 23:18:31 +000076 ((new_d,), dummy) = xmlrpclib.loads(xmlrpclib.dumps((d,),
Andrew M. Kuchlingbdb39012005-12-04 19:11:17 +000077 methodresponse=True))
Benjamin Petersonc9c0f202009-06-30 23:06:06 +000078 self.assertTrue(isinstance(new_d.value, str))
Andrew M. Kuchlingbdb39012005-12-04 19:11:17 +000079
80 # Check that the output of dumps() is still an 8-bit string
81 s = xmlrpclib.dumps((new_d,), methodresponse=True)
Benjamin Petersonc9c0f202009-06-30 23:06:06 +000082 self.assertTrue(isinstance(s, str))
Andrew M. Kuchlingbdb39012005-12-04 19:11:17 +000083
Thomas Wouters89f507f2006-12-13 04:49:30 +000084 def test_newstyle_class(self):
85 class T(object):
86 pass
87 t = T()
88 t.x = 100
89 t.y = "Hello"
90 ((t2,), dummy) = xmlrpclib.loads(xmlrpclib.dumps((t,)))
91 self.assertEquals(t2, t.__dict__)
92
Skip Montanaro3e7bba92001-10-19 16:06:52 +000093 def test_dump_big_long(self):
Guido van Rossume2a383d2007-01-15 16:59:06 +000094 self.assertRaises(OverflowError, xmlrpclib.dumps, (2**99,))
Skip Montanaro3e7bba92001-10-19 16:06:52 +000095
96 def test_dump_bad_dict(self):
97 self.assertRaises(TypeError, xmlrpclib.dumps, ({(1,2,3): 1},))
98
Guido van Rossumb5a755e2007-07-18 18:15:48 +000099 def test_dump_recursive_seq(self):
100 l = [1,2,3]
101 t = [3,4,5,l]
102 l.append(t)
103 self.assertRaises(TypeError, xmlrpclib.dumps, (l,))
104
105 def test_dump_recursive_dict(self):
106 d = {'1':1, '2':1}
107 t = {'3':3, 'd':d}
108 d['t'] = t
109 self.assertRaises(TypeError, xmlrpclib.dumps, (d,))
110
Skip Montanaro3e7bba92001-10-19 16:06:52 +0000111 def test_dump_big_int(self):
Christian Heimesa37d4c62007-12-04 23:02:19 +0000112 if sys.maxsize > 2**31-1:
Skip Montanaro3e7bba92001-10-19 16:06:52 +0000113 self.assertRaises(OverflowError, xmlrpclib.dumps,
Guido van Rossume2a383d2007-01-15 16:59:06 +0000114 (int(2**34),))
Skip Montanaro3e7bba92001-10-19 16:06:52 +0000115
Guido van Rossumb5a755e2007-07-18 18:15:48 +0000116 xmlrpclib.dumps((xmlrpclib.MAXINT, xmlrpclib.MININT))
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000117 self.assertRaises(OverflowError, xmlrpclib.dumps,
118 (xmlrpclib.MAXINT+1,))
119 self.assertRaises(OverflowError, xmlrpclib.dumps,
120 (xmlrpclib.MININT-1,))
Guido van Rossumb5a755e2007-07-18 18:15:48 +0000121
122 def dummy_write(s):
123 pass
124
125 m = xmlrpclib.Marshaller()
126 m.dump_int(xmlrpclib.MAXINT, dummy_write)
127 m.dump_int(xmlrpclib.MININT, dummy_write)
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000128 self.assertRaises(OverflowError, m.dump_int,
129 xmlrpclib.MAXINT+1, dummy_write)
130 self.assertRaises(OverflowError, m.dump_int,
131 xmlrpclib.MININT-1, dummy_write)
Guido van Rossumb5a755e2007-07-18 18:15:48 +0000132
Andrew M. Kuchling0b852032003-04-25 00:27:24 +0000133 def test_dump_none(self):
134 value = alist + [None]
135 arg1 = (alist + [None],)
136 strg = xmlrpclib.dumps(arg1, allow_none=True)
137 self.assertEquals(value,
138 xmlrpclib.loads(strg)[0][0])
139 self.assertRaises(TypeError, xmlrpclib.dumps, (arg1,))
140
Georg Brandlc8dcfb62009-02-13 10:50:01 +0000141 def test_get_host_info(self):
142 # see bug #3613, this raised a TypeError
143 transp = xmlrpc.client.Transport()
144 self.assertEquals(transp.get_host_info("user@host.tld"),
145 ('host.tld',
146 [('Authorization', 'Basic dXNlcg==')], {}))
147
148
Guido van Rossumb5a755e2007-07-18 18:15:48 +0000149class HelperTestCase(unittest.TestCase):
150 def test_escape(self):
151 self.assertEqual(xmlrpclib.escape("a&b"), "a&amp;b")
152 self.assertEqual(xmlrpclib.escape("a<b"), "a&lt;b")
153 self.assertEqual(xmlrpclib.escape("a>b"), "a&gt;b")
154
155class FaultTestCase(unittest.TestCase):
156 def test_repr(self):
157 f = xmlrpclib.Fault(42, 'Test Fault')
158 self.assertEqual(repr(f), "<Fault 42: 'Test Fault'>")
159 self.assertEqual(repr(f), str(f))
160
161 def test_dump_fault(self):
162 f = xmlrpclib.Fault(42, 'Test Fault')
163 s = xmlrpclib.dumps((f,))
164 (newf,), m = xmlrpclib.loads(s)
165 self.assertEquals(newf, {'faultCode': 42, 'faultString': 'Test Fault'})
166 self.assertEquals(m, None)
167
168 s = xmlrpclib.Marshaller().dumps(f)
169 self.assertRaises(xmlrpclib.Fault, xmlrpclib.loads, s)
170
Christian Heimeseda9e2b2007-12-08 22:32:47 +0000171 def test_dotted_attribute(self):
172 # this will raise AttirebuteError because code don't want us to use
173 # private methods
174 self.assertRaises(AttributeError,
Georg Brandl38eceaa2008-05-26 11:14:17 +0000175 xmlrpc.server.resolve_dotted_attribute, str, '__add')
Benjamin Petersonc9c0f202009-06-30 23:06:06 +0000176 self.assertTrue(xmlrpc.server.resolve_dotted_attribute(str, 'title'))
Guido van Rossumb5a755e2007-07-18 18:15:48 +0000177
178class DateTimeTestCase(unittest.TestCase):
179 def test_default(self):
180 t = xmlrpclib.DateTime()
181
182 def test_time(self):
183 d = 1181399930.036952
184 t = xmlrpclib.DateTime(d)
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000185 self.assertEqual(str(t),
186 time.strftime("%Y%m%dT%H:%M:%S", time.localtime(d)))
Guido van Rossumb5a755e2007-07-18 18:15:48 +0000187
188 def test_time_tuple(self):
189 d = (2007,6,9,10,38,50,5,160,0)
190 t = xmlrpclib.DateTime(d)
191 self.assertEqual(str(t), '20070609T10:38:50')
192
193 def test_time_struct(self):
194 d = time.localtime(1181399930.036952)
195 t = xmlrpclib.DateTime(d)
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000196 self.assertEqual(str(t), time.strftime("%Y%m%dT%H:%M:%S", d))
Guido van Rossumb5a755e2007-07-18 18:15:48 +0000197
198 def test_datetime_datetime(self):
199 d = datetime.datetime(2007,1,2,3,4,5)
200 t = xmlrpclib.DateTime(d)
201 self.assertEqual(str(t), '20070102T03:04:05')
202
Guido van Rossumb5a755e2007-07-18 18:15:48 +0000203 def test_repr(self):
204 d = datetime.datetime(2007,1,2,3,4,5)
205 t = xmlrpclib.DateTime(d)
206 val ="<DateTime '20070102T03:04:05' at %x>" % id(t)
207 self.assertEqual(repr(t), val)
208
209 def test_decode(self):
210 d = ' 20070908T07:11:13 '
211 t1 = xmlrpclib.DateTime()
212 t1.decode(d)
213 tref = xmlrpclib.DateTime(datetime.datetime(2007,9,8,7,11,13))
214 self.assertEqual(t1, tref)
215
216 t2 = xmlrpclib._datetime(d)
217 self.assertEqual(t1, tref)
218
219class BinaryTestCase(unittest.TestCase):
Guido van Rossum54a40cb2007-08-27 22:27:41 +0000220
221 # XXX What should str(Binary(b"\xff")) return? I'm chosing "\xff"
222 # for now (i.e. interpreting the binary data as Latin-1-encoded
223 # text). But this feels very unsatisfactory. Perhaps we should
224 # only define repr(), and return r"Binary(b'\xff')" instead?
225
Guido van Rossumb5a755e2007-07-18 18:15:48 +0000226 def test_default(self):
227 t = xmlrpclib.Binary()
228 self.assertEqual(str(t), '')
229
230 def test_string(self):
Guido van Rossum54a40cb2007-08-27 22:27:41 +0000231 d = b'\x01\x02\x03abc123\xff\xfe'
Guido van Rossumb5a755e2007-07-18 18:15:48 +0000232 t = xmlrpclib.Binary(d)
Guido van Rossum54a40cb2007-08-27 22:27:41 +0000233 self.assertEqual(str(t), str(d, "latin-1"))
Guido van Rossumb5a755e2007-07-18 18:15:48 +0000234
235 def test_decode(self):
Guido van Rossum54a40cb2007-08-27 22:27:41 +0000236 d = b'\x01\x02\x03abc123\xff\xfe'
Georg Brandlb54d8012009-06-04 09:11:51 +0000237 de = base64.encodebytes(d)
Guido van Rossumb5a755e2007-07-18 18:15:48 +0000238 t1 = xmlrpclib.Binary()
239 t1.decode(de)
Guido van Rossum54a40cb2007-08-27 22:27:41 +0000240 self.assertEqual(str(t1), str(d, "latin-1"))
Guido van Rossumb5a755e2007-07-18 18:15:48 +0000241
242 t2 = xmlrpclib._binary(de)
Guido van Rossum54a40cb2007-08-27 22:27:41 +0000243 self.assertEqual(str(t2), str(d, "latin-1"))
Guido van Rossumb5a755e2007-07-18 18:15:48 +0000244
245
Benjamin Petersone1cdfd72009-01-18 21:02:37 +0000246ADDR = PORT = URL = None
Skip Montanaro419abda2001-10-01 17:47:44 +0000247
Christian Heimesaf98da12008-01-27 15:18:18 +0000248# The evt is set twice. First when the server is ready to serve.
249# Second when the server has been shutdown. The user must clear
250# the event after it has been set the first time to catch the second set.
Kristján Valur Jónsson985fc6a2009-07-01 10:01:31 +0000251def http_server(evt, numrequests, requestHandler=None):
Guido van Rossumaf554a02007-08-16 23:48:43 +0000252 class TestInstanceClass:
253 def div(self, x, y):
Guido van Rossumaf554a02007-08-16 23:48:43 +0000254 return x // y
255
Christian Heimes255f53b2007-12-08 15:33:56 +0000256 def _methodHelp(self, name):
257 if name == 'div':
258 return 'This is the div function'
259
260 def my_function():
261 '''This is my function'''
262 return True
263
Georg Brandl38eceaa2008-05-26 11:14:17 +0000264 class MyXMLRPCServer(xmlrpc.server.SimpleXMLRPCServer):
Christian Heimesbbe741d2008-03-28 10:53:29 +0000265 def get_request(self):
266 # Ensure the socket is always non-blocking. On Linux, socket
267 # attributes are not inherited like they are on *BSD and Windows.
268 s, port = self.socket.accept()
269 s.setblocking(True)
270 return s, port
271
Kristján Valur Jónsson985fc6a2009-07-01 10:01:31 +0000272 if not requestHandler:
273 requestHandler = xmlrpc.server.SimpleXMLRPCRequestHandler
274 serv = MyXMLRPCServer(("localhost", 0), requestHandler,
Benjamin Petersone1cdfd72009-01-18 21:02:37 +0000275 logRequests=False, bind_and_activate=False)
Guido van Rossumaf554a02007-08-16 23:48:43 +0000276 try:
Guido van Rossumaf554a02007-08-16 23:48:43 +0000277 serv.server_bind()
Benjamin Petersone1cdfd72009-01-18 21:02:37 +0000278 global ADDR, PORT, URL
279 ADDR, PORT = serv.socket.getsockname()
280 #connect to IP address directly. This avoids socket.create_connection()
281 #trying to connect to to "localhost" using all address families, which
282 #causes slowdown e.g. on vista which supports AF_INET6. The server listens
283 #on AF_INET only.
284 URL = "http://%s:%d"%(ADDR, PORT)
Guido van Rossumaf554a02007-08-16 23:48:43 +0000285 serv.server_activate()
286 serv.register_introspection_functions()
287 serv.register_multicall_functions()
288 serv.register_function(pow)
289 serv.register_function(lambda x,y: x+y, 'add')
Christian Heimes255f53b2007-12-08 15:33:56 +0000290 serv.register_function(my_function)
Guido van Rossumaf554a02007-08-16 23:48:43 +0000291 serv.register_instance(TestInstanceClass())
Christian Heimesaf98da12008-01-27 15:18:18 +0000292 evt.set()
Guido van Rossumaf554a02007-08-16 23:48:43 +0000293
294 # handle up to 'numrequests' requests
295 while numrequests > 0:
296 serv.handle_request()
297 numrequests -= 1
298
299 except socket.timeout:
300 pass
301 finally:
302 serv.socket.close()
303 PORT = None
304 evt.set()
305
Kristján Valur Jónsson1f2a1ae2009-12-16 10:50:44 +0000306def http_multi_server(evt, numrequests, requestHandler=None):
307 class TestInstanceClass:
308 def div(self, x, y):
309 return x // y
310
311 def _methodHelp(self, name):
312 if name == 'div':
313 return 'This is the div function'
314
315 def my_function():
316 '''This is my function'''
317 return True
318
319 class MyXMLRPCServer(xmlrpc.server.MultiPathXMLRPCServer):
320 def get_request(self):
321 # Ensure the socket is always non-blocking. On Linux, socket
322 # attributes are not inherited like they are on *BSD and Windows.
323 s, port = self.socket.accept()
324 s.setblocking(True)
325 return s, port
326
327 if not requestHandler:
328 requestHandler = xmlrpc.server.SimpleXMLRPCRequestHandler
329 class MyRequestHandler(requestHandler):
330 rpc_paths = []
331
332 serv = MyXMLRPCServer(("localhost", 0), MyRequestHandler,
333 logRequests=False, bind_and_activate=False)
334 serv.socket.settimeout(3)
335 serv.server_bind()
336 try:
337 global ADDR, PORT, URL
338 ADDR, PORT = serv.socket.getsockname()
339 #connect to IP address directly. This avoids socket.create_connection()
340 #trying to connect to to "localhost" using all address families, which
341 #causes slowdown e.g. on vista which supports AF_INET6. The server listens
342 #on AF_INET only.
343 URL = "http://%s:%d"%(ADDR, PORT)
344 serv.server_activate()
345 paths = ["/foo", "/foo/bar"]
346 for path in paths:
347 d = serv.add_dispatcher(path, xmlrpc.server.SimpleXMLRPCDispatcher())
348 d.register_introspection_functions()
349 d.register_multicall_functions()
350 serv.get_dispatcher(paths[0]).register_function(pow)
351 serv.get_dispatcher(paths[1]).register_function(lambda x,y: x+y, 'add')
352 evt.set()
353
354 # handle up to 'numrequests' requests
355 while numrequests > 0:
356 serv.handle_request()
357 numrequests -= 1
358
359 except socket.timeout:
360 pass
361 finally:
362 serv.socket.close()
363 PORT = None
364 evt.set()
365
Christian Heimesaf98da12008-01-27 15:18:18 +0000366# This function prevents errors like:
367# <ProtocolError for localhost:57527/RPC2: 500 Internal Server Error>
368def is_unavailable_exception(e):
369 '''Returns True if the given ProtocolError is the product of a server-side
370 exception caused by the 'temporarily unavailable' response sometimes
371 given by operations on non-blocking sockets.'''
372
373 # sometimes we get a -1 error code and/or empty headers
Christian Heimes26855632008-01-27 23:50:43 +0000374 try:
375 if e.errcode == -1 or e.headers is None:
376 return True
377 exc_mess = e.headers.get('X-exception')
378 except AttributeError:
379 # Ignore socket.errors here.
380 exc_mess = str(e)
Guido van Rossumaf554a02007-08-16 23:48:43 +0000381
Christian Heimes26855632008-01-27 23:50:43 +0000382 if exc_mess and 'temporarily unavailable' in exc_mess.lower():
383 return True
Thomas Woutersed03b412007-08-28 21:37:11 +0000384
Kristján Valur Jónsson985fc6a2009-07-01 10:01:31 +0000385class BaseServerTestCase(unittest.TestCase):
386 requestHandler = None
Kristján Valur Jónssonf6087ca2009-07-12 22:45:18 +0000387 request_count = 1
Kristján Valur Jónsson1f2a1ae2009-12-16 10:50:44 +0000388 threadFunc = staticmethod(http_server)
Guido van Rossumaf554a02007-08-16 23:48:43 +0000389 def setUp(self):
Guido van Rossum61e21b52007-08-20 19:06:03 +0000390 # enable traceback reporting
Georg Brandl38eceaa2008-05-26 11:14:17 +0000391 xmlrpc.server.SimpleXMLRPCServer._send_traceback_header = True
Guido van Rossum61e21b52007-08-20 19:06:03 +0000392
Guido van Rossumaf554a02007-08-16 23:48:43 +0000393 self.evt = threading.Event()
Guido van Rossum61e21b52007-08-20 19:06:03 +0000394 # start server thread to handle requests
Kristján Valur Jónssonf6087ca2009-07-12 22:45:18 +0000395 serv_args = (self.evt, self.request_count, self.requestHandler)
Kristján Valur Jónsson1f2a1ae2009-12-16 10:50:44 +0000396 threading.Thread(target=self.threadFunc, args=serv_args).start()
Guido van Rossumaf554a02007-08-16 23:48:43 +0000397
Christian Heimesaf98da12008-01-27 15:18:18 +0000398 # wait for the server to be ready
399 self.evt.wait()
400 self.evt.clear()
Guido van Rossumaf554a02007-08-16 23:48:43 +0000401
402 def tearDown(self):
403 # wait on the server thread to terminate
Christian Heimes576bf652007-12-08 17:09:18 +0000404 self.evt.wait(4.0)
Benjamin Peterson1e637b72008-06-11 21:55:48 +0000405 if not self.evt.is_set():
Christian Heimes576bf652007-12-08 17:09:18 +0000406 self.evt.set()
407 stop_serving()
408 raise RuntimeError("timeout reached, test has failed")
Guido van Rossumaf554a02007-08-16 23:48:43 +0000409
Guido van Rossum61e21b52007-08-20 19:06:03 +0000410 # disable traceback reporting
Georg Brandl38eceaa2008-05-26 11:14:17 +0000411 xmlrpc.server.SimpleXMLRPCServer._send_traceback_header = False
Guido van Rossum61e21b52007-08-20 19:06:03 +0000412
Kristján Valur Jónsson985fc6a2009-07-01 10:01:31 +0000413class SimpleServerTestCase(BaseServerTestCase):
Guido van Rossumaf554a02007-08-16 23:48:43 +0000414 def test_simple1(self):
Guido van Rossum04110fb2007-08-24 16:32:05 +0000415 try:
Benjamin Petersone1cdfd72009-01-18 21:02:37 +0000416 p = xmlrpclib.ServerProxy(URL)
Guido van Rossum04110fb2007-08-24 16:32:05 +0000417 self.assertEqual(p.pow(6,8), 6**8)
Christian Heimes412dc9c2008-01-27 18:55:54 +0000418 except (xmlrpclib.ProtocolError, socket.error) as e:
419 # ignore failures due to non-blocking socket 'unavailable' errors
420 if not is_unavailable_exception(e):
421 # protocol error; provide additional information in test output
Christian Heimes26855632008-01-27 23:50:43 +0000422 self.fail("%s\n%s" % (e, getattr(e, "headers", "")))
Guido van Rossumaf554a02007-08-16 23:48:43 +0000423
Senthil Kumaranb3af08f2009-04-01 20:20:43 +0000424 def test_nonascii(self):
425 start_string = 'P\N{LATIN SMALL LETTER Y WITH CIRCUMFLEX}t'
426 end_string = 'h\N{LATIN SMALL LETTER O WITH HORN}n'
427 try:
428 p = xmlrpclib.ServerProxy(URL)
429 self.assertEqual(p.add(start_string, end_string),
430 start_string + end_string)
431 except (xmlrpclib.ProtocolError, socket.error) as e:
432 # ignore failures due to non-blocking socket 'unavailable' errors
433 if not is_unavailable_exception(e):
434 # protocol error; provide additional information in test output
435 self.fail("%s\n%s" % (e, getattr(e, "headers", "")))
436
Georg Brandlfceab5a2008-01-19 20:08:23 +0000437 # [ch] The test 404 is causing lots of false alarms.
438 def XXXtest_404(self):
Georg Brandl24420152008-05-26 16:32:26 +0000439 # send POST with http.client, it should return 404 header and
Christian Heimes255f53b2007-12-08 15:33:56 +0000440 # 'Not Found' message.
Benjamin Petersone1cdfd72009-01-18 21:02:37 +0000441 conn = httplib.client.HTTPConnection(ADDR, PORT)
Christian Heimes99d73f22007-12-08 16:13:06 +0000442 conn.request('POST', '/this-is-not-valid')
Christian Heimes255f53b2007-12-08 15:33:56 +0000443 response = conn.getresponse()
444 conn.close()
445
446 self.assertEqual(response.status, 404)
447 self.assertEqual(response.reason, 'Not Found')
448
Guido van Rossumaf554a02007-08-16 23:48:43 +0000449 def test_introspection1(self):
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000450 expected_methods = set(['pow', 'div', 'my_function', 'add',
451 'system.listMethods', 'system.methodHelp',
452 'system.methodSignature', 'system.multicall'])
Guido van Rossum04110fb2007-08-24 16:32:05 +0000453 try:
Benjamin Petersone1cdfd72009-01-18 21:02:37 +0000454 p = xmlrpclib.ServerProxy(URL)
Guido van Rossum04110fb2007-08-24 16:32:05 +0000455 meth = p.system.listMethods()
Guido van Rossum04110fb2007-08-24 16:32:05 +0000456 self.assertEqual(set(meth), expected_methods)
Christian Heimes412dc9c2008-01-27 18:55:54 +0000457 except (xmlrpclib.ProtocolError, socket.error) as e:
458 # ignore failures due to non-blocking socket 'unavailable' errors
459 if not is_unavailable_exception(e):
460 # protocol error; provide additional information in test output
Christian Heimes26855632008-01-27 23:50:43 +0000461 self.fail("%s\n%s" % (e, getattr(e, "headers", "")))
Christian Heimes412dc9c2008-01-27 18:55:54 +0000462
Guido van Rossumaf554a02007-08-16 23:48:43 +0000463
464 def test_introspection2(self):
Guido van Rossum04110fb2007-08-24 16:32:05 +0000465 try:
Christian Heimes255f53b2007-12-08 15:33:56 +0000466 # test _methodHelp()
Benjamin Petersone1cdfd72009-01-18 21:02:37 +0000467 p = xmlrpclib.ServerProxy(URL)
Guido van Rossum04110fb2007-08-24 16:32:05 +0000468 divhelp = p.system.methodHelp('div')
469 self.assertEqual(divhelp, 'This is the div function')
Christian Heimes412dc9c2008-01-27 18:55:54 +0000470 except (xmlrpclib.ProtocolError, socket.error) as e:
471 # ignore failures due to non-blocking socket 'unavailable' errors
472 if not is_unavailable_exception(e):
473 # protocol error; provide additional information in test output
Christian Heimes26855632008-01-27 23:50:43 +0000474 self.fail("%s\n%s" % (e, getattr(e, "headers", "")))
Guido van Rossumaf554a02007-08-16 23:48:43 +0000475
476 def test_introspection3(self):
Christian Heimes255f53b2007-12-08 15:33:56 +0000477 try:
478 # test native doc
Benjamin Petersone1cdfd72009-01-18 21:02:37 +0000479 p = xmlrpclib.ServerProxy(URL)
Christian Heimes255f53b2007-12-08 15:33:56 +0000480 myfunction = p.system.methodHelp('my_function')
481 self.assertEqual(myfunction, 'This is my function')
Christian Heimes412dc9c2008-01-27 18:55:54 +0000482 except (xmlrpclib.ProtocolError, socket.error) as e:
Christian Heimes255f53b2007-12-08 15:33:56 +0000483 # ignore failures due to non-blocking socket 'unavailable' errors
484 if not is_unavailable_exception(e):
485 # protocol error; provide additional information in test output
Christian Heimes26855632008-01-27 23:50:43 +0000486 self.fail("%s\n%s" % (e, getattr(e, "headers", "")))
Christian Heimes255f53b2007-12-08 15:33:56 +0000487
488 def test_introspection4(self):
Guido van Rossumaf554a02007-08-16 23:48:43 +0000489 # the SimpleXMLRPCServer doesn't support signatures, but
Thomas Woutersed03b412007-08-28 21:37:11 +0000490 # at least check that we can try making the call
Guido van Rossum04110fb2007-08-24 16:32:05 +0000491 try:
Benjamin Petersone1cdfd72009-01-18 21:02:37 +0000492 p = xmlrpclib.ServerProxy(URL)
Guido van Rossum04110fb2007-08-24 16:32:05 +0000493 divsig = p.system.methodSignature('div')
494 self.assertEqual(divsig, 'signatures not supported')
Christian Heimes412dc9c2008-01-27 18:55:54 +0000495 except (xmlrpclib.ProtocolError, socket.error) as e:
496 # ignore failures due to non-blocking socket 'unavailable' errors
497 if not is_unavailable_exception(e):
498 # protocol error; provide additional information in test output
Christian Heimes26855632008-01-27 23:50:43 +0000499 self.fail("%s\n%s" % (e, getattr(e, "headers", "")))
Guido van Rossumaf554a02007-08-16 23:48:43 +0000500
501 def test_multicall(self):
Guido van Rossum04110fb2007-08-24 16:32:05 +0000502 try:
Benjamin Petersone1cdfd72009-01-18 21:02:37 +0000503 p = xmlrpclib.ServerProxy(URL)
Guido van Rossum04110fb2007-08-24 16:32:05 +0000504 multicall = xmlrpclib.MultiCall(p)
505 multicall.add(2,3)
506 multicall.pow(6,8)
507 multicall.div(127,42)
508 add_result, pow_result, div_result = multicall()
509 self.assertEqual(add_result, 2+3)
510 self.assertEqual(pow_result, 6**8)
511 self.assertEqual(div_result, 127//42)
Christian Heimes412dc9c2008-01-27 18:55:54 +0000512 except (xmlrpclib.ProtocolError, socket.error) as e:
513 # ignore failures due to non-blocking socket 'unavailable' errors
514 if not is_unavailable_exception(e):
515 # protocol error; provide additional information in test output
Christian Heimes26855632008-01-27 23:50:43 +0000516 self.fail("%s\n%s" % (e, getattr(e, "headers", "")))
Guido van Rossumaf554a02007-08-16 23:48:43 +0000517
Christian Heimes255f53b2007-12-08 15:33:56 +0000518 def test_non_existing_multicall(self):
519 try:
Benjamin Petersone1cdfd72009-01-18 21:02:37 +0000520 p = xmlrpclib.ServerProxy(URL)
Christian Heimes255f53b2007-12-08 15:33:56 +0000521 multicall = xmlrpclib.MultiCall(p)
522 multicall.this_is_not_exists()
523 result = multicall()
524
525 # result.results contains;
Martin v. Löwis250ad612008-04-07 05:43:42 +0000526 # [{'faultCode': 1, 'faultString': '<class \'exceptions.Exception\'>:'
Christian Heimes255f53b2007-12-08 15:33:56 +0000527 # 'method "this_is_not_exists" is not supported'>}]
528
529 self.assertEqual(result.results[0]['faultCode'], 1)
530 self.assertEqual(result.results[0]['faultString'],
Martin v. Löwis250ad612008-04-07 05:43:42 +0000531 '<class \'Exception\'>:method "this_is_not_exists" '
Christian Heimes255f53b2007-12-08 15:33:56 +0000532 'is not supported')
Christian Heimes412dc9c2008-01-27 18:55:54 +0000533 except (xmlrpclib.ProtocolError, socket.error) as e:
Christian Heimes255f53b2007-12-08 15:33:56 +0000534 # ignore failures due to non-blocking socket 'unavailable' errors
535 if not is_unavailable_exception(e):
536 # protocol error; provide additional information in test output
Christian Heimes26855632008-01-27 23:50:43 +0000537 self.fail("%s\n%s" % (e, getattr(e, "headers", "")))
Christian Heimes255f53b2007-12-08 15:33:56 +0000538
Christian Heimesfe337bf2008-03-23 21:54:12 +0000539 def test_dotted_attribute(self):
540 # Raises an AttributeError because private methods are not allowed.
541 self.assertRaises(AttributeError,
Georg Brandl38eceaa2008-05-26 11:14:17 +0000542 xmlrpc.server.resolve_dotted_attribute, str, '__add')
Christian Heimesfe337bf2008-03-23 21:54:12 +0000543
Benjamin Petersonc9c0f202009-06-30 23:06:06 +0000544 self.assertTrue(xmlrpc.server.resolve_dotted_attribute(str, 'title'))
Christian Heimesfe337bf2008-03-23 21:54:12 +0000545 # Get the test to run faster by sending a request with test_simple1.
546 # This avoids waiting for the socket timeout.
547 self.test_simple1()
548
Kristján Valur Jónsson1f2a1ae2009-12-16 10:50:44 +0000549class MultiPathServerTestCase(BaseServerTestCase):
550 threadFunc = staticmethod(http_multi_server)
551 request_count = 2
552 def test_path1(self):
553 p = xmlrpclib.ServerProxy(URL+"/foo")
554 self.assertEqual(p.pow(6,8), 6**8)
555 self.assertRaises(xmlrpclib.Fault, p.add, 6, 8)
556 def test_path2(self):
557 p = xmlrpclib.ServerProxy(URL+"/foo/bar")
558 self.assertEqual(p.add(6,8), 6+8)
559 self.assertRaises(xmlrpclib.Fault, p.pow, 6, 8)
560
Kristján Valur Jónsson985fc6a2009-07-01 10:01:31 +0000561#A test case that verifies that a server using the HTTP/1.1 keep-alive mechanism
562#does indeed serve subsequent requests on the same connection
Kristján Valur Jónssonf6087ca2009-07-12 22:45:18 +0000563class BaseKeepaliveServerTestCase(BaseServerTestCase):
Kristján Valur Jónsson985fc6a2009-07-01 10:01:31 +0000564 #a request handler that supports keep-alive and logs requests into a
565 #class variable
566 class RequestHandler(xmlrpc.server.SimpleXMLRPCRequestHandler):
567 parentClass = xmlrpc.server.SimpleXMLRPCRequestHandler
568 protocol_version = 'HTTP/1.1'
569 myRequests = []
570 def handle(self):
571 self.myRequests.append([])
Kristján Valur Jónssonf6087ca2009-07-12 22:45:18 +0000572 self.reqidx = len(self.myRequests)-1
Kristján Valur Jónsson985fc6a2009-07-01 10:01:31 +0000573 return self.parentClass.handle(self)
574 def handle_one_request(self):
575 result = self.parentClass.handle_one_request(self)
Kristján Valur Jónssonf6087ca2009-07-12 22:45:18 +0000576 self.myRequests[self.reqidx].append(self.raw_requestline)
Kristján Valur Jónsson985fc6a2009-07-01 10:01:31 +0000577 return result
578
579 requestHandler = RequestHandler
580 def setUp(self):
581 #clear request log
582 self.RequestHandler.myRequests = []
583 return BaseServerTestCase.setUp(self)
584
Kristján Valur Jónssonf6087ca2009-07-12 22:45:18 +0000585#A test case that verifies that a server using the HTTP/1.1 keep-alive mechanism
586#does indeed serve subsequent requests on the same connection
587class KeepaliveServerTestCase1(BaseKeepaliveServerTestCase):
Kristján Valur Jónsson985fc6a2009-07-01 10:01:31 +0000588 def test_two(self):
589 p = xmlrpclib.ServerProxy(URL)
Kristján Valur Jónsson1bbb19a2009-07-11 21:57:16 +0000590 #do three requests.
Kristján Valur Jónsson985fc6a2009-07-01 10:01:31 +0000591 self.assertEqual(p.pow(6,8), 6**8)
592 self.assertEqual(p.pow(6,8), 6**8)
Kristján Valur Jónsson1bbb19a2009-07-11 21:57:16 +0000593 self.assertEqual(p.pow(6,8), 6**8)
594
595 #they should have all been handled by a single request handler
Kristján Valur Jónsson985fc6a2009-07-01 10:01:31 +0000596 self.assertEqual(len(self.RequestHandler.myRequests), 1)
Kristján Valur Jónsson1bbb19a2009-07-11 21:57:16 +0000597
598 #check that we did at least two (the third may be pending append
599 #due to thread scheduling)
600 self.assertGreaterEqual(len(self.RequestHandler.myRequests[-1]), 2)
Kristján Valur Jónsson985fc6a2009-07-01 10:01:31 +0000601
Kristján Valur Jónssonf6087ca2009-07-12 22:45:18 +0000602#test special attribute access on the serverproxy, through the __call__
603#function.
604class KeepaliveServerTestCase2(BaseKeepaliveServerTestCase):
605 #ask for two keepalive requests to be handled.
606 request_count=2
607
608 def test_close(self):
609 p = xmlrpclib.ServerProxy(URL)
610 #do some requests with close.
611 self.assertEqual(p.pow(6,8), 6**8)
612 self.assertEqual(p.pow(6,8), 6**8)
613 self.assertEqual(p.pow(6,8), 6**8)
614 p("close")() #this should trigger a new keep-alive request
615 self.assertEqual(p.pow(6,8), 6**8)
616 self.assertEqual(p.pow(6,8), 6**8)
617 self.assertEqual(p.pow(6,8), 6**8)
618
619 #they should have all been two request handlers, each having logged at least
620 #two complete requests
621 self.assertEqual(len(self.RequestHandler.myRequests), 2)
622 self.assertGreaterEqual(len(self.RequestHandler.myRequests[-1]), 2)
623 self.assertGreaterEqual(len(self.RequestHandler.myRequests[-2]), 2)
624
625 def test_transport(self):
626 p = xmlrpclib.ServerProxy(URL)
627 #do some requests with close.
628 self.assertEqual(p.pow(6,8), 6**8)
629 p("transport").close() #same as above, really.
630 self.assertEqual(p.pow(6,8), 6**8)
631 self.assertEqual(len(self.RequestHandler.myRequests), 2)
632
Kristján Valur Jónsson985fc6a2009-07-01 10:01:31 +0000633#A test case that verifies that gzip encoding works in both directions
634#(for a request and the response)
635class GzipServerTestCase(BaseServerTestCase):
636 #a request handler that supports keep-alive and logs requests into a
637 #class variable
638 class RequestHandler(xmlrpc.server.SimpleXMLRPCRequestHandler):
639 parentClass = xmlrpc.server.SimpleXMLRPCRequestHandler
640 protocol_version = 'HTTP/1.1'
641
642 def do_POST(self):
643 #store content of last request in class
644 self.__class__.content_length = int(self.headers["content-length"])
645 return self.parentClass.do_POST(self)
646 requestHandler = RequestHandler
647
648 class Transport(xmlrpclib.Transport):
649 #custom transport, stores the response length for our perusal
650 fake_gzip = False
651 def parse_response(self, response):
652 self.response_length=int(response.getheader("content-length", 0))
653 return xmlrpclib.Transport.parse_response(self, response)
654
655 def send_content(self, connection, body):
656 if self.fake_gzip:
657 #add a lone gzip header to induce decode error remotely
658 connection.putheader("Content-Encoding", "gzip")
659 return xmlrpclib.Transport.send_content(self, connection, body)
660
661 def test_gzip_request(self):
662 t = self.Transport()
663 t.encode_threshold = None
664 p = xmlrpclib.ServerProxy(URL, transport=t)
665 self.assertEqual(p.pow(6,8), 6**8)
666 a = self.RequestHandler.content_length
667 t.encode_threshold = 0 #turn on request encoding
668 self.assertEqual(p.pow(6,8), 6**8)
669 b = self.RequestHandler.content_length
670 self.assertTrue(a>b)
671
672 def test_bad_gzip_request(self):
673 t = self.Transport()
674 t.encode_threshold = None
675 t.fake_gzip = True
676 p = xmlrpclib.ServerProxy(URL, transport=t)
677 cm = self.assertRaisesRegexp(xmlrpclib.ProtocolError,
678 re.compile(r"\b400\b"))
679 with cm:
680 p.pow(6, 8)
681
682 def test_gsip_response(self):
683 t = self.Transport()
684 p = xmlrpclib.ServerProxy(URL, transport=t)
685 old = self.requestHandler.encode_threshold
686 self.requestHandler.encode_threshold = None #no encoding
687 self.assertEqual(p.pow(6,8), 6**8)
688 a = t.response_length
689 self.requestHandler.encode_threshold = 0 #always encode
690 self.assertEqual(p.pow(6,8), 6**8)
691 b = t.response_length
692 self.requestHandler.encode_threshold = old
693 self.assertTrue(a>b)
694
695#Test special attributes of the ServerProxy object
696class ServerProxyTestCase(unittest.TestCase):
697 def test_close(self):
698 p = xmlrpclib.ServerProxy(URL)
699 self.assertEqual(p('close')(), None)
700
701 def test_transport(self):
702 t = xmlrpclib.Transport()
703 p = xmlrpclib.ServerProxy(URL, transport=t)
704 self.assertEqual(p('transport'), t)
705
Guido van Rossum61e21b52007-08-20 19:06:03 +0000706# This is a contrived way to make a failure occur on the server side
707# in order to test the _send_traceback_header flag on the server
Barry Warsaw820c1202008-06-12 04:06:45 +0000708class FailingMessageClass(http.client.HTTPMessage):
709 def get(self, key, failobj=None):
Guido van Rossum61e21b52007-08-20 19:06:03 +0000710 key = key.lower()
711 if key == 'content-length':
712 return 'I am broken'
Barry Warsaw820c1202008-06-12 04:06:45 +0000713 return super().get(key, failobj)
Guido van Rossum61e21b52007-08-20 19:06:03 +0000714
715
716class FailingServerTestCase(unittest.TestCase):
717 def setUp(self):
718 self.evt = threading.Event()
719 # start server thread to handle requests
Guido van Rossum15c97462007-11-02 16:10:06 +0000720 serv_args = (self.evt, 1)
Guido van Rossum61e21b52007-08-20 19:06:03 +0000721 threading.Thread(target=http_server, args=serv_args).start()
722
Christian Heimesaf98da12008-01-27 15:18:18 +0000723 # wait for the server to be ready
724 self.evt.wait()
725 self.evt.clear()
Guido van Rossum61e21b52007-08-20 19:06:03 +0000726
727 def tearDown(self):
728 # wait on the server thread to terminate
729 self.evt.wait()
730 # reset flag
Georg Brandl38eceaa2008-05-26 11:14:17 +0000731 xmlrpc.server.SimpleXMLRPCServer._send_traceback_header = False
Guido van Rossum61e21b52007-08-20 19:06:03 +0000732 # reset message class
Barry Warsaw820c1202008-06-12 04:06:45 +0000733 default_class = http.client.HTTPMessage
734 xmlrpc.server.SimpleXMLRPCRequestHandler.MessageClass = default_class
Guido van Rossum61e21b52007-08-20 19:06:03 +0000735
736 def test_basic(self):
737 # check that flag is false by default
Georg Brandl38eceaa2008-05-26 11:14:17 +0000738 flagval = xmlrpc.server.SimpleXMLRPCServer._send_traceback_header
Guido van Rossum61e21b52007-08-20 19:06:03 +0000739 self.assertEqual(flagval, False)
740
Guido van Rossum04110fb2007-08-24 16:32:05 +0000741 # enable traceback reporting
Georg Brandl38eceaa2008-05-26 11:14:17 +0000742 xmlrpc.server.SimpleXMLRPCServer._send_traceback_header = True
Guido van Rossum04110fb2007-08-24 16:32:05 +0000743
744 # test a call that shouldn't fail just as a smoke test
745 try:
Benjamin Petersone1cdfd72009-01-18 21:02:37 +0000746 p = xmlrpclib.ServerProxy(URL)
Guido van Rossum04110fb2007-08-24 16:32:05 +0000747 self.assertEqual(p.pow(6,8), 6**8)
Christian Heimes412dc9c2008-01-27 18:55:54 +0000748 except (xmlrpclib.ProtocolError, socket.error) as e:
749 # ignore failures due to non-blocking socket 'unavailable' errors
750 if not is_unavailable_exception(e):
751 # protocol error; provide additional information in test output
Christian Heimes26855632008-01-27 23:50:43 +0000752 self.fail("%s\n%s" % (e, getattr(e, "headers", "")))
Guido van Rossum61e21b52007-08-20 19:06:03 +0000753
754 def test_fail_no_info(self):
755 # use the broken message class
Georg Brandl38eceaa2008-05-26 11:14:17 +0000756 xmlrpc.server.SimpleXMLRPCRequestHandler.MessageClass = FailingMessageClass
Guido van Rossum61e21b52007-08-20 19:06:03 +0000757
758 try:
Benjamin Petersone1cdfd72009-01-18 21:02:37 +0000759 p = xmlrpclib.ServerProxy(URL)
Guido van Rossum61e21b52007-08-20 19:06:03 +0000760 p.pow(6,8)
Christian Heimes412dc9c2008-01-27 18:55:54 +0000761 except (xmlrpclib.ProtocolError, socket.error) as e:
762 # ignore failures due to non-blocking socket 'unavailable' errors
Christian Heimes26855632008-01-27 23:50:43 +0000763 if not is_unavailable_exception(e) and hasattr(e, "headers"):
Christian Heimes412dc9c2008-01-27 18:55:54 +0000764 # The two server-side error headers shouldn't be sent back in this case
765 self.assertTrue(e.headers.get("X-exception") is None)
766 self.assertTrue(e.headers.get("X-traceback") is None)
Guido van Rossum61e21b52007-08-20 19:06:03 +0000767 else:
768 self.fail('ProtocolError not raised')
769
770 def test_fail_with_info(self):
771 # use the broken message class
Georg Brandl38eceaa2008-05-26 11:14:17 +0000772 xmlrpc.server.SimpleXMLRPCRequestHandler.MessageClass = FailingMessageClass
Guido van Rossum61e21b52007-08-20 19:06:03 +0000773
774 # Check that errors in the server send back exception/traceback
775 # info when flag is set
Georg Brandl38eceaa2008-05-26 11:14:17 +0000776 xmlrpc.server.SimpleXMLRPCServer._send_traceback_header = True
Guido van Rossum61e21b52007-08-20 19:06:03 +0000777
778 try:
Benjamin Petersone1cdfd72009-01-18 21:02:37 +0000779 p = xmlrpclib.ServerProxy(URL)
Guido van Rossum61e21b52007-08-20 19:06:03 +0000780 p.pow(6,8)
Christian Heimes412dc9c2008-01-27 18:55:54 +0000781 except (xmlrpclib.ProtocolError, socket.error) as e:
782 # ignore failures due to non-blocking socket 'unavailable' errors
Christian Heimes26855632008-01-27 23:50:43 +0000783 if not is_unavailable_exception(e) and hasattr(e, "headers"):
Christian Heimes412dc9c2008-01-27 18:55:54 +0000784 # We should get error info in the response
785 expected_err = "invalid literal for int() with base 10: 'I am broken'"
Barry Warsaw820c1202008-06-12 04:06:45 +0000786 self.assertEqual(e.headers.get("X-exception"), expected_err)
787 self.assertTrue(e.headers.get("X-traceback") is not None)
Guido van Rossum61e21b52007-08-20 19:06:03 +0000788 else:
789 self.fail('ProtocolError not raised')
790
R. David Murray0548ce02009-10-26 08:24:14 +0000791
792@contextlib.contextmanager
793def captured_stdout(encoding='utf-8'):
794 """A variation on support.captured_stdout() which gives a text stream
795 having a `buffer` attribute.
796 """
797 import io
798 orig_stdout = sys.stdout
799 sys.stdout = io.TextIOWrapper(io.BytesIO(), encoding=encoding)
800 try:
801 yield sys.stdout
802 finally:
803 sys.stdout = orig_stdout
804
805
Christian Heimes255f53b2007-12-08 15:33:56 +0000806class CGIHandlerTestCase(unittest.TestCase):
807 def setUp(self):
Georg Brandl38eceaa2008-05-26 11:14:17 +0000808 self.cgi = xmlrpc.server.CGIXMLRPCRequestHandler()
Christian Heimes255f53b2007-12-08 15:33:56 +0000809
810 def tearDown(self):
811 self.cgi = None
812
813 def test_cgi_get(self):
Walter Dörwaldb525e182009-04-26 21:39:21 +0000814 with support.EnvironmentVarGuard() as env:
Walter Dörwald155374d2009-05-01 19:58:58 +0000815 env['REQUEST_METHOD'] = 'GET'
Walter Dörwaldb525e182009-04-26 21:39:21 +0000816 # if the method is GET and no request_text is given, it runs handle_get
817 # get sysout output
R. David Murray0548ce02009-10-26 08:24:14 +0000818 with captured_stdout(encoding=self.cgi.encoding) as data_out:
Nick Coghlan6ead5522009-10-18 13:19:33 +0000819 self.cgi.handle_request()
Christian Heimes255f53b2007-12-08 15:33:56 +0000820
Walter Dörwaldb525e182009-04-26 21:39:21 +0000821 # parse Status header
Nick Coghlan6ead5522009-10-18 13:19:33 +0000822 data_out.seek(0)
823 handle = data_out.read()
Walter Dörwaldb525e182009-04-26 21:39:21 +0000824 status = handle.split()[1]
825 message = ' '.join(handle.split()[2:4])
Christian Heimes255f53b2007-12-08 15:33:56 +0000826
Walter Dörwaldb525e182009-04-26 21:39:21 +0000827 self.assertEqual(status, '400')
828 self.assertEqual(message, 'Bad Request')
Christian Heimes255f53b2007-12-08 15:33:56 +0000829
Christian Heimes255f53b2007-12-08 15:33:56 +0000830
831 def test_cgi_xmlrpc_response(self):
832 data = """<?xml version='1.0'?>
Nick Coghlan6ead5522009-10-18 13:19:33 +0000833 <methodCall>
834 <methodName>test_method</methodName>
835 <params>
836 <param>
837 <value><string>foo</string></value>
838 </param>
839 <param>
840 <value><string>bar</string></value>
841 </param>
842 </params>
843 </methodCall>
844 """
Christian Heimes255f53b2007-12-08 15:33:56 +0000845
Nick Coghlan6ead5522009-10-18 13:19:33 +0000846 with support.EnvironmentVarGuard() as env, \
R. David Murray0548ce02009-10-26 08:24:14 +0000847 captured_stdout(encoding=self.cgi.encoding) as data_out, \
Nick Coghlan6ead5522009-10-18 13:19:33 +0000848 support.captured_stdin() as data_in:
849 data_in.write(data)
850 data_in.seek(0)
Walter Dörwald155374d2009-05-01 19:58:58 +0000851 env['CONTENT_LENGTH'] = str(len(data))
Georg Brandlc7485062009-04-01 15:53:15 +0000852 self.cgi.handle_request()
Nick Coghlan6ead5522009-10-18 13:19:33 +0000853 data_out.seek(0)
Christian Heimes255f53b2007-12-08 15:33:56 +0000854
855 # will respond exception, if so, our goal is achieved ;)
Nick Coghlan6ead5522009-10-18 13:19:33 +0000856 handle = data_out.read()
Christian Heimes255f53b2007-12-08 15:33:56 +0000857
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000858 # start with 44th char so as not to get http header, we just
859 # need only xml
Christian Heimes255f53b2007-12-08 15:33:56 +0000860 self.assertRaises(xmlrpclib.Fault, xmlrpclib.loads, handle[44:])
861
Senthil Kumaranb3af08f2009-04-01 20:20:43 +0000862 # Also test the content-length returned by handle_request
863 # Using the same test method inorder to avoid all the datapassing
864 # boilerplate code.
865 # Test for bug: http://bugs.python.org/issue5040
866
867 content = handle[handle.find("<?xml"):]
868
869 self.assertEquals(
870 int(re.search('Content-Length: (\d+)', handle).group(1)),
871 len(content))
872
Senthil Kumaranb3af08f2009-04-01 20:20:43 +0000873
Antoine Pitroue03866f2009-10-30 17:58:27 +0000874@support.reap_threads
Guido van Rossumaf554a02007-08-16 23:48:43 +0000875def test_main():
876 xmlrpc_tests = [XMLRPCTestCase, HelperTestCase, DateTimeTestCase,
877 BinaryTestCase, FaultTestCase]
Benjamin Petersone1cdfd72009-01-18 21:02:37 +0000878 xmlrpc_tests.append(SimpleServerTestCase)
Kristján Valur Jónssonf6087ca2009-07-12 22:45:18 +0000879 xmlrpc_tests.append(KeepaliveServerTestCase1)
880 xmlrpc_tests.append(KeepaliveServerTestCase2)
Kristján Valur Jónssonaefde242009-07-19 22:29:24 +0000881 try:
882 import gzip
883 xmlrpc_tests.append(GzipServerTestCase)
884 except ImportError:
885 pass #gzip not supported in this build
Kristján Valur Jónsson1f2a1ae2009-12-16 10:50:44 +0000886 xmlrpc_tests.append(MultiPathServerTestCase)
Kristján Valur Jónsson985fc6a2009-07-01 10:01:31 +0000887 xmlrpc_tests.append(ServerProxyTestCase)
Benjamin Petersone1cdfd72009-01-18 21:02:37 +0000888 xmlrpc_tests.append(FailingServerTestCase)
889 xmlrpc_tests.append(CGIHandlerTestCase)
Guido van Rossumaf554a02007-08-16 23:48:43 +0000890
Benjamin Petersonee8712c2008-05-20 21:35:26 +0000891 support.run_unittest(*xmlrpc_tests)
Skip Montanaro419abda2001-10-01 17:47:44 +0000892
893if __name__ == "__main__":
894 test_main()