tests/test_scrollterm.py

   1 #!/usr/bin/env python3
   2
   3 #
   4 # Copyright (c) 2018 Vojtech Horky
   5 # All rights reserved.
   6 #
   7 # Redistribution and use in source and binary forms, with or without
   8 # modification, are permitted provided that the following conditions
   9 # are met:
  10 #
  11 # - Redistributions of source code must retain the above copyright
  12 #   notice, this list of conditions and the following disclaimer.
  13 # - Redistributions in binary form must reproduce the above copyright
  14 #   notice, this list of conditions and the following disclaimer in the
  15 #   documentation and/or other materials provided with the distribution.
  16 # - The name of the author may not be used to endorse or promote products
  17 #   derived from this software without specific prior written permission.
  18 #
  19 # THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR
  20 # IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
  21 # OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
  22 # IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT,
  23 # INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
  24 # NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
  25 # DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
  26 # THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
  27 # (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF
  28 # THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
  29 #
  30
  31 from nose.tools import eq_
  32 from htest.scrollterm import ScrollingTerminal
  33
  34 class TermCaptureGallery:
  35     def __init__(self, captures):
  36         self.captures = [ x.split('\n') for x in captures ]
  37
  38     def capture(self):
  39         if len(self.captures) == 0:
  40             raise StopIteration("Run out of captures.")
  41         return self.captures.pop(0)
  42
  43 def make_terminal(captures):
  44     gallery = TermCaptureGallery(captures)
  45     return ScrollingTerminal(gallery.capture, "_")
  46
  47
  48
  49 def test_capture():
  50     term = make_terminal([
  51         "a\nb\nc\nd",
  52         "b\nc\nd\ne",
  53     ])
  54     eq_(term.capture(), [ "a", "b", "c", "d" ])
  55     eq_(term.capture(), [ "b", "c", "d", "e" ])
  56
  57
  58 def check_intr_same_lines(expected, prev, curr):
  59     term = make_terminal([])
  60     msg = "_same_lines({}, {}) == {}".format(prev, curr, expected)
  61     assert term._same_lines(prev, curr) == expected, msg
  62
  63 def test_intr_same_lines():
  64     test_cases = [
  65         ( ( True, False ), ["a", "b"], ["a", "b"] ),
  66         ( ( False, False ), ["a", "b"], ["a", "bb"] ),
  67         ( ( True, True), ["a", "b_"], ["a", "bb"] ),
  68         ( ( True, True), ["a", "bb_"], ["a", "bb"] ),
  69         ( ( True, True), ["a", "bb_"], ["a", "bbbbb"] ),
  70     ]
  71     for expected, prev, curr in test_cases:
  72         yield check_intr_same_lines, expected, prev, curr
  73
  74
  75 def test_get_lines_once_simple():
  76     term = make_terminal([
  77         "a\nb\nc\nd",
  78         "a\nb\nc\nd",
  79         "b\nc\nd\ne",
  80         "d\ne\nf\ng",
  81     ])
  82     eq_(term.get_lines_once(), [ "a", "b", "c", "d" ])
  83     eq_(term.get_lines_once(), [])
  84     eq_(term.get_lines_once(), [ "e" ])
  85     eq_(term.get_lines_once(), [ "f", "g" ])
  86
  87 def test_get_lines_once_slow_line():
  88     term = make_terminal([
  89         "a\nb_\n\n",
  90         "a\nbbb\n\n",
  91         "a\nbbb\nc\n",
  92     ])
  93     eq_(term.get_lines_once(), [ "a", "b_" ])
  94     eq_(term.get_lines_once(), [ "bbb" ])
  95     eq_(term.get_lines_once(), [ "c" ])
  96