add benchmark for func getTopN

Author: LinuxSuRen

Makefile

@@ -3,4 +3,7 @@ build:
 	go build -o bin/yaml-readme .
 copy: build
 	cp bin/yaml-readme /usr/local/bin
-
+test:
+	go test ./...
+benchmark:
+	go test -benchmem -run=^$$ -bench ^BenchmarkGetTopN$$ github.com/linuxsuren/yaml-readme/function

function/github_test.go

@@ -373,18 +373,22 @@ func TestGetTopN(t *testing.T) {
 		count  int
 		expect map[string]int
 	}{{
-		name: "normal",
+		name:   "normal",
+		values: items,
+		count:  3,
+		expect: map[string]int{
+			"seven": 11,
+			"six":   10,
+			"five":  9,
+		},
+	}, {
+		name: "less data",
 		values: map[string]int{
-			"linuxsuren": 1,
-			"fake":       2,
-			"test":       3,
-			"rick":       4,
+			"linuxsuren": 2,
 		},
-		count: 3,
+		count: 2,
 		expect: map[string]int{
-			"rick": 4,
-			"test": 3,
-			"fake": 2,
+			"linuxsuren": 2,
 		},
 	}}
 	for _, tt := range tests {
@@ -394,3 +398,23 @@ func TestGetTopN(t *testing.T) {
 		})
 	}
 }
+
+var items = map[string]int{
+	"linuxsuren": 1,
+	"fake":       2,
+	"test":       3,
+	"rick":       4,
+	"one":        5,
+	"two":        6,
+	"three":      7,
+	"four":       8,
+	"five":       9,
+	"six":        10,
+	"seven":      11,
+}
+
+func BenchmarkGetTopN(b *testing.B) {
+	for i := 0; i < b.N; i++ {
+		getTopN(items, 3)
+	}
+}

main_test.go

@@ -6,8 +6,9 @@ import (
 	"io/ioutil"
 	"reflect"
 	"testing"
+
+	"github.com/stretchr/testify/assert"
 )
-import "github.com/stretchr/testify/assert"
 
 func Test_sortBy(t *testing.T) {
 	type args struct {
@@ -302,6 +303,7 @@ printGHTable
 printHelp
 printPages
 printStarHistory
+printStargazers
 printToc
 printVisitorCount
 render