This commit is contained in:
Dimitri John Ledkov 2014-12-18 21:11:00 +00:00
parent dae4c18c9e
commit b8cf7b113e

View File

@ -14,7 +14,10 @@
# OTHER TORTIOUS ACTION, ARISING OUT OF OR IN CONNECTION WITH THE USE OR # OTHER TORTIOUS ACTION, ARISING OUT OF OR IN CONNECTION WITH THE USE OR
# PERFORMANCE OF THIS SOFTWARE. # PERFORMANCE OF THIS SOFTWARE.
import StringIO try:
from StringIO import StringIO
except:
from io import StringIO
import sys import sys
from ubuntutools.logger import Logger from ubuntutools.logger import Logger
@ -23,8 +26,8 @@ from ubuntutools.test import unittest
class LoggerTestCase(unittest.TestCase): class LoggerTestCase(unittest.TestCase):
def setUp(self): def setUp(self):
Logger.stdout = StringIO.StringIO() Logger.stdout = StringIO()
Logger.stderr = StringIO.StringIO() Logger.stderr = StringIO()
self._script_name = Logger.script_name self._script_name = Logger.script_name
Logger.script_name = 'test' Logger.script_name = 'test'
self._verbose = Logger.verbose self._verbose = Logger.verbose