92e0a58793a637c232ad1410e764be14aefde5bc
1 #
2 # Copyright (c) 2003 Richard Jones, rjones@ekit-inc.com
3 # This module is free software, and you may redistribute it and/or modify
4 # under the same terms as Python, so long as this copyright message and
5 # disclaimer are retained in their original form.
6 #
7 # This module is distributed in the hope that it will be useful,
8 # but WITHOUT ANY WARRANTY; without even the implied warranty of
9 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.
10 #
11 # $Id: test_cgi.py,v 1.16 2003-05-09 01:47:50 richard Exp $
13 import unittest, os, shutil, errno, sys, difflib, cgi, re
15 from roundup.cgi import client
16 from roundup import init, instance, password, hyperdb, date
18 class FileUpload:
19 def __init__(self, content, filename):
20 self.content = content
21 self.filename = filename
23 def makeForm(args):
24 form = cgi.FieldStorage()
25 for k,v in args.items():
26 if type(v) is type([]):
27 [form.list.append(cgi.MiniFieldStorage(k, x)) for x in v]
28 elif isinstance(v, FileUpload):
29 x = cgi.MiniFieldStorage(k, v.content)
30 x.filename = v.filename
31 form.list.append(x)
32 else:
33 form.list.append(cgi.MiniFieldStorage(k, v))
34 return form
36 class config:
37 TRACKER_NAME = 'testing testing'
38 TRACKER_WEB = 'http://testing.testing/'
40 class FormTestCase(unittest.TestCase):
41 def setUp(self):
42 self.dirname = '_test_cgi_form'
43 try:
44 shutil.rmtree(self.dirname)
45 except OSError, error:
46 if error.errno not in (errno.ENOENT, errno.ESRCH): raise
47 # create the instance
48 init.install(self.dirname, 'templates/classic')
49 init.write_select_db(self.dirname, 'anydbm')
50 init.initialise(self.dirname, 'sekrit')
51 # check we can load the package
52 self.instance = instance.open(self.dirname)
53 # and open the database
54 self.db = self.instance.open('admin')
55 self.db.user.create(username='Chef', address='chef@bork.bork.bork',
56 realname='Bork, Chef', roles='User')
57 self.db.user.create(username='mary', address='mary@test',
58 roles='User', realname='Contrary, Mary')
60 test = self.instance.dbinit.Class(self.db, "test",
61 string=hyperdb.String(), number=hyperdb.Number(),
62 boolean=hyperdb.Boolean(), link=hyperdb.Link('test'),
63 multilink=hyperdb.Multilink('test'), date=hyperdb.Date(),
64 interval=hyperdb.Interval())
66 # compile the labels re
67 classes = '|'.join(self.db.classes.keys())
68 self.FV_SPECIAL = re.compile(client.Client.FV_LABELS%classes,
69 re.VERBOSE)
71 def parseForm(self, form, classname='test', nodeid=None):
72 cl = client.Client(self.instance, None, {'PATH_INFO':'/'},
73 makeForm(form))
74 cl.classname = classname
75 cl.nodeid = nodeid
76 cl.db = self.db
77 return cl.parsePropsFromForm()
79 def tearDown(self):
80 self.db.close()
81 try:
82 shutil.rmtree(self.dirname)
83 except OSError, error:
84 if error.errno not in (errno.ENOENT, errno.ESRCH): raise
86 #
87 # form label extraction
88 #
89 def tl(self, s, c, i, a, p):
90 m = self.FV_SPECIAL.match(s)
91 self.assertNotEqual(m, None)
92 d = m.groupdict()
93 self.assertEqual(d['classname'], c)
94 self.assertEqual(d['id'], i)
95 for action in 'required add remove link note file'.split():
96 if a == action:
97 self.assertNotEqual(d[action], None)
98 else:
99 self.assertEqual(d[action], None)
100 self.assertEqual(d['propname'], p)
102 def testLabelMatching(self):
103 self.tl('<propname>', None, None, None, '<propname>')
104 self.tl(':required', None, None, 'required', None)
105 self.tl(':confirm:<propname>', None, None, 'confirm', '<propname>')
106 self.tl(':add:<propname>', None, None, 'add', '<propname>')
107 self.tl(':remove:<propname>', None, None, 'remove', '<propname>')
108 self.tl(':link:<propname>', None, None, 'link', '<propname>')
109 self.tl('test1:<prop>', 'test', '1', None, '<prop>')
110 self.tl('test1:required', 'test', '1', 'required', None)
111 self.tl('test1:add:<prop>', 'test', '1', 'add', '<prop>')
112 self.tl('test1:remove:<prop>', 'test', '1', 'remove', '<prop>')
113 self.tl('test1:link:<prop>', 'test', '1', 'link', '<prop>')
114 self.tl('test1:confirm:<prop>', 'test', '1', 'confirm', '<prop>')
115 self.tl('test-1:<prop>', 'test', '-1', None, '<prop>')
116 self.tl('test-1:required', 'test', '-1', 'required', None)
117 self.tl('test-1:add:<prop>', 'test', '-1', 'add', '<prop>')
118 self.tl('test-1:remove:<prop>', 'test', '-1', 'remove', '<prop>')
119 self.tl('test-1:link:<prop>', 'test', '-1', 'link', '<prop>')
120 self.tl('test-1:confirm:<prop>', 'test', '-1', 'confirm', '<prop>')
121 self.tl(':note', None, None, 'note', None)
122 self.tl(':file', None, None, 'file', None)
124 #
125 # Empty form
126 #
127 def testNothing(self):
128 self.assertEqual(self.parseForm({}), ({('test', None): {}}, []))
130 def testNothingWithRequired(self):
131 self.assertRaises(ValueError, self.parseForm, {':required': 'string'})
132 self.assertRaises(ValueError, self.parseForm,
133 {':required': 'title,status', 'status':'1'}, 'issue')
134 self.assertRaises(ValueError, self.parseForm,
135 {':required': ['title','status'], 'status':'1'}, 'issue')
136 self.assertRaises(ValueError, self.parseForm,
137 {':required': 'status', 'status':''}, 'issue')
138 self.assertRaises(ValueError, self.parseForm,
139 {':required': 'nosy', 'nosy':''}, 'issue')
141 #
142 # Nonexistant edit
143 #
144 def testEditNonexistant(self):
145 self.assertRaises(IndexError, self.parseForm, {'boolean': ''},
146 'test', '1')
148 #
149 # String
150 #
151 def testEmptyString(self):
152 self.assertEqual(self.parseForm({'string': ''}),
153 ({('test', None): {}}, []))
154 self.assertEqual(self.parseForm({'string': ' '}),
155 ({('test', None): {}}, []))
156 self.assertRaises(ValueError, self.parseForm, {'string': ['', '']})
158 def testSetString(self):
159 self.assertEqual(self.parseForm({'string': 'foo'}),
160 ({('test', None): {'string': 'foo'}}, []))
161 self.assertEqual(self.parseForm({'string': 'a\r\nb\r\n'}),
162 ({('test', None): {'string': 'a\nb'}}, []))
163 nodeid = self.db.issue.create(title='foo')
164 self.assertEqual(self.parseForm({'title': 'foo'}, 'issue', nodeid),
165 ({('issue', nodeid): {}}, []))
167 def testEmptyStringSet(self):
168 nodeid = self.db.issue.create(title='foo')
169 self.assertEqual(self.parseForm({'title': ''}, 'issue', nodeid),
170 ({('issue', nodeid): {'title': None}}, []))
171 nodeid = self.db.issue.create(title='foo')
172 self.assertEqual(self.parseForm({'title': ' '}, 'issue', nodeid),
173 ({('issue', nodeid): {'title': None}}, []))
175 def testFileUpload(self):
176 file = FileUpload('foo', 'foo.txt')
177 self.assertEqual(self.parseForm({'content': file}, 'file'),
178 ({('file', None): {'content': 'foo', 'name': 'foo.txt',
179 'type': 'text/plain'}}, []))
181 #
182 # Link
183 #
184 def testEmptyLink(self):
185 self.assertEqual(self.parseForm({'link': ''}),
186 ({('test', None): {}}, []))
187 self.assertEqual(self.parseForm({'link': ' '}),
188 ({('test', None): {}}, []))
189 self.assertRaises(ValueError, self.parseForm, {'link': ['', '']})
190 self.assertEqual(self.parseForm({'link': '-1'}),
191 ({('test', None): {}}, []))
193 def testSetLink(self):
194 self.assertEqual(self.parseForm({'status': 'unread'}, 'issue'),
195 ({('issue', None): {'status': '1'}}, []))
196 self.assertEqual(self.parseForm({'status': '1'}, 'issue'),
197 ({('issue', None): {'status': '1'}}, []))
198 nodeid = self.db.issue.create(status='unread')
199 self.assertEqual(self.parseForm({'status': 'unread'}, 'issue', nodeid),
200 ({('issue', nodeid): {}}, []))
202 def testUnsetLink(self):
203 nodeid = self.db.issue.create(status='unread')
204 self.assertEqual(self.parseForm({'status': '-1'}, 'issue', nodeid),
205 ({('issue', nodeid): {'status': None}}, []))
207 def testInvalidLinkValue(self):
208 # XXX This is not the current behaviour - should we enforce this?
209 # self.assertRaises(IndexError, self.parseForm,
210 # {'status': '4'}))
211 self.assertRaises(ValueError, self.parseForm, {'link': 'frozzle'})
212 self.assertRaises(ValueError, self.parseForm, {'status': 'frozzle'},
213 'issue')
215 #
216 # Multilink
217 #
218 def testEmptyMultilink(self):
219 self.assertEqual(self.parseForm({'nosy': ''}),
220 ({('test', None): {}}, []))
221 self.assertEqual(self.parseForm({'nosy': ' '}),
222 ({('test', None): {}}, []))
224 def testSetMultilink(self):
225 self.assertEqual(self.parseForm({'nosy': '1'}, 'issue'),
226 ({('issue', None): {'nosy': ['1']}}, []))
227 self.assertEqual(self.parseForm({'nosy': 'admin'}, 'issue'),
228 ({('issue', None): {'nosy': ['1']}}, []))
229 self.assertEqual(self.parseForm({'nosy': ['1','2']}, 'issue'),
230 ({('issue', None): {'nosy': ['1','2']}}, []))
231 self.assertEqual(self.parseForm({'nosy': '1,2'}, 'issue'),
232 ({('issue', None): {'nosy': ['1','2']}}, []))
233 self.assertEqual(self.parseForm({'nosy': 'admin,2'}, 'issue'),
234 ({('issue', None): {'nosy': ['1','2']}}, []))
236 def testMixedMultilink(self):
237 form = cgi.FieldStorage()
238 form.list.append(cgi.MiniFieldStorage('nosy', '1,2'))
239 form.list.append(cgi.MiniFieldStorage('nosy', '3'))
240 cl = client.Client(self.instance, None, {'PATH_INFO':'/'}, form)
241 cl.classname = 'issue'
242 cl.nodeid = None
243 cl.db = self.db
244 self.assertEqual(cl.parsePropsFromForm(),
245 ({('issue', None): {'nosy': ['1','2', '3']}}, []))
247 def testEmptyMultilinkSet(self):
248 nodeid = self.db.issue.create(nosy=['1','2'])
249 self.assertEqual(self.parseForm({'nosy': ''}, 'issue', nodeid),
250 ({('issue', nodeid): {'nosy': []}}, []))
251 nodeid = self.db.issue.create(nosy=['1','2'])
252 self.assertEqual(self.parseForm({'nosy': ' '}, 'issue', nodeid),
253 ({('issue', nodeid): {'nosy': []}}, []))
254 self.assertEqual(self.parseForm({'nosy': '1,2'}, 'issue', nodeid),
255 ({('issue', nodeid): {}}, []))
257 def testInvalidMultilinkValue(self):
258 # XXX This is not the current behaviour - should we enforce this?
259 # self.assertRaises(IndexError, self.parseForm,
260 # {'nosy': '4'}))
261 self.assertRaises(ValueError, self.parseForm, {'nosy': 'frozzle'},
262 'issue')
263 self.assertRaises(ValueError, self.parseForm, {'nosy': '1,frozzle'},
264 'issue')
265 self.assertRaises(ValueError, self.parseForm, {'multilink': 'frozzle'})
267 def testMultilinkAdd(self):
268 nodeid = self.db.issue.create(nosy=['1'])
269 # do nothing
270 self.assertEqual(self.parseForm({':add:nosy': ''}, 'issue', nodeid),
271 ({('issue', nodeid): {}}, []))
273 # do something ;)
274 self.assertEqual(self.parseForm({':add:nosy': '2'}, 'issue', nodeid),
275 ({('issue', nodeid): {'nosy': ['1','2']}}, []))
276 self.assertEqual(self.parseForm({':add:nosy': '2,mary'}, 'issue',
277 nodeid), ({('issue', nodeid): {'nosy': ['1','2','4']}}, []))
278 self.assertEqual(self.parseForm({':add:nosy': ['2','3']}, 'issue',
279 nodeid), ({('issue', nodeid): {'nosy': ['1','2','3']}}, []))
281 def testMultilinkAddNew(self):
282 self.assertEqual(self.parseForm({':add:nosy': ['2','3']}, 'issue'),
283 ({('issue', None): {'nosy': ['2','3']}}, []))
285 def testMultilinkRemove(self):
286 nodeid = self.db.issue.create(nosy=['1','2'])
287 # do nothing
288 self.assertEqual(self.parseForm({':remove:nosy': ''}, 'issue', nodeid),
289 ({('issue', nodeid): {}}, []))
291 # do something ;)
292 self.assertEqual(self.parseForm({':remove:nosy': '1'}, 'issue',
293 nodeid), ({('issue', nodeid): {'nosy': ['2']}}, []))
294 self.assertEqual(self.parseForm({':remove:nosy': 'admin,2'},
295 'issue', nodeid), ({('issue', nodeid): {'nosy': []}}, []))
296 self.assertEqual(self.parseForm({':remove:nosy': ['1','2']},
297 'issue', nodeid), ({('issue', nodeid): {'nosy': []}}, []))
299 # add and remove
300 self.assertEqual(self.parseForm({':add:nosy': ['3'],
301 ':remove:nosy': ['1','2']},
302 'issue', nodeid), ({('issue', nodeid): {'nosy': ['3']}}, []))
304 # remove one that doesn't exist?
305 self.assertRaises(ValueError, self.parseForm, {':remove:nosy': '4'},
306 'issue', nodeid)
308 def testMultilinkRetired(self):
309 self.db.user.retire('2')
310 self.assertEqual(self.parseForm({'nosy': ['2','3']}, 'issue'),
311 ({('issue', None): {'nosy': ['2','3']}}, []))
312 nodeid = self.db.issue.create(nosy=['1','2'])
313 self.assertEqual(self.parseForm({':remove:nosy': '2'}, 'issue',
314 nodeid), ({('issue', nodeid): {'nosy': ['1']}}, []))
315 self.assertEqual(self.parseForm({':add:nosy': '3'}, 'issue', nodeid),
316 ({('issue', nodeid): {'nosy': ['1','2','3']}}, []))
318 def testAddRemoveNonexistant(self):
319 self.assertRaises(ValueError, self.parseForm, {':remove:foo': '2'},
320 'issue')
321 self.assertRaises(ValueError, self.parseForm, {':add:foo': '2'},
322 'issue')
324 #
325 # Password
326 #
327 def testEmptyPassword(self):
328 self.assertEqual(self.parseForm({'password': ''}, 'user'),
329 ({('user', None): {}}, []))
330 self.assertEqual(self.parseForm({'password': ''}, 'user'),
331 ({('user', None): {}}, []))
332 self.assertRaises(ValueError, self.parseForm, {'password': ['', '']},
333 'user')
334 self.assertRaises(ValueError, self.parseForm, {'password': 'foo',
335 ':confirm:password': ['', '']}, 'user')
337 def testSetPassword(self):
338 self.assertEqual(self.parseForm({'password': 'foo',
339 ':confirm:password': 'foo'}, 'user'),
340 ({('user', None): {'password': 'foo'}}, []))
342 def testSetPasswordConfirmBad(self):
343 self.assertRaises(ValueError, self.parseForm, {'password': 'foo'},
344 'user')
345 self.assertRaises(ValueError, self.parseForm, {'password': 'foo',
346 ':confirm:password': 'bar'}, 'user')
348 def testEmptyPasswordNotSet(self):
349 nodeid = self.db.user.create(username='1',
350 password=password.Password('foo'))
351 self.assertEqual(self.parseForm({'password': ''}, 'user', nodeid),
352 ({('user', nodeid): {}}, []))
353 nodeid = self.db.user.create(username='2',
354 password=password.Password('foo'))
355 self.assertEqual(self.parseForm({'password': '',
356 ':confirm:password': ''}, 'user', nodeid),
357 ({('user', nodeid): {}}, []))
359 #
360 # Boolean
361 #
362 def testEmptyBoolean(self):
363 self.assertEqual(self.parseForm({'boolean': ''}),
364 ({('test', None): {}}, []))
365 self.assertEqual(self.parseForm({'boolean': ' '}),
366 ({('test', None): {}}, []))
367 self.assertRaises(ValueError, self.parseForm, {'boolean': ['', '']})
369 def testSetBoolean(self):
370 self.assertEqual(self.parseForm({'boolean': 'yes'}),
371 ({('test', None): {'boolean': 1}}, []))
372 self.assertEqual(self.parseForm({'boolean': 'a\r\nb\r\n'}),
373 ({('test', None): {'boolean': 0}}, []))
374 nodeid = self.db.test.create(boolean=1)
375 self.assertEqual(self.parseForm({'boolean': 'yes'}, 'test', nodeid),
376 ({('test', nodeid): {}}, []))
377 nodeid = self.db.test.create(boolean=0)
378 self.assertEqual(self.parseForm({'boolean': 'no'}, 'test', nodeid),
379 ({('test', nodeid): {}}, []))
381 def testEmptyBooleanSet(self):
382 nodeid = self.db.test.create(boolean=0)
383 self.assertEqual(self.parseForm({'boolean': ''}, 'test', nodeid),
384 ({('test', nodeid): {'boolean': None}}, []))
385 nodeid = self.db.test.create(boolean=1)
386 self.assertEqual(self.parseForm({'boolean': ' '}, 'test', nodeid),
387 ({('test', nodeid): {'boolean': None}}, []))
389 #
390 # Number
391 #
392 def testEmptyNumber(self):
393 self.assertEqual(self.parseForm({'number': ''}),
394 ({('test', None): {}}, []))
395 self.assertEqual(self.parseForm({'number': ' '}),
396 ({('test', None): {}}, []))
397 self.assertRaises(ValueError, self.parseForm, {'number': ['', '']})
399 def testInvalidNumber(self):
400 self.assertRaises(ValueError, self.parseForm, {'number': 'hi, mum!'})
402 def testSetNumber(self):
403 self.assertEqual(self.parseForm({'number': '1'}),
404 ({('test', None): {'number': 1}}, []))
405 self.assertEqual(self.parseForm({'number': '\n0\n'}),
406 ({('test', None): {'number': 0}}, []))
407 nodeid = self.db.test.create(number=1)
408 self.assertEqual(self.parseForm({'number': '1'}, 'test', nodeid),
409 ({('test', nodeid): {}}, []))
410 nodeid = self.db.test.create(number=0)
411 self.assertEqual(self.parseForm({'number': '0'}, 'test', nodeid),
412 ({('test', nodeid): {}}, []))
414 def testEmptyNumberSet(self):
415 nodeid = self.db.test.create(number=0)
416 self.assertEqual(self.parseForm({'number': ''}, 'test', nodeid),
417 ({('test', nodeid): {'number': None}}, []))
418 nodeid = self.db.test.create(number=1)
419 self.assertEqual(self.parseForm({'number': ' '}, 'test', nodeid),
420 ({('test', nodeid): {'number': None}}, []))
422 #
423 # Date
424 #
425 def testEmptyDate(self):
426 self.assertEqual(self.parseForm({'date': ''}),
427 ({('test', None): {}}, []))
428 self.assertEqual(self.parseForm({'date': ' '}),
429 ({('test', None): {}}, []))
430 self.assertRaises(ValueError, self.parseForm, {'date': ['', '']})
432 def testInvalidDate(self):
433 self.assertRaises(ValueError, self.parseForm, {'date': '12'})
435 def testSetDate(self):
436 self.assertEqual(self.parseForm({'date': '2003-01-01'}),
437 ({('test', None): {'date': date.Date('2003-01-01')}}, []))
438 nodeid = self.db.test.create(date=date.Date('2003-01-01'))
439 self.assertEqual(self.parseForm({'date': '2003-01-01'}, 'test',
440 nodeid), ({('test', nodeid): {}}, []))
442 def testEmptyDateSet(self):
443 nodeid = self.db.test.create(date=date.Date('.'))
444 self.assertEqual(self.parseForm({'date': ''}, 'test', nodeid),
445 ({('test', nodeid): {'date': None}}, []))
446 nodeid = self.db.test.create(date=date.Date('1970-01-01.00:00:00'))
447 self.assertEqual(self.parseForm({'date': ' '}, 'test', nodeid),
448 ({('test', nodeid): {'date': None}}, []))
450 #
451 # Test multiple items in form
452 #
453 def testMultiple(self):
454 self.assertEqual(self.parseForm({'string': 'a', 'issue-1@title': 'b'}),
455 ({('test', None): {'string': 'a'},
456 ('issue', '-1'): {'title': 'b'}
457 }, []))
459 def testMultipleExistingContext(self):
460 nodeid = self.db.test.create()
461 self.assertEqual(self.parseForm({'string': 'a', 'issue-1@title': 'b'},
462 'test', nodeid),({('test', nodeid): {'string': 'a'},
463 ('issue', '-1'): {'title': 'b'}}, []))
465 def testLinking(self):
466 self.assertEqual(self.parseForm({
467 'string': 'a',
468 'issue-1@add@nosy': '1',
469 'issue-2@link@superseder': 'issue-1',
470 }),
471 ({('test', None): {'string': 'a'},
472 ('issue', '-1'): {'nosy': ['1']},
473 ('issue', '-2'): {}
474 },
475 [('issue', '-2', 'superseder', [('issue', '-1')])
476 ]
477 )
478 )
480 def testLinkBadDesignator(self):
481 self.assertRaises(ValueError, self.parseForm,
482 {'test-1@link@link': 'blah'})
483 self.assertRaises(ValueError, self.parseForm,
484 {'test-1@link@link': 'issue'})
486 def testLinkNotLink(self):
487 self.assertRaises(ValueError, self.parseForm,
488 {'test-1@link@boolean': 'issue-1'})
489 self.assertRaises(ValueError, self.parseForm,
490 {'test-1@link@string': 'issue-1'})
492 def testBackwardsCompat(self):
493 res = self.parseForm({':note': 'spam'}, 'issue')
494 date = res[0][('msg', '-1')]['date']
495 self.assertEqual(res, ({('issue', None): {}, ('msg', '-1'):
496 {'content': 'spam', 'author': '1', 'date': date}},
497 [('issue', None, 'messages', [('msg', '-1')])]))
498 file = FileUpload('foo', 'foo.txt')
499 self.assertEqual(self.parseForm({':file': file}, 'issue'),
500 ({('issue', None): {}, ('file', '-1'): {'content': 'foo',
501 'name': 'foo.txt', 'type': 'text/plain'}},
502 [('issue', None, 'files', [('file', '-1')])]))
504 def suite():
505 l = [unittest.makeSuite(FormTestCase),
506 ]
507 return unittest.TestSuite(l)
509 def run():
510 runner = unittest.TextTestRunner()
511 unittest.main(testRunner=runner)
513 if __name__ == '__main__':
514 run()
516 # vim: set filetype=python ts=4 sw=4 et si