mirror of
https://github.com/goharbor/harbor.git
synced 2024-11-27 04:35:16 +01:00
Port tests to github.com/stretchr/testify/suite
Signed-off-by: Nathan Lowe <public@nlowe.me>
This commit is contained in:
parent
8b174825c1
commit
f03ed9b14c
@ -8,9 +8,14 @@ import (
|
||||
"github.com/goharbor/harbor/src/pkg/retention/policy/rule"
|
||||
"github.com/goharbor/harbor/src/pkg/retention/res"
|
||||
"github.com/stretchr/testify/require"
|
||||
"github.com/stretchr/testify/suite"
|
||||
)
|
||||
|
||||
func TestEvaluator_New(t *testing.T) {
|
||||
type EvaluatorTestSuite struct {
|
||||
suite.Suite
|
||||
}
|
||||
|
||||
func (e *EvaluatorTestSuite) TestNew() {
|
||||
tests := []struct {
|
||||
Name string
|
||||
args rule.Parameters
|
||||
@ -23,7 +28,7 @@ func TestEvaluator_New(t *testing.T) {
|
||||
}
|
||||
|
||||
for _, tt := range tests {
|
||||
t.Run(tt.Name, func(t *testing.T) {
|
||||
e.T().Run(tt.Name, func(t *testing.T) {
|
||||
e := New(tt.args).(*evaluator)
|
||||
|
||||
require.Equal(t, tt.expectedX, e.x)
|
||||
@ -31,7 +36,7 @@ func TestEvaluator_New(t *testing.T) {
|
||||
}
|
||||
}
|
||||
|
||||
func TestEvaluator_Process(t *testing.T) {
|
||||
func (e *EvaluatorTestSuite) TestProcess() {
|
||||
now := time.Now().UTC()
|
||||
data := []*res.Candidate{
|
||||
{PushedTime: now.Add(time.Duration(1*-24) * time.Hour).Unix()},
|
||||
@ -57,7 +62,7 @@ func TestEvaluator_Process(t *testing.T) {
|
||||
}
|
||||
|
||||
for _, tt := range tests {
|
||||
t.Run(fmt.Sprintf("%d days - should keep %d", tt.days, tt.expected), func(t *testing.T) {
|
||||
e.T().Run(fmt.Sprintf("%d days - should keep %d", tt.days, tt.expected), func(t *testing.T) {
|
||||
e := New(rule.Parameters{ParameterX: tt.days})
|
||||
|
||||
result, err := e.Process(data)
|
||||
@ -67,3 +72,7 @@ func TestEvaluator_Process(t *testing.T) {
|
||||
})
|
||||
}
|
||||
}
|
||||
|
||||
func TestEvaluatorSuite(t *testing.T) {
|
||||
suite.Run(t, &EvaluatorTestSuite{})
|
||||
}
|
||||
|
Loading…
Reference in New Issue
Block a user