add: tests

Author: abhijeetmohanan

main.go

@@ -25,8 +25,14 @@ func main() {
 
 	log.Printf("Streaming Byte Size is %d Kb", bsvalue)
 
-	source_map_kv := utils.ParseInput("source", *source_ftp)
-	destination_map_kv := utils.ParseInput("destination", *destination_ftp)
+	source_map_kv, err := utils.ParseInput("source", *source_ftp)
+	if err != nil {
+		log.Fatalf("Failed to parse input %v", err)
+	}
+	destination_map_kv, err := utils.ParseInput("destination", *destination_ftp)
+	if err != nil {
+		log.Fatalf("Failed to parse input %v", err)
+	}
 
 	if source_map_kv["scheme"] == "ftp" && destination_map_kv["scheme"] == "ftp" {
 		log.Println("Source and Destination are ftp endpoints")
@@ -35,7 +41,7 @@ func main() {
 		utils.FtpParamsValidator(source_map_kv)
 		utils.FtpParamsValidator(destination_map_kv)
 
-		// Create Coonection and start streaming
+		// Create Connection and start streaming
 
 		utils.FtpClientHandler(source_map_kv, destination_map_kv, bsvalue)
 	}

utils/ftphandler.go

@@ -75,13 +75,13 @@ func connectToSFTP(user, password, host string) (*sftp.Client, error) {
 
 	conn, err := ssh.Dial("tcp", host, config)
 	if err != nil {
-		log.Fatalf("failed to dial: %w", err)
+		log.Fatalf("failed to dial: %v", err)
 		return nil, err
 	}
 
 	client, err := sftp.NewClient(conn)
 	if err != nil {
-		log.Fatalf("failed to create SFTP client: %w", err)
+		log.Fatalf("failed to create SFTP client: %v", err)
 		return nil, err
 	}
 	log.Printf("SFTP Client initalized %s", host)

utils/parsers.go

@@ -5,7 +5,7 @@ import (
 	"net/url"
 )
 
-func ParseInput(key string, furl string) map[string]string {
+func ParseInput(key string, furl string) (map[string]string, error) {
 	// Define return variable
 
 	params := make(map[string]string)
@@ -14,7 +14,7 @@ func ParseInput(key string, furl string) map[string]string {
 	// Parse the URLs
 	sourceURL, err := url.Parse(furl)
 	if err != nil {
-		log.Panicf("Error parsing source URL: %v\n", err)
+		return params, err
 	}
 
 	// Output the parsed URLs
@@ -37,7 +37,7 @@ func ParseInput(key string, furl string) map[string]string {
 
 	log.Println(key + ":: " + parameters)
 
-	return params
+	return params, nil
 }
 
 func NullChecker(key string, ftpurl string) {
@@ -48,7 +48,7 @@ func NullChecker(key string, ftpurl string) {
 }
 
 func SchemeValidator(scheme string) bool {
-	log.Println("scheme: %s", scheme)
+	log.Printf("scheme: %s", scheme)
 	if scheme == "ftp" {
 		return true
 	}

utils/parsers_test.go

@@ -0,0 +1,17 @@
+package utils
+
+import "testing"
+
+func TestParseInput(t *testing.T) {
+	out, err := ParseInput("source", "sftp://user:password@host/myfile/out.txt")
+	if err != nil {
+		t.Error("failed parsing URL")
+	}
+
+	got := len(out)
+	want := 5
+
+	if got != want {
+		t.Errorf("Wanted %d parameters git %d parameters ", want, got)
+	}
+}