tests: added CLI tests

Author: deven96

cmd/cmd_test.go

@@ -0,0 +1,27 @@
+package cmd
+
+import (
+	. "github.com/smartystreets/goconvey/convey"
+	"testing"
+)
+
+func TestCmd(t *testing.T) {
+	Convey("CLI [Test listing engines]\n", t, func() {
+		args := []string{"engines", "list"}
+		rootCmd.SetArgs(args)
+		So(rootCmd.Execute(), ShouldBeNil)
+	})
+
+	Convey("CLI [Test current version]\n", t, func() {
+		args := []string{"version"}
+		rootCmd.SetArgs(args)
+		So(rootCmd.Execute(), ShouldBeNil)
+	})
+
+	Convey("CLI [Test clearing cache]\n", t, func() {
+		args := []string{"clear-cache"}
+		rootCmd.SetArgs(args)
+		So(rootCmd.Execute(), ShouldBeNil)
+	})
+
+}

engine/engine_test.go

@@ -4,23 +4,24 @@ import (
 	"fmt"
 	"strings"
 	"testing"
-	// _ "github.com/go-phie/gophie/cmd"
 )
 
 func testResults(t *testing.T, engine Engine) {
 	counter := map[string]int{}
 	var result SearchResult
 	var searchTerm string
 	fmt.Println(engine.String())
-	if !strings.HasPrefix(engine.String(), "TvSeries") {
-		if strings.HasPrefix(engine.String(), "Anime") || strings.HasPrefix(engine.String(), "Takanime") {
-			searchTerm = "attack on titan"
-		} else {
-			searchTerm = "jumanji"
-		}
-	} else {
-		// search for the flash for movie series
+	// different search terms on engines
+	switch {
+	case strings.HasPrefix(engine.String(), "TvSeries"):
 		searchTerm = "devs"
+	case strings.HasPrefix(engine.String(), "TakanimeList"),
+		strings.HasPrefix(engine.String(), "AnimeOut"):
+		searchTerm = "attack on titans"
+	case strings.HasPrefix(engine.String(), "KDramaHood"):
+		searchTerm = "flower of evil"
+	default:
+		searchTerm = "jumanji"
 	}
 	result = engine.Search(searchTerm)
 
@@ -37,7 +38,7 @@ func testResults(t *testing.T, engine Engine) {
 			}
 			if movie.IsSeries == false {
 				downloadlink := strings.ToLower(movie.DownloadLink.String())
-				if !(strings.HasSuffix(downloadlink, "1") || strings.HasSuffix(downloadlink, ".mp4") || strings.Contains(downloadlink, ".mkv") || strings.Contains(downloadlink, ".avi") || strings.Contains(downloadlink, ".webm") || strings.Contains(downloadlink, "freeload") || strings.Contains(downloadlink, "download_token=") || strings.Contains(downloadlink, "mycoolmoviez") || strings.Contains(downloadlink, "server")) {
+				if !(strings.HasSuffix(downloadlink, "1") || strings.HasSuffix(downloadlink, ".mp4") || strings.Contains(downloadlink, ".mkv") || strings.Contains(downloadlink, ".avi") || strings.Contains(downloadlink, ".webm") || strings.Contains(downloadlink, "freeload") || strings.Contains(downloadlink, "download_token=") || strings.Contains(downloadlink, "mycoolmoviez") || strings.Contains(downloadlink, "server") || strings.Contains(downloadlink, "kdramahood")) {
 					t.Errorf("Could not obtain link for single movie, linked returned is %v", downloadlink)
 				}
 			}