You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
76 lines
2.7 KiB
76 lines
2.7 KiB
import sys
|
|
import os
|
|
import logging
|
|
import unittest
|
|
from unittest import TestCase
|
|
#from mock import MagicMock
|
|
|
|
from couchpotato.core.plugins.browser import FileBrowser
|
|
|
|
CHROOT_DIR = '/tmp/'
|
|
|
|
class FileBrowserChrootedTest(TestCase):
|
|
def setUp(self):
|
|
self.b = FileBrowser()
|
|
|
|
# TODO : remove scrutch:
|
|
self.b.soft_chroot = CHROOT_DIR
|
|
self.b.soft_chroot_enabled = True
|
|
|
|
# Logger
|
|
logger = logging.getLogger()
|
|
logger.setLevel(logging.DEBUG)
|
|
# To screen
|
|
hdlr = logging.StreamHandler(sys.stderr)
|
|
formatter = logging.Formatter('%(asctime)s %(levelname)s %(message)s', '%m-%d %H:%M:%S')
|
|
hdlr.setFormatter(formatter)
|
|
#logger.addHandler(hdlr)
|
|
|
|
def test_soft_chroot_enabled(self):
|
|
self.assertTrue( self.b.soft_chroot_enabled)
|
|
|
|
def test_soft_chroot_is_subdir(self):
|
|
self.assertFalse( self.b.soft_chroot_is_subdir('') )
|
|
self.assertFalse( self.b.soft_chroot_is_subdir(None) )
|
|
|
|
self.assertTrue( self.b.soft_chroot_is_subdir(CHROOT_DIR) )
|
|
noslash = CHROOT_DIR[:-1]
|
|
self.assertTrue( self.b.soft_chroot_is_subdir(noslash) )
|
|
|
|
self.assertTrue( self.b.soft_chroot_is_subdir(CHROOT_DIR + 'come') )
|
|
|
|
def test_soft_chroot_add(self):
|
|
with self.assertRaises(ValueError):
|
|
self.b.soft_chroot_add('no_leading_slash')
|
|
|
|
self.assertEqual( self.b.soft_chroot_add(None), CHROOT_DIR )
|
|
self.assertEqual( self.b.soft_chroot_add(''), CHROOT_DIR )
|
|
|
|
self.assertEqual( self.b.soft_chroot_add('/asdf'), CHROOT_DIR + 'asdf' )
|
|
|
|
def test_soft_chroot_cut(self):
|
|
with self.assertRaises(ValueError): self.b.soft_chroot_cut(None)
|
|
with self.assertRaises(ValueError): self.b.soft_chroot_cut('')
|
|
|
|
self.assertEqual( self.b.soft_chroot_cut(CHROOT_DIR + 'asdf'), '/asdf' )
|
|
self.assertEqual( self.b.soft_chroot_cut(CHROOT_DIR), '/' )
|
|
self.assertEqual( self.b.soft_chroot_cut(CHROOT_DIR.rstrip(os.path.sep)), '/' )
|
|
|
|
def test_view__chrooted_path_none(self):
|
|
#def view(self, path = '/', show_hidden = True, **kwargs):
|
|
r = self.b.view(None)
|
|
self.assertEqual(r['home'], '/')
|
|
self.assertEqual(r['parent'], '/')
|
|
self.assertTrue(r['is_root'])
|
|
|
|
def test_view__chrooted_path_chroot(self):
|
|
#def view(self, path = '/', show_hidden = True, **kwargs):
|
|
for path, parent in [('/asdf','/'), (CHROOT_DIR, '/'), ('/mnk/123/t', '/mnk/123/')]:
|
|
r = self.b.view(path)
|
|
path_strip = path
|
|
if (path.endswith(os.path.sep)):
|
|
path_strip = path_strip.rstrip(os.path.sep)
|
|
|
|
self.assertEqual(r['home'], '/')
|
|
self.assertEqual(r['parent'], parent)
|
|
self.assertFalse(r['is_root'])
|
|
|