summaryrefslogtreecommitdiffstats
path: root/tests/other/test_requestFactory.py
diff options
context:
space:
mode:
Diffstat (limited to 'tests/other/test_requestFactory.py')
-rw-r--r--tests/other/test_requestFactory.py38
1 files changed, 38 insertions, 0 deletions
diff --git a/tests/other/test_requestFactory.py b/tests/other/test_requestFactory.py
new file mode 100644
index 000000000..751e7f03b
--- /dev/null
+++ b/tests/other/test_requestFactory.py
@@ -0,0 +1,38 @@
+# -*- coding: utf-8 -*-
+
+from tests.helper.Stubs import Core
+
+from pyload.plugins.network.CurlRequest import CurlRequest
+from pyload.network.RequestFactory import RequestFactory
+
+class TestRequestFactory:
+
+ @classmethod
+ def setUpClass(cls):
+ cls.req = RequestFactory(Core())
+
+ def test_get_request(self):
+ req = self.req.getRequest()
+
+ new_req = self.req.getRequest(req.getContext())
+ assert new_req.getContext() == req.getContext()
+
+ cj = CurlRequest.CONTEXT_CLASS()
+ assert self.req.getRequest(cj).context is cj
+
+ def test_get_request_class(self):
+
+ self.req.getRequest(None, CurlRequest)
+
+ def test_get_download(self):
+ dl = self.req.getDownloadRequest()
+ dl.close()
+
+ # with given request
+ req = self.req.getRequest()
+ dl = self.req.getDownloadRequest(req)
+
+ assert req.context is dl.context
+ assert req.options is dl.options
+
+ dl.close() \ No newline at end of file