tests: Add tests for download

Author: danez

go.mod

@@ -4,6 +4,7 @@ go 1.15
 
 require (
 	github.com/dustin/go-humanize v1.0.0
+	github.com/gorilla/mux v1.8.0
 	github.com/mholt/archiver/v3 v3.5.0
 	github.com/spf13/cobra v1.1.1
 )

go.sum

@@ -73,6 +73,8 @@ github.com/google/renameio v0.1.0/go.mod h1:KWCgfxg9yswjAJkECMjeO8J8rahYeXnNhOm4
 github.com/googleapis/gax-go/v2 v2.0.4/go.mod h1:0Wqv26UfaUD9n4G6kQubkQ+KchISgw+vpHVxEJEs9eg=
 github.com/googleapis/gax-go/v2 v2.0.5/go.mod h1:DWXyrwAJ9X0FpwwEdw+IPEYBICEFu5mhpdKc/us6bOk=
 github.com/gopherjs/gopherjs v0.0.0-20181017120253-0766667cb4d1/go.mod h1:wJfORRmW1u3UXTncJ5qlYoELFm8eSnnEO6hX4iZ3EWY=
+github.com/gorilla/mux v1.8.0 h1:i40aqfkR1h2SlN9hojwV5ZA91wcXFOvkdNIeFDP5koI=
+github.com/gorilla/mux v1.8.0/go.mod h1:DVbg23sWSpFRCP0SfiEN6jmj59UnW/n46BH5rLB71So=
 github.com/gorilla/websocket v1.4.2/go.mod h1:YR8l580nyteQvAITg2hZ9XVh4b55+EU/adAjf1fMHhE=
 github.com/grpc-ecosystem/go-grpc-middleware v1.0.0/go.mod h1:FiyG127CGDf3tlThmgyCl78X/SZQqEOJBCDaAfeWzPs=
 github.com/grpc-ecosystem/go-grpc-prometheus v1.2.0/go.mod h1:8NvIoxWQoOIhqOTXgfV/d3M/q6VIi02HzZEHgUlZvzk=

nsd/download/download_test.go

@@ -0,0 +1,56 @@
+package download
+
+import (
+	"net/http"
+	"net/http/httptest"
+	"os"
+	"testing"
+
+	"github.com/gorilla/mux"
+)
+
+func TestDownloadTgz(t *testing.T) {
+	router := mux.NewRouter()
+	router.PathPrefix("/").Handler(http.FileServer(http.Dir("./testdata/")))
+
+	server := httptest.NewServer(router)
+	defer server.Close()
+
+	filePath, err := File(server.URL + "/basic.tar.gz")
+	if filePath == "" {
+		t.Errorf("File() returned no path")
+	}
+	defer os.Remove(filePath)
+
+	if err != nil {
+		t.Errorf("File() did error %s", err)
+	}
+
+	info, err := os.Stat(filePath)
+	if os.IsNotExist(err) || info.IsDir() {
+		t.Errorf("File() returned path to file, but file does not exist")
+	}
+}
+
+func TestDownloadZip(t *testing.T) {
+	router := mux.NewRouter()
+	router.PathPrefix("/").Handler(http.FileServer(http.Dir("./testdata/")))
+
+	server := httptest.NewServer(router)
+	defer server.Close()
+
+	filePath, err := File(server.URL + "/basic.zip")
+	if filePath == "" {
+		t.Errorf("File() returned no path")
+	}
+	defer os.Remove(filePath)
+
+	if err != nil {
+		t.Errorf("File() did error %s", err)
+	}
+
+	info, err := os.Stat(filePath)
+	if os.IsNotExist(err) || info.IsDir() {
+		t.Errorf("File() returned path to file, but file does not exist")
+	}
+}