add test

Author: caimaoy

test_mini_spider.py

@@ -57,13 +57,21 @@ def test_error_url_(self):
 
 class DownloadWorkerTest(unittest.TestCase):
 
+    fm = ''.join(['file:///', os.path.join(FILE_DIR, 'test_download_file')])
+    to = os.path.join(FILE_DIR, 'test_download_file_to')
+
     # 初始化工作
     def setUp(self):
         pass
 
     # 退出清理工作
     def tearDown(self):
         pass
+        '''
+        if os.path.exists(self.to):
+            os.remove(self.to)
+        '''
+
 
     @httprettified
     def test_get_url_text(self):
@@ -92,11 +100,24 @@ def test_get_url_text_404(self):
         dw = mini.DownloadWorker('http://www.test.com', '.', 'reg', 0, 3, 1, 1)
         dw.get_url_text()
 
-
     def test_get_url_text_wrong_url(self):
         dw = mini.DownloadWorker('http://wrong.url.me', '.', 'reg', 0, 3, 1, 1)
         dw.get_url_text()
 
+    def test_download_file(self):
+        dw = mini.DownloadWorker('http://no.url.me', '.', 'reg', 0, 3, 1, 1)
+        to = dw.url_to_localfile(self.fm)
+        def del_file(to):
+            if os.path.exists(to):
+                os.remove(to)
+        del_file(to)
+        dw.download_file(self.fm)
+        self.assertTrue(os.path.exists(dw.url_to_localfile(self.fm)))
+        dw.download_file(self.fm)
+        self.assertTrue(os.path.exists(dw.url_to_localfile(self.fm)))
+        del_file(to)
+
+
 class SpiderManagerTest(unittest.TestCase):
 
     # 初始化工作
@@ -122,8 +143,5 @@ def test_spidermanage_ini(self):
         )
 
 
-
-
-
 if __name__ == '__main__':
     unittest.main()