95 lines
3 KiB
Python
95 lines
3 KiB
Python
#!/usr/bin/python
|
|
# Copyright (C) 2022 Jelmer Vernooij <jelmer@jelmer.uk>
|
|
#
|
|
# This program is free software; you can redistribute it and/or modify
|
|
# it under the terms of the GNU General Public License as published by
|
|
# the Free Software Foundation; either version 2 of the License, or
|
|
# (at your option) any later version.
|
|
#
|
|
# This program is distributed in the hope that it will be useful,
|
|
# but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
# GNU General Public License for more details.
|
|
#
|
|
# You should have received a copy of the GNU General Public License
|
|
# along with this program; if not, write to the Free Software
|
|
# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
|
|
|
|
import os
|
|
import sys
|
|
import tempfile
|
|
from unittest import TestCase
|
|
|
|
from ognibuild.logs import (
|
|
copy_output,
|
|
redirect_output,
|
|
rotate_logfile,
|
|
DirectoryLogManager,
|
|
)
|
|
|
|
|
|
class TestCopyOutput(TestCase):
|
|
|
|
def test_no_tee(self):
|
|
with tempfile.TemporaryDirectory() as td:
|
|
p = os.path.join(td, 'foo.log')
|
|
with copy_output(p, tee=False):
|
|
sys.stdout.write('lala\n')
|
|
sys.stdout.flush()
|
|
with open(p, 'r') as f:
|
|
self.assertEqual('lala\n', f.read())
|
|
|
|
def test_tee(self):
|
|
with tempfile.TemporaryDirectory() as td:
|
|
p = os.path.join(td, 'foo.log')
|
|
with copy_output(p, tee=True):
|
|
sys.stdout.write('lala\n')
|
|
sys.stdout.flush()
|
|
with open(p, 'r') as f:
|
|
self.assertEqual('lala\n', f.read())
|
|
|
|
|
|
class TestRedirectOutput(TestCase):
|
|
|
|
def test_simple(self):
|
|
with tempfile.TemporaryDirectory() as td:
|
|
p = os.path.join(td, 'foo.log')
|
|
with open(p, 'w') as f:
|
|
with redirect_output(f):
|
|
sys.stdout.write('lala\n')
|
|
sys.stdout.flush()
|
|
with open(p, 'r') as f:
|
|
self.assertEqual('lala\n', f.read())
|
|
|
|
|
|
class TestRotateLogfile(TestCase):
|
|
|
|
def test_does_not_exist(self):
|
|
with tempfile.TemporaryDirectory() as td:
|
|
p = os.path.join(td, 'foo.log')
|
|
rotate_logfile(p)
|
|
self.assertEqual([], os.listdir(td))
|
|
|
|
def test_simple(self):
|
|
with tempfile.TemporaryDirectory() as td:
|
|
p = os.path.join(td, 'foo.log')
|
|
with open(p, 'w') as f:
|
|
f.write('contents\n')
|
|
rotate_logfile(p)
|
|
self.assertEqual(['foo.log.1'], os.listdir(td))
|
|
|
|
|
|
class TestLogManager(TestCase):
|
|
|
|
def test_simple(self):
|
|
with tempfile.TemporaryDirectory() as td:
|
|
p = os.path.join(td, 'foo.log')
|
|
lm = DirectoryLogManager(p, mode='redirect')
|
|
|
|
def writesomething():
|
|
sys.stdout.write('foo\n')
|
|
sys.stdout.flush()
|
|
fn = lm.wrap(writesomething)
|
|
fn()
|
|
with open(p, 'r') as f:
|
|
self.assertEqual('foo\n', f.read())
|