Add: tests

.github/workflows/tests.yml

@@ -0,0 +1,15 @@
+name: tests
+
+on:
+ push:
+ branches:
+ - feature/*
+ - main
+
+jobs:
+ tests:
+ uses: nzbgetcom/nzbget-extensions/.github/workflows/python-tests.yml@main
+ with:
+ python-versions: "3.2 3.3 3.4 3.5 3.6 3.7 3.8 3.9 3.10 3.11 3.12"
+ supported-python-versions: "3.8 3.9 3.10 3.11 3.12"
+ test-script: tests.py

.gitignore

@@ -0,0 +1 @@
+tmp

tests.py

@@ -0,0 +1,100 @@
+#!/usr/bin/env python3
+#
+# Test for FakeDetector queue/post-processing script for NZBGet.
+#
+# Copyright (C) 2024 Denis <[email protected]>
+#
+# This program is free software; you can redistribute it and/or modify it
+# under the terms of the GNU Lesser General Public License as published by
+# the Free Software Foundation; either version 3 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 Lesser General Public License for more details.
+#
+# You should have received a copy of the GNU Lesser General Public License
+# along with the program. If not, see <http://www.gnu.org/licenses/>.
+#
+
+import sys
+from os.path import dirname
+import os
+import subprocess
+import unittest
+import shutil
+
+POSTPROCESS_SUCCESS=93
+POSTPROCESS_NONE=95
+POSTPROCESS_ERROR=94
+
+unrar = os.environ.get('unrar', 'unrar')
+
+root_dir = dirname(__file__)
+test_data_dir = root_dir + '/test_data/'
+tmp_dir = root_dir + '/tmp/'
+with_password_rar = 'with_password_123.rar'
+without_password_rar = 'without_password.rar'
+
+host = '127.0.0.1'
+username = 'TestUser'
+password = 'TestPassword'
+port = '6789'
+
+def get_python(): 
+ if os.name == 'nt':
+ return 'python'
+ return 'python3'
+
+def run_script():
+ sys.stdout.flush()
+ proc = subprocess.Popen([get_python(), root_dir + '/main.py'], stdout=subprocess.PIPE, stderr=subprocess.PIPE, env=os.environ.copy())
+ out, err = proc.communicate()
+ proc.pid
+ ret_code = proc.returncode
+ return (out.decode(), int(ret_code), err.decode())
+
+def set_defaults_env():
+ # NZBGet global options
+ os.environ['NZBNA_EVENT'] = 'FILE_DOWNLOADED'
+ os.environ['NZBPP_DIRECTORY'] = tmp_dir
+ os.environ['NZBOP_ARTICLECACHE'] = '8'
+ os.environ['NZBPO_PASSACTION'] = 'PASSACTION'
+ os.environ['NZBOP_CONTROLPORT'] = port
+ os.environ['NZBOP_CONTROLIP'] = host
+ os.environ['NZBOP_CONTROLUSERNAME'] = username
+ os.environ['NZBOP_CONTROLPASSWORD'] = password
+ os.environ['NZBPR_PASSWORDDETECTOR_HASPASSWORD'] = 'no'
+
+ # script options
+ os.environ['NZBNA_CATEGORY'] = 'Movies'
+ os.environ['NZBNA_DIRECTORY'] = tmp_dir
+ os.environ['NZBNA_NZBNAME'] = 'TestNZB'
+ os.environ['NZBPR_FAKEDETECTOR_SORTED'] = 'yes'
+ os.environ['NZBOP_TEMPDIR'] = tmp_dir
+ os.environ['NZBOP_UNRARCMD'] = unrar
+
+class Tests(unittest.TestCase):
+
+ def test_without_password(self):
+ os.mkdir(tmp_dir)
+ set_defaults_env()
+ shutil.copyfile(test_data_dir + without_password_rar, tmp_dir + without_password_rar)
+ os.environ['NZBNA_NZBID'] = without_password_rar
+ [out, _, _] = run_script()
+ self.assertTrue('Password found in' not in out)
+
+ def test_with_password(self):
+ os.mkdir(tmp_dir)
+ set_defaults_env()
+ shutil.copyfile(test_data_dir + with_password_rar, tmp_dir + with_password_rar)
+ os.environ['NZBNA_NZBID'] = with_password_rar
+ [out, _, _] = run_script()
+ self.assertTrue('Password found in' in out)
+
+ def __del__(self):
+ shutil.rmtree(tmp_dir)
+
+if __name__ == '__main__':
+ unittest.main()