PappyProxy/pappyproxy/tests/test_crypto.py

import os
import pytest
import random
import string
from pappyproxy.session import Session
from pappyproxy.crypto import Crypto
from pappyproxy.config import PappyConfig

@pytest.fixture
def conf():
    c = PappyConfig()
    return c

@pytest.fixture
def crypt():
    c = Crypto(conf())
    return c

@pytest.fixture
def tmpname():
    cns = string.ascii_lowercase + string.ascii_uppercase + string.digits
    tn = ''
    for i in xrange(8):
        tn += cns[random.randint(0,len(cns)-1)]
    return tn

tmpdir = '/tmp/test_crypto'+tmpname()
tmpfiles = ['cmdhistory', 'config.json', 'data.db']
tmp_pass = 'fugyeahbruh'

def stub_files():
    enter_tmpdir()
    for sf in tmpfiles:
        with os.fdopen(os.open(sf, os.O_CREAT, 0o0600), 'r'):
            pass

def enter_tmpdir():
    if not os.path.isdir(tmpdir):
        os.mkdir(tmpdir)
    os.chdir(tmpdir)

def test_decrypt_tmpdir():
    enter_tmpdir()
    c = crypt()
     
    # Stub out the password, working with stdout is a pain with pytest
    c.password = tmp_pass
    
    c.decrypt_project()
    assert os.path.isdir(os.path.join(os.getcwd(), '../crypt'))

def test_decrypt_copy_files():
    enter_tmpdir()
    stub_files()
    c = crypt()
     
    # Stub out the password, working with stdout is a pain with pytest
    c.password = tmp_pass
    
    c.decrypt_project()
    for tf in tmpfiles:
        assert os.path.isfile(os.path.join(os.getcwd(),tf))