touchstone/errors_test.go at main · xmidt-org/touchstone · 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
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
// SPDX-FileCopyrightText: 2022 Comcast Cable Communications Management, LLC
// SPDX-License-Identifier: Apache-2.0

package touchstone

import (
	"errors"
	"testing"

	"github.com/prometheus/client_golang/prometheus"
	"github.com/stretchr/testify/suite"
)

type AsAlreadyRegisteredErrorTestSuite struct {
	suite.Suite
}

func (suite *AsAlreadyRegisteredErrorTestSuite) TestNil() {
	suite.Nil(AsAlreadyRegisteredError(nil))
}

func (suite *AsAlreadyRegisteredErrorTestSuite) TestAlreadyRegisteredError() {
	input := prometheus.AlreadyRegisteredError{
		ExistingCollector: prometheus.NewCounter(prometheus.CounterOpts{
			Name: "already_registered",
			Help: "existing",
		}),
		NewCollector: prometheus.NewCounter(prometheus.CounterOpts{
			Name: "already_registered",
			Help: "new",
		}),
	}

	result := AsAlreadyRegisteredError(input)
	suite.Require().NotNil(result)
	suite.Equal(input, *result)
}

func (suite *AsAlreadyRegisteredErrorTestSuite) TestNotAlreadyRegisteredError() {
	input := errors.New("this is not an AlreadyRegisteredError")
	suite.Nil(AsAlreadyRegisteredError(input))
}

func TestAsAlreadyRegisteredError(t *testing.T) {
	suite.Run(t, new(AsAlreadyRegisteredErrorTestSuite))
}

type ExistingCollectorTestSuite struct {
	suite.Suite
}

func (suite *ExistingCollectorTestSuite) TestNilError() {
	suite.Run("NilTarget", func() {
		suite.Nil(ExistingCollector(nil, nil))
	})

	suite.Run("WithTarget", func() {
		var cv *prometheus.CounterVec
		suite.Nil(ExistingCollector(&cv, nil))
		suite.Nil(cv)
	})
}

func (suite *ExistingCollectorTestSuite) TestNotAlreadyRegisteredError() {
	suite.Run("NilTarget", func() {
		input := errors.New("this is not an AlreadyRegisteredError")
		suite.Equal(input, ExistingCollector(nil, input))
	})

	suite.Run("WithTarget", func() {
		input := errors.New("this is not an AlreadyRegisteredError")
		var cv *prometheus.CounterVec
		suite.Equal(input, ExistingCollector(nil, input))
		suite.Nil(cv)
	})
}

func (suite *ExistingCollectorTestSuite) TestAlreadyRegisteredError() {
	suite.Run("NilTarget", func() {
		input := prometheus.AlreadyRegisteredError{
			ExistingCollector: prometheus.NewCounter(prometheus.CounterOpts{
				Name: "already_registered",
				Help: "existing",
			}),
			NewCollector: prometheus.NewCounter(prometheus.CounterOpts{
				Name: "already_registered",
				Help: "new",
			}),
		}

		suite.Panics(func() {
			ExistingCollector(nil, input)
		})
	})

	suite.Run("WithTarget", func() {
		input := prometheus.AlreadyRegisteredError{
			ExistingCollector: prometheus.NewCounter(prometheus.CounterOpts{
				Name: "already_registered",
				Help: "existing",
			}),
			NewCollector: prometheus.NewCounter(prometheus.CounterOpts{
				Name: "already_registered",
				Help: "new",
			}),
		}

		var c prometheus.Counter
		suite.Nil(ExistingCollector(&c, input))
		suite.Equal(input.ExistingCollector, c)
	})
}

func TestExistingCollector(t *testing.T) {
	suite.Run(t, new(ExistingCollectorTestSuite))
}