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.

30 lines
1.0 KiB

"""
tests.test_nzbstuff - Testing functions in nzbstuff.py
"""
import sabnzbd.nzbstuff as nzbstuff
class TestNzbstuff:
def test_scan_passwords(self):
file_names = {
"my_awesome_nzb_file{{password}}": "password",
"file_with_text_after_pw{{passw0rd}}_[180519]": "passw0rd",
"file_without_pw": None,
"file_with_multiple_pw{{first-pw}}_{{second-pw}}": "first-pw",
}
for file_name, password in file_names.items():
assert nzbstuff.scan_password(file_name)[1] == password
def test_scan_passwords_filenames(self):
file_names = {
"my_awesome_nzb_file{{password}}": "my_awesome_nzb_file",
"file_with_text_after_pw{{passw0rd}}_[310313]": "file_with_text_after_pw",
"file_without_pw": "file_without_pw",
"file_with_multiple_pw{{first-pw}}_{{second-pw}}": "file_with_multiple_pw",
}
for file_name, clean_file_name in file_names.items():
assert nzbstuff.scan_password(file_name)[0] == clean_file_name