Mercurial > p > roundup > code
annotate test/test_rest.py @ 5585:8725c09802b8 REST-rebased
Added test cases for the element URI methods
committer: Ralf Schlatterbeck <rsc@runtux.com>
| author | Chau Nguyen <dangchau1991@yahoo.com> |
|---|---|
| date | Wed, 30 Jan 2019 10:26:35 +0100 |
| parents | c65d98a16780 |
| children | 8f2fbc88e155 |
| rev | line source |
|---|---|
| 5583 | 1 import unittest, os, shutil, errno, sys, difflib, cgi, re |
| 2 | |
| 3 from xmlrpclib import MultiCall | |
| 4 from roundup.cgi.exceptions import * | |
| 5 from roundup import init, instance, password, hyperdb, date | |
| 6 from roundup.rest import RestfulInstance | |
| 7 from roundup.backends import list_backends | |
| 8 from roundup.hyperdb import String | |
| 9 from roundup.cgi import TranslationService, client | |
| 10 | |
| 11 import db_test_base | |
| 12 | |
| 13 NEEDS_INSTANCE = 1 | |
| 14 | |
| 15 class TestCase(unittest.TestCase): | |
| 16 | |
| 17 backend = None | |
| 18 | |
| 19 def setUp(self): | |
| 20 self.dirname = '_test_rest' | |
| 21 # set up and open a tracker | |
| 22 self.instance = db_test_base.setupTracker(self.dirname, self.backend) | |
| 23 | |
| 24 # open the database | |
| 25 self.db = self.instance.open('admin') | |
| 26 | |
| 27 # Get user id (user4 maybe). Used later to get data from db. | |
| 28 self.joeid = self.db.user.create( | |
| 29 username='joe', | |
| 30 password=password.Password('random'), | |
| 31 address='random@home.org', | |
| 32 realname='Joe Random', | |
| 33 roles='User' | |
| 34 ) | |
| 35 | |
| 36 self.db.commit() | |
| 37 self.db.close() | |
| 38 self.db = self.instance.open('joe') | |
| 39 | |
| 40 self.db.tx_Source = 'web' | |
| 41 | |
| 42 self.db.issue.addprop(tx_Source=hyperdb.String()) | |
| 43 self.db.msg.addprop(tx_Source=hyperdb.String()) | |
| 44 | |
| 45 self.db.post_init() | |
| 46 | |
| 47 thisdir = os.path.dirname(__file__) | |
| 48 vars = {} | |
| 49 execfile(os.path.join(thisdir, "tx_Source_detector.py"), vars) | |
| 50 vars['init'](self.db) | |
| 51 | |
| 52 env = { | |
| 53 'PATH_INFO': 'http://localhost/rounduptest/rest/', | |
| 54 'HTTP_HOST': 'localhost', | |
| 55 'TRACKER_NAME': 'rounduptest' | |
| 56 } | |
| 57 dummy_client = client.Client(self.instance, None, env, [], None) | |
| 58 | |
| 59 self.server = RestfulInstance(dummy_client, self.db) | |
| 60 | |
| 61 def tearDown(self): | |
| 62 self.db.close() | |
| 63 try: | |
| 64 shutil.rmtree(self.dirname) | |
| 65 except OSError, error: | |
| 66 if error.errno not in (errno.ENOENT, errno.ESRCH): | |
| 67 raise | |
| 68 | |
| 69 def testGet(self): | |
| 70 """ | |
| 71 Retrieve all three users | |
| 72 obtain data for 'joe' | |
| 73 """ | |
| 74 # Retrieve all three users. | |
| 75 code, results = self.server.get_collection('user', {}) | |
| 76 self.assertEqual(code, 200) | |
| 77 self.assertEqual(len(results), 3) | |
| 78 | |
| 79 # Obtain data for 'joe'. | |
| 80 code, results = self.server.get_element('user', self.joeid, {}) | |
| 81 self.assertEqual(code, 200) | |
| 82 self.assertEqual(results['attributes']['username'], 'joe') | |
| 83 self.assertEqual(results['attributes']['realname'], 'Joe Random') | |
| 84 | |
|
5585
8725c09802b8
Added test cases for the element URI methods
Chau Nguyen <dangchau1991@yahoo.com>
parents:
5583
diff
changeset
|
85 # Obtain data for 'joe'. |
|
8725c09802b8
Added test cases for the element URI methods
Chau Nguyen <dangchau1991@yahoo.com>
parents:
5583
diff
changeset
|
86 code, results = self.server.get_attribute( |
|
8725c09802b8
Added test cases for the element URI methods
Chau Nguyen <dangchau1991@yahoo.com>
parents:
5583
diff
changeset
|
87 'user', self.joeid, 'username', {} |
|
8725c09802b8
Added test cases for the element URI methods
Chau Nguyen <dangchau1991@yahoo.com>
parents:
5583
diff
changeset
|
88 ) |
|
8725c09802b8
Added test cases for the element URI methods
Chau Nguyen <dangchau1991@yahoo.com>
parents:
5583
diff
changeset
|
89 self.assertEqual(code, 200) |
|
8725c09802b8
Added test cases for the element URI methods
Chau Nguyen <dangchau1991@yahoo.com>
parents:
5583
diff
changeset
|
90 self.assertEqual(results['data'], 'joe') |
|
8725c09802b8
Added test cases for the element URI methods
Chau Nguyen <dangchau1991@yahoo.com>
parents:
5583
diff
changeset
|
91 |
| 5583 | 92 def testPut(self): |
| 93 """ | |
| 94 Change joe's 'realname' | |
| 95 Check if we can't change admin's detail | |
| 96 """ | |
|
5585
8725c09802b8
Added test cases for the element URI methods
Chau Nguyen <dangchau1991@yahoo.com>
parents:
5583
diff
changeset
|
97 # change Joe's realname via attribute uri |
|
8725c09802b8
Added test cases for the element URI methods
Chau Nguyen <dangchau1991@yahoo.com>
parents:
5583
diff
changeset
|
98 form = cgi.FieldStorage() |
|
8725c09802b8
Added test cases for the element URI methods
Chau Nguyen <dangchau1991@yahoo.com>
parents:
5583
diff
changeset
|
99 form.list = [ |
|
8725c09802b8
Added test cases for the element URI methods
Chau Nguyen <dangchau1991@yahoo.com>
parents:
5583
diff
changeset
|
100 cgi.MiniFieldStorage('data', 'Joe Doe Doe') |
|
8725c09802b8
Added test cases for the element URI methods
Chau Nguyen <dangchau1991@yahoo.com>
parents:
5583
diff
changeset
|
101 ] |
|
8725c09802b8
Added test cases for the element URI methods
Chau Nguyen <dangchau1991@yahoo.com>
parents:
5583
diff
changeset
|
102 code, results = self.server.put_attribute( |
|
8725c09802b8
Added test cases for the element URI methods
Chau Nguyen <dangchau1991@yahoo.com>
parents:
5583
diff
changeset
|
103 'user', self.joeid, 'realname', form |
|
8725c09802b8
Added test cases for the element URI methods
Chau Nguyen <dangchau1991@yahoo.com>
parents:
5583
diff
changeset
|
104 ) |
|
8725c09802b8
Added test cases for the element URI methods
Chau Nguyen <dangchau1991@yahoo.com>
parents:
5583
diff
changeset
|
105 code, results = self.server.get_attribute( |
|
8725c09802b8
Added test cases for the element URI methods
Chau Nguyen <dangchau1991@yahoo.com>
parents:
5583
diff
changeset
|
106 'user', self.joeid, 'realname', {} |
|
8725c09802b8
Added test cases for the element URI methods
Chau Nguyen <dangchau1991@yahoo.com>
parents:
5583
diff
changeset
|
107 ) |
|
8725c09802b8
Added test cases for the element URI methods
Chau Nguyen <dangchau1991@yahoo.com>
parents:
5583
diff
changeset
|
108 self.assertEqual(code, 200) |
|
8725c09802b8
Added test cases for the element URI methods
Chau Nguyen <dangchau1991@yahoo.com>
parents:
5583
diff
changeset
|
109 self.assertEqual(results['data'], 'Joe Doe Doe') |
|
8725c09802b8
Added test cases for the element URI methods
Chau Nguyen <dangchau1991@yahoo.com>
parents:
5583
diff
changeset
|
110 |
| 5583 | 111 # Reset joe's 'realname'. |
| 112 form = cgi.FieldStorage() | |
| 113 form.list = [ | |
| 114 cgi.MiniFieldStorage('realname', 'Joe Doe') | |
| 115 ] | |
| 116 code, results = self.server.put_element('user', self.joeid, form) | |
| 117 code, results = self.server.get_element('user', self.joeid, {}) | |
| 118 self.assertEqual(code, 200) | |
| 119 self.assertEqual(results['attributes']['realname'], 'Joe Doe') | |
| 120 | |
| 121 # check we can't change admin's details | |
| 122 self.assertRaises( | |
| 123 Unauthorised, | |
| 124 self.server.put_element, 'user', '1', form | |
| 125 ) | |
| 126 | |
| 127 def testPost(self): | |
| 128 """ | |
| 129 Post a new issue with title: foo | |
| 130 Verify the information of the created issue | |
| 131 """ | |
| 132 form = cgi.FieldStorage() | |
| 133 form.list = [ | |
| 134 cgi.MiniFieldStorage('title', 'foo') | |
| 135 ] | |
| 136 code, results = self.server.post_collection('issue', form) | |
| 137 self.assertEqual(code, 201) | |
| 138 issueid = results['id'] | |
| 139 code, results = self.server.get_element('issue', issueid, {}) | |
| 140 self.assertEqual(code, 200) | |
| 141 self.assertEqual(results['attributes']['title'], 'foo') | |
| 142 self.assertEqual(self.db.issue.get(issueid, "tx_Source"), 'web') | |
| 143 | |
| 144 def testPostFile(self): | |
| 145 """ | |
| 146 Post a new file with content: hello\r\nthere | |
| 147 Verify the information of the created file | |
| 148 """ | |
| 149 form = cgi.FieldStorage() | |
| 150 form.list = [ | |
| 151 cgi.MiniFieldStorage('content', 'hello\r\nthere') | |
| 152 ] | |
| 153 code, results = self.server.post_collection('file', form) | |
| 154 self.assertEqual(code, 201) | |
| 155 fileid = results['id'] | |
| 156 code, results = self.server.get_element('file', fileid, {}) | |
| 157 self.assertEqual(code, 200) | |
| 158 self.assertEqual(results['attributes']['content'], 'hello\r\nthere') | |
| 159 | |
| 160 def testAuthDeniedPut(self): | |
| 161 """ | |
| 162 Test unauthorized PUT request | |
| 163 """ | |
| 164 # Wrong permissions (caught by roundup security module). | |
| 165 form = cgi.FieldStorage() | |
| 166 form.list = [ | |
| 167 cgi.MiniFieldStorage('realname', 'someone') | |
| 168 ] | |
| 169 self.assertRaises( | |
| 170 Unauthorised, | |
| 171 self.server.put_element, 'user', '1', form | |
| 172 ) | |
| 173 | |
| 174 def testAuthDeniedPost(self): | |
| 175 """ | |
| 176 Test unauthorized POST request | |
| 177 """ | |
| 178 form = cgi.FieldStorage() | |
| 179 form.list = [ | |
| 180 cgi.MiniFieldStorage('username', 'blah') | |
| 181 ] | |
| 182 self.assertRaises( | |
| 183 Unauthorised, | |
| 184 self.server.post_collection, 'user', form | |
| 185 ) | |
| 186 | |
| 187 def testAuthAllowedPut(self): | |
| 188 """ | |
| 189 Test authorized PUT request | |
| 190 """ | |
| 191 self.db.setCurrentUser('admin') | |
| 192 form = cgi.FieldStorage() | |
| 193 form.list = [ | |
| 194 cgi.MiniFieldStorage('realname', 'someone') | |
| 195 ] | |
| 196 try: | |
| 197 try: | |
| 198 self.server.put_element('user', '2', form) | |
| 199 except Unauthorised, err: | |
| 200 self.fail('raised %s' % err) | |
| 201 finally: | |
| 202 self.db.setCurrentUser('joe') | |
| 203 | |
| 204 def testAuthAllowedPost(self): | |
| 205 """ | |
| 206 Test authorized POST request | |
| 207 """ | |
| 208 self.db.setCurrentUser('admin') | |
| 209 form = cgi.FieldStorage() | |
| 210 form.list = [ | |
| 211 cgi.MiniFieldStorage('username', 'blah') | |
| 212 ] | |
| 213 try: | |
| 214 try: | |
| 215 self.server.post_collection('user', form) | |
| 216 except Unauthorised, err: | |
| 217 self.fail('raised %s' % err) | |
| 218 finally: | |
| 219 self.db.setCurrentUser('joe') | |
| 220 | |
|
5585
8725c09802b8
Added test cases for the element URI methods
Chau Nguyen <dangchau1991@yahoo.com>
parents:
5583
diff
changeset
|
221 def testDeleteAttributeUri(self): |
|
8725c09802b8
Added test cases for the element URI methods
Chau Nguyen <dangchau1991@yahoo.com>
parents:
5583
diff
changeset
|
222 """ |
|
8725c09802b8
Added test cases for the element URI methods
Chau Nguyen <dangchau1991@yahoo.com>
parents:
5583
diff
changeset
|
223 Test Delete an attribute |
|
8725c09802b8
Added test cases for the element URI methods
Chau Nguyen <dangchau1991@yahoo.com>
parents:
5583
diff
changeset
|
224 """ |
|
8725c09802b8
Added test cases for the element URI methods
Chau Nguyen <dangchau1991@yahoo.com>
parents:
5583
diff
changeset
|
225 # create a new issue with userid 1 in the nosy list |
|
8725c09802b8
Added test cases for the element URI methods
Chau Nguyen <dangchau1991@yahoo.com>
parents:
5583
diff
changeset
|
226 issue_id = self.db.issue.create(title='foo', nosy=['1']) |
|
8725c09802b8
Added test cases for the element URI methods
Chau Nguyen <dangchau1991@yahoo.com>
parents:
5583
diff
changeset
|
227 |
|
8725c09802b8
Added test cases for the element URI methods
Chau Nguyen <dangchau1991@yahoo.com>
parents:
5583
diff
changeset
|
228 # remove the title and nosy |
|
8725c09802b8
Added test cases for the element URI methods
Chau Nguyen <dangchau1991@yahoo.com>
parents:
5583
diff
changeset
|
229 code, results = self.server.delete_attribute( |
|
8725c09802b8
Added test cases for the element URI methods
Chau Nguyen <dangchau1991@yahoo.com>
parents:
5583
diff
changeset
|
230 'issue', issue_id, 'title', {} |
|
8725c09802b8
Added test cases for the element URI methods
Chau Nguyen <dangchau1991@yahoo.com>
parents:
5583
diff
changeset
|
231 ) |
|
8725c09802b8
Added test cases for the element URI methods
Chau Nguyen <dangchau1991@yahoo.com>
parents:
5583
diff
changeset
|
232 self.assertEqual(code, 200) |
|
8725c09802b8
Added test cases for the element URI methods
Chau Nguyen <dangchau1991@yahoo.com>
parents:
5583
diff
changeset
|
233 |
|
8725c09802b8
Added test cases for the element URI methods
Chau Nguyen <dangchau1991@yahoo.com>
parents:
5583
diff
changeset
|
234 code, results = self.server.delete_attribute( |
|
8725c09802b8
Added test cases for the element URI methods
Chau Nguyen <dangchau1991@yahoo.com>
parents:
5583
diff
changeset
|
235 'issue', issue_id, 'nosy', {} |
|
8725c09802b8
Added test cases for the element URI methods
Chau Nguyen <dangchau1991@yahoo.com>
parents:
5583
diff
changeset
|
236 ) |
|
8725c09802b8
Added test cases for the element URI methods
Chau Nguyen <dangchau1991@yahoo.com>
parents:
5583
diff
changeset
|
237 self.assertEqual(code, 200) |
|
8725c09802b8
Added test cases for the element URI methods
Chau Nguyen <dangchau1991@yahoo.com>
parents:
5583
diff
changeset
|
238 |
|
8725c09802b8
Added test cases for the element URI methods
Chau Nguyen <dangchau1991@yahoo.com>
parents:
5583
diff
changeset
|
239 # verify the result |
|
8725c09802b8
Added test cases for the element URI methods
Chau Nguyen <dangchau1991@yahoo.com>
parents:
5583
diff
changeset
|
240 code, results = self.server.get_element('issue', issue_id, {}) |
|
8725c09802b8
Added test cases for the element URI methods
Chau Nguyen <dangchau1991@yahoo.com>
parents:
5583
diff
changeset
|
241 self.assertEqual(code, 200) |
|
8725c09802b8
Added test cases for the element URI methods
Chau Nguyen <dangchau1991@yahoo.com>
parents:
5583
diff
changeset
|
242 self.assertEqual(len(results['attributes']['nosy']), 0) |
|
8725c09802b8
Added test cases for the element URI methods
Chau Nguyen <dangchau1991@yahoo.com>
parents:
5583
diff
changeset
|
243 self.assertListEqual(results['attributes']['nosy'], []) |
|
8725c09802b8
Added test cases for the element URI methods
Chau Nguyen <dangchau1991@yahoo.com>
parents:
5583
diff
changeset
|
244 self.assertEqual(results['attributes']['title'], None) |
|
8725c09802b8
Added test cases for the element URI methods
Chau Nguyen <dangchau1991@yahoo.com>
parents:
5583
diff
changeset
|
245 |
| 5583 | 246 def testPatchAdd(self): |
| 247 """ | |
| 248 Test Patch op 'Add' | |
| 249 """ | |
| 250 # create a new issue with userid 1 in the nosy list | |
| 251 issue_id = self.db.issue.create(title='foo', nosy=['1']) | |
| 252 | |
| 253 # add userid 2 to the nosy list | |
| 254 form = cgi.FieldStorage() | |
| 255 form.list = [ | |
| 256 cgi.MiniFieldStorage('op', 'add'), | |
| 257 cgi.MiniFieldStorage('nosy', '2') | |
| 258 ] | |
| 259 code, results = self.server.patch_element('issue', issue_id, form) | |
| 260 self.assertEqual(code, 200) | |
| 261 | |
| 262 # verify the result | |
| 263 code, results = self.server.get_element('issue', issue_id, {}) | |
| 264 self.assertEqual(code, 200) | |
| 265 self.assertEqual(len(results['attributes']['nosy']), 2) | |
| 266 self.assertListEqual(results['attributes']['nosy'], ['1', '2']) | |
| 267 | |
| 268 def testPatchReplace(self): | |
| 269 """ | |
| 270 Test Patch op 'Replace' | |
| 271 """ | |
| 272 # create a new issue with userid 1 in the nosy list and status = 1 | |
| 273 issue_id = self.db.issue.create(title='foo', nosy=['1'], status='1') | |
| 274 | |
| 275 # replace userid 2 to the nosy list and status = 3 | |
| 276 form = cgi.FieldStorage() | |
| 277 form.list = [ | |
| 278 cgi.MiniFieldStorage('op', 'replace'), | |
| 279 cgi.MiniFieldStorage('nosy', '2'), | |
| 280 cgi.MiniFieldStorage('status', '3') | |
| 281 ] | |
| 282 code, results = self.server.patch_element('issue', issue_id, form) | |
| 283 self.assertEqual(code, 200) | |
| 284 | |
| 285 # verify the result | |
| 286 code, results = self.server.get_element('issue', issue_id, {}) | |
| 287 self.assertEqual(code, 200) | |
| 288 self.assertEqual(results['attributes']['status'], '3') | |
| 289 self.assertEqual(len(results['attributes']['nosy']), 1) | |
| 290 self.assertListEqual(results['attributes']['nosy'], ['2']) | |
| 291 | |
| 292 def testPatchRemoveAll(self): | |
| 293 """ | |
| 294 Test Patch Action 'Remove' | |
| 295 """ | |
| 296 # create a new issue with userid 1 in the nosy list | |
| 297 issue_id = self.db.issue.create(title='foo', nosy=['1', '2']) | |
| 298 | |
| 299 # remove the nosy list and the title | |
| 300 form = cgi.FieldStorage() | |
| 301 form.list = [ | |
| 302 cgi.MiniFieldStorage('op', 'remove'), | |
| 303 cgi.MiniFieldStorage('nosy', ''), | |
| 304 cgi.MiniFieldStorage('title', '') | |
| 305 ] | |
| 306 code, results = self.server.patch_element('issue', issue_id, form) | |
| 307 self.assertEqual(code, 200) | |
| 308 | |
| 309 # verify the result | |
| 310 code, results = self.server.get_element('issue', issue_id, {}) | |
| 311 self.assertEqual(code, 200) | |
| 312 self.assertEqual(results['attributes']['title'], None) | |
| 313 self.assertEqual(len(results['attributes']['nosy']), 0) | |
| 314 self.assertEqual(results['attributes']['nosy'], []) | |
| 315 | |
| 316 def test_suite(): | |
| 317 suite = unittest.TestSuite() | |
| 318 for l in list_backends(): | |
| 319 dct = dict(backend=l) | |
| 320 subcls = type(TestCase)('TestCase_%s' % l, (TestCase,), dct) | |
| 321 suite.addTest(unittest.makeSuite(subcls)) | |
| 322 return suite | |
| 323 | |
| 324 if __name__ == '__main__': | |
| 325 runner = unittest.TextTestRunner() | |
| 326 unittest.main(testRunner=runner) |