From bba7dd8ee43a0315f11f6d4dd1e6dfe84efeae46 Mon Sep 17 00:00:00 2001 From: Felix Palta Date: Sun, 4 Dec 2016 23:43:07 +0300 Subject: [PATCH] Add test for drip with invalid numbytes parameter --- test_httpbin.py | 6 ++++++ 1 file changed, 6 insertions(+) diff --git a/test_httpbin.py b/test_httpbin.py index 4f6a322..a567e54 100755 --- a/test_httpbin.py +++ b/test_httpbin.py @@ -231,6 +231,12 @@ class HttpbinTestCase(unittest.TestCase): self.assertEqual(len(response.get_data()), 400) self.assertEqual(response.status_code, 200) + def test_drip_with_invalid_numbytes(self): + for bad_num in -1, 0: + uri = '/drip?numbytes={0}&duration=2&delay=1'.format(bad_num) + response = self.app.get(uri) + self.assertEqual(response.status_code, 400) + def test_drip_with_custom_code(self): response = self.app.get('/drip?numbytes=400&duration=2&code=500') self.assertEqual(response.content_length, 400)