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.
199 lines
8.1 KiB
199 lines
8.1 KiB
#!/usr/bin/python3 -OO
|
|
# Copyright 2007-2020 The SABnzbd-Team <team@sabnzbd.org>
|
|
#
|
|
# 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 2
|
|
# 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, write to the Free Software
|
|
# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
|
|
|
|
"""
|
|
tests.test_misc - Testing functions in misc.py
|
|
"""
|
|
|
|
import datetime
|
|
|
|
from sabnzbd import lang
|
|
from sabnzbd import misc
|
|
from sabnzbd.config import ConfigCat
|
|
from tests.testhelper import *
|
|
|
|
|
|
class TestMisc:
|
|
def assertTime(self, offset, age):
|
|
assert offset == misc.calc_age(age, trans=True)
|
|
assert offset == misc.calc_age(age, trans=False)
|
|
|
|
def test_timeformat24h(self):
|
|
assert "%H:%M:%S" == misc.time_format("%H:%M:%S")
|
|
assert "%H:%M" == misc.time_format("%H:%M")
|
|
|
|
@set_config({"ampm": True})
|
|
def test_timeformatampm(self):
|
|
misc.HAVE_AMPM = True
|
|
assert "%I:%M:%S %p" == misc.time_format("%H:%M:%S")
|
|
assert "%I:%M %p" == misc.time_format("%H:%M")
|
|
|
|
def test_calc_age(self):
|
|
date = datetime.datetime.now()
|
|
m = date - datetime.timedelta(minutes=1)
|
|
h = date - datetime.timedelta(hours=1)
|
|
d = date - datetime.timedelta(days=1)
|
|
self.assertTime("1m", m)
|
|
self.assertTime("1h", h)
|
|
self.assertTime("1d", d)
|
|
|
|
def test_monthrange(self):
|
|
# Dynamic dates would be a problem
|
|
assert 12 == len(list(misc.monthrange(datetime.date(2018, 1, 1), datetime.date(2018, 12, 31))))
|
|
assert 2 == len(list(misc.monthrange(datetime.date(2019, 1, 1), datetime.date(2019, 2, 1))))
|
|
|
|
def test_safe_lower(self):
|
|
assert "all caps" == misc.safe_lower("ALL CAPS")
|
|
assert "" == misc.safe_lower(None)
|
|
|
|
def test_cmp(self):
|
|
assert misc.cmp(1, 2) < 0
|
|
assert misc.cmp(2, 1) > 0
|
|
assert misc.cmp(1, 1) == 0
|
|
|
|
def test_cat_to_opts(self):
|
|
# Need to create the Default category
|
|
# Otherwise it will try to save the config
|
|
ConfigCat("*", {})
|
|
assert ("*", "", "Default", -100) == misc.cat_to_opts("*")
|
|
assert ("*", "", "Default", -100) == misc.cat_to_opts("Nonsense")
|
|
assert ("*", 1, "Default", -100) == misc.cat_to_opts("*", pp=1)
|
|
assert ("*", 1, "test.py", -100) == misc.cat_to_opts("*", pp=1, script="test.py")
|
|
|
|
def test_wildcard_to_re(self):
|
|
assert "\\\\\\^\\$\\.\\[" == misc.wildcard_to_re("\\^$.[")
|
|
assert "\\]\\(\\)\\+.\\|\\{\\}.*" == misc.wildcard_to_re("]()+?|{}*")
|
|
|
|
def test_cat_convert(self):
|
|
# TODO: Make test
|
|
pass
|
|
|
|
def test_convert_version(self):
|
|
assert (3010099, False) == misc.convert_version("3.1.0")
|
|
assert (3010099, False) == misc.convert_version("3.1.0BlaBla")
|
|
assert (3010001, True) == misc.convert_version("3.1.0Alpha1")
|
|
assert (3010041, True) == misc.convert_version("3.1.0Beta1")
|
|
assert (3010081, True) == misc.convert_version("3.1.0RC1")
|
|
assert (3010194, True) == misc.convert_version("3.1.1RC14")
|
|
|
|
def test_from_units(self):
|
|
assert 100.0 == misc.from_units("100")
|
|
assert 1024.0 == misc.from_units("1KB")
|
|
assert 1048576.0 == misc.from_units("1024KB")
|
|
assert 1048576.0 == misc.from_units("1024Kb")
|
|
assert 1048576.0 == misc.from_units("1024kB")
|
|
assert 1048576.0 == misc.from_units("1MB")
|
|
assert 1073741824.0 == misc.from_units("1GB")
|
|
assert 1125899906842624.0 == misc.from_units("1P")
|
|
|
|
def test_to_units(self):
|
|
assert "1 K" == misc.to_units(1024)
|
|
assert "1 KBla" == misc.to_units(1024, postfix="Bla")
|
|
assert "1.0 M" == misc.to_units(1024 * 1024)
|
|
assert "1.0 M" == misc.to_units(1024 * 1024 + 10)
|
|
assert "10.0 M" == misc.to_units(1024 * 1024 * 10)
|
|
assert "100.0 M" == misc.to_units(1024 * 1024 * 100)
|
|
assert "9.8 G" == misc.to_units(1024 * 1024 * 10000)
|
|
assert "1024.0 P" == misc.to_units(1024 ** 6)
|
|
|
|
def test_unit_back_and_forth(self):
|
|
assert 100 == misc.from_units(misc.to_units(100))
|
|
assert 1024 == misc.from_units(misc.to_units(1024))
|
|
assert 1024 ** 3 == misc.from_units(misc.to_units(1024 ** 3))
|
|
|
|
def test_caller_name(self):
|
|
@set_config({"log_level": 0})
|
|
def test_wrapper(skip):
|
|
return misc.caller_name(skip=skip)
|
|
|
|
@set_config({"log_level": 2})
|
|
def test_wrapper_2(skip):
|
|
return misc.caller_name(skip=skip)
|
|
|
|
# No logging on lower-level
|
|
assert "N/A" == test_wrapper(1)
|
|
assert "N/A" == test_wrapper(2)
|
|
assert "N/A" == test_wrapper(3)
|
|
|
|
# Wrappers originate from the set_config-wrapper
|
|
assert "test_wrapper_2" in test_wrapper_2(1)
|
|
assert "wrapper_func" in test_wrapper_2(2)
|
|
|
|
def test_split_host(self):
|
|
assert (None, None) == misc.split_host(None)
|
|
assert (None, None) == misc.split_host("")
|
|
assert ("sabnzbd.org", 123) == misc.split_host("sabnzbd.org:123")
|
|
assert ("sabnzbd.org", None) == misc.split_host("sabnzbd.org")
|
|
assert ("127.0.0.1", 566) == misc.split_host("127.0.0.1:566")
|
|
assert ("[::1]", 1234) == misc.split_host("[::1]:1234")
|
|
assert ("[2001:db8::8080]", None) == misc.split_host("[2001:db8::8080]")
|
|
|
|
@set_config({"cleanup_list": [".exe", ".nzb"]})
|
|
def test_on_cleanup_list(self):
|
|
assert misc.on_cleanup_list("test.exe")
|
|
assert misc.on_cleanup_list("TEST.EXE")
|
|
assert misc.on_cleanup_list("longExeFIlanam.EXe")
|
|
assert not misc.on_cleanup_list("testexe")
|
|
assert misc.on_cleanup_list("test.nzb")
|
|
assert not misc.on_cleanup_list("test.nzb", skip_nzb=True)
|
|
assert not misc.on_cleanup_list("test.exe.lnk")
|
|
|
|
def test_format_time_string(self):
|
|
assert "0 seconds" == misc.format_time_string(None)
|
|
assert "0 seconds" == misc.format_time_string("Test")
|
|
assert "0 seconds" == misc.format_time_string(0)
|
|
assert "1 sec" == misc.format_time_string(1)
|
|
assert "10 seconds" == misc.format_time_string(10)
|
|
assert "1 min" == misc.format_time_string(60)
|
|
assert "1 hour 1 min 1 sec" == misc.format_time_string(60 * 60 + 60 + 1)
|
|
assert "1 day 59 seconds" == misc.format_time_string(86400 + 59)
|
|
assert "2 days 2 hours 2 seconds" == misc.format_time_string(2 * 86400 + 2 * 60 * 60 + 2)
|
|
|
|
def test_format_time_string_locale(self):
|
|
# Have to set the languages, if it was compiled
|
|
locale_dir = os.path.join(SAB_BASE_DIR, "..", sabnzbd.constants.DEF_LANGUAGE)
|
|
if not os.path.exists(locale_dir):
|
|
pytest.mark.skip("No language files compiled")
|
|
|
|
lang.set_locale_info("SABnzbd", locale_dir)
|
|
lang.set_language("de")
|
|
assert "1 Sekunde" == misc.format_time_string(1)
|
|
assert "10 Sekunden" == misc.format_time_string(10)
|
|
assert "1 Minuten" == misc.format_time_string(60)
|
|
assert "1 Stunde 1 Minuten 1 Sekunde" == misc.format_time_string(60 * 60 + 60 + 1)
|
|
assert "1 Tag 59 Sekunden" == misc.format_time_string(86400 + 59)
|
|
assert "2 Tage 2 Stunden 2 Sekunden" == misc.format_time_string(2 * 86400 + 2 * 60 * 60 + 2)
|
|
|
|
def test_int_conv(self):
|
|
assert 0 == misc.int_conv("0")
|
|
assert 10 == misc.int_conv("10")
|
|
assert 10 == misc.int_conv(10)
|
|
assert 10 == misc.int_conv(10.0)
|
|
assert 0 == misc.int_conv(None)
|
|
assert 1 == misc.int_conv(True)
|
|
assert 0 == misc.int_conv(object)
|
|
|
|
def test_create_https_certificates(self):
|
|
cert_file = "test.cert"
|
|
key_file = "test.key"
|
|
assert misc.create_https_certificates(cert_file, key_file)
|
|
assert os.path.exists(cert_file)
|
|
assert os.path.exists(key_file)
|
|
|
|
# Remove files
|
|
os.unlink("test.cert")
|
|
os.unlink("test.key")
|
|
|