class CriticalPartTest(LinkbeanTestCase): @mock.patch('urllib2.urlopen') def setUp(self, urlopen_mock): self.url = 'http://test.com/file.tst' self.file = mock.Mock(headers={'Accept-Ranges': 'bytes', 'Content-Length': '1000'}, read=mock.Mock(return_value='1'), code=200, url=self.url) urlopen_mock.return_value = self.file try: os.mkdir("tests/downloads") except OSError: pass self.cpart = CriticalPart("tests/downloads", "http://test.com/file.tst", 1000) def tearDown(self): try: shutil.rmtree('./tests/downloads') except OSError: pass def test_init(self): self.assertEqual(self.cpart.url, "http://test.com/file.tst") self.assertEqual(self.cpart.download_folder, "tests/downloads") self.assertEqual(self.cpart.start_at, 0) self.assertEqual(self.cpart.length, 1000) self.assertEqual(self.cpart.file_size, 1000) self.assertEqual(self.cpart.file, self.file) def test_download(self): self.cpart.download() fl = file(self.cpart.get_filename(), 'r') content = fl.read() fl.close() self.assertEqual(content, '1')
def setUp(self, urlopen_mock): self.url = 'http://test.com/file.tst' self.file = mock.Mock(headers={'Accept-Ranges': 'bytes', 'Content-Length': '1000'}, read=mock.Mock(return_value='1'), code=200, url=self.url) urlopen_mock.return_value = self.file try: os.mkdir("tests/downloads") except OSError: pass self.cpart = CriticalPart("tests/downloads", "http://test.com/file.tst", 1000)