From 873f91be7d3316e93672731e2f3d2da02b41fca6 Mon Sep 17 00:00:00 2001 From: RaNaN Date: Fri, 19 Jul 2013 19:36:05 +0200 Subject: new plugin type and refactored request classes --- tests/other/test_requestFactory.py | 30 ++++++++++++++++++++++++++++++ 1 file changed, 30 insertions(+) create mode 100644 tests/other/test_requestFactory.py (limited to 'tests/other/test_requestFactory.py') diff --git a/tests/other/test_requestFactory.py b/tests/other/test_requestFactory.py new file mode 100644 index 000000000..734d9c1fb --- /dev/null +++ b/tests/other/test_requestFactory.py @@ -0,0 +1,30 @@ +# -*- 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() + + 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 + dl = self.req.getDownloadRequest(self.req.getRequest()) + dl.close() \ No newline at end of file -- cgit v1.2.3