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