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