129 lines
3.4 KiB
Go
129 lines
3.4 KiB
Go
// Copyright 2019 Drone.IO Inc. All rights reserved.
|
|
// Use of this source code is governed by the Drone Non-Commercial License
|
|
// that can be found in the LICENSE file.
|
|
|
|
package hook
|
|
|
|
import (
|
|
"context"
|
|
"testing"
|
|
|
|
"github.com/drone/drone/mock"
|
|
"github.com/drone/drone/mock/mockscm"
|
|
"github.com/drone/drone/core"
|
|
"github.com/drone/go-scm/scm"
|
|
|
|
"github.com/golang/mock/gomock"
|
|
)
|
|
|
|
var noContext = context.Background()
|
|
|
|
func TestCreate(t *testing.T) {
|
|
controller := gomock.NewController(t)
|
|
defer controller.Finish()
|
|
|
|
mockUser := &core.User{}
|
|
mockHooks := []*scm.Hook{}
|
|
mockRepo := &core.Repository{
|
|
Namespace: "octocat",
|
|
Name: "hello-world",
|
|
Slug: "octocat/hello-world",
|
|
Signer: "abc123",
|
|
}
|
|
|
|
hook := &scm.HookInput{
|
|
Name: "drone",
|
|
Target: "https://drone.company.com/hook",
|
|
Secret: "abc123",
|
|
Events: scm.HookEvents{
|
|
Branch: true,
|
|
PullRequest: true,
|
|
Push: true,
|
|
Tag: true,
|
|
},
|
|
}
|
|
|
|
mockRenewer := mock.NewMockRenewer(controller)
|
|
mockRenewer.EXPECT().Renew(gomock.Any(), mockUser, false).Return(nil)
|
|
|
|
mockRepos := mockscm.NewMockRepositoryService(controller)
|
|
mockRepos.EXPECT().ListHooks(gomock.Any(), "octocat/hello-world", gomock.Any()).Return(mockHooks, nil, nil)
|
|
mockRepos.EXPECT().CreateHook(gomock.Any(), "octocat/hello-world", hook).Return(nil, nil, nil)
|
|
|
|
client := new(scm.Client)
|
|
client.Repositories = mockRepos
|
|
|
|
service := New(client, "https://drone.company.com", mockRenewer)
|
|
err := service.Create(noContext, mockUser, mockRepo)
|
|
if err != nil {
|
|
t.Error(err)
|
|
}
|
|
}
|
|
|
|
func TestCreate_RenewErr(t *testing.T) {
|
|
controller := gomock.NewController(t)
|
|
defer controller.Finish()
|
|
|
|
mockUser := &core.User{}
|
|
|
|
mockRenewer := mock.NewMockRenewer(controller)
|
|
mockRenewer.EXPECT().Renew(gomock.Any(), mockUser, false).Return(scm.ErrNotAuthorized)
|
|
|
|
service := New(nil, "https://drone.company.com", mockRenewer)
|
|
err := service.Create(noContext, mockUser, nil)
|
|
if err != scm.ErrNotAuthorized {
|
|
t.Errorf("Want not authorized error, got %v", err)
|
|
}
|
|
}
|
|
|
|
func TestDelete(t *testing.T) {
|
|
controller := gomock.NewController(t)
|
|
defer controller.Finish()
|
|
|
|
mockUser := &core.User{}
|
|
mockHooks := []*scm.Hook{
|
|
{
|
|
ID: "1",
|
|
Name: "drone",
|
|
Target: "https://drone.company.com/hook",
|
|
},
|
|
}
|
|
mockRepo := &core.Repository{
|
|
Namespace: "octocat",
|
|
Name: "hello-world",
|
|
Slug: "octocat/hello-world",
|
|
Signer: "abc123",
|
|
}
|
|
|
|
mockRenewer := mock.NewMockRenewer(controller)
|
|
mockRenewer.EXPECT().Renew(gomock.Any(), mockUser, false).Return(nil)
|
|
|
|
mockRepos := mockscm.NewMockRepositoryService(controller)
|
|
mockRepos.EXPECT().ListHooks(gomock.Any(), "octocat/hello-world", gomock.Any()).Return(mockHooks, nil, nil)
|
|
mockRepos.EXPECT().DeleteHook(gomock.Any(), "octocat/hello-world", "1").Return(nil, nil)
|
|
|
|
client := new(scm.Client)
|
|
client.Repositories = mockRepos
|
|
|
|
service := New(client, "https://drone.company.com", mockRenewer)
|
|
err := service.Delete(noContext, mockUser, mockRepo)
|
|
if err != nil {
|
|
t.Error(err)
|
|
}
|
|
}
|
|
|
|
func TestDelete_RenewErr(t *testing.T) {
|
|
controller := gomock.NewController(t)
|
|
defer controller.Finish()
|
|
|
|
mockUser := &core.User{}
|
|
|
|
mockRenewer := mock.NewMockRenewer(controller)
|
|
mockRenewer.EXPECT().Renew(gomock.Any(), mockUser, false).Return(scm.ErrNotAuthorized)
|
|
|
|
service := New(nil, "https://drone.company.com", mockRenewer)
|
|
err := service.Delete(noContext, mockUser, nil)
|
|
if err != scm.ErrNotAuthorized {
|
|
t.Errorf("Want not authorized error, got %v", err)
|
|
}
|
|
}
|