From cb3d7263b2ec3844a5e729efffa10d227fbbc460 Mon Sep 17 00:00:00 2001
From: hantmac <hantmac@outlook.com>
Date: Tue, 3 Sep 2019 12:47:18 +0800
Subject: [PATCH] bug fix:fix go test ./.. bug

---
 backend/mock/node_test.go  |  6 ++++
 backend/utils/file_test.go | 71 +++++++++++++++++++++++++++++++++-----
 2 files changed, 69 insertions(+), 8 deletions(-)

diff --git a/backend/mock/node_test.go b/backend/mock/node_test.go
index cc2f94e5..eca321ad 100644
--- a/backend/mock/node_test.go
+++ b/backend/mock/node_test.go
@@ -41,6 +41,12 @@ func init() {
 	app.DELETE("/tasks/:id", DeleteTask)                           // 删除任务
 	app.GET("/tasks/:id/results",GetTaskResults)                  // 任务结果
 	app.GET("/tasks/:id/results/download", DownloadTaskResultsCsv) // 下载任务结果
+	app.GET("/spiders", GetSpiderList)              // 爬虫列表
+	app.GET("/spiders/:id", GetSpider)              // 爬虫详情
+	app.POST("/spiders/:id", PostSpider)            // 修改爬虫
+	app.DELETE("/spiders/:id",DeleteSpider)        // 删除爬虫
+	app.GET("/spiders/:id/tasks",GetSpiderTasks)   // 爬虫任务列表
+	app.GET("/spiders/:id/dir",GetSpiderDir)       // 爬虫目录
 }
 
 //mock test, test data in ./mock
diff --git a/backend/utils/file_test.go b/backend/utils/file_test.go
index 484366f5..64f2df6d 100644
--- a/backend/utils/file_test.go
+++ b/backend/utils/file_test.go
@@ -1,8 +1,12 @@
 package utils
 
 import (
+	"archive/zip"
 	. "github.com/smartystreets/goconvey/convey"
+	"io"
+	"log"
 	"os"
+	"runtime/debug"
 	"testing"
 )
 
@@ -38,9 +42,13 @@ func TestIsDir(t *testing.T) {
 }
 
 func TestCompress(t *testing.T) {
-	var pathString = "../utils"
+	err := os.Mkdir("testCompress", os.ModePerm)
+	if err != nil {
+		t.Error("create testCompress failed")
+	}
+	var pathString = "testCompress"
 	var files []*os.File
-	var disPath = "../utils/test"
+	var disPath = "testCompress"
 	file, err := os.Open(pathString)
 	if err != nil {
 		t.Error("open source path failed")
@@ -52,15 +60,60 @@ func TestCompress(t *testing.T) {
 			So(er, ShouldEqual, nil)
 		})
 	})
+	os.RemoveAll("testCompress")
 
 }
-
-// 测试之前需存在有效的test(.zip)文件
-func TestDeCompress(t *testing.T) {
-	var tmpFilePath = "./test"
-	tmpFile, err := os.OpenFile(tmpFilePath, os.O_RDONLY, 0777)
+func Zip(zipFile string, fileList []string) error {
+	// 创建 zip 包文件
+	fw, err := os.Create(zipFile)
 	if err != nil {
-		t.Fatal("open zip file failed")
+		log.Fatal()
+	}
+	defer fw.Close()
+
+	// 实例化新的 zip.Writer
+	zw := zip.NewWriter(fw)
+	defer func() {
+		// 检测一下是否成功关闭
+		if err := zw.Close(); err != nil {
+			log.Fatalln(err)
+		}
+	}()
+
+	for _, fileName := range fileList {
+		fr, err := os.Open(fileName)
+		if err != nil {
+			return err
+		}
+		fi, err := fr.Stat()
+		if err != nil {
+			return err
+		}
+		// 写入文件的头信息
+		fh, err := zip.FileInfoHeader(fi)
+		w, err := zw.CreateHeader(fh)
+		if err != nil {
+			return err
+		}
+		// 写入文件内容
+		_, err = io.Copy(w, fr)
+		if err != nil {
+			return err
+		}
+	}
+	return nil
+}
+
+func TestDeCompress(t *testing.T) {
+	err := os.Mkdir("testDeCompress", os.ModePerm)
+	err = Zip("demo.zip", []string{})
+	if err != nil {
+		t.Error("create zip file failed")
+	}
+	tmpFile, err := os.OpenFile("demo.zip", os.O_RDONLY, 0777)
+	if err != nil {
+		debug.PrintStack()
+		t.Error("open demo.zip failed")
 	}
 	var dstPath = "./testDeCompress"
 	Convey("Test DeCopmress func", t, func() {
@@ -68,5 +121,7 @@ func TestDeCompress(t *testing.T) {
 		err := DeCompress(tmpFile, dstPath)
 		So(err, ShouldEqual, nil)
 	})
+	os.RemoveAll("testDeCompress")
+	os.Remove("demo.zip")
 
 }