diff --git a/beat/web/libraries/tests/tests_api.py b/beat/web/libraries/tests/tests_api.py index 5cb8717487b5edfc3814190d503eaf8f043b2332..1a26fa1eeed7acf0667b12cb45ef336ce42d0696 100644 --- a/beat/web/libraries/tests/tests_api.py +++ b/beat/web/libraries/tests/tests_api.py @@ -835,7 +835,7 @@ class LibraryUpdate(LibrariesAPIBase): response = self.client.put(self.url) self.checkResponse(response, 200, content_type="application/json") - def test_successfull_update(self): + def test_successful_update(self): self.login_jackdoe() code = b"import numpy as np" @@ -867,7 +867,54 @@ class LibraryUpdate(LibrariesAPIBase): ) self.assertEqual(storage.code.load(), code) - def test_successfull_update_description_only(self): + def test_successful_update_with_specific_return_field(self): + self.login_jackdoe() + + code = b"""import numpy as np""" + + response = self.client.put( + self.url, + json.dumps( + { + "description": "blah", + "declaration": LibrariesAPIBase.UPDATE, + "code": code, + } + ), + content_type="application/json", + QUERY_STRING="fields=code", + ) + + self.checkResponse(response, 200, content_type="application/json") + answer = response.json() + self.assertEqual(len(answer), 1) + self.assertTrue("code" in answer) + + def test_successful_update_with_specific_return_several_fields(self): + self.login_jackdoe() + + code = b"""import numpy as np""" + + response = self.client.put( + self.url, + json.dumps( + { + "description": "blah", + "declaration": LibrariesAPIBase.UPDATE, + "code": code, + } + ), + content_type="application/json", + QUERY_STRING="fields=code,description", + ) + + self.checkResponse(response, 200, content_type="application/json") + answer = response.json() + self.assertEqual(len(answer), 2) + self.assertTrue("code" in answer) + self.assertTrue("description" in answer) + + def test_successful_update_description_only(self): self.login_jackdoe() response = self.client.put( @@ -883,7 +930,7 @@ class LibraryUpdate(LibrariesAPIBase): ) self.assertEqual(library.description, b"blah") - def test_successfull_update_declaration_only(self): + def test_successful_update_declaration_only(self): self.login_jackdoe() response = self.client.put( @@ -905,7 +952,7 @@ class LibraryUpdate(LibrariesAPIBase): json.loads(storage.json.load()), json.loads(LibrariesAPIBase.UPDATE) ) - def test_successfull_update_code_only(self): + def test_successful_update_code_only(self): self.login_jackdoe() code = b"import numpy as np"