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.
89 lines
3.1 KiB
89 lines
3.1 KiB
5 years ago
|
#!/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_config - Testing functions in config.py
|
||
|
|
"""
|
||
|
|
|
||
|
|
import sabnzbd.config as config
|
||
|
|
|
||
|
|
|
||
|
|
class TestValidators:
|
||
|
|
def test_clean_nice_ionice_parameters_allowed(self):
|
||
|
|
""" Allowed nice and ionice parameters
|
||
|
|
https://linux.die.net/man/1/nice
|
||
|
|
https://linux.die.net/man/1/ionice
|
||
|
|
"""
|
||
|
|
|
||
|
|
def assert_allowed(inp_value):
|
||
|
|
""" Helper function to check for block """
|
||
|
|
msg, value = config.clean_nice_ionice_parameters(inp_value)
|
||
|
|
assert msg is None
|
||
|
|
assert value == inp_value
|
||
|
|
|
||
|
|
# nice
|
||
|
|
assert_allowed("-n1")
|
||
|
|
assert_allowed("-n-11")
|
||
|
|
assert_allowed("-n 3")
|
||
|
|
assert_allowed("-n -4")
|
||
|
|
assert_allowed("--adjustment=11")
|
||
|
|
assert_allowed("--adjustment=-7")
|
||
|
|
assert_allowed("--adjustment 20")
|
||
|
|
assert_allowed("--adjustment -8")
|
||
|
|
|
||
|
|
# ionice
|
||
|
|
assert_allowed("-c1")
|
||
|
|
assert_allowed("-c-11")
|
||
|
|
assert_allowed("-c 3")
|
||
|
|
assert_allowed("-c -4")
|
||
|
|
assert_allowed("--classdata=1")
|
||
|
|
assert_allowed("--classdata=-11")
|
||
|
|
assert_allowed("--classdata 3")
|
||
|
|
assert_allowed("--classdata -4")
|
||
|
|
assert_allowed("-t")
|
||
|
|
assert_allowed("--ignore")
|
||
|
|
assert_allowed("-c 11 -n 12 -t")
|
||
|
|
assert_allowed("-c 11 --classdata=12 --ignore")
|
||
|
|
assert_allowed("--ignore -n9 --class=7")
|
||
|
|
assert_allowed("-t -n9 -c7")
|
||
|
|
|
||
|
|
def test_clean_nice_ionice_parameters_blocked(self):
|
||
|
|
""" Should all be blocked """
|
||
|
|
|
||
|
|
def assert_blocked(inp_value):
|
||
|
|
""" Helper function to check for block """
|
||
|
|
msg, value = config.clean_nice_ionice_parameters(inp_value)
|
||
|
|
assert msg
|
||
|
|
assert msg.startswith("Incorrect parameter")
|
||
|
|
assert value is None
|
||
|
|
|
||
|
|
assert_blocked("-ca")
|
||
|
|
assert_blocked("-t11")
|
||
|
|
assert_blocked("-p 11")
|
||
|
|
assert_blocked("123")
|
||
|
|
assert_blocked("/bin/sh /tmp/test.sh")
|
||
|
|
assert_blocked("'/evil.sh' 11")
|
||
|
|
assert_blocked("; 11")
|
||
|
|
assert_blocked("python evil.py")
|
||
|
|
assert_blocked("-n5 /bin/echo 666")
|
||
|
|
assert_blocked("4 && test.sh")
|
||
|
|
assert_blocked("-t | bla.py")
|
||
|
|
assert_blocked("5 || now")
|
||
|
|
assert_blocked("echo 'how;now;brown;cow'")
|
||
|
|
assert_blocked("-c'echo'")
|
||
|
|
assert_blocked("--classdata=;/bin/echo")
|