def test_encode_payload_raises_value_error_key_is_required(self):
encode_args = ['encode', 'name=Vader', 'job=Sith']
parser = build_argparser()
args = parser.parse_args(encode_args)
with pytest.raises(ValueError) as excinfo:
encode_payload(args)
assert 'Key is required when encoding' in str(excinfo.value)
def test_encode_payload_raises_value_error_key_is_required(self):
encode_args = ["encode", "name=Vader", "job=Sith"]
parser = build_argparser()
args = parser.parse_args(encode_args)
with pytest.raises(ValueError) as excinfo:
encode_payload(args)
assert "Key is required when encoding" in str(excinfo.value)
def test_encode_decode(self, key, name, job, exp, verify):
encode_args = [
'--key={0}'.format(key),
'encode',
'name={0}'.format(name),
'job={0}'.format(job),
]
if exp:
encode_args.append('exp={0}'.format(exp))
if verify:
encode_args.append('verify={0}'.format(verify))
parser = build_argparser()
parsed_encode_args = parser.parse_args(encode_args)
token = encode_payload(parsed_encode_args)
assert token is not None
assert token != ''
decode_args = [
'--key={0}'.format(key),
'decode',
token
]
parser = build_argparser()
parsed_decode_args = parser.parse_args(decode_args)
actual = json.loads(decode_payload(parsed_decode_args))
expected = {
'job': job,
'name': name,
}
assert actual['name'] == expected['name']
assert actual['job'] == expected['job']
def test_encode_header_raises_value_error_bad_dict(self):
encode_args = [
"--key=secret",
"--header=dfsfd",
"encode",
"name=Vader",
"job=Sith",
]
parser = build_argparser()
args = parser.parse_args(encode_args)
with pytest.raises(ValueError) as excinfo:
encode_payload(args)
assert "Error loading header:" in str(excinfo.value)
def test_encode_decode(self, key, header, name, job, exp, verify):
encode_args = [
"--key={}".format(key),
"--header={}".format(header),
"encode",
"name={}".format(name),
"job={}".format(job),
]
if exp:
encode_args.append("exp={}".format(exp))
if verify:
encode_args.append("verify={}".format(verify))
parser = build_argparser()
parsed_encode_args = parser.parse_args(encode_args)
token = encode_payload(parsed_encode_args)
assert token is not None
assert token != ""
decode_args = ["--key={}".format(key), "decode", token]
parser = build_argparser()
parsed_decode_args = parser.parse_args(decode_args)
actual = json.loads(decode_payload(parsed_decode_args))
expected = {"job": job, "name": name}
assert actual["name"] == expected["name"]
assert actual["job"] == expected["job"]
def test_encode_decode(self, key, name, job, exp, verify):
encode_args = [
'--key={0}'.format(key),
'encode',
'name={0}'.format(name),
'job={0}'.format(job),
]
if exp:
encode_args.append('exp={0}'.format(exp))
if verify:
encode_args.append('verify={0}'.format(verify))
parser = build_argparser()
parsed_encode_args = parser.parse_args(encode_args)
token = encode_payload(parsed_encode_args)
assert token is not None
assert token != ''
decode_args = ['--key={0}'.format(key), 'decode', token]
parser = build_argparser()
parsed_decode_args = parser.parse_args(decode_args)
actual = json.loads(decode_payload(parsed_decode_args))
expected = {
'job': job,
'name': name,
}
assert actual['name'] == expected['name']
assert actual['job'] == expected['job']
def test_encode_kid_header(self):
encode_args = ['--key=1234', 'encode', '--kid=0001', 'name=Vader']
parser = build_argparser()
parsed_encode_args = parser.parse_args(encode_args)
token = encode_payload(parsed_encode_args)
assert token is not None
assert token is not ''
actual = jwt.get_unverified_header(token)
assert actual['kid'] == '0001'
def test_decode_payload_terminal_tty(self, monkeypatch):
encode_args = ["--key=secret-key", "encode", "name=hello-world"]
parser = build_argparser()
parsed_encode_args = parser.parse_args(encode_args)
token = encode_payload(parsed_encode_args)
decode_args = ["--key=secret-key", "decode"]
parsed_decode_args = parser.parse_args(decode_args)
monkeypatch.setattr(sys.stdin, "isatty", lambda: True)
monkeypatch.setattr(sys.stdin, "readline", lambda: token)
actual = json.loads(decode_payload(parsed_decode_args))
assert actual["name"] == "hello-world"
def test_decode_payload_terminal_tty(self, monkeypatch):
encode_args = [
'--key=secret-key',
'encode',
'name=hello-world',
]
parser = build_argparser()
parsed_encode_args = parser.parse_args(encode_args)
token = encode_payload(parsed_encode_args)
decode_args = ['--key=secret-key', 'decode']
parsed_decode_args = parser.parse_args(decode_args)
monkeypatch.setattr(sys.stdin, 'isatty', lambda: True)
monkeypatch.setattr(sys.stdin, 'readline', lambda: token)
actual = json.loads(decode_payload(parsed_decode_args))
assert actual['name'] == 'hello-world'
def test_decode_payload_terminal_tty(self, monkeypatch):
encode_args = [
'--key=secret-key',
'encode',
'name=hello-world',
]
parser = build_argparser()
parsed_encode_args = parser.parse_args(encode_args)
token = encode_payload(parsed_encode_args)
decode_args = ['--key=secret-key', 'decode']
parsed_decode_args = parser.parse_args(decode_args)
monkeypatch.setattr(sys.stdin, 'isatty', lambda: True)
monkeypatch.setattr(sys.stdin, 'readline', lambda: token)
actual = json.loads(decode_payload(parsed_decode_args))
assert actual['name'] == 'hello-world'
