142 lines
5.7 KiB
Python
142 lines
5.7 KiB
Python
# -*- coding: utf-8 -*-
|
|
#
|
|
# This file is part of Deluge and is licensed under GNU General Public License 3.0, or later, with
|
|
# the additional special exception to link portions of this program with the OpenSSL library.
|
|
# See LICENSE for more details.
|
|
#
|
|
|
|
from __future__ import print_function, unicode_literals
|
|
|
|
import pytest
|
|
from twisted.trial import unittest
|
|
|
|
import deluge.component as component
|
|
from deluge.configmanager import ConfigManager
|
|
from deluge.ui.translations_util import setup_translations
|
|
|
|
from . import common
|
|
from .basetest import BaseTestCase
|
|
|
|
libs_available = True
|
|
# Allow running other tests without GTKUI dependencies available
|
|
try:
|
|
from deluge.ui.gtkui.mainwindow import MainWindow
|
|
from deluge.ui.gtkui.gtkui import DEFAULT_PREFS
|
|
from deluge.ui.gtkui.files_tab import FilesTab
|
|
|
|
except ImportError as err:
|
|
libs_available = False
|
|
import traceback
|
|
traceback.print_exc()
|
|
|
|
setup_translations()
|
|
|
|
|
|
@pytest.mark.gtkui
|
|
class FilesTabTestCase(BaseTestCase):
|
|
|
|
def set_up(self):
|
|
if libs_available is False:
|
|
raise unittest.SkipTest('GTKUI dependencies not available')
|
|
|
|
common.set_tmp_config_dir()
|
|
ConfigManager('gtkui.conf', defaults=DEFAULT_PREFS)
|
|
self.mainwindow = MainWindow()
|
|
self.filestab = FilesTab()
|
|
self.t_id = '1'
|
|
self.filestab.torrent_id = self.t_id
|
|
self.index = 1
|
|
|
|
def tear_down(self):
|
|
return component.shutdown()
|
|
|
|
def print_treestore(self, title, treestore):
|
|
root = treestore.get_iter_root()
|
|
level = 1
|
|
|
|
def p_level(s, l):
|
|
print('%s%s' % (' ' * l, s))
|
|
|
|
def _print_treestore_children(i, lvl):
|
|
while i:
|
|
p_level(treestore[i][0], lvl)
|
|
if treestore.iter_children(i):
|
|
_print_treestore_children(treestore.iter_children(i), lvl + 2)
|
|
i = treestore.iter_next(i)
|
|
|
|
print('\n%s' % title)
|
|
_print_treestore_children(root, level)
|
|
print('')
|
|
|
|
def verify_treestore(self, treestore, tree):
|
|
|
|
def _verify_treestore(itr, tree_values):
|
|
i = 0
|
|
while itr:
|
|
values = tree_values[i]
|
|
if treestore[itr][0] != values[0]:
|
|
return False
|
|
if treestore.iter_children(itr):
|
|
if not _verify_treestore(treestore.iter_children(itr), values[1]):
|
|
return False
|
|
itr = treestore.iter_next(itr)
|
|
i += 1
|
|
return True
|
|
return _verify_treestore(treestore.get_iter_root(), tree)
|
|
|
|
def test_files_tab(self):
|
|
self.filestab.files_list[self.t_id] = ({'index': 0, 'path': '1/test_10.txt', 'offset': 0, 'size': 13},
|
|
{'index': 1, 'path': 'test_100.txt', 'offset': 13, 'size': 14})
|
|
self.filestab.update_files()
|
|
self.filestab._on_torrentfilerenamed_event(self.t_id, self.index, '2/test_100.txt')
|
|
|
|
ret = self.verify_treestore(self.filestab.treestore, [['1/', [['test_10.txt']]], ['2/', [['test_100.txt']]]])
|
|
if not ret:
|
|
self.print_treestore('Treestore not expected:', self.filestab.treestore)
|
|
self.assertTrue(ret)
|
|
|
|
def test_files_tab2(self):
|
|
self.filestab.files_list[self.t_id] = ({'index': 0, 'path': '1/1/test_10.txt', 'offset': 0, 'size': 13},
|
|
{'index': 1, 'path': 'test_100.txt', 'offset': 13, 'size': 14})
|
|
self.filestab.update_files()
|
|
self.filestab._on_torrentfilerenamed_event(self.t_id, self.index, '1/1/test_100.txt')
|
|
|
|
ret = self.verify_treestore(self.filestab.treestore, [['1/', [['1/', [['test_100.txt'], ['test_10.txt']]]]]])
|
|
if not ret:
|
|
self.print_treestore('Treestore not expected:', self.filestab.treestore)
|
|
self.assertTrue(ret)
|
|
|
|
def test_files_tab3(self):
|
|
self.filestab.files_list[self.t_id] = ({'index': 0, 'path': '1/test_10.txt', 'offset': 0, 'size': 13},
|
|
{'index': 1, 'path': 'test_100.txt', 'offset': 13, 'size': 14})
|
|
self.filestab.update_files()
|
|
self.filestab._on_torrentfilerenamed_event(self.t_id, self.index, '1/test_100.txt')
|
|
|
|
ret = self.verify_treestore(self.filestab.treestore, [['1/', [['test_100.txt'], ['test_10.txt']]]])
|
|
if not ret:
|
|
self.print_treestore('Treestore not expected:', self.filestab.treestore)
|
|
self.assertTrue(ret)
|
|
|
|
def test_files_tab4(self):
|
|
self.filestab.files_list[self.t_id] = ({'index': 0, 'path': '1/test_10.txt', 'offset': 0, 'size': 13},
|
|
{'index': 1, 'path': '1/test_100.txt', 'offset': 13, 'size': 14})
|
|
self.filestab.update_files()
|
|
self.filestab._on_torrentfilerenamed_event(self.t_id, self.index, '1/2/test_100.txt')
|
|
|
|
ret = self.verify_treestore(self.filestab.treestore, [['1/', [['2/', [['test_100.txt']]],
|
|
['test_10.txt']]]])
|
|
if not ret:
|
|
self.print_treestore('Treestore not expected:', self.filestab.treestore)
|
|
self.assertTrue(ret)
|
|
|
|
def test_files_tab5(self):
|
|
self.filestab.files_list[self.t_id] = ({'index': 0, 'path': '1/test_10.txt', 'offset': 0, 'size': 13},
|
|
{'index': 1, 'path': '2/test_100.txt', 'offset': 13, 'size': 14})
|
|
self.filestab.update_files()
|
|
self.filestab._on_torrentfilerenamed_event(self.t_id, self.index, '1/test_100.txt')
|
|
|
|
ret = self.verify_treestore(self.filestab.treestore, [['1/', [['test_100.txt'], ['test_10.txt']]]])
|
|
if not ret:
|
|
self.print_treestore('Treestore not expected:', self.filestab.treestore)
|
|
self.assertTrue(ret)
|