Fix test on Python 3: vault code expects bytes
This commit is contained in:
parent
174ecb843c
commit
f58f0c62e1
1 changed files with 6 additions and 6 deletions
|
@ -78,17 +78,17 @@ class TestVaultLib(unittest.TestCase):
|
||||||
def test_format_output(self):
|
def test_format_output(self):
|
||||||
v = VaultLib('ansible')
|
v = VaultLib('ansible')
|
||||||
v.cipher_name = "TEST"
|
v.cipher_name = "TEST"
|
||||||
sensitive_data = "ansible"
|
sensitive_data = b"ansible"
|
||||||
data = v._format_output(sensitive_data)
|
data = v._format_output(sensitive_data)
|
||||||
lines = data.split(b'\n')
|
lines = data.split(b'\n')
|
||||||
assert len(lines) > 1, "failed to properly add header"
|
assert len(lines) > 1, "failed to properly add header"
|
||||||
header = to_unicode(lines[0])
|
header = to_bytes(lines[0])
|
||||||
assert header.endswith(';TEST'), "header does end with cipher name"
|
assert header.endswith(b';TEST'), "header does end with cipher name"
|
||||||
header_parts = header.split(';')
|
header_parts = header.split(b';')
|
||||||
assert len(header_parts) == 3, "header has the wrong number of parts"
|
assert len(header_parts) == 3, "header has the wrong number of parts"
|
||||||
assert header_parts[0] == '$ANSIBLE_VAULT', "header does not start with $ANSIBLE_VAULT"
|
assert header_parts[0] == b'$ANSIBLE_VAULT', "header does not start with $ANSIBLE_VAULT"
|
||||||
assert header_parts[1] == v.b_version, "header version is incorrect"
|
assert header_parts[1] == v.b_version, "header version is incorrect"
|
||||||
assert header_parts[2] == 'TEST', "header does end with cipher name"
|
assert header_parts[2] == b'TEST', "header does end with cipher name"
|
||||||
|
|
||||||
def test_split_header(self):
|
def test_split_header(self):
|
||||||
v = VaultLib('ansible')
|
v = VaultLib('ansible')
|
||||||
|
|
Loading…
Reference in a new issue