test/test_cgi.py

   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.7 2003-02-12 06:41:58 richard Exp $
  12
  13 import unittest, os, shutil, errno, sys, difflib, cgi
  14
  15 from roundup.cgi import client
  16 from roundup import init, instance, password, hyperdb, date
  17
  18 def makeForm(args):
  19     form = cgi.FieldStorage()
  20     for k,v in args.items():
  21         if type(v) is type([]):
  22             [form.list.append(cgi.MiniFieldStorage(k, x)) for x in v]
  23         else:
  24             form.list.append(cgi.MiniFieldStorage(k, v))
  25     return form
  26
  27 class config:
  28     TRACKER_NAME = 'testing testing'
  29     TRACKER_WEB = 'http://testing.testing/'
  30
  31 class FormTestCase(unittest.TestCase):
  32     def setUp(self):
  33         self.dirname = '_test_cgi_form'
  34         try:
  35             shutil.rmtree(self.dirname)
  36         except OSError, error:
  37             if error.errno not in (errno.ENOENT, errno.ESRCH): raise
  38         # create the instance
  39         init.install(self.dirname, 'classic', 'anydbm')
  40         init.initialise(self.dirname, 'sekrit')
  41         # check we can load the package
  42         self.instance = instance.open(self.dirname)
  43         # and open the database
  44         self.db = self.instance.open('admin')
  45         self.db.user.create(username='Chef', address='chef@bork.bork.bork',
  46             realname='Bork, Chef', roles='User')
  47         self.db.user.create(username='mary', address='mary@test',
  48             roles='User', realname='Contrary, Mary')
  49
  50         test = self.instance.dbinit.Class(self.db, "test",
  51             string=hyperdb.String(),
  52             boolean=hyperdb.Boolean(), link=hyperdb.Link('test'),
  53             multilink=hyperdb.Multilink('test'), date=hyperdb.Date(),
  54             interval=hyperdb.Interval())
  55
  56     def parseForm(self, form, classname='test', nodeid=None):
  57         cl = client.Client(self.instance, None, {'PATH_INFO':'/'},
  58             makeForm(form))
  59         cl.classname = classname
  60         cl.nodeid = nodeid
  61         cl.db = self.db
  62         return cl.parsePropsFromForm()
  63
  64     def tearDown(self):
  65         self.db.close()
  66         try:
  67             shutil.rmtree(self.dirname)
  68         except OSError, error:
  69             if error.errno not in (errno.ENOENT, errno.ESRCH): raise
  70
  71     #
  72     # Empty form
  73     #
  74     def testNothing(self):
  75         self.assertEqual(self.parseForm({}), {'test': {}})
  76
  77     def testNothingWithRequired(self):
  78         self.assertRaises(ValueError, self.parseForm, {':required': 'string'})
  79         self.assertRaises(ValueError, self.parseForm,
  80             {':required': 'title,status', 'status':'1'}, 'issue')
  81         self.assertRaises(ValueError, self.parseForm,
  82             {':required': ['title','status'], 'status':'1'}, 'issue')
  83         self.assertRaises(ValueError, self.parseForm,
  84             {':required': 'status', 'status':''}, 'issue')
  85         self.assertRaises(ValueError, self.parseForm,
  86             {':required': 'nosy', 'nosy':''}, 'issue')
  87
  88     #
  89     # Nonexistant edit
  90     #
  91     def testEditNonexistant(self):
  92         self.assertRaises(IndexError, self.parseForm, {'boolean': ''},
  93             'test', '1')
  94
  95     #
  96     # String
  97     #
  98     def testEmptyString(self):
  99         self.assertEqual(self.parseForm({'string': ''}), {'test': {}})
 100         self.assertEqual(self.parseForm({'string': ' '}), {'test': {}})
 101         self.assertRaises(ValueError, self.parseForm, {'string': ['', '']})
 102
 103     def testSetString(self):
 104         self.assertEqual(self.parseForm({'string': 'foo'}),
 105             {'test': {'string': 'foo'}})
 106         self.assertEqual(self.parseForm({'string': 'a\r\nb\r\n'}),
 107             {'test': {'string': 'a\nb'}})
 108         nodeid = self.db.issue.create(title='foo')
 109         self.assertEqual(self.parseForm({'title': 'foo'}, 'issue', nodeid),
 110             {'issue'+nodeid: {}})
 111
 112     def testEmptyStringSet(self):
 113         nodeid = self.db.issue.create(title='foo')
 114         self.assertEqual(self.parseForm({'title': ''}, 'issue', nodeid),
 115             {'issue'+nodeid: {'title': None}})
 116         nodeid = self.db.issue.create(title='foo')
 117         self.assertEqual(self.parseForm({'title': ' '}, 'issue', nodeid),
 118             {'issue'+nodeid: {'title': None}})
 119
 120     #
 121     # Link
 122     #
 123     def testEmptyLink(self):
 124         self.assertEqual(self.parseForm({'link': ''}), {'test': {}})
 125         self.assertEqual(self.parseForm({'link': ' '}), {'test': {}})
 126         self.assertRaises(ValueError, self.parseForm, {'link': ['', '']})
 127         self.assertEqual(self.parseForm({'link': '-1'}), {'test': {}})
 128
 129     def testSetLink(self):
 130         self.assertEqual(self.parseForm({'status': 'unread'}, 'issue'),
 131             {'issue': {'status': '1'}})
 132         self.assertEqual(self.parseForm({'status': '1'}, 'issue'),
 133             {'issue': {'status': '1'}})
 134         nodeid = self.db.issue.create(status='unread')
 135         self.assertEqual(self.parseForm({'status': 'unread'}, 'issue', nodeid),
 136             {'issue'+nodeid: {}})
 137
 138     def testUnsetLink(self):
 139         nodeid = self.db.issue.create(status='unread')
 140         self.assertEqual(self.parseForm({'status': '-1'}, 'issue', nodeid),
 141             {'issue'+nodeid: {'status': None}})
 142
 143     def testInvalidLinkValue(self):
 144 # XXX This is not the current behaviour - should we enforce this?
 145 #        self.assertRaises(IndexError, self.parseForm,
 146 #            {'status': '4'}))
 147         self.assertRaises(ValueError, self.parseForm, {'link': 'frozzle'})
 148
 149         self.assertRaises(ValueError, self.parseForm, {'link': 'frozzle'})
 150
 151     #
 152     # Multilink
 153     #
 154     def testEmptyMultilink(self):
 155         self.assertEqual(self.parseForm({'nosy': ''}), {'test': {}})
 156         self.assertEqual(self.parseForm({'nosy': ' '}), {'test': {}})
 157
 158     def testSetMultilink(self):
 159         self.assertEqual(self.parseForm({'nosy': '1'}, 'issue'),
 160             {'issue': {'nosy': ['1']}})
 161         self.assertEqual(self.parseForm({'nosy': 'admin'}, 'issue'),
 162             {'issue': {'nosy': ['1']}})
 163         self.assertEqual(self.parseForm({'nosy': ['1','2']}, 'issue'),
 164             {'issue': {'nosy': ['1','2']}})
 165         self.assertEqual(self.parseForm({'nosy': '1,2'}, 'issue'),
 166             {'issue': {'nosy': ['1','2']}})
 167         self.assertEqual(self.parseForm({'nosy': 'admin,2'}, 'issue'),
 168             {'issue': {'nosy': ['1','2']}})
 169
 170     def testEmptyMultilinkSet(self):
 171         nodeid = self.db.issue.create(nosy=['1','2'])
 172         self.assertEqual(self.parseForm({'nosy': ''}, 'issue', nodeid),
 173             {'issue'+nodeid: {'nosy': []}})
 174         nodeid = self.db.issue.create(nosy=['1','2'])
 175         self.assertEqual(self.parseForm({'nosy': ' '}, 'issue', nodeid),
 176             {'issue'+nodeid: {'nosy': []}})
 177         self.assertEqual(self.parseForm({'nosy': '1,2'}, 'issue', nodeid),
 178             {'issue'+nodeid: {}})
 179
 180     def testInvalidMultilinkValue(self):
 181 # XXX This is not the current behaviour - should we enforce this?
 182 #        self.assertRaises(IndexError, self.parseForm,
 183 #            {'nosy': '4'}))
 184         self.assertRaises(ValueError, self.parseForm, {'nosy': 'frozzle'},
 185             'issue')
 186         self.assertRaises(ValueError, self.parseForm, {'nosy': '1,frozzle'},
 187             'issue')
 188         self.assertRaises(ValueError, self.parseForm, {'multilink': 'frozzle'})
 189
 190     def testMultilinkAdd(self):
 191         nodeid = self.db.issue.create(nosy=['1'])
 192         # do nothing
 193         self.assertEqual(self.parseForm({':add:nosy': ''}, 'issue', nodeid),
 194             {'issue'+nodeid: {}})
 195
 196         # do something ;)
 197         self.assertEqual(self.parseForm({':add:nosy': '2'}, 'issue', nodeid),
 198             {'issue'+nodeid: {'nosy': ['1','2']}})
 199         self.assertEqual(self.parseForm({':add:nosy': '2,mary'}, 'issue',
 200             nodeid), {'issue'+nodeid: {'nosy': ['1','2','4']}})
 201         self.assertEqual(self.parseForm({':add:nosy': ['2','3']}, 'issue',
 202             nodeid), {'issue'+nodeid: {'nosy': ['1','2','3']}})
 203
 204     def testMultilinkAddNew(self):
 205         self.assertEqual(self.parseForm({':add:nosy': ['2','3']}, 'issue'),
 206             {'issue': {'nosy': ['2','3']}})
 207
 208     def testMultilinkRemove(self):
 209         nodeid = self.db.issue.create(nosy=['1','2'])
 210         # do nothing
 211         self.assertEqual(self.parseForm({':remove:nosy': ''}, 'issue', nodeid),
 212             {'issue'+nodeid: {}})
 213
 214         # do something ;)
 215         self.assertEqual(self.parseForm({':remove:nosy': '1'}, 'issue',
 216             nodeid), {'issue'+nodeid: {'nosy': ['2']}})
 217         self.assertEqual(self.parseForm({':remove:nosy': 'admin,2'},
 218             'issue', nodeid), {'issue'+nodeid: {'nosy': []}})
 219         self.assertEqual(self.parseForm({':remove:nosy': ['1','2']},
 220             'issue', nodeid), {'issue'+nodeid: {'nosy': []}})
 221
 222         # remove one that doesn't exist?
 223         self.assertRaises(ValueError, self.parseForm, {':remove:nosy': '4'},
 224             'issue', nodeid)
 225
 226     def testMultilinkRetired(self):
 227         self.db.user.retire('2')
 228         self.assertEqual(self.parseForm({'nosy': ['2','3']}, 'issue'),
 229             {'issue': {'nosy': ['2','3']}})
 230         nodeid = self.db.issue.create(nosy=['1','2'])
 231         self.assertEqual(self.parseForm({':remove:nosy': '2'}, 'issue',
 232             nodeid), {'issue'+nodeid: {'nosy': ['1']}})
 233         self.assertEqual(self.parseForm({':add:nosy': '3'}, 'issue', nodeid),
 234             {'issue'+nodeid: {'nosy': ['1','2','3']}})
 235
 236     def testAddRemoveNonexistant(self):
 237         self.assertRaises(ValueError, self.parseForm, {':remove:foo': '2'},
 238             'issue')
 239         self.assertRaises(ValueError, self.parseForm, {':add:foo': '2'},
 240             'issue')
 241
 242     #
 243     # Password
 244     #
 245     def testEmptyPassword(self):
 246         self.assertEqual(self.parseForm({'password': ''}, 'user'),
 247             {'user': {}})
 248         self.assertEqual(self.parseForm({'password': ''}, 'user'),
 249             {'user': {}})
 250         self.assertRaises(ValueError, self.parseForm, {'password': ['', '']},
 251             'user')
 252         self.assertRaises(ValueError, self.parseForm, {'password': 'foo',
 253             'password:confirm': ['', '']}, 'user')
 254
 255     def testSetPassword(self):
 256         self.assertEqual(self.parseForm({'password': 'foo',
 257             'password:confirm': 'foo'}, 'user'), {'user': {'password': 'foo'}})
 258
 259     def testSetPasswordConfirmBad(self):
 260         self.assertRaises(ValueError, self.parseForm, {'password': 'foo'},
 261             'user')
 262         self.assertRaises(ValueError, self.parseForm, {'password': 'foo',
 263             'password:confirm': 'bar'}, 'user')
 264
 265     def testEmptyPasswordNotSet(self):
 266         nodeid = self.db.user.create(username='1',
 267             password=password.Password('foo'))
 268         self.assertEqual(self.parseForm({'password': ''}, 'user', nodeid),
 269             {'user'+nodeid: {}})
 270         nodeid = self.db.user.create(username='2',
 271             password=password.Password('foo'))
 272         self.assertEqual(self.parseForm({'password': '',
 273             'password:confirm': ''}, 'user', nodeid),
 274             {'user'+nodeid: {}})
 275
 276     #
 277     # Boolean
 278     #
 279     def testEmptyBoolean(self):
 280         self.assertEqual(self.parseForm({'boolean': ''}), {'test': {}})
 281         self.assertEqual(self.parseForm({'boolean': ' '}), {'test': {}})
 282         self.assertRaises(ValueError, self.parseForm, {'boolean': ['', '']})
 283
 284     def testSetBoolean(self):
 285         self.assertEqual(self.parseForm({'boolean': 'yes'}),
 286             {'test': {'boolean': 1}})
 287         self.assertEqual(self.parseForm({'boolean': 'a\r\nb\r\n'}),
 288             {'test': {'boolean': 0}})
 289         nodeid = self.db.test.create(boolean=1)
 290         self.assertEqual(self.parseForm({'boolean': 'yes'}, 'test', nodeid),
 291             {'test'+nodeid: {}})
 292         nodeid = self.db.test.create(boolean=0)
 293         self.assertEqual(self.parseForm({'boolean': 'no'}, 'test', nodeid),
 294             {'test'+nodeid: {}})
 295
 296     def testEmptyBooleanSet(self):
 297         nodeid = self.db.test.create(boolean=0)
 298         self.assertEqual(self.parseForm({'boolean': ''}, 'test', nodeid),
 299             {'test'+nodeid: {'boolean': None}})
 300         nodeid = self.db.test.create(boolean=1)
 301         self.assertEqual(self.parseForm({'boolean': ' '}, 'test', nodeid),
 302             {'test'+nodeid: {'boolean': None}})
 303
 304     #
 305     # Date
 306     #
 307     def testEmptyDate(self):
 308         self.assertEqual(self.parseForm({'date': ''}), {'test': {}})
 309         self.assertEqual(self.parseForm({'date': ' '}), {'test': {}})
 310         self.assertRaises(ValueError, self.parseForm, {'date': ['', '']})
 311
 312     def testSetDate(self):
 313         self.assertEqual(self.parseForm({'date': '2003-01-01'}),
 314             {'test': {'date': date.Date('2003-01-01')}})
 315         nodeid = self.db.test.create(date=date.Date('2003-01-01'))
 316         self.assertEqual(self.parseForm({'date': '2003-01-01'}, 'test',
 317             nodeid), {'test'+nodeid: {}})
 318
 319     def testEmptyDateSet(self):
 320         nodeid = self.db.test.create(date=date.Date('.'))
 321         self.assertEqual(self.parseForm({'date': ''}, 'test', nodeid),
 322             {'test'+nodeid: {'date': None}})
 323         nodeid = self.db.test.create(date=date.Date('1970-01-01.00:00:00'))
 324         self.assertEqual(self.parseForm({'date': ' '}, 'test', nodeid),
 325             {'test'+nodeid: {'date': None}})
 326
 327     #
 328     # Test multiple items in form
 329     #
 330     def testMultiple(self):
 331         self.assertEqual(self.parseForm({'string': 'a', 'issue@title': 'b'}),
 332             {'test': {'string': 'a'}, 'issue': {'title': 'b'}})
 333         nodeid = self.db.test.create()
 334         self.assertEqual(self.parseForm({'string': 'a', 'issue@title': 'b'},
 335             'test', nodeid),
 336             {'test1': {'string': 'a'}, 'issue': {'title': 'b'}})
 337
 338 def suite():
 339     l = [unittest.makeSuite(FormTestCase),
 340     ]
 341     return unittest.TestSuite(l)
 342
 343
 344 # vim: set filetype=python ts=4 sw=4 et si