Add graphql tests
This commit is contained in:
35
tests/test_graphql.py
Normal file
35
tests/test_graphql.py
Normal file
@@ -0,0 +1,35 @@
|
||||
import pytest
|
||||
import json
|
||||
from fbchat._graphql import ConcatJSONDecoder, queries_to_json, response_to_json
|
||||
|
||||
|
||||
@pytest.mark.parametrize(
|
||||
"content,result",
|
||||
[
|
||||
("", []),
|
||||
('{"a":"b"}', [{"a": "b"}]),
|
||||
('{"a":"b"}{"b":"c"}', [{"a": "b"}, {"b": "c"}]),
|
||||
(' \n{"a": "b" } \n { "b" \n\n : "c" }', [{"a": "b"}, {"b": "c"}]),
|
||||
],
|
||||
)
|
||||
def test_concat_json_decoder(content, result):
|
||||
assert result == json.loads(content, cls=ConcatJSONDecoder)
|
||||
|
||||
|
||||
def test_queries_to_json():
|
||||
assert {"q0": "A", "q1": "B", "q2": "C"} == json.loads(
|
||||
queries_to_json("A", "B", "C")
|
||||
)
|
||||
|
||||
|
||||
def test_response_to_json():
|
||||
data = (
|
||||
'{"q1":{"data":{"b":"c"}}}\r\n'
|
||||
'{"q0":{"response":[1,2]}}\r\n'
|
||||
"{\n"
|
||||
' "successful_results": 2,\n'
|
||||
' "error_results": 0,\n'
|
||||
' "skipped_results": 0\n'
|
||||
"}"
|
||||
)
|
||||
assert [[1, 2], {"b": "c"}] == response_to_json(data)
|
Reference in New Issue
Block a user