1 files changed, 67 insertions, 0 deletions
diff --git a/deluge/tests/daemon_base.py b/deluge/tests/daemon_base.py
new file mode 100644
index 0000000..707570f
--- /dev/null
+++ b/deluge/tests/daemon_base.py
@@ -0,0 +1,67 @@
+#
+# 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.
+#
+
+import pytest
+from twisted.internet import defer
+from twisted.internet.error import CannotListenError
+
+import deluge.component as component
+
+from . import common
+
+
+@pytest.mark.usefixtures('config_dir')
+class DaemonBase:
+    def common_set_up(self):
+        self.listen_port = 58900
+        self.core = None
+        return component.start()
+
+    def terminate_core(self, *args):
+        if args[0] is not None:
+            if hasattr(args[0], 'getTraceback'):
+                print('terminate_core: Errback Exception: %s' % args[0].getTraceback())
+
+        if not self.core.killed:
+            d = self.core.kill()
+            return d
+
+    @defer.inlineCallbacks
+    def start_core(
+        self,
+        arg,
+        custom_script='',
+        logfile='',
+        print_stdout=True,
+        print_stderr=True,
+        timeout=5,
+        port_range=10,
+        extra_callbacks=None,
+    ):
+        logfile = f'daemon_{self.id()}.log' if logfile == '' else logfile
+
+        for dummy in range(port_range):
+            try:
+                d, self.core = common.start_core(
+                    listen_port=self.listen_port,
+                    logfile=logfile,
+                    timeout=timeout,
+                    timeout_msg='Timeout!',
+                    custom_script=custom_script,
+                    print_stdout=print_stdout,
+                    print_stderr=print_stderr,
+                    extra_callbacks=extra_callbacks,
+                    config_directory=self.config_dir,
+                )
+                yield d
+            except CannotListenError as ex:
+                exception_error = ex
+                self.listen_port += 1
+            except (KeyboardInterrupt, SystemExit):
+                raise
+            else:
+                return
+        raise exception_error