127 lines
3.1 KiB
Go
127 lines
3.1 KiB
Go
package pipeline
|
|
|
|
import (
|
|
"context"
|
|
"errors"
|
|
"io"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/stretchr/testify/mock"
|
|
"github.com/urfave/cli/v3"
|
|
|
|
"go.woodpecker-ci.org/woodpecker/v3/woodpecker-go/woodpecker"
|
|
"go.woodpecker-ci.org/woodpecker/v3/woodpecker-go/woodpecker/mocks"
|
|
)
|
|
|
|
func TestPipelinePurge(t *testing.T) {
|
|
tests := []struct {
|
|
name string
|
|
repoID int64
|
|
args []string
|
|
pipelinesKeep []*woodpecker.Pipeline
|
|
pipelines []*woodpecker.Pipeline
|
|
mockDeleteError error
|
|
wantDelete int
|
|
wantErr error
|
|
}{
|
|
{
|
|
name: "success with no pipelines to purge",
|
|
repoID: 1,
|
|
args: []string{"purge", "--older-than", "1h", "repo/name"},
|
|
pipelinesKeep: []*woodpecker.Pipeline{
|
|
{Number: 1},
|
|
},
|
|
pipelines: []*woodpecker.Pipeline{},
|
|
},
|
|
{
|
|
name: "success with pipelines to purge",
|
|
repoID: 1,
|
|
args: []string{"purge", "--older-than", "1h", "repo/name"},
|
|
pipelinesKeep: []*woodpecker.Pipeline{
|
|
{Number: 1},
|
|
},
|
|
pipelines: []*woodpecker.Pipeline{
|
|
{Number: 1},
|
|
{Number: 2},
|
|
{Number: 3},
|
|
},
|
|
wantDelete: 2,
|
|
},
|
|
{
|
|
name: "error on invalid duration",
|
|
repoID: 1,
|
|
args: []string{"purge", "--older-than", "invalid", "repo/name"},
|
|
wantErr: errors.New("time: invalid duration \"invalid\""),
|
|
},
|
|
{
|
|
name: "continue on 422 error",
|
|
repoID: 1,
|
|
args: []string{"purge", "--older-than", "1h", "repo/name"},
|
|
pipelinesKeep: []*woodpecker.Pipeline{
|
|
{Number: 1},
|
|
},
|
|
pipelines: []*woodpecker.Pipeline{
|
|
{Number: 1},
|
|
{Number: 2},
|
|
{Number: 3},
|
|
},
|
|
wantDelete: 2,
|
|
mockDeleteError: &woodpecker.ClientError{
|
|
StatusCode: 422,
|
|
Message: "test error",
|
|
},
|
|
},
|
|
}
|
|
|
|
for _, tt := range tests {
|
|
t.Run(tt.name, func(t *testing.T) {
|
|
mockClient := mocks.NewClient(t)
|
|
mockClient.On("RepoLookup", mock.Anything).Maybe().Return(&woodpecker.Repo{ID: tt.repoID}, nil)
|
|
|
|
mockClient.On("PipelineList", mock.Anything, mock.Anything).Return(func(_ int64, opt woodpecker.PipelineListOptions) ([]*woodpecker.Pipeline, error) {
|
|
// Return keep pipelines for first call
|
|
if opt.Before.IsZero() {
|
|
if opt.Page == 1 {
|
|
return tt.pipelinesKeep, nil
|
|
}
|
|
return []*woodpecker.Pipeline{}, nil
|
|
}
|
|
|
|
// Return pipelines to purge for calls with Before filter
|
|
if !opt.Before.IsZero() {
|
|
if opt.Page == 1 {
|
|
return tt.pipelines, nil
|
|
}
|
|
return []*woodpecker.Pipeline{}, nil
|
|
}
|
|
|
|
return []*woodpecker.Pipeline{}, nil
|
|
}).Maybe()
|
|
|
|
if tt.mockDeleteError != nil {
|
|
mockClient.On("PipelineDelete", tt.repoID, mock.Anything).Return(tt.mockDeleteError)
|
|
} else if tt.wantDelete > 0 {
|
|
mockClient.On("PipelineDelete", tt.repoID, mock.Anything).Return(nil).Times(tt.wantDelete)
|
|
}
|
|
|
|
command := pipelinePurgeCmd
|
|
command.Writer = io.Discard
|
|
command.Action = func(_ context.Context, c *cli.Command) error {
|
|
err := pipelinePurge(c, mockClient)
|
|
|
|
if tt.wantErr != nil {
|
|
assert.EqualError(t, err, tt.wantErr.Error())
|
|
return nil
|
|
}
|
|
|
|
assert.NoError(t, err)
|
|
|
|
return nil
|
|
}
|
|
|
|
_ = command.Run(context.Background(), tt.args)
|
|
})
|
|
}
|
|
}
|