# Copyright (C) 2009, 2010 Roman Zimbelmann <romanz@lavabit.com>
#
# 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 3 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, see <http://www.gnu.org/licenses/>.
import os.path
import sys
rangerpath = os.path.join(os.path.dirname(__file__), '..')
if sys.path[1] != rangerpath:
sys.path[1:1] = [rangerpath]
import unittest
import curses
from random import randint
from ranger.gui.displayable import Displayable, DisplayableContainer
from testlib import Fake, OK, raise_ok, TODO
class TestWithFakeCurses(unittest.TestCase):
def setUp(self):
self.win = Fake()
self.fm = Fake()
self.env = Fake()
self.settings = Fake()
self.initdict = {'win': self.win, 'settings': self.settings,
'fm': self.fm, 'env': self.env}
self.disp = Displayable(**self.initdict)
self.disc = DisplayableContainer(**self.initdict)
self.disc.add_child(self.disp)
hei, wid = 100, 100
self.env.termsize = (hei, wid)
def tearDown(self):
self.disp.destroy()
self.disc.destroy()
def test_colorscheme(self):
# Using a color method implies change of window attributes
disp = self.disp
disp.win.chgat = raise_ok
disp.win.attrset = raise_ok
self.assertRaises(OK, disp.color, 'a', 'b')
self.assertRaises(OK, disp.color_at, 0, 0, 0, 'a', 'b')
self.assertRaises(OK, disp.color_reset)
def test_focused_object(self):
d1 = Displayable(**self.initdict)
d2 = DisplayableContainer(**self.initdict)
for obj in (Displayable(**self.initdict) for x in range(5)):
d2.add_child(obj)
d3 = DisplayableContainer(**self.initdict)
for obj in (Displayable(**self.initdict) for x in range(5)):
d3.add_child(obj)
for obj in (d1, d2, d3):
self.disc.add_child(obj)
d3.container[3].focused = True
self.assertEqual(self.disc._get_focused_obj(), d3.container[3])
d3.container[3].focused = False
d2.container[0].focused = True
self.assertEqual(self.disc._get_focused_obj(), d2.container[0])
gWin = None
class TestDisplayableWithCurses(unittest.TestCase):
def setUp(self):
global gWin
if not gWin:
gWin = curses.initscr()
self.win = gWin
curses.cbreak()
curses.noecho()
curses.start_color()
curses.use_default_colors()
self.fm = Fake()
self.env = Fake()
self.settings = Fake()
self.initdict = {'win': self.win, 'settings': self.settings,
'fm': self.fm, 'env': self.env}
self.disp = Displayable(**self.initdict)
self.disc = DisplayableContainer(**self.initdict)
self.disc.add_child(self.disp)
self.env.termsize = self.win.getmaxyx()
def tearDown(self):
self.disp.destroy()
curses.nocbreak()
curses.echo()
curses.endwin()
@TODO
def test_boundaries(self):
disp = self.disp
hei, wid = self.env.termsize
self.assertRaises(ValueError, disp.resize, 0, 0, hei + 1, wid)
self.assertRaises(ValueError, disp.resize, 0, 0, hei, wid + 1)
self.assertRaises(ValueError, disp.resize, -1, 0, hei, wid)
self.assertRaises(ValueError, disp.resize, 0, -1, hei, wid)
for i in range(1000):
box = [int(randint(0, hei) * 0.2), int(randint(0, wid) * 0.2)]
box.append(randint(0, hei - box[0]))
box.append(randint(0, wid - box[1]))
def in_box(y, x):
return (y >= box[1] and y < box[1] + box[3]) and \
(x >= box[0] and x < box[0] + box[2])
disp.resize(*box)
self.assertEqual(box, [disp.y, disp.x, disp.hei, disp.wid],
"Resizing failed for some reason on loop " + str(i))
for y, x in zip(range(10), range(10)):
is_in_box = in_box(y, x)
point1 = (y, x)
self.assertEqual(is_in_box, point1 in disp)
point2 = Fake()
point2.x = x
point2.y = y
self.assertEqual(is_in_box, point2 in disp)
def test_click(self):
self.disp.click = raise_ok
hei, wid = self.env.termsize
for i in range(50):
winwid = randint(2, wid-1)
winhei = randint(2, hei-1)
self.disc.resize(0, 0, hei, wid)
self.disp.resize(0, 0, winhei, winwid)
fakepos = Fake()
fakepos.x = winwid - 2
fakepos.y = winhei - 2
self.assertRaises(OK, self.disc.click, fakepos)
fakepos.x = winwid
fakepos.y = winhei
self.disc.click(fakepos)
if <if __name__ == '__main__': from __init__ import init; init()
from os.path import realpath, join, dirname
from ranger import fsobject
from ranger.fsobject.file import File
from ranger.fsobject.directory import Directory
from ranger.shared.settings import SettingsAware
SettingsAware._setup()
TESTDIR = realpath(join(dirname(__file__), 'testdir'))
TESTFILE = join(TESTDIR, 'testfile5234148')
NONEXISTANT_DIR = join(TESTDIR, 'nonexistant')
import unittest
class Test1(unittest.TestCase):
def test_initial_condition(self):
# Check for the expected initial condition
dir = Directory(TESTDIR)
self.assertEqual(dir.path, TESTDIR)
self.assertFalse(dir.content_loaded)
self.assertEqual(dir.filenames, None)
self.assertEqual(dir.files, None)
self.assertRaises(fsobject.NotLoadedYet, len, dir)
self.assertRaises(fsobject.NotLoadedYet, dir.__getitem__, 0)
def test_after_content_loaded(self):
import os
# Check whether the directory has the correct list of filenames.
dir = Directory(TESTDIR)
dir.load_content()
self.assertTrue(dir.exists)
self.assertEqual(type(dir.filenames), list)
# Get the filenames you expect it to have and sort both before
# comparing. I don't expect any order after only loading the filenames.
assumed_filenames = os.listdir(TESTDIR)
assumed_filenames = list(map(lambda str: os.path.join(TESTDIR, str),
assumed_filenames))
assumed_filenames.sort()
dir.filenames.sort()
self.assertTrue(len(dir) > 0)
self.assertEqual(dir.filenames, assumed_filenames)
# build a file object for each file in the list assumed_filenames
# and find exactly one equivalent in dir.files
for name in assumed_filenames:
f = File(name)
f.load()
for dirfile in dir.files:
if (f.path == dirfile.path and f.stat == dirfile.stat):
break
else:
self.fail("couldn't find file {0}".format(name))
def test_nonexistant_dir(self):
dir = Directory(NONEXISTANT_DIR)
dir.load_content()
self.assertTrue(dir.content_loaded)
self.assertFalse(dir.exists)
self.assertFalse(dir.accessible)
self.assertEqual(dir.filenames, None)
self.assertRaises(fsobject.NotLoadedYet, len, dir)
self.assertRaises(fsobject.NotLoadedYet, dir.__getitem__, 0)
def test_load_if_outdated(self):
import os
import time
# modify the directory. If the time between the last modification
# was within the filesystems resolution of mtime, we should have a reload
def modify_dir():
open(TESTFILE, 'w').close()
os.unlink(TESTFILE)
def mtime():
return os.stat(TESTDIR).st_mtime
dir = Directory(TESTDIR)
dir.load()
# If the modification happens to be in the same second as the
# last modification, it will result in mtime having the same
# integer value. So we wait until the resolution is exceeded
# and mtime differs.
old_mtime = mtime()
for i in range(50):
modify_dir()
if old_mtime != mtime(): break
time.sleep(0.1)
else:
# fail after 5 seconds of trying
self.fail(
"Cannot perform test: mtime of TESTDIR is not being updated.")
self.assertTrue(dir.load_if_outdated())
if __name__ == '__main__':
unittest.main()