95 lines
2.6 KiB
Python
95 lines
2.6 KiB
Python
from fbchat._poll import Poll, PollOption
|
|
|
|
|
|
def test_poll_option_from_graphql_unvoted():
|
|
data = {
|
|
"id": "123456789",
|
|
"text": "abc",
|
|
"total_count": 0,
|
|
"viewer_has_voted": "false",
|
|
"voters": [],
|
|
}
|
|
assert PollOption(
|
|
text="abc", vote=False, voters=[], votes_count=0, id="123456789"
|
|
) == PollOption._from_graphql(data)
|
|
|
|
|
|
def test_poll_option_from_graphql_voted():
|
|
data = {
|
|
"id": "123456789",
|
|
"text": "abc",
|
|
"total_count": 2,
|
|
"viewer_has_voted": "true",
|
|
"voters": ["1234", "2345"],
|
|
}
|
|
assert PollOption(
|
|
text="abc", vote=True, voters=["1234", "2345"], votes_count=2, id="123456789"
|
|
) == PollOption._from_graphql(data)
|
|
|
|
|
|
def test_poll_option_from_graphql_alternate_format():
|
|
# Format received when fetching poll options
|
|
data = {
|
|
"id": "123456789",
|
|
"text": "abc",
|
|
"viewer_has_voted": True,
|
|
"voters": {
|
|
"count": 2,
|
|
"edges": [{"node": {"id": "1234"}}, {"node": {"id": "2345"}}],
|
|
},
|
|
}
|
|
assert PollOption(
|
|
text="abc", vote=True, voters=["1234", "2345"], votes_count=2, id="123456789"
|
|
) == PollOption._from_graphql(data)
|
|
|
|
|
|
def test_poll_from_graphql(session):
|
|
data = {
|
|
"id": "123456789",
|
|
"text": "Some poll",
|
|
"total_count": 5,
|
|
"viewer_has_voted": "true",
|
|
"options": [
|
|
{
|
|
"id": "1111",
|
|
"text": "Abc",
|
|
"total_count": 1,
|
|
"viewer_has_voted": "true",
|
|
"voters": ["1234"],
|
|
},
|
|
{
|
|
"id": "2222",
|
|
"text": "Def",
|
|
"total_count": 2,
|
|
"viewer_has_voted": "false",
|
|
"voters": ["2345", "3456"],
|
|
},
|
|
{
|
|
"id": "3333",
|
|
"text": "Ghi",
|
|
"total_count": 0,
|
|
"viewer_has_voted": "false",
|
|
"voters": [],
|
|
},
|
|
],
|
|
}
|
|
assert Poll(
|
|
session=session,
|
|
question="Some poll",
|
|
options=[
|
|
PollOption(
|
|
text="Abc", vote=True, voters=["1234"], votes_count=1, id="1111"
|
|
),
|
|
PollOption(
|
|
text="Def",
|
|
vote=False,
|
|
voters=["2345", "3456"],
|
|
votes_count=2,
|
|
id="2222",
|
|
),
|
|
PollOption(text="Ghi", vote=False, voters=[], votes_count=0, id="3333"),
|
|
],
|
|
options_count=5,
|
|
id=123456789,
|
|
) == Poll._from_graphql(session, data)
|