diff --git a/api.py b/api.py index ea23b59e8f7cd07257f107b70da5213e061d7d45..d130b0632cdd30733b548b9390d167d80ba635a2 100644 --- a/api.py +++ b/api.py @@ -18,6 +18,10 @@ def startup(): def read_root(): return {'success': True} +@app.put('/bounce_back') +def read_root(par1=None, par2=None): + return {'success': True, 'params': {'par1': par1, 'par2': par2}} + @app.get('/models') def list_active_models(): return session.describe_loaded_models() diff --git a/tests/test_api.py b/tests/test_api.py index e4091cb21e6807d4b596ad7d60ff6732ec14b9d1..ae2529f8455d741c8e544d4665d4db0da5fc7640 100644 --- a/tests/test_api.py +++ b/tests/test_api.py @@ -41,6 +41,12 @@ class TestApiFromAutomatedClient(TestServerBaseClass): resp = requests.get(self.uri, ) self.assertEqual(resp.status_code, 200) + def test_bounceback_parameters(self): + resp = requests.put(self.uri + 'bounce_back', params={'par1': 'hello'}) + self.assertEqual(resp.status_code, 200, resp.json()) + self.assertEqual(resp.json()['params']['par1'], 'hello', resp.json()) + self.assertEqual(resp.json()['params']['par2'], None, resp.json()) + def test_list_empty_loaded_models(self): resp = requests.get(self.uri + 'models') self.assertEqual(resp.status_code, 200)