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'