catapult/third_party/typ/typ/tests/printer_test.py

# Copyright 2014 Dirk Pranke. All rights reserved.
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
#    http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.

import unittest

from typ.printer import Printer


class TestPrinter(unittest.TestCase):

    def setUp(self):
        # 'Invalid name' pylint: disable=C0103
        self.out = []

    def print_(self, msg, end='\n'):
        self.out.append(msg + end)

    def test_basic(self):
        pr = Printer(self.print_, False, 80)
        pr.update('foo')
        pr.flush()
        self.assertEqual(self.out, ['foo', '\n'])

    def test_elide(self):
        pr = Printer(self.print_, False, 8)
        pr.update('hello world')
        pr.flush()
        self.assertEqual(self.out, ['h...d', '\n'])

    def test_overwrite(self):
        pr = Printer(self.print_, True, 80)
        pr.update('hello world')
        pr.update('goodbye world')
        pr.flush()
        self.assertEqual(self.out,
                         ['hello world',
                          '\r           \r',
                          'goodbye world',
                          '\n'])

    def test_last_line_flushed_when_not_overwriting(self):
        pr = Printer(self.print_, False, 80)
        pr.update('foo\nbar')
        pr.update('baz')
        pr.flush()
        self.assertEqual(self.out,
                         ['foo\nbar',
                          '\n',
                          'baz',
                          '\n'])