fix

Author: dmkjfs

cmd/main.go

@@ -2,10 +2,16 @@ package main
 
 import (
 	"fmt"
+	"os"
 
 	"cli/internal"
 )
 
 func main () {
-    fmt.Print(internal.Respond())
+    if len(os.Args) != 2 {
+        fmt.Println("Error parsing argument: path specified incorrectly")
+    } else {
+        path := os.Args[1]
+        fmt.Print(internal.Respond(path))
+    }
 }

internal/utils.go

@@ -2,25 +2,20 @@ package internal
 
 import (
 	"fmt"
-	"os"
 	"strconv"
 	"strings"
 
 	core "github.com/statloc/core"
 )
 
-func Respond() string {
-    if len(os.Args) != 2 {
-        return "Error parsing argument: path specified incorrectly"
-    } else {
-        path := os.Args[1]
+func Respond(path string) (result string) {
         response, err := core.GetStatistics(path)
 
         if err != nil {
             return fmt.Sprintf("ERROR: path \"%s\" is not found!!!\n", path)
         }
 
-        result := fmt.Sprintf(
+        result = fmt.Sprintf(
 `🗣️ Languages
 %s
 ⚡ Components
@@ -35,8 +30,7 @@ Languages: %d   LOC: %d   Files %d
             response.Total.Files,
         )
 
-        return result
-    }
+        return
 }
 
 func GetTable(

internal/utils_test.go

@@ -11,16 +11,31 @@ import (
 	"github.com/stretchr/testify/assert"
 )
 
-func TestGetTable(t *testing.T) {
+func TestRespond(t *testing.T) {
     file, _ := os.Open("../testdata/results.txt")
 	defer file.Close() // nolint:errcheck
 
+	response := internal.Respond("../testdata")
+
+	// go line by line
+	scanner := bufio.NewScanner(file)
+	for scanner.Scan() {
+        println(response, scanner.Text())
+		assert.True(t, strings.Contains(response, scanner.Text()))
+	}
+}
+
+func TestGetTable(t *testing.T) {
+    file, _ := os.Open("../testdata/languages.txt")
+	defer file.Close() // nolint:errcheck
+
 	statistics, _ := core.GetStatistics("../testdata")
 	table := internal.GetTable(statistics.Languages, 5, 3, 5)
 
 	// go line by line
 	scanner := bufio.NewScanner(file)
 	for scanner.Scan() {
+        println(table, scanner.Text())
 		assert.True(t, strings.Contains(table, scanner.Text()))
 	}
 }

testdata/languages.txt

@@ -0,0 +1,8 @@
++--------+-----+-------+
+| Title  | LOC | Files |
++--------+-----+-------+
+| Rust   | 8   | 2     |
+| Go     | 8   | 1     |
+| Python | 4   | 2     |
+| C++    | 8   | 2     |
++--------+-----+-------+

testdata/results.txt

@@ -1,9 +1,20 @@
+🗣️ Languages
 +--------+-----+-------+
 | Title  | LOC | Files |
 +--------+-----+-------+
-| Tests  | 6   | 2     |
-| Rust   | 4   | 1     |
+| Rust   | 8   | 2     |
 | Go     | 8   | 1     |
-| Python | 2   | 1     |
-| Total  | 24  | 4     |
+| Python | 4   | 2     |
+| C++    | 8   | 2     |
 +--------+-----+-------+
+
+⚡ Components
++-------------+-----+-------+
+| Title       | LOC | Files |
++-------------+-----+-------+
+| Tests       | 10  | 3     |
+| Controllers | 8   | 2     |
++-------------+-----+-------+
+
+📊 Total statistics
+Languages: 4   LOC: 28   Files 7

testdata/somedir/controllers/main.cpp

@@ -0,0 +1,3 @@
+int main() {
+    return 0;
+}

testdata/somedir/main.py

@@ -0,0 +1 @@
+print("Hello world!")

testdata/somedir/main.rs

@@ -0,0 +1,3 @@
+fn main() {
+    println!("Hello World!");
+}

testdata/tests/controllers/main.cpp

@@ -0,0 +1,3 @@
+int main() {
+    return 0;
+}