darkkirb/harness-drone
1
0
Fork 0
Code Issues Pull requests Projects Releases Wiki Activity

Fix unit tests to work with new data model

Browse source
This commit is contained in:
Ben Schumacher 2015-06-24 12:14:13 -06:00
parent e728349265
commit cf6f41fc9b
3 changed files with 75 additions and 75 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

10
pkg/server/badge_test.go
View file

@ -56,8 +56,8 @@ func TestBadges(t *testing.T) {
ctx.Set("datastore", store) ctx.Set("datastore", store)
ctx.Set("repo", repo) ctx.Set("repo", repo)
commit := &common.Commit{State: test.state} commit := &common.Build{Status: test.state}
store.On("CommitLast", repo, test.branch).Return(commit, test.err).Once() store.On("BuildLast", repo, test.branch).Return(commit, test.err).Once()
GetBadge(ctx) GetBadge(ctx)
g.Assert(rw.Code).Equal(200) g.Assert(rw.Code).Equal(200)
@ -77,10 +77,10 @@ func TestBadges(t *testing.T) {
ctx.Set("datastore", store) ctx.Set("datastore", store)
ctx.Set("repo", repo) ctx.Set("repo", repo)
commits := []*common.Commit{ commits := []*common.Build{
&common.Commit{State: test.state}, &common.Build{Status: test.state},
} }
store.On("CommitList", repo, mock.AnythingOfType("int"), mock.AnythingOfType("int")).Return(commits, test.err).Once() store.On("BuildList", repo, mock.AnythingOfType("int"), mock.AnythingOfType("int")).Return(commits, test.err).Once()
GetCC(ctx) GetCC(ctx)
// in an error scenario (ie no build exists) we should // in an error scenario (ie no build exists) we should

2
pkg/server/user_test.go
View file

@ -59,7 +59,7 @@ func TestUser(t *testing.T) {
g.Assert(rw.Code).Equal(200) g.Assert(rw.Code).Equal(200)
g.Assert(out.Login).Equal(user.Login) g.Assert(out.Login).Equal(user.Login)
g.Assert(out.Email).Equal(in.Email) g.Assert(out.Email).Equal(in.Email)
g.Assert(out.Gravatar).Equal("7194e8d48fa1d2b689f99443b767316c") g.Assert(out.Avatar).Equal("7194e8d48fa1d2b689f99443b767316c")
}) })
g.It("should put, error", func() { g.It("should put, error", func() {

138
pkg/store/mock/mock.go
View file

@ -4,52 +4,52 @@ import (
"io" "io"
"github.com/drone/drone/Godeps/_workspace/src/github.com/stretchr/testify/mock" "github.com/drone/drone/Godeps/_workspace/src/github.com/stretchr/testify/mock"
common "github.com/drone/drone/pkg/types" "github.com/drone/drone/pkg/types"
) )
type Store struct { type Store struct {
mock.Mock mock.Mock
} }
func (m *Store) User(id int64) (*common.User, error) { func (m *Store) User(id int64) (*types.User, error) {
ret := m.Called(id) ret := m.Called(id)
var r0 *common.User var r0 *types.User
if ret.Get(0) != nil { if ret.Get(0) != nil {
r0 = ret.Get(0).(*common.User) r0 = ret.Get(0).(*types.User)
} }
r1 := ret.Error(1) r1 := ret.Error(1)
return r0, r1 return r0, r1
} }
func (m *Store) UserLogin(_a0 string) (*common.User, error) { func (m *Store) UserLogin(_a0 string) (*types.User, error) {
ret := m.Called(_a0) ret := m.Called(_a0)
var r0 *common.User var r0 *types.User
if ret.Get(0) != nil { if ret.Get(0) != nil {
r0 = ret.Get(0).(*common.User) r0 = ret.Get(0).(*types.User)
} }
r1 := ret.Error(1) r1 := ret.Error(1)
return r0, r1 return r0, r1
} }
func (m *Store) UserList() ([]*common.User, error) { func (m *Store) UserList() ([]*types.User, error) {
ret := m.Called() ret := m.Called()
var r0 []*common.User var r0 []*types.User
if ret.Get(0) != nil { if ret.Get(0) != nil {
r0 = ret.Get(0).([]*common.User) r0 = ret.Get(0).([]*types.User)
} }
r1 := ret.Error(1) r1 := ret.Error(1)
return r0, r1 return r0, r1
} }
func (m *Store) UserFeed(_a0 *common.User, _a1 int, _a2 int) ([]*common.RepoCommit, error) { func (m *Store) UserFeed(_a0 *types.User, _a1 int, _a2 int) ([]*types.RepoCommit, error) {
ret := m.Called(_a0, _a1, _a2) ret := m.Called(_a0, _a1, _a2)
var r0 []*common.RepoCommit var r0 []*types.RepoCommit
if ret.Get(0) != nil { if ret.Get(0) != nil {
r0 = ret.Get(0).([]*common.RepoCommit) r0 = ret.Get(0).([]*types.RepoCommit)
} }
r1 := ret.Error(1) r1 := ret.Error(1)
@ -63,75 +63,75 @@ func (m *Store) UserCount() (int, error) {
return r0, r1 return r0, r1
} }
func (m *Store) AddUser(_a0 *common.User) error { func (m *Store) AddUser(_a0 *types.User) error {
ret := m.Called(_a0) ret := m.Called(_a0)
r0 := ret.Error(0) r0 := ret.Error(0)
return r0 return r0
} }
func (m *Store) SetUser(_a0 *common.User) error { func (m *Store) SetUser(_a0 *types.User) error {
ret := m.Called(_a0) ret := m.Called(_a0)
r0 := ret.Error(0) r0 := ret.Error(0)
return r0 return r0
} }
func (m *Store) DelUser(_a0 *common.User) error { func (m *Store) DelUser(_a0 *types.User) error {
ret := m.Called(_a0) ret := m.Called(_a0)
r0 := ret.Error(0) r0 := ret.Error(0)
return r0 return r0
} }
func (m *Store) Token(_a0 int64) (*common.Token, error) { func (m *Store) Token(_a0 int64) (*types.Token, error) {
ret := m.Called(_a0) ret := m.Called(_a0)
var r0 *common.Token var r0 *types.Token
if ret.Get(0) != nil { if ret.Get(0) != nil {
r0 = ret.Get(0).(*common.Token) r0 = ret.Get(0).(*types.Token)
} }
r1 := ret.Error(1) r1 := ret.Error(1)
return r0, r1 return r0, r1
} }
func (m *Store) TokenLabel(_a0 *common.User, _a1 string) (*common.Token, error) { func (m *Store) TokenLabel(_a0 *types.User, _a1 string) (*types.Token, error) {
ret := m.Called(_a0, _a1) ret := m.Called(_a0, _a1)
var r0 *common.Token var r0 *types.Token
if ret.Get(0) != nil { if ret.Get(0) != nil {
r0 = ret.Get(0).(*common.Token) r0 = ret.Get(0).(*types.Token)
} }
r1 := ret.Error(1) r1 := ret.Error(1)
return r0, r1 return r0, r1
} }
func (m *Store) TokenList(_a0 *common.User) ([]*common.Token, error) { func (m *Store) TokenList(_a0 *types.User) ([]*types.Token, error) {
ret := m.Called(_a0) ret := m.Called(_a0)
var r0 []*common.Token var r0 []*types.Token
if ret.Get(0) != nil { if ret.Get(0) != nil {
r0 = ret.Get(0).([]*common.Token) r0 = ret.Get(0).([]*types.Token)
} }
r1 := ret.Error(1) r1 := ret.Error(1)
return r0, r1 return r0, r1
} }
func (m *Store) AddToken(_a0 *common.Token) error { func (m *Store) AddToken(_a0 *types.Token) error {
ret := m.Called(_a0) ret := m.Called(_a0)
r0 := ret.Error(0) r0 := ret.Error(0)
return r0 return r0
} }
func (m *Store) DelToken(_a0 *common.Token) error { func (m *Store) DelToken(_a0 *types.Token) error {
ret := m.Called(_a0) ret := m.Called(_a0)
r0 := ret.Error(0) r0 := ret.Error(0)
return r0 return r0
} }
func (m *Store) Starred(_a0 *common.User, _a1 *common.Repo) (bool, error) { func (m *Store) Starred(_a0 *types.User, _a1 *types.Repo) (bool, error) {
ret := m.Called(_a0, _a1) ret := m.Called(_a0, _a1)
r0 := ret.Get(0).(bool) r0 := ret.Get(0).(bool)
@ -139,173 +139,173 @@ func (m *Store) Starred(_a0 *common.User, _a1 *common.Repo) (bool, error) {
return r0, r1 return r0, r1
} }
func (m *Store) AddStar(_a0 *common.User, _a1 *common.Repo) error { func (m *Store) AddStar(_a0 *types.User, _a1 *types.Repo) error {
ret := m.Called(_a0, _a1) ret := m.Called(_a0, _a1)
r0 := ret.Error(0) r0 := ret.Error(0)
return r0 return r0
} }
func (m *Store) DelStar(_a0 *common.User, _a1 *common.Repo) error { func (m *Store) DelStar(_a0 *types.User, _a1 *types.Repo) error {
ret := m.Called(_a0, _a1) ret := m.Called(_a0, _a1)
r0 := ret.Error(0) r0 := ret.Error(0)
return r0 return r0
} }
func (m *Store) Repo(id int64) (*common.Repo, error) { func (m *Store) Repo(id int64) (*types.Repo, error) {
ret := m.Called(id) ret := m.Called(id)
var r0 *common.Repo var r0 *types.Repo
if ret.Get(0) != nil { if ret.Get(0) != nil {
r0 = ret.Get(0).(*common.Repo) r0 = ret.Get(0).(*types.Repo)
} }
r1 := ret.Error(1) r1 := ret.Error(1)
return r0, r1 return r0, r1
} }
func (m *Store) RepoName(owner string, name string) (*common.Repo, error) { func (m *Store) RepoName(owner string, name string) (*types.Repo, error) {
ret := m.Called(owner, name) ret := m.Called(owner, name)
var r0 *common.Repo var r0 *types.Repo
if ret.Get(0) != nil { if ret.Get(0) != nil {
r0 = ret.Get(0).(*common.Repo) r0 = ret.Get(0).(*types.Repo)
} }
r1 := ret.Error(1) r1 := ret.Error(1)
return r0, r1 return r0, r1
} }
func (m *Store) RepoList(_a0 *common.User) ([]*common.Repo, error) { func (m *Store) RepoList(_a0 *types.User) ([]*types.Repo, error) {
ret := m.Called(_a0) ret := m.Called(_a0)
var r0 []*common.Repo var r0 []*types.Repo
if ret.Get(0) != nil { if ret.Get(0) != nil {
r0 = ret.Get(0).([]*common.Repo) r0 = ret.Get(0).([]*types.Repo)
} }
r1 := ret.Error(1) r1 := ret.Error(1)
return r0, r1 return r0, r1
} }
func (m *Store) AddRepo(_a0 *common.Repo) error { func (m *Store) AddRepo(_a0 *types.Repo) error {
ret := m.Called(_a0) ret := m.Called(_a0)
r0 := ret.Error(0) r0 := ret.Error(0)
return r0 return r0
} }
func (m *Store) SetRepo(_a0 *common.Repo) error { func (m *Store) SetRepo(_a0 *types.Repo) error {
ret := m.Called(_a0) ret := m.Called(_a0)
r0 := ret.Error(0) r0 := ret.Error(0)
return r0 return r0
} }
func (m *Store) DelRepo(_a0 *common.Repo) error { func (m *Store) DelRepo(_a0 *types.Repo) error {
ret := m.Called(_a0) ret := m.Called(_a0)
r0 := ret.Error(0) r0 := ret.Error(0)
return r0 return r0
} }
func (m *Store) Commit(_a0 int64) (*common.Commit, error) { func (m *Store) Build(_a0 int64) (*types.Build, error) {
ret := m.Called(_a0) ret := m.Called(_a0)
var r0 *common.Commit var r0 *types.Build
if ret.Get(0) != nil { if ret.Get(0) != nil {
r0 = ret.Get(0).(*common.Commit) r0 = ret.Get(0).(*types.Build)
} }
r1 := ret.Error(1) r1 := ret.Error(1)
return r0, r1 return r0, r1
} }
func (m *Store) CommitSeq(_a0 *common.Repo, _a1 int) (*common.Commit, error) { func (m *Store) BuildNumber(_a0 *types.Repo, _a1 int) (*types.Build, error) {
ret := m.Called(_a0, _a1) ret := m.Called(_a0, _a1)
var r0 *common.Commit var r0 *types.Build
if ret.Get(0) != nil { if ret.Get(0) != nil {
r0 = ret.Get(0).(*common.Commit) r0 = ret.Get(0).(*types.Build)
} }
r1 := ret.Error(1) r1 := ret.Error(1)
return r0, r1 return r0, r1
} }
func (m *Store) CommitLast(_a0 *common.Repo, _a1 string) (*common.Commit, error) { func (m *Store) BuildLast(_a0 *types.Repo, _a1 string) (*types.Build, error) {
ret := m.Called(_a0, _a1) ret := m.Called(_a0, _a1)
var r0 *common.Commit var r0 *types.Build
if ret.Get(0) != nil { if ret.Get(0) != nil {
r0 = ret.Get(0).(*common.Commit) r0 = ret.Get(0).(*types.Build)
} }
r1 := ret.Error(1) r1 := ret.Error(1)
return r0, r1 return r0, r1
} }
func (m *Store) CommitList(_a0 *common.Repo, _a1 int, _a2 int) ([]*common.Commit, error) { func (m *Store) BuildList(_a0 *types.Repo, _a1 int, _a2 int) ([]*types.Build, error) {
ret := m.Called(_a0, _a1, _a2) ret := m.Called(_a0, _a1, _a2)
var r0 []*common.Commit var r0 []*types.Build
if ret.Get(0) != nil { if ret.Get(0) != nil {
r0 = ret.Get(0).([]*common.Commit) r0 = ret.Get(0).([]*types.Build)
} }
r1 := ret.Error(1) r1 := ret.Error(1)
return r0, r1 return r0, r1
} }
func (m *Store) AddCommit(_a0 *common.Commit) error { func (m *Store) AddBuild(_a0 *types.Build) error {
ret := m.Called(_a0) ret := m.Called(_a0)
r0 := ret.Error(0) r0 := ret.Error(0)
return r0 return r0
} }
func (m *Store) SetCommit(_a0 *common.Commit) error { func (m *Store) SetBuild(_a0 *types.Build) error {
ret := m.Called(_a0) ret := m.Called(_a0)
r0 := ret.Error(0) r0 := ret.Error(0)
return r0 return r0
} }
func (m *Store) KillCommits() error { func (m *Store) KillBuilds() error {
ret := m.Called() ret := m.Called()
r0 := ret.Error(0) r0 := ret.Error(0)
return r0 return r0
} }
func (m *Store) Build(_a0 int64) (*common.Build, error) { func (m *Store) Job(_a0 int64) (*types.Job, error) {
ret := m.Called(_a0) ret := m.Called(_a0)
var r0 *common.Build var r0 *types.Job
if ret.Get(0) != nil { if ret.Get(0) != nil {
r0 = ret.Get(0).(*common.Build) r0 = ret.Get(0).(*types.Job)
} }
r1 := ret.Error(1) r1 := ret.Error(1)
return r0, r1 return r0, r1
} }
func (m *Store) BuildSeq(_a0 *common.Commit, _a1 int) (*common.Build, error) { func (m *Store) JobNumber(_a0 *types.Build, _a1 int) (*types.Job, error) {
ret := m.Called(_a0, _a1) ret := m.Called(_a0, _a1)
var r0 *common.Build var r0 *types.Job
if ret.Get(0) != nil { if ret.Get(0) != nil {
r0 = ret.Get(0).(*common.Build) r0 = ret.Get(0).(*types.Job)
} }
r1 := ret.Error(1) r1 := ret.Error(1)
return r0, r1 return r0, r1
} }
func (m *Store) BuildList(_a0 *common.Commit) ([]*common.Build, error) { func (m *Store) JobList(_a0 *types.Build) ([]*types.Job, error) {
ret := m.Called(_a0) ret := m.Called(_a0)
var r0 []*common.Build var r0 []*types.Job
if ret.Get(0) != nil { if ret.Get(0) != nil {
r0 = ret.Get(0).([]*common.Build) r0 = ret.Get(0).([]*types.Job)
} }
r1 := ret.Error(1) r1 := ret.Error(1)
return r0, r1 return r0, r1
} }
func (m *Store) SetBuild(_a0 *common.Build) error { func (m *Store) SetJob(_a0 *types.Job) error {
ret := m.Called(_a0) ret := m.Called(_a0)
r0 := ret.Error(0) r0 := ret.Error(0)
@ -352,7 +352,7 @@ func (m *Store) DelBlob(path string) error {
return r0 return r0
} }
func (m *Store) Agent(_a0 *common.Commit) (string, error) { func (m *Store) Agent(_a0 *types.Build) (string, error) {
ret := m.Called(_a0) ret := m.Called(_a0)
r0 := ret.Get(0).(string) r0 := ret.Get(0).(string)
@ -360,7 +360,7 @@ func (m *Store) Agent(_a0 *common.Commit) (string, error) {
return r0, r1 return r0, r1
} }
func (m *Store) SetAgent(_a0 *common.Commit, _a1 string) error { func (m *Store) SetAgent(_a0 *types.Build, _a1 string) error {
ret := m.Called(_a0, _a1) ret := m.Called(_a0, _a1)
r0 := ret.Error(0) r0 := ret.Error(0)