mirror of https://github.com/jtesta/ssh-audit.git
Share output spying for tests.
This commit is contained in:
parent
e60d4ff809
commit
58a943bed9
|
@ -1,10 +1,35 @@
|
||||||
#!/usr/bin/env python
|
#!/usr/bin/env python
|
||||||
# -*- coding: utf-8 -*-
|
# -*- coding: utf-8 -*-
|
||||||
import os, sys
|
import pytest, os, sys, io
|
||||||
import pytest
|
|
||||||
|
|
||||||
|
if sys.version_info[0] == 2:
|
||||||
|
import StringIO
|
||||||
|
StringIO = StringIO.StringIO
|
||||||
|
else:
|
||||||
|
StringIO = io.StringIO
|
||||||
|
|
||||||
|
|
||||||
@pytest.fixture(scope='module')
|
@pytest.fixture(scope='module')
|
||||||
def ssh_audit():
|
def ssh_audit():
|
||||||
__rdir = os.path.join(os.path.dirname(os.path.abspath(__file__)), '..')
|
__rdir = os.path.join(os.path.dirname(os.path.abspath(__file__)), '..')
|
||||||
sys.path.append(os.path.abspath(__rdir))
|
sys.path.append(os.path.abspath(__rdir))
|
||||||
return __import__('ssh-audit')
|
return __import__('ssh-audit')
|
||||||
|
|
||||||
|
|
||||||
|
class _OutputSpy(list):
|
||||||
|
def begin(self):
|
||||||
|
self.__out = StringIO()
|
||||||
|
self.__old_stdout = sys.stdout
|
||||||
|
sys.stdout = self.__out
|
||||||
|
|
||||||
|
def flush(self):
|
||||||
|
lines = self.__out.getvalue().splitlines()
|
||||||
|
sys.stdout = self.__old_stdout
|
||||||
|
self.__out = None
|
||||||
|
return lines
|
||||||
|
|
||||||
|
|
||||||
|
@pytest.fixture(scope='module')
|
||||||
|
def output_spy():
|
||||||
|
return _OutputSpy()
|
||||||
|
|
Loading…
Reference in New Issue