Merge pull request #16 from erizocosmico/fix/lang

add language to result of analysis
yaohui-wyh · May 23, 2019 · c5379f7 · c5379f7
2 parents b9c5a4a + 37c92a8
commit c5379f7
Show file tree

Hide file tree

Showing 2 changed files with 19 additions and 0 deletions.
diff --git a/file.go b/file.go
@@ -49,6 +49,7 @@ func AnalyzeReader(filename string, language *Language, file io.Reader, opts *Cl
 
 	clocFile := &ClocFile{
 		Name: filename,
+		Lang: language.Name,
 	}
 
 	isFirstLine := true

diff --git a/file_test.go b/file_test.go
@@ -39,6 +39,9 @@ class A:
 	if clocFile.Code != 3 {
 		t.Errorf("invalid logic. code=%v", clocFile.Code)
 	}
+	if clocFile.Lang != "Python" {
+		t.Errorf("invalid logic. lang=%v", clocFile.Lang)
+	}
 }
 
 func TestAnalayzeFile4PythonInvalid(t *testing.T) {
@@ -72,6 +75,9 @@ class A:
 	if clocFile.Code != 3 {
 		t.Errorf("invalid logic. code=%v", clocFile.Code)
 	}
+	if clocFile.Lang != "Python" {
+		t.Errorf("invalid logic. lang=%v", clocFile.Lang)
+	}
 }
 
 func TestAnalayzeFile4Go(t *testing.T) {
@@ -106,6 +112,9 @@ func main() {
 	if clocFile.Code != 4 {
 		t.Errorf("invalid logic. code=%v", clocFile.Code)
 	}
+	if clocFile.Lang != "Go" {
+		t.Errorf("invalid logic. lang=%v", clocFile.Lang)
+	}
 }
 
 func TestAnalayzeFile4GoWithOnelineBlockComment(t *testing.T) {
@@ -141,6 +150,9 @@ func main() {
 	if clocFile.Code != 6 {
 		t.Errorf("invalid logic. code=%v", clocFile.Code)
 	}
+	if clocFile.Lang != "Go" {
+		t.Errorf("invalid logic. lang=%v", clocFile.Lang)
+	}
 }
 
 func TestAnalayzeFile4GoWithCommentInnerBlockComment(t *testing.T) {
@@ -174,6 +186,9 @@ func main() {
 	if clocFile.Code != 5 {
 		t.Errorf("invalid logic. code=%v", clocFile.Code)
 	}
+	if clocFile.Lang != "Go" {
+		t.Errorf("invalid logic. lang=%v", clocFile.Lang)
+	}
 }
 
 func TestAnalayzeReader(t *testing.T) {
@@ -200,4 +215,7 @@ class A:
 	if clocFile.Code != 3 {
 		t.Errorf("invalid logic. code=%v", clocFile.Code)
 	}
+	if clocFile.Lang != "Python" {
+		t.Errorf("invalid logic. lang=%v", clocFile.Lang)
+	}
 }