Add assertion

backend/test/test_graphql.py

@@ -172,15 +172,7 @@ class GraphQlTests(LiveServerTestCase):
               role{{name}}
             }}}}'''.format(username, role_name)
         self.client.execute(querystring)
-        querystring = '''
-        mutation{{
-          removeRoleFromUser(
-            username: "{0}",
-            role: "{1}"){{ ok }}
-            }}'''.format(username, role_name)
-        result = loads(self.client.execute(querystring))
-        self.assertTrue(result["data"]["removeRoleFromUser"]["ok"])
-        querystring = '''
+        get_user_querystring = '''
         {{
           getUser(username: "{0}"){{
             roles{{
@@ -189,7 +181,17 @@ class GraphQlTests(LiveServerTestCase):
             }}
         }}
         '''.format(username).strip()
+        result = loads(self.client.execute(get_user_querystring))
+        self.assertEqual(1, len(result["data"]["getUser"]["roles"]["edges"]))
+        querystring = '''
+        mutation{{
+          removeRoleFromUser(
+            username: "{0}",
+            role: "{1}"){{ ok }}
+            }}'''.format(username, role_name)
         result = loads(self.client.execute(querystring))
+        self.assertTrue(result["data"]["removeRoleFromUser"]["ok"])
+        result = loads(self.client.execute(get_user_querystring))
         self.assertEqual(0, len(result["data"]["getUser"]["roles"]["edges"]))