34 lines
1.6 KiB
Python
34 lines
1.6 KiB
Python
|
import pytest
|
||
|
|
import shutil
|
||
|
|
import subprocess
|
||
|
|
|
||
|
|
def test_warrant_canary_usage():
|
||
|
|
### Ensure that the script provides output.
|
||
|
|
procState=subprocess.run(['./warrant_canary','-h'],capture_output=True,timeout=3)
|
||
|
|
assert procState.returncode == 0 and procState.stderr == b''
|
||
|
|
|
||
|
|
def test_warrant_canary_seed():
|
||
|
|
### Test that we can seed a warrant canary file
|
||
|
|
# procState=subprocess.run(['./warrant_canary','-s'],capture_output=True,timeout=3)
|
||
|
|
try:
|
||
|
|
with open('canary.asc') as fh:
|
||
|
|
# assert procState.returncode == 0 and procState.stderr == b'' and 'BEGIN PGP SIGNATURE' in fh.read() and b'Success' in procState.stderr
|
||
|
|
assert 'BEGIN PGP SIGNATURE' in fh.read()
|
||
|
|
except:
|
||
|
|
assert False
|
||
|
|
|
||
|
|
def test_warrant_canary_basic_verify():
|
||
|
|
### Test that we can verify with no arguments
|
||
|
|
procState=subprocess.run(['./warrant_canary','-V'],capture_output=True,timeout=3)
|
||
|
|
assert procState.returncode == 0 and procState.stderr == b'' and b'Good signature' in procState.stdout
|
||
|
|
|
||
|
|
def test_warrant_canary_file_verify():
|
||
|
|
### Test that we can verify with a file
|
||
|
|
procState=subprocess.run(['./warrant_canary','-c','./canary.asc','-V'],capture_output=True,timeout=3)
|
||
|
|
assert procState.returncode == 0 and procState.stderr == b'' and b'Good signature' in procState.stdout
|
||
|
|
|
||
|
|
def test_warrant_canary_url_verify():
|
||
|
|
### Test that we can verify with a web address
|
||
|
|
procState=subprocess.run(['./warrant_canary','-c','https://cryptostorm.is/canary.txt','-k','E9C7C942','-K','pgp.mit.edu','-V'],capture_output=True,timeout=30)
|
||
|
|
assert procState.returncode == 0 and procState.stderr == b'' and b'Good signature' in procState.stdout
|