covbadger/main_test.go at master · better/covbadger · 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
90
91
92
93
94
95
96
97
98
package main

import (
	"io/ioutil"
	"os"
	"strings"
	"testing"
)

var expected string = `<svg xmlns="http://www.w3.org/2000/svg" width="96" height="20">
    <title>90</title>
    <desc>Generated with covbadger (https://github.com/imsky/covbadger)</desc>
    <linearGradient id="smooth" x2="0" y2="100%">
        <stop offset="0" stop-color="#bbb" stop-opacity=".1" />
        <stop offset="1" stop-opacity=".1" />
    </linearGradient>
    <rect rx="3" width="96" height="20" fill="#555" />
    <rect rx="3" x="60" width="36" height="20" fill="#97ca00" />
    <rect x="60" width="4" height="20" fill="#97ca00" />
    <rect rx="3" width="96" height="20" fill="url(#smooth)" />
    <g fill="#fff" text-anchor="middle" font-family="DejaVu Sans,Verdana,sans-serif" font-size="11">
        <text x="30" y="15" fill="#010101" fill-opacity=".3">coverage</text>
        <text x="30" y="14">coverage</text>
        <text x="78" y="15" fill="#010101" fill-opacity=".3">90%</text>
        <text x="78" y="14">90%</text>
    </g>
</svg>`

func TestRenderBadge(t *testing.T) {
	var err error
	badge, _ := RenderBadge(90)

	if badge != expected {
		t.Errorf("RenderBadge output is incorrect")
	}

	badge, _ = RenderBadge(100)

	if strings.Contains(badge, colors["brightgreen"]) != true {
		t.Errorf("Incorrect color for coverage badge, expected brightgreen")
	}

	badge, _ = RenderBadge(70)

	if strings.Contains(badge, colors["yellow"]) != true {
		t.Errorf("Incorrect color for coverage badge, expected yellow")
	}

	badge, _ = RenderBadge(60)

	if strings.Contains(badge, colors["orange"]) != true {
		t.Errorf("Incorrect color for coverage badge, expected orange")
	}

	badge, _ = RenderBadge(40)

	if strings.Contains(badge, colors["red"]) != true {
		t.Errorf("Incorrect color for coverage badge, expected red")
		t.Errorf(badge)
	}

	badge, err = RenderBadge(101)

	if err == nil {
		t.Errorf("Invalid coverage: greater than 100%%")
	}

	badge, err = RenderBadge(-1)

	if err == nil {
		t.Errorf("Invalid coverage: less than 0%%")
	}
}

func TestCovbadger(t *testing.T) {
	defer func() {
		if r := recover(); r == nil {
			t.Errorf("Invalid coverage value did not throw an error")
		}
	}()

	main()
	Run([]string{"99"})

	userInput := []byte("90.09")
	tmpfile, _ := ioutil.TempFile("", "covbadger-stdin")
	defer os.Remove(tmpfile.Name())
	tmpfile.Write(userInput)
	tmpfile.Seek(0, 0)

	_stdin := os.Stdin
	defer func() { os.Stdin = _stdin }()
	os.Stdin = tmpfile
	Run([]string{"-"})
	tmpfile.Close()

	Run([]string{"-99"})
}