todo/todo_test.go at master · stretchr/todo · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
package to

import (
	"fmt"
	"testing"
	"time"

	"github.com/stretchr/testify/assert"
)

func TestTODOExpires(t *testing.T) {
	expired := false
	printed := ""
	expire = func(msg string) {
		expired = true
		printed = fmt.Sprintf("%s: TODO Expired: - %s", callerInfo(), msg)
	}

	Do(time.Now().Format(shortForm), "Testing expiration!")
	assert.True(t, expired, "Do function should have expired!")
	assert.Equal(t, "todo_test.go:16: TODO Expired: - Testing expiration!", printed)
}

func TestTODONotExpires(t *testing.T) {
	expired := false
	expire = func(msg string) {
		expired = true
	}
	Do(time.Now().AddDate(0, 0, 1).Format(shortForm), "Testing expiration!")
	assert.False(t, expired, "Do function should not have expired!")
}

func TestTODOCacheDate(t *testing.T) {
	expired := false
	expire = func(msg string) {
		expired = true
	}

	now := time.Now()
	parsed, _ := time.Parse(shortForm, now.Format(shortForm))

	Do(now.Format(shortForm), "Testing expiration!")
	assert.True(t, expired, "Do function should have expired!")
	assert.Equal(t, times[now.Format(shortForm)], parsed)

	Do(now.Format(shortForm), "Testing expiration!")
	assert.True(t, expired, "Do function should have expired!")

}

func BenchmarkTODOCachedParse(b *testing.B) {

	b.StopTimer()

	benchTimes := map[string]time.Time{}
	by := time.Now().Format(shortForm)

	b.StartTimer()

	for i := 0; i < b.N; i++ {
		var byDate time.Time
		ok := false
		var err error
		if byDate, ok = benchTimes[by]; !ok {
			byDate, err = time.Parse(shortForm, by)
			if err != nil {
				panic(fmt.Sprintf("Unable to parse Do date: %s", err))
			}
			benchTimes[by] = byDate
		}
	}
}

func BenchmarkTODOParseAlways(b *testing.B) {

	b.StopTimer()

	by := time.Now().Format(shortForm)

	b.StartTimer()

	for i := 0; i < b.N; i++ {
		byDate, err := time.Parse(shortForm, by)
		if err != nil {
			panic(fmt.Sprintf("Unable to parse Do date: %s", err))
		}
		_ = byDate
	}
}