Commit fcd4d66

committed

Allow a wordpress api request to specify certain status codes it wants to allow/handle in response.

1 parent 60fec3f commit fcd4d66Copy full SHA for fcd4d66

File tree

+21

-1

lines changed

+21

-1

lines changed

Lines changed: 18 additions & 0 deletions

Original file line number	Diff line number	Diff line change
`@@ -438,6 +438,24 @@ def test_APIPostBadData(self):`
`438`	`438`	`with self.assertRaises(UserWarning):`
`439`	`439`	`self.wpapi.post('posts', data)`
`440`	`440`
	`441`	`+ def test_APIPostBadDataHandleBadStatus(self):`
	`442`	`+ """`
	`443`	`+ Test handling explicitly a bad status code for a request.`
	`444`	`+ """`
	`445`	`+ nonce = "%f\u00ae" % random.random()`
	`446`	`+`
	`447`	`+ data = {`
	`448`	`+ 'a': nonce`
	`449`	`+ }`
	`450`	`+`
	`451`	`+ response = self.wpapi.post('posts', data, handle_status_codes=[400])`
	`452`	`+ self.assertEqual(response.status_code, 400)`
	`453`	`+`
	`454`	`+ # If we don't specify a correct status code to handle we should`
	`455`	`+ # still expect an exception`
	`456`	`+ with self.assertRaises(UserWarning):`
	`457`	`+ self.wpapi.post('posts', data, handle_status_codes=[404])`
	`458`	`+`
`441`	`459`	`def test_APIPostMedia(self):`
`442`	`460`	`img_path = 'tests/data/test.jpg'`
`443`	`461`	`with open(img_path, 'rb') as test_file:`

Lines changed: 3 additions & 1 deletion

Original file line number	Diff line number	Diff line change
`@@ -219,6 +219,8 @@ def __request(self, method, endpoint, data, **kwargs):`
`219`	`219`	`# enforce utf-8 encoded binary`
`220`	`220`	`data = StrUtils.to_binary(data)`
`221`	`221`
	`222`	`+ handle_status_codes = kwargs.pop('handle_status_codes', [])`
	`223`	`+`
`222`	`224`	`response = self.requester.request(`
`223`	`225`	`method=method,`
`224`	`226`	`url=endpoint_url,`
`@@ -227,7 +229,7 @@ def __request(self, method, endpoint, data, **kwargs):`
`227`	`229`	`**kwargs`
`228`	`230`	`)`
`229`	`231`
`230`		`- if response.status_code not in [200, 201, 202]:`
	`232`	`+ if response.status_code not in [200, 201, 202]+handle_status_codes:`
`231`	`233`	`self.request_post_mortem(response)`
`232`	`234`
`233`	`235`	`return response`

Comments

(0)