Upstream/harbor
1
0
Fork 0
mirror of https://github.com/goharbor/harbor.git synced 2025-03-02 10:41:59 +01:00
Code Issues Projects Releases Wiki Activity

fix dayspl/daysps test

Browse Source 
Signed-off-by: Ziming Zhang <zziming@vmware.com>
Change-Id: I3c654e305f774c2329109888f68e708efeca721e
This commit is contained in:
Ziming Zhang 2019-09-20 18:32:01 +08:00
parent afe81a8b3b
commit f6ac2687fa
2 changed files with 35 additions and 35 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

36
src/pkg/retention/policy/rule/dayspl/evaluator_test.go
View File

@ -17,12 +17,12 @@ package dayspl
import (
"errors"
"fmt"
"github.com/stretchr/testify/assert"
"testing"
"time"
"github.com/goharbor/harbor/src/pkg/art"
"github.com/goharbor/harbor/src/pkg/retention/policy/rule"
"github.com/stretchr/testify/assert"
"github.com/stretchr/testify/require"
"github.com/stretchr/testify/suite"
)
@ -55,14 +55,14 @@ func (e *EvaluatorTestSuite) TestNew() {
func (e *EvaluatorTestSuite) TestProcess() {
now := time.Now().UTC()
data := []*art.Candidate{
{PulledTime: daysAgo(now, 1)},
{PulledTime: daysAgo(now, 2)},
{PulledTime: daysAgo(now, 3)},
{PulledTime: daysAgo(now, 4)},
{PulledTime: daysAgo(now, 5)},
{PulledTime: daysAgo(now, 10)},
{PulledTime: daysAgo(now, 20)},
{PulledTime: daysAgo(now, 30)},
{PulledTime: daysAgo(now, 1, time.Hour)},
{PulledTime: daysAgo(now, 2, time.Hour)},
{PulledTime: daysAgo(now, 3, time.Hour)},
{PulledTime: daysAgo(now, 4, time.Hour)},
{PulledTime: daysAgo(now, 5, time.Hour)},
{PulledTime: daysAgo(now, 10, time.Hour)},
{PulledTime: daysAgo(now, 20, time.Hour)},
{PulledTime: daysAgo(now, 30, time.Hour)},
}
tests := []struct {
@ -71,13 +71,13 @@ func (e *EvaluatorTestSuite) TestProcess() {
minPullTime int64
}{
{n: 0, expected: 0, minPullTime: 0},
{n: 1, expected: 1, minPullTime: daysAgo(now, 1)},
{n: 2, expected: 2, minPullTime: daysAgo(now, 2)},
{n: 3, expected: 3, minPullTime: daysAgo(now, 3)},
{n: 4, expected: 4, minPullTime: daysAgo(now, 4)},
{n: 5, expected: 5, minPullTime: daysAgo(now, 5)},
{n: 15, expected: 6, minPullTime: daysAgo(now, 10)},
{n: 90, expected: 8, minPullTime: daysAgo(now, 30)},
{n: 1, expected: 1, minPullTime: daysAgo(now, 1, 0)},
{n: 2, expected: 2, minPullTime: daysAgo(now, 2, 0)},
{n: 3, expected: 3, minPullTime: daysAgo(now, 3, 0)},
{n: 4, expected: 4, minPullTime: daysAgo(now, 4, 0)},
{n: 5, expected: 5, minPullTime: daysAgo(now, 5, 0)},
{n: 15, expected: 6, minPullTime: daysAgo(now, 10, 0)},
{n: 90, expected: 8, minPullTime: daysAgo(now, 30, 0)},
}
for _, tt := range tests {
@ -120,6 +120,6 @@ func TestEvaluatorSuite(t *testing.T) {
suite.Run(t, &EvaluatorTestSuite{})
}
func daysAgo(from time.Time, n int) int64 {
return from.Add(time.Duration(-1*24*n) * time.Hour).Unix()
func daysAgo(from time.Time, n int, offset time.Duration) int64 {
return from.Add(time.Duration(-1*24*n)*time.Hour + offset).Unix()
}

34
src/pkg/retention/policy/rule/daysps/evaluator_test.go
View File

@ -55,14 +55,14 @@ func (e *EvaluatorTestSuite) TestNew() {
func (e *EvaluatorTestSuite) TestProcess() {
now := time.Now().UTC()
data := []*art.Candidate{
{PushedTime: daysAgo(now, 1)},
{PushedTime: daysAgo(now, 2)},
{PushedTime: daysAgo(now, 3)},
{PushedTime: daysAgo(now, 4)},
{PushedTime: daysAgo(now, 5)},
{PushedTime: daysAgo(now, 10)},
{PushedTime: daysAgo(now, 20)},
{PushedTime: daysAgo(now, 30)},
{PushedTime: daysAgo(now, 1, time.Hour)},
{PushedTime: daysAgo(now, 2, time.Hour)},
{PushedTime: daysAgo(now, 3, time.Hour)},
{PushedTime: daysAgo(now, 4, time.Hour)},
{PushedTime: daysAgo(now, 5, time.Hour)},
{PushedTime: daysAgo(now, 10, time.Hour)},
{PushedTime: daysAgo(now, 20, time.Hour)},
{PushedTime: daysAgo(now, 30, time.Hour)},
}
tests := []struct {
@ -71,13 +71,13 @@ func (e *EvaluatorTestSuite) TestProcess() {
minPushTime int64
}{
{n: 0, expected: 0, minPushTime: 0},
{n: 1, expected: 1, minPushTime: daysAgo(now, 1)},
{n: 2, expected: 2, minPushTime: daysAgo(now, 2)},
{n: 3, expected: 3, minPushTime: daysAgo(now, 3)},
{n: 4, expected: 4, minPushTime: daysAgo(now, 4)},
{n: 5, expected: 5, minPushTime: daysAgo(now, 5)},
{n: 15, expected: 6, minPushTime: daysAgo(now, 10)},
{n: 90, expected: 8, minPushTime: daysAgo(now, 30)},
{n: 1, expected: 1, minPushTime: daysAgo(now, 1, 0)},
{n: 2, expected: 2, minPushTime: daysAgo(now, 2, 0)},
{n: 3, expected: 3, minPushTime: daysAgo(now, 3, 0)},
{n: 4, expected: 4, minPushTime: daysAgo(now, 4, 0)},
{n: 5, expected: 5, minPushTime: daysAgo(now, 5, 0)},
{n: 15, expected: 6, minPushTime: daysAgo(now, 10, 0)},
{n: 90, expected: 8, minPushTime: daysAgo(now, 30, 0)},
}
for _, tt := range tests {
@ -120,6 +120,6 @@ func TestEvaluatorSuite(t *testing.T) {
suite.Run(t, &EvaluatorTestSuite{})
}
func daysAgo(from time.Time, n int) int64 {
return from.Add(time.Duration(-1*24*n) * time.Hour).Unix()
func daysAgo(from time.Time, n int, offset time.Duration) int64 {
return from.Add(time.Duration(-1*24*n)*time.Hour + offset).Unix()
}