| import pytest |
| import json |
|
|
| from extract import extract |
|
|
| def test_pure_string(): |
| inp = 'abc' |
| token, user, redirect, info = extract(inp) |
| assert token == inp |
| assert user is None |
| assert redirect is None |
| assert info is None |
|
|
| def test_json_format_01(): |
| inp_s = '{"a":"b"}' |
| token, user, redirect, info = extract(inp_s) |
| inp = json.loads(inp_s) |
| assert token == None |
| assert user == None |
| assert redirect is None |
| assert info == inp |
|
|
| def test_json_format_02(): |
| inp_s = '{"token":"b"}' |
| token, user, redirect, info = extract(inp_s) |
| inp = json.loads(inp_s) |
| assert token == inp['token'] |
| assert user == None |
| assert redirect is None |
| assert info == inp |
|
|
| def test_json_format_03(): |
| inp_s = '{"user":"xb"}' |
| token, user, redirect, info = extract(inp_s) |
| inp = json.loads(inp_s) |
| assert token == None |
| assert user == inp['user'] |
| assert redirect is None |
| assert info == inp |
|
|
| def test_json_format_04(): |
| inp_s = '{"user":"t1", "token":"t2"}' |
| token, user, redirect, info = extract(inp_s) |
| inp = json.loads(inp_s) |
| assert token == inp['token'] |
| assert user == inp['user'] |
| assert redirect is None |
| assert info == inp |
|
|
| def test_json_format_05(): |
| inp_s = '{"user":"t1", "token":"t2", "redirect":"xx"}' |
| token, user, redirect, info = extract(inp_s) |
| inp = json.loads(inp_s) |
| assert token == inp['token'] |
| assert user == inp['user'] |
| assert redirect == inp['redirect'] |
| assert info == inp |
|
|
