diff --git a/test/test_graphql.py b/test/test_graphql.py index d8c2bc3da709ba3a99c921de7badf664fa3d401d..900285a53ee8a6c4f958445938aa73923885397e 100644 --- a/test/test_graphql.py +++ b/test/test_graphql.py @@ -21,21 +21,12 @@ class GraphQlTests(LiveServerTestCase): users = User.query.filter_by(username="graphqluser").delete() db_session.commit() + def test_query_create_and_get_user(self): username = "graphqluser" password = "randompassword" email = "user@example.com" - querystring = ''' - mutation{{ - createUser( - username: "{0}", - password: "{1}", - email: "{2}"){{ - user{{ - username, - email}}}}}} - '''.format(username, password, email).strip() - result = loads(self.client.execute(querystring)) + result = self._create_user(username, password, email) self.assertNotIn("error", result, "Creating user failed due to an error") self.assertIn("data", result) self.assertIn("createUser", result["data"]) @@ -57,3 +48,16 @@ class GraphQlTests(LiveServerTestCase): user = result["data"]["getUser"] self.assertEqual(user["username"], username) self.assertEqual(user["email"], email) + + def _create_user(self, username, password, email): + querystring = ''' + mutation{{ + createUser( + username: "{0}", + password: "{1}", + email: "{2}"){{ + user{{ + username, + email}}}}}} + '''.format(username, password, email).strip() + return loads(self.client.execute(querystring))