Exemple #1
0
class TestUrlOpener(TestCase):
    """Unit test for cygapt.url_opener
    """

    def setUp(self):
        TestCase.setUp(self)
        self.obj = CygAptURLopener(True)

    def test__init__(self):
        self.assertTrue(isinstance(self.obj, CygAptURLopener))

    def testHttp_error_default(self):
        f = TemporaryFile()
        errcode = 404
        self.obj.http_error_default("url", f, errcode, "errmsg", "headers")
        f.close()
        self.assertEqual(self.obj.getErrorCode(), errcode)

    def testDlProgress(self):
        old_stdout = sys.stdout
        buf = StringIO()
        sys.stdout = buf
        self.obj.dlProgress(1, 512, 1024)
        sys.stdout = old_stdout
        buf.seek(0)
        out = buf.readline()
        buf.close()
        expect_out = "[====================>                   ]\r"

        self.assertEqual(out, expect_out)
Exemple #2
0
class TestUrlOpener(TestCase):
    """Unit test for cygapt.url_opener
    """
    def setUp(self):
        TestCase.setUp(self)
        self.obj = CygAptURLopener(True)

    def test__init__(self):
        self.assertTrue(isinstance(self.obj, CygAptURLopener))

    def testHttp_error_default(self):
        f = TemporaryFile()
        errcode = 404
        self.obj.http_error_default("url", f, errcode, "errmsg", "headers")
        f.close()
        self.assertEqual(self.obj.getErrorCode(), errcode)

    def testDlProgress(self):
        old_stdout = sys.stdout
        buf = StringIO()
        sys.stdout = buf
        self.obj.dlProgress(1, 512, 1024)
        sys.stdout = old_stdout
        buf.seek(0)
        out = buf.readline()
        buf.close()
        expect_out = "[====================>                   ]\r"

        self.assertEqual(out, expect_out)