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)