Alternative pattern for API unit tests
This commit is contained in:
Ben Schumacher
parent
6006699109
commit
e033a1a4a5
4 changed files with 441 additions and 2 deletions
|
|
@ -1,14 +1,15 @@
|
|||
package datastore
|
||||
|
||||
import (
|
||||
"errors"
|
||||
"io"
|
||||
|
||||
"github.com/drone/drone/common"
|
||||
)
|
||||
|
||||
var (
|
||||
ErrConflict = "Key not unique"
|
||||
ErrNotFound = "Key not found"
|
||||
ErrConflict = errors.New("Key not unique")
|
||||
ErrKeyNotFound = errors.New("Key not found")
|
||||
)
|
||||
|
||||
type Datastore interface {
|
||||
|
|
|
|||
304
mocks/Datastore.go
Normal file
304
mocks/Datastore.go
Normal file
|
|
@ -0,0 +1,304 @@
|
|||
package mocks
|
||||
|
||||
import "github.com/stretchr/testify/mock"
|
||||
|
||||
import "io"
|
||||
import "github.com/drone/drone/common"
|
||||
|
||||
type Datastore struct {
|
||||
mock.Mock
|
||||
}
|
||||
|
||||
func (m *Datastore) User(_a0 string) (*common.User, error) {
|
||||
ret := m.Called(_a0)
|
||||
|
||||
var r0 *common.User
|
||||
if ret.Get(0) != nil {
|
||||
r0 = ret.Get(0).(*common.User)
|
||||
}
|
||||
r1 := ret.Error(1)
|
||||
|
||||
return r0, r1
|
||||
}
|
||||
func (m *Datastore) UserCount() (int, error) {
|
||||
ret := m.Called()
|
||||
|
||||
r0 := ret.Get(0).(int)
|
||||
r1 := ret.Error(1)
|
||||
|
||||
return r0, r1
|
||||
}
|
||||
func (m *Datastore) UserList() ([]*common.User, error) {
|
||||
ret := m.Called()
|
||||
|
||||
var r0 []*common.User
|
||||
if ret.Get(0) != nil {
|
||||
r0 = ret.Get(0).([]*common.User)
|
||||
}
|
||||
r1 := ret.Error(1)
|
||||
|
||||
return r0, r1
|
||||
}
|
||||
func (m *Datastore) SetUser(_a0 *common.User) error {
|
||||
ret := m.Called(_a0)
|
||||
|
||||
r0 := ret.Error(0)
|
||||
|
||||
return r0
|
||||
}
|
||||
func (m *Datastore) SetUserNotExists(_a0 *common.User) error {
|
||||
ret := m.Called(_a0)
|
||||
|
||||
r0 := ret.Error(0)
|
||||
|
||||
return r0
|
||||
}
|
||||
func (m *Datastore) DelUser(_a0 *common.User) error {
|
||||
ret := m.Called(_a0)
|
||||
|
||||
r0 := ret.Error(0)
|
||||
|
||||
return r0
|
||||
}
|
||||
func (m *Datastore) Token(_a0 string, _a1 string) (*common.Token, error) {
|
||||
ret := m.Called(_a0, _a1)
|
||||
|
||||
var r0 *common.Token
|
||||
if ret.Get(0) != nil {
|
||||
r0 = ret.Get(0).(*common.Token)
|
||||
}
|
||||
r1 := ret.Error(1)
|
||||
|
||||
return r0, r1
|
||||
}
|
||||
func (m *Datastore) TokenList(_a0 string) ([]*common.Token, error) {
|
||||
ret := m.Called(_a0)
|
||||
|
||||
var r0 []*common.Token
|
||||
if ret.Get(0) != nil {
|
||||
r0 = ret.Get(0).([]*common.Token)
|
||||
}
|
||||
r1 := ret.Error(1)
|
||||
|
||||
return r0, r1
|
||||
}
|
||||
func (m *Datastore) SetToken(_a0 *common.Token) error {
|
||||
ret := m.Called(_a0)
|
||||
|
||||
r0 := ret.Error(0)
|
||||
|
||||
return r0
|
||||
}
|
||||
func (m *Datastore) DelToken(_a0 *common.Token) error {
|
||||
ret := m.Called(_a0)
|
||||
|
||||
r0 := ret.Error(0)
|
||||
|
||||
return r0
|
||||
}
|
||||
func (m *Datastore) Subscribed(_a0 string, _a1 string) (bool, error) {
|
||||
ret := m.Called(_a0, _a1)
|
||||
|
||||
r0 := ret.Get(0).(bool)
|
||||
r1 := ret.Error(1)
|
||||
|
||||
return r0, r1
|
||||
}
|
||||
func (m *Datastore) SetSubscriber(_a0 string, _a1 string) error {
|
||||
ret := m.Called(_a0, _a1)
|
||||
|
||||
r0 := ret.Error(0)
|
||||
|
||||
return r0
|
||||
}
|
||||
func (m *Datastore) DelSubscriber(_a0 string, _a1 string) error {
|
||||
ret := m.Called(_a0, _a1)
|
||||
|
||||
r0 := ret.Error(0)
|
||||
|
||||
return r0
|
||||
}
|
||||
func (m *Datastore) Repo(_a0 string) (*common.Repo, error) {
|
||||
ret := m.Called(_a0)
|
||||
|
||||
var r0 *common.Repo
|
||||
if ret.Get(0) != nil {
|
||||
r0 = ret.Get(0).(*common.Repo)
|
||||
}
|
||||
r1 := ret.Error(1)
|
||||
|
||||
return r0, r1
|
||||
}
|
||||
func (m *Datastore) RepoList(_a0 string) ([]*common.Repo, error) {
|
||||
ret := m.Called(_a0)
|
||||
|
||||
var r0 []*common.Repo
|
||||
if ret.Get(0) != nil {
|
||||
r0 = ret.Get(0).([]*common.Repo)
|
||||
}
|
||||
r1 := ret.Error(1)
|
||||
|
||||
return r0, r1
|
||||
}
|
||||
func (m *Datastore) RepoParams(_a0 string) (map[string]string, error) {
|
||||
ret := m.Called(_a0)
|
||||
|
||||
var r0 map[string]string
|
||||
if ret.Get(0) != nil {
|
||||
r0 = ret.Get(0).(map[string]string)
|
||||
}
|
||||
r1 := ret.Error(1)
|
||||
|
||||
return r0, r1
|
||||
}
|
||||
func (m *Datastore) RepoKeypair(_a0 string) (*common.Keypair, error) {
|
||||
ret := m.Called(_a0)
|
||||
|
||||
var r0 *common.Keypair
|
||||
if ret.Get(0) != nil {
|
||||
r0 = ret.Get(0).(*common.Keypair)
|
||||
}
|
||||
r1 := ret.Error(1)
|
||||
|
||||
return r0, r1
|
||||
}
|
||||
func (m *Datastore) SetRepo(_a0 *common.Repo) error {
|
||||
ret := m.Called(_a0)
|
||||
|
||||
r0 := ret.Error(0)
|
||||
|
||||
return r0
|
||||
}
|
||||
func (m *Datastore) SetRepoNotExists(_a0 *common.User, _a1 *common.Repo) error {
|
||||
ret := m.Called(_a0, _a1)
|
||||
|
||||
r0 := ret.Error(0)
|
||||
|
||||
return r0
|
||||
}
|
||||
func (m *Datastore) SetRepoParams(_a0 string, _a1 map[string]string) error {
|
||||
ret := m.Called(_a0, _a1)
|
||||
|
||||
r0 := ret.Error(0)
|
||||
|
||||
return r0
|
||||
}
|
||||
func (m *Datastore) SetRepoKeypair(_a0 string, _a1 *common.Keypair) error {
|
||||
ret := m.Called(_a0, _a1)
|
||||
|
||||
r0 := ret.Error(0)
|
||||
|
||||
return r0
|
||||
}
|
||||
func (m *Datastore) DelRepo(_a0 *common.Repo) error {
|
||||
ret := m.Called(_a0)
|
||||
|
||||
r0 := ret.Error(0)
|
||||
|
||||
return r0
|
||||
}
|
||||
func (m *Datastore) Build(_a0 string, _a1 int) (*common.Build, error) {
|
||||
ret := m.Called(_a0, _a1)
|
||||
|
||||
var r0 *common.Build
|
||||
if ret.Get(0) != nil {
|
||||
r0 = ret.Get(0).(*common.Build)
|
||||
}
|
||||
r1 := ret.Error(1)
|
||||
|
||||
return r0, r1
|
||||
}
|
||||
func (m *Datastore) BuildList(_a0 string) ([]*common.Build, error) {
|
||||
ret := m.Called(_a0)
|
||||
|
||||
var r0 []*common.Build
|
||||
if ret.Get(0) != nil {
|
||||
r0 = ret.Get(0).([]*common.Build)
|
||||
}
|
||||
r1 := ret.Error(1)
|
||||
|
||||
return r0, r1
|
||||
}
|
||||
func (m *Datastore) BuildLast(_a0 string) (*common.Build, error) {
|
||||
ret := m.Called(_a0)
|
||||
|
||||
var r0 *common.Build
|
||||
if ret.Get(0) != nil {
|
||||
r0 = ret.Get(0).(*common.Build)
|
||||
}
|
||||
r1 := ret.Error(1)
|
||||
|
||||
return r0, r1
|
||||
}
|
||||
func (m *Datastore) SetBuild(_a0 string, _a1 *common.Build) error {
|
||||
ret := m.Called(_a0, _a1)
|
||||
|
||||
r0 := ret.Error(0)
|
||||
|
||||
return r0
|
||||
}
|
||||
func (m *Datastore) Status(_a0 string, _a1 int, _a2 string) (*common.Status, error) {
|
||||
ret := m.Called(_a0, _a1, _a2)
|
||||
|
||||
var r0 *common.Status
|
||||
if ret.Get(0) != nil {
|
||||
r0 = ret.Get(0).(*common.Status)
|
||||
}
|
||||
r1 := ret.Error(1)
|
||||
|
||||
return r0, r1
|
||||
}
|
||||
func (m *Datastore) StatusList(_a0 string, _a1 int) ([]*common.Status, error) {
|
||||
ret := m.Called(_a0, _a1)
|
||||
|
||||
var r0 []*common.Status
|
||||
if ret.Get(0) != nil {
|
||||
r0 = ret.Get(0).([]*common.Status)
|
||||
}
|
||||
r1 := ret.Error(1)
|
||||
|
||||
return r0, r1
|
||||
}
|
||||
func (m *Datastore) SetStatus(_a0 string, _a1 int, _a2 *common.Status) error {
|
||||
ret := m.Called(_a0, _a1, _a2)
|
||||
|
||||
r0 := ret.Error(0)
|
||||
|
||||
return r0
|
||||
}
|
||||
func (m *Datastore) LogReader(_a0 string, _a1 int, _a2 int) (io.Reader, error) {
|
||||
ret := m.Called(_a0, _a1, _a2)
|
||||
|
||||
r0 := ret.Get(0).(io.Reader)
|
||||
r1 := ret.Error(1)
|
||||
|
||||
return r0, r1
|
||||
}
|
||||
func (m *Datastore) SetLogs(_a0 string, _a1 int, _a2 int, _a3 []byte) error {
|
||||
ret := m.Called(_a0, _a1, _a2, _a3)
|
||||
|
||||
r0 := ret.Error(0)
|
||||
|
||||
return r0
|
||||
}
|
||||
func (m *Datastore) SetBuildState(_a0 string, _a1 *common.Build) error {
|
||||
ret := m.Called(_a0, _a1)
|
||||
|
||||
r0 := ret.Error(0)
|
||||
|
||||
return r0
|
||||
}
|
||||
func (m *Datastore) SetBuildStatus(_a0 string, _a1 int, _a2 *common.Status) error {
|
||||
ret := m.Called(_a0, _a1, _a2)
|
||||
|
||||
r0 := ret.Error(0)
|
||||
|
||||
return r0
|
||||
}
|
||||
func (m *Datastore) SetBuildTask(_a0 string, _a1 int, _a2 *common.Task) error {
|
||||
ret := m.Called(_a0, _a1, _a2)
|
||||
|
||||
r0 := ret.Error(0)
|
||||
|
||||
return r0
|
||||
}
|
||||
115
server/badge_test.go
Normal file
115
server/badge_test.go
Normal file
|
|
@ -0,0 +1,115 @@
|
|||
package server
|
||||
|
||||
import (
|
||||
"bufio"
|
||||
"net"
|
||||
"net/http"
|
||||
"net/http/httptest"
|
||||
"testing"
|
||||
|
||||
"github.com/drone/drone/common"
|
||||
"github.com/drone/drone/mocks"
|
||||
. "github.com/franela/goblin"
|
||||
"github.com/gin-gonic/gin"
|
||||
)
|
||||
|
||||
type RecorderImpl struct {
|
||||
recorder *httptest.ResponseRecorder
|
||||
}
|
||||
|
||||
func (ri RecorderImpl) Header() http.Header {
|
||||
return ri.recorder.Header()
|
||||
}
|
||||
|
||||
func (ri RecorderImpl) Write(buf []byte) (int, error) {
|
||||
return ri.recorder.Write(buf)
|
||||
}
|
||||
|
||||
func (ri RecorderImpl) WriteHeader(code int) {
|
||||
ri.recorder.WriteHeader(code)
|
||||
}
|
||||
|
||||
func (ri RecorderImpl) CloseNotify() <-chan bool {
|
||||
return http.ResponseWriter(ri.recorder).(http.CloseNotifier).CloseNotify()
|
||||
}
|
||||
|
||||
func (ri RecorderImpl) Flush() {
|
||||
ri.recorder.Flush()
|
||||
}
|
||||
|
||||
func (ri RecorderImpl) Hijack() (net.Conn, *bufio.ReadWriter, error) {
|
||||
return http.ResponseWriter(ri.recorder).(http.Hijacker).Hijack()
|
||||
}
|
||||
|
||||
func (ri RecorderImpl) Size() int {
|
||||
return ri.recorder.Body.Len()
|
||||
}
|
||||
|
||||
func (ri RecorderImpl) Status() int {
|
||||
return ri.recorder.Code
|
||||
}
|
||||
|
||||
func (ri RecorderImpl) WriteHeaderNow() {
|
||||
// no-op?
|
||||
}
|
||||
|
||||
func (ri RecorderImpl) Written() bool {
|
||||
return false
|
||||
}
|
||||
|
||||
func TestBadage(t *testing.T) {
|
||||
g := Goblin(t)
|
||||
g.Describe("Badge", func() {
|
||||
// token := common.Token{
|
||||
// Kind: "github",
|
||||
// Login: "Freya",
|
||||
// Label: "github",
|
||||
// Repos: []string{"Freya/Hello-World"},
|
||||
// Scopes: []string{},
|
||||
// Expiry: 0,
|
||||
// Issued: 0,
|
||||
// }
|
||||
|
||||
var ds *mocks.Datastore
|
||||
var ctx gin.Context
|
||||
g.BeforeEach(func() {
|
||||
router := gin.New()
|
||||
ds = new(mocks.Datastore)
|
||||
ctx = gin.Context{Engine: router}
|
||||
ctx.Set("datastore", ds)
|
||||
})
|
||||
|
||||
g.AfterEach(func() {
|
||||
})
|
||||
|
||||
g.It("should provide SVG response", func() {
|
||||
w := new(RecorderImpl)
|
||||
w.recorder = httptest.NewRecorder()
|
||||
ctx.Writer = w
|
||||
|
||||
repo := &common.Repo{Owner: "Freya", Name: "Hello-World"}
|
||||
ctx.Set("repo", repo)
|
||||
|
||||
// TODO(benschumacher) expand this a lot.
|
||||
GetBadge(&ctx)
|
||||
g.Assert(w.Status()).Equal(200)
|
||||
|
||||
// Check the variety of states
|
||||
for _, state := range []string{common.StateError, common.StateFailure, common.StateKilled, common.StatePending, common.StateRunning, common.StateSuccess} {
|
||||
repo.Last = &common.Build{State: state}
|
||||
ctx.Set("repo", repo)
|
||||
|
||||
GetBadge(&ctx)
|
||||
g.Assert(w.Status()).Equal(200)
|
||||
}
|
||||
})
|
||||
g.It("should provide CCTray response") /*, func() {
|
||||
w := httptest.NewRecorder()
|
||||
ctx.Writer = w
|
||||
|
||||
repo := &common.Repo{Owner: "Freya", Name: "Hello-World"}
|
||||
ctx.Set("repo", repo)
|
||||
|
||||
}*/
|
||||
})
|
||||
}
|
||||
19
server/mocks_test.go
Normal file
19
server/mocks_test.go
Normal file
|
|
@ -0,0 +1,19 @@
|
|||
package server
|
||||
|
||||
import (
|
||||
"net/http"
|
||||
|
||||
"github.com/drone/drone/common"
|
||||
)
|
||||
|
||||
type MockSession struct {
|
||||
Token *common.Token
|
||||
}
|
||||
|
||||
func (s MockSession) GenerateToken(*common.Token) (string, error) {
|
||||
return "session", nil
|
||||
}
|
||||
|
||||
func (s MockSession) GetLogin(*http.Request) *common.Token {
|
||||
return s.Token
|
||||
}
|
||||
Loading…
Reference in a new issue