import pytest
import json

from extract import extract

def test_pure_string():
    inp = 'abc'
    token, user, info = extract(inp)
    assert token == inp
    assert user is None
    assert info is None

def test_json_format_01():
    inp_s = '{"a":"b"}'
    token, user, info = extract(inp_s)
    inp = json.loads(inp_s)
    assert token == None
    assert user == None
    assert info == inp

def test_json_format_02():
    inp_s = '{"token":"b"}'
    token, user, info = extract(inp_s)
    inp = json.loads(inp_s)
    assert token == inp['token']
    assert user == None
    assert info == inp

def test_json_format_03():
    inp_s = '{"user":"xb"}'
    token, user, info = extract(inp_s)
    inp = json.loads(inp_s)
    assert token == None
    assert user == inp['user']
    assert info == inp

def test_json_format_04():
    inp_s = '{"user":"t1", "token":"t2"}'
    token, user, info = extract(inp_s)
    inp = json.loads(inp_s)
    assert token == inp['token']
    assert user == inp['user']
    assert info == inp