From 23c80e11581ebac8cebd93c3fb01dd448d15000a Mon Sep 17 00:00:00 2001
From: Andy Pan <panjf2000@gmail.com>
Date: Sun, 15 Jul 2018 20:50:10 +0800
Subject: [PATCH] add codecov test

---
 .travis.yml            | 10 +++++--
 ants_benchmark_test.go |  4 +--
 ants_test.go           | 66 ++++++++++++++++++++++++++----------------
 3 files changed, 51 insertions(+), 29 deletions(-)

diff --git a/.travis.yml b/.travis.yml
index c7e9cc6..2e9b7ad 100644
--- a/.travis.yml
+++ b/.travis.yml
@@ -6,6 +6,12 @@ go:
   - "1.10.x"
   - master
 
+before_install:
+  - go get -t -v ./...
+
 script:
-  - go test -v ./...
-  - go test -bench=. -benchmem=true -run=none ./...
\ No newline at end of file
+  - go test -race -coverprofile=coverage.txt -covermode=atomic
+  - go test -bench=. -benchmem=true -run=none ./...
+
+after_success:
+  - bash <(curl -s https://codecov.io/bash)
\ No newline at end of file
diff --git a/ants_benchmark_test.go b/ants_benchmark_test.go
index 7ef3e89..ea22d3d 100644
--- a/ants_benchmark_test.go
+++ b/ants_benchmark_test.go
@@ -42,9 +42,9 @@ const (
 	YiB // 1208925819614629174706176
 )
 const (
-	RunTimes = 1000000
+	RunTimes = 1000
 	Param    = 10
-	AntsSize = 100000
+	AntsSize = 100
 )
 
 func demoFunc() error {
diff --git a/ants_test.go b/ants_test.go
index 11059ec..2b8dcb9 100644
--- a/ants_test.go
+++ b/ants_test.go
@@ -30,9 +30,29 @@ import (
 	"github.com/panjf2000/ants"
 )
 
-var n = 1000000
+var n = 1000
 
-func TestDefaultPool(t *testing.T) {
+func TestAntsPoolWithFunc(t *testing.T) {
+	var wg sync.WaitGroup
+	p, _ := ants.NewPoolWithFunc(AntsSize, func(i interface{}) error {
+		demoPoolFunc(i)
+		wg.Done()
+		return nil
+	})
+	defer p.Release()
+
+	for i := 0; i < n; i++ {
+		wg.Add(1)
+		p.Serve(n)
+	}
+	wg.Wait()
+
+	mem := runtime.MemStats{}
+	runtime.ReadMemStats(&mem)
+	t.Logf("memory usage:%d", mem.TotalAlloc/GiB)
+}
+
+func TestAntsPool(t *testing.T) {
 	defer ants.Release()
 	var wg sync.WaitGroup
 	for i := 0; i < n; i++ {
@@ -45,10 +65,6 @@ func TestDefaultPool(t *testing.T) {
 	}
 	wg.Wait()
 
-	//t.Logf("pool capacity:%d", ants.Cap())
-	//t.Logf("free workers number:%d", ants.Free())
-
-	t.Logf("running workers number:%d", ants.Running())
 	mem := runtime.MemStats{}
 	runtime.ReadMemStats(&mem)
 	t.Logf("memory usage:%d MB", mem.TotalAlloc/MiB)
@@ -70,27 +86,27 @@ func TestNoPool(t *testing.T) {
 	t.Logf("memory usage:%d MB", mem.TotalAlloc/MiB)
 }
 
-// func TestAntsPoolWithFunc(t *testing.T) {
-// 	var wg sync.WaitGroup
-// 	p, _ := ants.NewPoolWithFunc(50000, func(i interface{}) error {
-// 		demoPoolFunc(i)
-// 		wg.Done()
-// 		return nil
-// 	})
-// 	for i := 0; i < n; i++ {
-// 		wg.Add(1)
-// 		p.Serve(n)
-// 	}
-// 	wg.Wait()
+func TestCodeCov(t *testing.T) {
+	defer ants.Release()
+	for i := 0; i < n; i++ {
+		ants.Submit(demoFunc)
+	}
+	t.Logf("pool, capacity:%d", ants.Cap())
+	t.Logf("pool, running workers number:%d", ants.Running())
+	t.Logf("pool, free workers number:%d", ants.Free())
 
-// 	//t.Logf("pool capacity:%d", ants.Cap())
-// 	//t.Logf("free workers number:%d", ants.Free())
+	p, _ := ants.NewPoolWithFunc(AntsSize, demoPoolFunc)
+	defer p.Release()
+	for i := 0; i < n; i++ {
+		p.Serve(n)
+	}
+	t.Logf("pool with func, capacity:%d", p.Cap())
+	t.Logf("pool with func, running workers number:%d", p.Running())
+	t.Logf("pool with func, free workers number:%d", p.Free())
+	p.ReSize(AntsSize / 2)
+	t.Logf("pool with func, after resize, capacity:%d", p.Cap())
 
-// 	t.Logf("running workers number:%d", p.Running())
-// 	mem := runtime.MemStats{}
-// 	runtime.ReadMemStats(&mem)
-// 	t.Logf("memory usage:%d", mem.TotalAlloc/GiB)
-// }
+}
 
 // func TestNoPool(t *testing.T) {
 // 	var wg sync.WaitGroup