def test_kitchen_sink(self):
parser = Parser()
doc = parser.parse_query(
"""
query queryName($foo: ComplexType, $site: Site) {
whoever123is: node(id: [123, 456]) {
id ,
... on User @defer {
field2 {
id ,
alias: field1(first:10, after:$foo,) @include(if: $foo) {
id,
...frag
}
}
}
}
}
"""
)
self.assertEqual(doc.definitions_size, 1)
self.assertEqual(doc.definitions[0].operation, "query")
self.assertEqual(doc.definitions[0].name, "queryName")
self.assertEqual(doc.definitions[0].variables_size, 2)
self.assertEqual(doc.definitions[0].definitions[0].variable.value, "foo")
self.assertEqual(doc.definitions[0].definitions[0].type, "ComplexType")
self.assertEqual(doc.definitions[0].directives_size, 0)
self.assertEqual(doc.definitions[0].selection_set.selection_set_size, 1)
self.assertEqual(doc.definitions[0].selection_set.fields[0].arguments_size, 1)
self.assertEqual(doc.definitions[0].selection_set.fields[0].arguments["id"], [123, 456])
self.assertEqual(doc.definitions[0].selection_set.fields[0].name, "node")
self.assertEqual(doc.definitions[0].selection_set.fields[0].selection_set.selection_set_size, 2)
self.assertEqual(doc.definitions[0].selection_set.fields[0].selection_set.fields[0].arguments_size, 0)
self.assertEqual(doc.definitions[0].selection_set.fields[0].selection_set.fields[0].name, "id")
self.assertEqual(doc.definitions[0].selection_set.fields[0].selection_set.fields[1].condition, "User")
def test_with_args_default(self):
parser = Parser()
doc = parser.parse_query("""query myQuery($someTest: Boolean = 1) {
experimentalField @skip(if: $someTest)
}
""")
self.assertEqual(doc.definitions_size, 1)
self.assertIsInstance(doc.definitions[0], Operation)
self.assertEqual(doc.definitions[0].operation, "query")
self.assertEqual(doc.definitions[0].name, "myQuery")
self.assertEqual(doc.definitions[0].variables_size, 1)
self.assertEqual(doc.definitions[0].definitions[0].variable.value,
"someTest")
self.assertEqual(doc.definitions[0].definitions[0].type, "Boolean")
self.assertEqual(doc.definitions[0].definitions[0].default_value, True)
self.assertEqual(doc.definitions[0].directives_size, 0)
self.assertEqual(doc.definitions[0].selection_set.selection_set_size,
1)
self.assertEqual(
doc.definitions[0].selection_set.fields[0].arguments_size, 0)
self.assertEqual(doc.definitions[0].selection_set.fields[0].name,
"experimentalField")
self.assertEqual(
doc.definitions[0].selection_set.fields[0].directives_size, 1)
self.assertEqual(
doc.definitions[0].selection_set.fields[0].directives[0].name,
"skip")
self.assertEqual(
doc.definitions[0].selection_set.fields[0].directives[0].
arguments["if"], "someTest")
def test_with_args(self):
parser = Parser()
doc = parser.parse_query(
"""query RootQuery {
user(id: 12356) {
id,
name,
isViewerFriend
}
}
"""
)
self.assertEqual(doc.definitions_size, 1)
self.assertEqual(doc.definitions[0].operation, "query")
self.assertEqual(doc.definitions[0].name, "RootQuery")
self.assertEqual(doc.definitions[0].variables_size, 0)
self.assertEqual(doc.definitions[0].directives_size, 0)
self.assertEqual(doc.definitions[0].selection_set.selection_set_size, 1)
self.assertEqual(doc.definitions[0].selection_set.fields[0].arguments_size, 1)
self.assertEqual(doc.definitions[0].selection_set.fields[0].arguments, {"id": 12356})
self.assertEqual(doc.definitions[0].selection_set.fields[0].name, "user")
self.assertEqual(doc.definitions[0].selection_set.fields[0].selection_set.selection_set_size, 3)
self.assertEqual(doc.definitions[0].selection_set.fields[0].selection_set.fields[0].arguments_size, 0)
self.assertEqual(doc.definitions[0].selection_set.fields[0].selection_set.fields[0].name, "id")
self.assertEqual(doc.definitions[0].selection_set.fields[0].selection_set.fields[1].name, "name")
self.assertEqual(doc.definitions[0].selection_set.fields[0].selection_set.fields[2].name, "isViewerFriend")
def test_with_object_arg(self):
parser = Parser()
doc = parser.parse_query("""query RootQuery {
user(details: { name: "Hello world", score: 1.0 }) {
id
}
}
""")
self.assertEqual(doc.definitions_size, 1)
self.assertEqual(doc.definitions[0].operation, "query")
self.assertEqual(doc.definitions[0].name, "RootQuery")
self.assertEqual(doc.definitions[0].variables_size, 0)
self.assertEqual(doc.definitions[0].directives_size, 0)
self.assertEqual(doc.definitions[0].selection_set.selection_set_size,
1)
self.assertEqual(
doc.definitions[0].selection_set.fields[0].arguments_size, 1)
self.assertEqual(
doc.definitions[0].selection_set.fields[0].arguments['details']
['name'], "Hello world")
self.assertEqual(
doc.definitions[0].selection_set.fields[0].arguments['details']
['score'], 1.0)
self.assertEqual(doc.definitions[0].selection_set.fields[0].name,
"user")
self.assertEqual(
doc.definitions[0].selection_set.fields[0].selection_set.
selection_set_size, 1)
self.assertEqual(
doc.definitions[0].selection_set.fields[0].selection_set.fields[0].
arguments_size, 0)
self.assertEqual(
doc.definitions[0].selection_set.fields[0].selection_set.fields[0].
name, "id")
def test_basic(self):
parser = Parser()
doc = parser.parse_query("""query RootQuery{
viewer{
id
}
}
""")
self.assertEqual(doc.definitions_size, 1)
self.assertEqual(doc.definitions[0].operation, "query")
self.assertEqual(doc.definitions[0].name, "RootQuery")
self.assertEqual(doc.definitions[0].variables_size, 0)
self.assertEqual(doc.definitions[0].directives_size, 0)
self.assertEqual(doc.definitions[0].selection_set.selection_set_size,
1)
self.assertEqual(
doc.definitions[0].selection_set.fields[0].arguments_size, 0)
self.assertEqual(doc.definitions[0].selection_set.fields[0].name,
"viewer")
self.assertEqual(
doc.definitions[0].selection_set.fields[0].selection_set.
selection_set_size, 1)
self.assertEqual(
doc.definitions[0].selection_set.fields[0].selection_set.fields[0].
arguments_size, 0)
self.assertEqual(
doc.definitions[0].selection_set.fields[0].selection_set.fields[0].
name, "id")
def test_with_array_arg(self):
parser = Parser()
doc = parser.parse_query("""query RootQuery {
user(status: [1, 2]) {
id
}
}
""")
self.assertEqual(doc.definitions_size, 1)
self.assertEqual(doc.definitions[0].operation, "query")
self.assertEqual(doc.definitions[0].name, "RootQuery")
self.assertEqual(doc.definitions[0].variables_size, 0)
self.assertEqual(doc.definitions[0].directives_size, 0)
self.assertEqual(doc.definitions[0].selection_set.selection_set_size,
1)
self.assertEqual(
doc.definitions[0].selection_set.fields[0].arguments_size, 1)
self.assertEqual(doc.definitions[0].selection_set.fields[0].arguments,
{"status": [1, 2]})
self.assertEqual(doc.definitions[0].selection_set.fields[0].name,
"user")
self.assertEqual(
doc.definitions[0].selection_set.fields[0].selection_set.
selection_set_size, 1)
self.assertEqual(
doc.definitions[0].selection_set.fields[0].selection_set.fields[0].
arguments_size, 0)
self.assertEqual(
doc.definitions[0].selection_set.fields[0].selection_set.fields[0].
name, "id")
def test_with_float_arg(self):
parser = Parser()
doc = parser.parse_query("""query RootQuery {
user(age: 21.2) {
id
}
}
""")
self.assertEqual(doc.definitions_size, 1)
self.assertEqual(doc.definitions[0].operation, "query")
self.assertEqual(doc.definitions[0].name, "RootQuery")
self.assertEqual(doc.definitions[0].variables_size, 0)
self.assertEqual(doc.definitions[0].directives_size, 0)
self.assertEqual(doc.definitions[0].selection_set.selection_set_size,
1)
self.assertEqual(
doc.definitions[0].selection_set.fields[0].arguments_size, 1)
self.assertEqual(doc.definitions[0].selection_set.fields[0].arguments,
{"age": 21.2})
self.assertEqual(doc.definitions[0].selection_set.fields[0].name,
"user")
self.assertEqual(
doc.definitions[0].selection_set.fields[0].selection_set.
selection_set_size, 1)
self.assertEqual(
doc.definitions[0].selection_set.fields[0].selection_set.fields[0].
arguments_size, 0)
self.assertEqual(
doc.definitions[0].selection_set.fields[0].selection_set.fields[0].
name, "id")
def test_with_variables(self):
parser = Parser()
doc = parser.parse_query(
"""query getZuckProfile($devicePicSize: Int) {
user(id: 4) {
id
name
profilePic(size: $devicePicSize)
}
}
"""
)
self.assertEqual(doc.definitions_size, 1)
self.assertEqual(doc.definitions[0].operation, "query")
self.assertEqual(doc.definitions[0].name, "getZuckProfile")
self.assertEqual(doc.definitions[0].variables_size, 1)
self.assertEqual(doc.definitions[0].definitions[0].variable.value, "devicePicSize")
self.assertEqual(doc.definitions[0].definitions[0].type, "Int")
self.assertEqual(doc.definitions[0].directives_size, 0)
self.assertEqual(doc.definitions[0].selection_set.selection_set_size, 1)
self.assertEqual(doc.definitions[0].selection_set.fields[0].arguments_size, 1)
self.assertEqual(doc.definitions[0].selection_set.fields[0].arguments["id"], 4)
self.assertEqual(doc.definitions[0].selection_set.fields[0].name, "user")
self.assertEqual(doc.definitions[0].selection_set.fields[0].selection_set.selection_set_size, 3)
self.assertEqual(doc.definitions[0].selection_set.fields[0].selection_set.fields[0].arguments_size, 0)
self.assertEqual(doc.definitions[0].selection_set.fields[0].selection_set.fields[0].name, "id")
self.assertEqual(doc.definitions[0].selection_set.fields[0].selection_set.fields[1].arguments_size, 0)
self.assertEqual(doc.definitions[0].selection_set.fields[0].selection_set.fields[1].name, "name")
self.assertEqual(doc.definitions[0].selection_set.fields[0].selection_set.fields[2].arguments_size, 1)
self.assertEqual(doc.definitions[0].selection_set.fields[0].selection_set.fields[2].name, "profilePic")
self.assertEqual(
doc.definitions[0].selection_set.fields[0].selection_set.fields[2].arguments["size"], "devicePicSize"
)
def test_raises_exceptions(self):
parser = Parser()
with self.assertRaises(GraphQLException):
doc = parser.parse_query(
"""
abcdef
"""
)
def test_with_fragments(self):
parser = Parser()
doc = parser.parse_query(
"""query withFragments {
user(id: 4) {
friends(first: 10) {
...friendFields
}
mutualFriends(first: 10) {
...friendFields
}
}
}
fragment friendFields on User {
id
name
profilePic(size: 50)
}
"""
)
self.assertEqual(doc.definitions_size, 2)
self.assertIsInstance(doc.definitions[0], Operation)
self.assertEqual(doc.definitions[0].operation, "query")
self.assertEqual(doc.definitions[0].name, "withFragments")
self.assertEqual(doc.definitions[0].variables_size, 0)
self.assertEqual(doc.definitions[0].directives_size, 0)
self.assertEqual(doc.definitions[0].selection_set.selection_set_size, 1)
self.assertEqual(doc.definitions[0].selection_set.fields[0].arguments_size, 1)
self.assertEqual(doc.definitions[0].selection_set.fields[0].arguments["id"], 4)
self.assertEqual(doc.definitions[0].selection_set.fields[0].name, "user")
self.assertEqual(doc.definitions[0].selection_set.fields[0].selection_set.selection_set_size, 2)
self.assertEqual(doc.definitions[0].selection_set.fields[0].selection_set.fields[0].arguments_size, 1)
self.assertEqual(doc.definitions[0].selection_set.fields[0].selection_set.fields[0].arguments, {"first": 10})
self.assertEqual(doc.definitions[0].selection_set.fields[0].selection_set.fields[0].name, "friends")
self.assertEqual(
doc.definitions[0].selection_set.fields[0].selection_set.fields[0].selection_set.selection_set_size, 1
)
self.assertEqual(
doc.definitions[0].selection_set.fields[0].selection_set.fields[0].selection_set.frags[0], "friendFields"
)
self.assertEqual(doc.definitions[0].selection_set.fields[0].selection_set.fields[1].arguments_size, 1)
self.assertEqual(doc.definitions[0].selection_set.fields[0].selection_set.fields[0].arguments, {"first": 10})
self.assertEqual(doc.definitions[0].selection_set.fields[0].selection_set.fields[1].name, "mutualFriends")
self.assertEqual(
doc.definitions[0].selection_set.fields[0].selection_set.fields[1].selection_set.frags[0], "friendFields"
)
self.assertIsInstance(doc.definitions[1], Fragment)
self.assertEqual(doc.definitions[1].name, "friendFields")
self.assertEqual(doc.definitions[1].selection_set.selection_set_size, 3)
self.assertEqual(doc.definitions[1].selection_set.fields[0].name, "id")
self.assertEqual(doc.definitions[1].selection_set.fields[1].name, "name")
self.assertEqual(doc.definitions[1].selection_set.fields[2].name, "profilePic")
self.assertEqual(doc.definitions[1].selection_set.fields[2].arguments_size, 1)
self.assertEqual(doc.definitions[1].selection_set.fields[2].arguments, {"size": 50})
def test_mutation(self):
parser = Parser()
doc = parser.parse_query("""
mutation likeStory {
like(story: 123) @defer {
story {
id
}
}
}
""")
self.assertEqual(doc.definitions_size, 1)
self.assertEqual(doc.definitions[0].operation, "mutation")
def test_with_variables(self):
parser = Parser()
doc = parser.parse_query("""query getZuckProfile($devicePicSize: Int) {
user(id: 4) {
id
name
profilePic(size: $devicePicSize)
}
}
""")
self.assertEqual(doc.definitions_size, 1)
self.assertEqual(doc.definitions[0].operation, "query")
self.assertEqual(doc.definitions[0].name, "getZuckProfile")
self.assertEqual(doc.definitions[0].variables_size, 1)
self.assertEqual(doc.definitions[0].definitions[0].variable.value,
"devicePicSize")
self.assertEqual(doc.definitions[0].definitions[0].type, "Int")
self.assertEqual(doc.definitions[0].directives_size, 0)
self.assertEqual(doc.definitions[0].selection_set.selection_set_size,
1)
self.assertEqual(
doc.definitions[0].selection_set.fields[0].arguments_size, 1)
self.assertEqual(
doc.definitions[0].selection_set.fields[0].arguments["id"], 4)
self.assertEqual(doc.definitions[0].selection_set.fields[0].name,
"user")
self.assertEqual(
doc.definitions[0].selection_set.fields[0].selection_set.
selection_set_size, 3)
self.assertEqual(
doc.definitions[0].selection_set.fields[0].selection_set.fields[0].
arguments_size, 0)
self.assertEqual(
doc.definitions[0].selection_set.fields[0].selection_set.fields[0].
name, "id")
self.assertEqual(
doc.definitions[0].selection_set.fields[0].selection_set.fields[1].
arguments_size, 0)
self.assertEqual(
doc.definitions[0].selection_set.fields[0].selection_set.fields[1].
name, "name")
self.assertEqual(
doc.definitions[0].selection_set.fields[0].selection_set.fields[2].
arguments_size, 1)
self.assertEqual(
doc.definitions[0].selection_set.fields[0].selection_set.fields[2].
name, "profilePic")
self.assertEqual(
doc.definitions[0].selection_set.fields[0].selection_set.fields[2].
arguments["size"], "devicePicSize")
def test_with_type_conditions(self):
parser = Parser()
doc = parser.parse_query(
"""query FragmentTyping {
profiles(handles: ["zuck", "cocacola"]) {
handle
...userFragment
...pageFragment
}
}
fragment userFragment on User {
friends {
count
}
}
fragment pageFragment on Page {
likers {
count
}
}
"""
)
self.assertEqual(doc.definitions_size, 3)
self.assertIsInstance(doc.definitions[0], Operation)
self.assertEqual(doc.definitions[0].operation, "query")
self.assertEqual(doc.definitions[0].name, "FragmentTyping")
self.assertEqual(doc.definitions[0].variables_size, 0)
self.assertEqual(doc.definitions[0].directives_size, 0)
self.assertEqual(doc.definitions[0].selection_set.selection_set_size, 1)
self.assertEqual(doc.definitions[0].selection_set.fields[0].arguments_size, 1)
self.assertEqual(doc.definitions[0].selection_set.fields[0].arguments["handles"], ["zuck", "cocacola"])
self.assertEqual(doc.definitions[0].selection_set.fields[0].name, "profiles")
self.assertEqual(doc.definitions[0].selection_set.fields[0].selection_set.selection_set_size, 3)
self.assertEqual(doc.definitions[0].selection_set.fields[0].selection_set.fields[0].arguments_size, 0)
self.assertEqual(doc.definitions[0].selection_set.fields[0].selection_set.fields[0].name, "handle")
self.assertEqual(doc.definitions[0].selection_set.fields[0].selection_set.frags[0], "userFragment")
self.assertEqual(doc.definitions[0].selection_set.fields[0].selection_set.frags[1], "pageFragment")
self.assertIsInstance(doc.definitions[1], Fragment)
self.assertEqual(doc.definitions[1].name, "userFragment")
self.assertEqual(doc.definitions[1].condition, "User")
self.assertEqual(doc.definitions[1].selection_set.selection_set_size, 1)
self.assertEqual(doc.definitions[1].selection_set.fields[0].name, "friends")
self.assertIsInstance(doc.definitions[2], Fragment)
self.assertEqual(doc.definitions[2].name, "pageFragment")
self.assertEqual(doc.definitions[2].condition, "Page")
self.assertEqual(doc.definitions[2].selection_set.selection_set_size, 1)
self.assertEqual(doc.definitions[2].selection_set.fields[0].name, "likers")
def test_mutation(self):
parser = Parser()
doc = parser.parse_query(
"""
mutation likeStory {
like(story: 123) @defer {
story {
id
}
}
}
"""
)
self.assertEqual(doc.definitions_size, 1)
self.assertEqual(doc.definitions[0].operation, "mutation")
def test_kitchen_sink(self):
parser = Parser()
doc = parser.parse_query("""
query queryName($foo: ComplexType, $site: Site) {
whoever123is: node(id: [123, 456]) {
id ,
... on User @defer {
field2 {
id ,
alias: field1(first:10, after:$foo,) @include(if: $foo) {
id,
...frag
}
}
}
}
}
""")
self.assertEqual(doc.definitions_size, 1)
self.assertEqual(doc.definitions[0].operation, "query")
self.assertEqual(doc.definitions[0].name, "queryName")
self.assertEqual(doc.definitions[0].variables_size, 2)
self.assertEqual(doc.definitions[0].definitions[0].variable.value,
"foo")
self.assertEqual(doc.definitions[0].definitions[0].type, "ComplexType")
self.assertEqual(doc.definitions[0].directives_size, 0)
self.assertEqual(doc.definitions[0].selection_set.selection_set_size,
1)
self.assertEqual(
doc.definitions[0].selection_set.fields[0].arguments_size, 1)
self.assertEqual(
doc.definitions[0].selection_set.fields[0].arguments["id"],
[123, 456])
self.assertEqual(doc.definitions[0].selection_set.fields[0].name,
"node")
self.assertEqual(
doc.definitions[0].selection_set.fields[0].selection_set.
selection_set_size, 2)
self.assertEqual(
doc.definitions[0].selection_set.fields[0].selection_set.fields[0].
arguments_size, 0)
self.assertEqual(
doc.definitions[0].selection_set.fields[0].selection_set.fields[0].
name, "id")
self.assertEqual(
doc.definitions[0].selection_set.fields[0].selection_set.fields[1].
condition, "User")
def test_with_web_example(self):
parser = Parser()
doc = parser.parse_query(
"""query RootQuery {
user(id: 3500401) {
id,
name,
isViewerFriend,
profilePicture(size: 50) {
uri,
width,
height
}
}
}
"""
)
self.assertEqual(doc.definitions_size, 1)
self.assertEqual(doc.definitions[0].operation, "query")
self.assertEqual(doc.definitions[0].name, "RootQuery")
self.assertEqual(doc.definitions[0].variables_size, 0)
self.assertEqual(doc.definitions[0].directives_size, 0)
self.assertEqual(doc.definitions[0].selection_set.selection_set_size, 1)
self.assertEqual(doc.definitions[0].selection_set.fields[0].arguments_size, 1)
self.assertEqual(doc.definitions[0].selection_set.fields[0].arguments, {"id": 3500401})
self.assertEqual(doc.definitions[0].selection_set.fields[0].name, "user")
self.assertEqual(doc.definitions[0].selection_set.fields[0].selection_set.selection_set_size, 4)
self.assertEqual(doc.definitions[0].selection_set.fields[0].selection_set.fields[0].arguments_size, 0)
self.assertEqual(doc.definitions[0].selection_set.fields[0].selection_set.fields[0].name, "id")
self.assertEqual(doc.definitions[0].selection_set.fields[0].selection_set.fields[1].name, "name")
self.assertEqual(doc.definitions[0].selection_set.fields[0].selection_set.fields[2].name, "isViewerFriend")
self.assertEqual(doc.definitions[0].selection_set.fields[0].selection_set.fields[3].name, "profilePicture")
self.assertEqual(doc.definitions[0].selection_set.fields[0].selection_set.fields[3].arguments_size, 1)
self.assertEqual(doc.definitions[0].selection_set.fields[0].selection_set.fields[3].arguments, {"size": 50})
self.assertEqual(
doc.definitions[0].selection_set.fields[0].selection_set.fields[3].selection_set.selection_set_size, 3
)
self.assertEqual(
doc.definitions[0].selection_set.fields[0].selection_set.fields[3].selection_set.fields[0].name, "uri"
)
self.assertEqual(
doc.definitions[0].selection_set.fields[0].selection_set.fields[3].selection_set.fields[1].name, "width"
)
self.assertEqual(
doc.definitions[0].selection_set.fields[0].selection_set.fields[3].selection_set.fields[2].name, "height"
)
def test_basic(self):
parser = Parser()
doc = parser.parse_query(
"""query RootQuery{
viewer{
id
}
}
"""
)
self.assertEqual(doc.definitions_size, 1)
self.assertEqual(doc.definitions[0].operation, "query")
self.assertEqual(doc.definitions[0].name, "RootQuery")
self.assertEqual(doc.definitions[0].variables_size, 0)
self.assertEqual(doc.definitions[0].directives_size, 0)
self.assertEqual(doc.definitions[0].selection_set.selection_set_size, 1)
self.assertEqual(doc.definitions[0].selection_set.fields[0].arguments_size, 0)
self.assertEqual(doc.definitions[0].selection_set.fields[0].name, "viewer")
self.assertEqual(doc.definitions[0].selection_set.fields[0].selection_set.selection_set_size, 1)
self.assertEqual(doc.definitions[0].selection_set.fields[0].selection_set.fields[0].arguments_size, 0)
self.assertEqual(doc.definitions[0].selection_set.fields[0].selection_set.fields[0].name, "id")
def test_with_directives(self):
parser = Parser()
doc = parser.parse_query(
"""query myQuery($someTest: Boolean) {
experimentalField @skip(if: $someTest)
}
"""
)
self.assertEqual(doc.definitions_size, 1)
self.assertIsInstance(doc.definitions[0], Operation)
self.assertEqual(doc.definitions[0].operation, "query")
self.assertEqual(doc.definitions[0].name, "myQuery")
self.assertEqual(doc.definitions[0].variables_size, 1)
self.assertEqual(doc.definitions[0].definitions[0].variable.value, "someTest")
self.assertEqual(doc.definitions[0].definitions[0].type, "Boolean")
self.assertEqual(doc.definitions[0].directives_size, 0)
self.assertEqual(doc.definitions[0].selection_set.selection_set_size, 1)
self.assertEqual(doc.definitions[0].selection_set.fields[0].arguments_size, 0)
self.assertEqual(doc.definitions[0].selection_set.fields[0].name, "experimentalField")
self.assertEqual(doc.definitions[0].selection_set.fields[0].directives_size, 1)
self.assertEqual(doc.definitions[0].selection_set.fields[0].directives[0].name, "skip")
self.assertEqual(doc.definitions[0].selection_set.fields[0].directives[0].arguments["if"], "someTest")
def test_with_array_arg(self):
parser = Parser()
doc = parser.parse_query(
"""query RootQuery {
user(status: [1, 2]) {
id
}
}
"""
)
self.assertEqual(doc.definitions_size, 1)
self.assertEqual(doc.definitions[0].operation, "query")
self.assertEqual(doc.definitions[0].name, "RootQuery")
self.assertEqual(doc.definitions[0].variables_size, 0)
self.assertEqual(doc.definitions[0].directives_size, 0)
self.assertEqual(doc.definitions[0].selection_set.selection_set_size, 1)
self.assertEqual(doc.definitions[0].selection_set.fields[0].arguments_size, 1)
self.assertEqual(doc.definitions[0].selection_set.fields[0].arguments, {"status": [1, 2]})
self.assertEqual(doc.definitions[0].selection_set.fields[0].name, "user")
self.assertEqual(doc.definitions[0].selection_set.fields[0].selection_set.selection_set_size, 1)
self.assertEqual(doc.definitions[0].selection_set.fields[0].selection_set.fields[0].arguments_size, 0)
self.assertEqual(doc.definitions[0].selection_set.fields[0].selection_set.fields[0].name, "id")
def test_with_float_arg(self):
parser = Parser()
doc = parser.parse_query(
"""query RootQuery {
user(age: 21.2) {
id
}
}
"""
)
self.assertEqual(doc.definitions_size, 1)
self.assertEqual(doc.definitions[0].operation, "query")
self.assertEqual(doc.definitions[0].name, "RootQuery")
self.assertEqual(doc.definitions[0].variables_size, 0)
self.assertEqual(doc.definitions[0].directives_size, 0)
self.assertEqual(doc.definitions[0].selection_set.selection_set_size, 1)
self.assertEqual(doc.definitions[0].selection_set.fields[0].arguments_size, 1)
self.assertEqual(doc.definitions[0].selection_set.fields[0].arguments, {"age": 21.2})
self.assertEqual(doc.definitions[0].selection_set.fields[0].name, "user")
self.assertEqual(doc.definitions[0].selection_set.fields[0].selection_set.selection_set_size, 1)
self.assertEqual(doc.definitions[0].selection_set.fields[0].selection_set.fields[0].arguments_size, 0)
self.assertEqual(doc.definitions[0].selection_set.fields[0].selection_set.fields[0].name, "id")
def test_with_args(self):
parser = Parser()
doc = parser.parse_query("""query RootQuery {
user(id: 12356) {
id,
name,
isViewerFriend
}
}
""")
self.assertEqual(doc.definitions_size, 1)
self.assertEqual(doc.definitions[0].operation, "query")
self.assertEqual(doc.definitions[0].name, "RootQuery")
self.assertEqual(doc.definitions[0].variables_size, 0)
self.assertEqual(doc.definitions[0].directives_size, 0)
self.assertEqual(doc.definitions[0].selection_set.selection_set_size,
1)
self.assertEqual(
doc.definitions[0].selection_set.fields[0].arguments_size, 1)
self.assertEqual(doc.definitions[0].selection_set.fields[0].arguments,
{"id": 12356})
self.assertEqual(doc.definitions[0].selection_set.fields[0].name,
"user")
self.assertEqual(
doc.definitions[0].selection_set.fields[0].selection_set.
selection_set_size, 3)
self.assertEqual(
doc.definitions[0].selection_set.fields[0].selection_set.fields[0].
arguments_size, 0)
self.assertEqual(
doc.definitions[0].selection_set.fields[0].selection_set.fields[0].
name, "id")
self.assertEqual(
doc.definitions[0].selection_set.fields[0].selection_set.fields[1].
name, "name")
self.assertEqual(
doc.definitions[0].selection_set.fields[0].selection_set.fields[2].
name, "isViewerFriend")
def test_with_object_arg(self):
parser = Parser()
doc = parser.parse_query(
"""query RootQuery {
user(details: { name: "Hello world", score: 1.0 }) {
id
}
}
"""
)
self.assertEqual(doc.definitions_size, 1)
self.assertEqual(doc.definitions[0].operation, "query")
self.assertEqual(doc.definitions[0].name, "RootQuery")
self.assertEqual(doc.definitions[0].variables_size, 0)
self.assertEqual(doc.definitions[0].directives_size, 0)
self.assertEqual(doc.definitions[0].selection_set.selection_set_size, 1)
self.assertEqual(doc.definitions[0].selection_set.fields[0].arguments_size, 1)
self.assertEqual(doc.definitions[0].selection_set.fields[0].arguments["details"]["name"], "Hello world")
self.assertEqual(doc.definitions[0].selection_set.fields[0].arguments["details"]["score"], 1.0)
self.assertEqual(doc.definitions[0].selection_set.fields[0].name, "user")
self.assertEqual(doc.definitions[0].selection_set.fields[0].selection_set.selection_set_size, 1)
self.assertEqual(doc.definitions[0].selection_set.fields[0].selection_set.fields[0].arguments_size, 0)
self.assertEqual(doc.definitions[0].selection_set.fields[0].selection_set.fields[0].name, "id")
def test_with_nested_fragments(self):
parser = Parser()
doc = parser.parse_query("""query withNestedFragments {
user(id: 4) {
friends(first: 10) {
...friendFields
}
mutualFriends(first: 10) {
...friendFields
}
}
}
fragment friendFields on User {
id
name
...standardProfilePic
}
fragment standardProfilePic on User {
profilePic(size: 50)
}
""")
self.assertEqual(doc.definitions_size, 3)
self.assertIsInstance(doc.definitions[0], Operation)
self.assertEqual(doc.definitions[0].operation, "query")
self.assertEqual(doc.definitions[0].name, "withNestedFragments")
self.assertEqual(doc.definitions[0].variables_size, 0)
self.assertEqual(doc.definitions[0].directives_size, 0)
self.assertEqual(doc.definitions[0].selection_set.selection_set_size,
1)
self.assertEqual(
doc.definitions[0].selection_set.fields[0].arguments_size, 1)
self.assertEqual(
doc.definitions[0].selection_set.fields[0].arguments["id"], 4)
self.assertEqual(doc.definitions[0].selection_set.fields[0].name,
"user")
self.assertEqual(
doc.definitions[0].selection_set.fields[0].selection_set.
selection_set_size, 2)
self.assertEqual(
doc.definitions[0].selection_set.fields[0].selection_set.fields[0].
arguments_size, 1)
self.assertEqual(
doc.definitions[0].selection_set.fields[0].selection_set.fields[0].
arguments, {"first": 10})
self.assertEqual(
doc.definitions[0].selection_set.fields[0].selection_set.fields[0].
name, "friends")
self.assertEqual(
doc.definitions[0].selection_set.fields[0].selection_set.fields[0].
selection_set.selection_set_size, 1)
self.assertEqual(
doc.definitions[0].selection_set.fields[0].selection_set.fields[0].
selection_set.frags[0], "friendFields")
self.assertEqual(
doc.definitions[0].selection_set.fields[0].selection_set.fields[1].
arguments_size, 1)
self.assertEqual(
doc.definitions[0].selection_set.fields[0].selection_set.fields[0].
arguments, {"first": 10})
self.assertEqual(
doc.definitions[0].selection_set.fields[0].selection_set.fields[1].
name, "mutualFriends")
self.assertEqual(
doc.definitions[0].selection_set.fields[0].selection_set.fields[1].
selection_set.frags[0], "friendFields")
self.assertIsInstance(doc.definitions[1], Fragment)
self.assertEqual(doc.definitions[1].name, "friendFields")
self.assertEqual(doc.definitions[1].selection_set.selection_set_size,
3)
self.assertEqual(doc.definitions[1].selection_set.fields[0].name, "id")
self.assertEqual(doc.definitions[1].selection_set.fields[1].name,
"name")
self.assertEqual(doc.definitions[1].selection_set.frags[0],
"standardProfilePic")
self.assertIsInstance(doc.definitions[2], Fragment)
self.assertEqual(doc.definitions[2].name, "standardProfilePic")
self.assertEqual(doc.definitions[2].selection_set.selection_set_size,
1)
self.assertEqual(doc.definitions[2].selection_set.fields[0].name,
"profilePic")
def test_with_inline_fragments(self):
parser = Parser()
doc = parser.parse_query("""query inlineFragmentTyping {
profiles(handles: ["zuck", "cocacola"]) {
handle
... on User {
friends {
count
}
}
... on Page {
likers {
count
}
}
}
}
""")
self.assertEqual(doc.definitions_size, 1)
self.assertIsInstance(doc.definitions[0], Operation)
self.assertEqual(doc.definitions[0].operation, "query")
self.assertEqual(doc.definitions[0].name, "inlineFragmentTyping")
self.assertEqual(doc.definitions[0].variables_size, 0)
self.assertEqual(doc.definitions[0].directives_size, 0)
self.assertEqual(doc.definitions[0].selection_set.selection_set_size,
1)
self.assertEqual(
doc.definitions[0].selection_set.fields[0].arguments_size, 1)
self.assertEqual(
doc.definitions[0].selection_set.fields[0].arguments["handles"],
["zuck", "cocacola"])
self.assertEqual(doc.definitions[0].selection_set.fields[0].name,
"profiles")
self.assertEqual(
doc.definitions[0].selection_set.fields[0].selection_set.
selection_set_size, 3)
self.assertEqual(
doc.definitions[0].selection_set.fields[0].selection_set.fields[0].
arguments_size, 0)
self.assertEqual(
doc.definitions[0].selection_set.fields[0].selection_set.fields[0].
name, "handle")
self.assertEqual(
doc.definitions[0].selection_set.fields[0].selection_set.fields[1].
condition, "User")
self.assertEqual(
doc.definitions[0].selection_set.fields[0].selection_set.fields[1].
selection_set.selection_set_size, 1)
self.assertEqual(
doc.definitions[0].selection_set.fields[0].selection_set.fields[1].
selection_set.fields[0].name, "friends")
self.assertEqual(
doc.definitions[0].selection_set.fields[0].selection_set.fields[1].
selection_set.fields[0].selection_set.selection_set_size, 1)
self.assertEqual(
doc.definitions[0].selection_set.fields[0].selection_set.fields[1].
selection_set.fields[0].selection_set.fields[0].name, "count")
self.assertEqual(
doc.definitions[0].selection_set.fields[0].selection_set.fields[2].
condition, "Page")
self.assertEqual(
doc.definitions[0].selection_set.fields[0].selection_set.fields[2].
selection_set.selection_set_size, 1)
self.assertEqual(
doc.definitions[0].selection_set.fields[0].selection_set.fields[2].
selection_set.fields[0].name, "likers")
self.assertEqual(
doc.definitions[0].selection_set.fields[0].selection_set.fields[2].
selection_set.fields[0].selection_set.selection_set_size, 1)
self.assertEqual(
doc.definitions[0].selection_set.fields[0].selection_set.fields[2].
selection_set.fields[0].selection_set.fields[0].name, "count")
def test_with_web_example(self):
parser = Parser()
doc = parser.parse_query("""query RootQuery {
user(id: 3500401) {
id,
name,
isViewerFriend,
profilePicture(size: 50) {
uri,
width,
height
}
}
}
""")
self.assertEqual(doc.definitions_size, 1)
self.assertEqual(doc.definitions[0].operation, "query")
self.assertEqual(doc.definitions[0].name, "RootQuery")
self.assertEqual(doc.definitions[0].variables_size, 0)
self.assertEqual(doc.definitions[0].directives_size, 0)
self.assertEqual(doc.definitions[0].selection_set.selection_set_size,
1)
self.assertEqual(
doc.definitions[0].selection_set.fields[0].arguments_size, 1)
self.assertEqual(doc.definitions[0].selection_set.fields[0].arguments,
{"id": 3500401})
self.assertEqual(doc.definitions[0].selection_set.fields[0].name,
"user")
self.assertEqual(
doc.definitions[0].selection_set.fields[0].selection_set.
selection_set_size, 4)
self.assertEqual(
doc.definitions[0].selection_set.fields[0].selection_set.fields[0].
arguments_size, 0)
self.assertEqual(
doc.definitions[0].selection_set.fields[0].selection_set.fields[0].
name, "id")
self.assertEqual(
doc.definitions[0].selection_set.fields[0].selection_set.fields[1].
name, "name")
self.assertEqual(
doc.definitions[0].selection_set.fields[0].selection_set.fields[2].
name, "isViewerFriend")
self.assertEqual(
doc.definitions[0].selection_set.fields[0].selection_set.fields[3].
name, "profilePicture")
self.assertEqual(
doc.definitions[0].selection_set.fields[0].selection_set.fields[3].
arguments_size, 1)
self.assertEqual(
doc.definitions[0].selection_set.fields[0].selection_set.fields[3].
arguments, {"size": 50})
self.assertEqual(
doc.definitions[0].selection_set.fields[0].selection_set.fields[3].
selection_set.selection_set_size, 3)
self.assertEqual(
doc.definitions[0].selection_set.fields[0].selection_set.fields[3].
selection_set.fields[0].name, "uri")
self.assertEqual(
doc.definitions[0].selection_set.fields[0].selection_set.fields[3].
selection_set.fields[1].name, "width")
self.assertEqual(
doc.definitions[0].selection_set.fields[0].selection_set.fields[3].
selection_set.fields[2].name, "height")
def test_raises_exceptions(self):
parser = Parser()
with self.assertRaises(GraphQLException):
doc = parser.parse_query("""
abcdef
""")
def main():
f = open("tests/relay-todo.graphql", 'r')
query = f.read()
parser = Parser()
document = parser.parse_query(query)
print(document)
