woodpecker/server/store/datastore/repo_test.go

480 lines
11 KiB
Go
Raw Normal View History

2018-02-19 22:24:10 +00:00
// Copyright 2018 Drone.IO Inc.
2018-03-21 13:02:17 +00:00
//
2018-02-19 22:24:10 +00:00
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
2018-03-21 13:02:17 +00:00
//
2018-02-19 22:24:10 +00:00
// http://www.apache.org/licenses/LICENSE-2.0
2018-03-21 13:02:17 +00:00
//
2018-02-19 22:24:10 +00:00
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.
package datastore
2015-09-30 01:21:17 +00:00
import (
"testing"
"time"
2015-09-30 01:21:17 +00:00
"github.com/franela/goblin"
"github.com/stretchr/testify/assert"
"github.com/woodpecker-ci/woodpecker/server/model"
2015-09-30 01:21:17 +00:00
)
func TestRepos(t *testing.T) {
store, closer := newTestStore(t, new(model.Repo), new(model.User), new(model.Build))
defer closer()
2015-09-30 01:21:17 +00:00
g := goblin.Goblin(t)
g.Describe("Repo", func() {
// before each test be sure to purge the package
// table data from the database.
g.BeforeEach(func() {
_, err := store.engine.Exec("DELETE FROM builds")
g.Assert(err).IsNil()
_, err = store.engine.Exec("DELETE FROM repos")
g.Assert(err).IsNil()
_, err = store.engine.Exec("DELETE FROM users")
g.Assert(err).IsNil()
2015-09-30 01:21:17 +00:00
})
g.It("Should Set a Repo", func() {
repo := model.Repo{
2015-09-30 01:21:17 +00:00
UserID: 1,
FullName: "bradrydzewski/test",
2015-09-30 01:21:17 +00:00
Owner: "bradrydzewski",
Name: "test",
2015-09-30 01:21:17 +00:00
}
err1 := store.CreateRepo(&repo)
err2 := store.UpdateRepo(&repo)
getRepo, err3 := store.GetRepo(repo.ID)
2016-04-23 11:27:28 +00:00
g.Assert(err1).IsNil()
g.Assert(err2).IsNil()
g.Assert(err3).IsNil()
g.Assert(repo.ID).Equal(getRepo.ID)
2015-09-30 01:21:17 +00:00
})
g.It("Should Add a Repo", func() {
repo := model.Repo{
2015-09-30 01:21:17 +00:00
UserID: 1,
FullName: "bradrydzewski/test",
2015-09-30 01:21:17 +00:00
Owner: "bradrydzewski",
Name: "test",
2015-09-30 01:21:17 +00:00
}
err := store.CreateRepo(&repo)
g.Assert(err).IsNil()
2015-09-30 01:21:17 +00:00
g.Assert(repo.ID != 0).IsTrue()
})
g.It("Should Get a Repo by ID", func() {
repo := model.Repo{
2015-09-30 01:21:17 +00:00
UserID: 1,
FullName: "bradrydzewski/test",
2015-09-30 01:21:17 +00:00
Owner: "bradrydzewski",
Name: "test",
2015-09-30 01:21:17 +00:00
}
g.Assert(store.CreateRepo(&repo)).IsNil()
getrepo, err := store.GetRepo(repo.ID)
g.Assert(err).IsNil()
2015-09-30 01:21:17 +00:00
g.Assert(repo.ID).Equal(getrepo.ID)
g.Assert(repo.UserID).Equal(getrepo.UserID)
g.Assert(repo.Owner).Equal(getrepo.Owner)
g.Assert(repo.Name).Equal(getrepo.Name)
})
g.It("Should Get a Repo by Name", func() {
repo := model.Repo{
2015-09-30 01:21:17 +00:00
UserID: 1,
FullName: "bradrydzewski/test",
2015-09-30 01:21:17 +00:00
Owner: "bradrydzewski",
Name: "test",
2015-09-30 01:21:17 +00:00
}
g.Assert(store.CreateRepo(&repo)).IsNil()
getrepo, err := store.GetRepoName(repo.FullName)
g.Assert(err).IsNil()
2015-09-30 01:21:17 +00:00
g.Assert(repo.ID).Equal(getrepo.ID)
g.Assert(repo.UserID).Equal(getrepo.UserID)
g.Assert(repo.Owner).Equal(getrepo.Owner)
g.Assert(repo.Name).Equal(getrepo.Name)
})
g.It("Should Enforce Unique Repo Name", func() {
repo1 := model.Repo{
2015-09-30 01:21:17 +00:00
UserID: 1,
FullName: "bradrydzewski/test",
2015-09-30 01:21:17 +00:00
Owner: "bradrydzewski",
Name: "test",
2015-09-30 01:21:17 +00:00
}
repo2 := model.Repo{
2015-09-30 01:21:17 +00:00
UserID: 2,
FullName: "bradrydzewski/test",
2015-09-30 01:21:17 +00:00
Owner: "bradrydzewski",
Name: "test",
2015-09-30 01:21:17 +00:00
}
err1 := store.CreateRepo(&repo1)
err2 := store.CreateRepo(&repo2)
g.Assert(err1).IsNil()
g.Assert(err2 == nil).IsFalse()
2015-09-30 01:21:17 +00:00
})
})
}
2017-07-14 19:58:38 +00:00
2017-07-15 16:51:02 +00:00
func TestRepoList(t *testing.T) {
store, closer := newTestStore(t, new(model.Repo), new(model.User), new(model.Perm))
defer closer()
2017-07-15 16:51:02 +00:00
user := &model.User{
Login: "joe",
Email: "foo@bar.com",
Token: "e42080dddf012c718e476da161d21ad5",
}
assert.NoError(t, store.CreateUser(user))
2017-07-15 16:51:02 +00:00
repo1 := &model.Repo{
Owner: "bradrydzewski",
Name: "test",
FullName: "bradrydzewski/test",
RemoteID: "1",
2017-07-15 16:51:02 +00:00
}
repo2 := &model.Repo{
Owner: "test",
Name: "test",
FullName: "test/test",
RemoteID: "2",
2017-07-15 16:51:02 +00:00
}
repo3 := &model.Repo{
Owner: "octocat",
Name: "hello-world",
FullName: "octocat/hello-world",
RemoteID: "3",
2017-07-15 16:51:02 +00:00
}
assert.NoError(t, store.CreateRepo(repo1))
assert.NoError(t, store.CreateRepo(repo2))
assert.NoError(t, store.CreateRepo(repo3))
2017-07-15 16:51:02 +00:00
for _, perm := range []*model.Perm{
{UserID: user.ID, Repo: repo1},
{UserID: user.ID, Repo: repo2},
} {
assert.NoError(t, store.PermUpsert(perm))
}
2017-07-15 16:51:02 +00:00
repos, err := store.RepoList(user, false)
if err != nil {
t.Error(err)
return
}
if got, want := len(repos), 2; got != want {
t.Errorf("Want %d repositories, got %d", want, got)
}
if got, want := repos[0].ID, repo1.ID; got != want {
t.Errorf("Want repository id %d, got %d", want, got)
}
if got, want := repos[1].ID, repo2.ID; got != want {
t.Errorf("Want repository id %d, got %d", want, got)
}
}
func TestOwnedRepoList(t *testing.T) {
store, closer := newTestStore(t, new(model.Repo), new(model.User), new(model.Perm))
defer closer()
user := &model.User{
Login: "joe",
Email: "foo@bar.com",
Token: "e42080dddf012c718e476da161d21ad5",
}
assert.NoError(t, store.CreateUser(user))
repo1 := &model.Repo{
Owner: "bradrydzewski",
Name: "test",
FullName: "bradrydzewski/test",
RemoteID: "1",
}
repo2 := &model.Repo{
Owner: "test",
Name: "test",
FullName: "test/test",
RemoteID: "2",
}
repo3 := &model.Repo{
Owner: "octocat",
Name: "hello-world",
FullName: "octocat/hello-world",
RemoteID: "3",
}
repo4 := &model.Repo{
Owner: "demo",
Name: "demo",
FullName: "demo/demo",
RemoteID: "4",
}
assert.NoError(t, store.CreateRepo(repo1))
assert.NoError(t, store.CreateRepo(repo2))
assert.NoError(t, store.CreateRepo(repo3))
assert.NoError(t, store.CreateRepo(repo4))
for _, perm := range []*model.Perm{
{UserID: user.ID, Repo: repo1, Push: true, Admin: false},
{UserID: user.ID, Repo: repo2, Push: false, Admin: true},
{UserID: user.ID, Repo: repo3},
{UserID: user.ID, Repo: repo4},
} {
assert.NoError(t, store.PermUpsert(perm))
}
repos, err := store.RepoList(user, true)
2017-07-15 16:51:02 +00:00
if err != nil {
t.Error(err)
return
}
if got, want := len(repos), 2; got != want {
t.Errorf("Want %d repositories, got %d", want, got)
}
if got, want := repos[0].ID, repo1.ID; got != want {
t.Errorf("Want repository id %d, got %d", want, got)
}
if got, want := repos[1].ID, repo2.ID; got != want {
t.Errorf("Want repository id %d, got %d", want, got)
}
}
func TestRepoCount(t *testing.T) {
store, closer := newTestStore(t, new(model.Repo))
defer closer()
2017-07-15 16:51:02 +00:00
repo1 := &model.Repo{
Owner: "bradrydzewski",
Name: "test",
FullName: "bradrydzewski/test",
2017-07-25 19:32:22 +00:00
IsActive: true,
2017-07-15 16:51:02 +00:00
}
repo2 := &model.Repo{
Owner: "test",
Name: "test",
FullName: "test/test",
2017-07-25 19:32:22 +00:00
IsActive: true,
}
repo3 := &model.Repo{
Owner: "test",
Name: "test-ui",
FullName: "test/test-ui",
2017-07-25 19:32:22 +00:00
IsActive: false,
2017-07-15 16:51:02 +00:00
}
assert.NoError(t, store.CreateRepo(repo1))
assert.NoError(t, store.CreateRepo(repo2))
assert.NoError(t, store.CreateRepo(repo3))
2017-07-15 16:51:02 +00:00
count, _ := store.GetRepoCount()
if got, want := count, int64(2); got != want {
2017-07-15 16:51:02 +00:00
t.Errorf("Want %d repositories, got %d", want, got)
}
}
2017-07-14 19:58:38 +00:00
func TestRepoBatch(t *testing.T) {
store, closer := newTestStore(t, new(model.Repo), new(model.User), new(model.Perm), new(model.Redirection))
defer closer()
if !assert.NoError(t, store.CreateRepo(&model.Repo{
RemoteID: "5",
2017-07-14 19:58:38 +00:00
UserID: 1,
FullName: "foo/bar",
Owner: "foo",
Name: "bar",
2017-07-25 19:32:22 +00:00
IsActive: true,
})) {
2017-07-14 19:58:38 +00:00
return
}
repos := []*model.Repo{
{
RemoteID: "5",
UserID: 1,
FullName: "foo/bar",
Owner: "foo",
Name: "bar",
IsActive: true,
Perm: &model.Perm{
UserID: 1,
Pull: true,
Push: true,
Admin: true,
Synced: time.Now().Unix(),
},
},
{
RemoteID: "6",
UserID: 1,
FullName: "bar/baz",
Owner: "bar",
Name: "baz",
IsActive: true,
},
{
RemoteID: "7",
UserID: 1,
FullName: "baz/qux",
Owner: "baz",
Name: "qux",
IsActive: true,
2017-07-14 19:58:38 +00:00
},
{
RemoteID: "8",
UserID: 0, // not activated repos do hot have a user id assigned
FullName: "baz/notes",
Owner: "baz",
Name: "notes",
IsActive: false,
},
}
if !assert.NoError(t, store.RepoBatch(repos)) {
2017-07-14 19:58:38 +00:00
return
}
// check DB state
perm, err := store.PermFind(&model.User{ID: 1}, repos[0])
assert.NoError(t, err)
assert.True(t, perm.Admin)
repo := &model.Repo{
RemoteID: "5",
FullName: "foo/bar",
Owner: "foo",
Name: "bar",
Perm: &model.Perm{
UserID: 1,
Pull: true,
Push: true,
Admin: false,
Synced: time.Now().Unix(),
},
}
assert.NoError(t, store.RepoBatch([]*model.Repo{repo}))
assert.EqualValues(t, repos[0].ID, repo.ID)
// check current DB state
_, err = store.engine.ID(repo.ID).Get(repo)
assert.NoError(t, err)
assert.True(t, repo.IsActive)
perm, err = store.PermFind(&model.User{ID: 1}, repos[0])
assert.NoError(t, err)
assert.False(t, perm.Admin)
allRepos := make([]*model.Repo, 0, 4)
assert.NoError(t, store.engine.Find(&allRepos))
assert.Len(t, allRepos, 4)
count, err := store.GetRepoCount()
assert.NoError(t, err)
assert.EqualValues(t, 3, count)
2017-07-14 19:58:38 +00:00
}
2017-07-15 16:51:02 +00:00
func TestRepoCrud(t *testing.T) {
store, closer := newTestStore(t,
new(model.Repo),
new(model.User),
new(model.Perm),
new(model.Build),
new(model.BuildConfig),
new(model.Logs),
new(model.Proc),
new(model.File),
new(model.Secret),
new(model.Registry),
new(model.Config),
new(model.Redirection))
defer closer()
2017-07-15 16:51:02 +00:00
repo := model.Repo{
UserID: 1,
FullName: "bradrydzewski/test",
2017-07-15 16:51:02 +00:00
Owner: "bradrydzewski",
Name: "test",
2017-07-15 16:51:02 +00:00
}
assert.NoError(t, store.CreateRepo(&repo))
build := model.Build{
RepoID: repo.ID,
}
proc := model.Proc{
Name: "a proc",
2017-07-15 16:51:02 +00:00
}
assert.NoError(t, store.CreateBuild(&build, &proc))
// create unrelated
repoUnrelated := model.Repo{
UserID: 2,
FullName: "x/x",
Owner: "x",
Name: "x",
2017-07-15 16:51:02 +00:00
}
assert.NoError(t, store.CreateRepo(&repoUnrelated))
buildUnrelated := model.Build{
RepoID: repoUnrelated.ID,
2017-07-15 16:51:02 +00:00
}
procUnrelated := model.Proc{
Name: "a unrelated proc",
}
assert.NoError(t, store.CreateBuild(&buildUnrelated, &procUnrelated))
_, err := store.GetRepo(repo.ID)
assert.NoError(t, err)
assert.NoError(t, store.DeleteRepo(&repo))
_, err = store.GetRepo(repo.ID)
assert.Error(t, err)
procCount, err := store.engine.Count(new(model.Proc))
assert.NoError(t, err)
assert.EqualValues(t, 1, procCount)
buildCount, err := store.engine.Count(new(model.Build))
assert.NoError(t, err)
assert.EqualValues(t, 1, buildCount)
2017-07-15 16:51:02 +00:00
}
func TestRepoRedirection(t *testing.T) {
store, closer := newTestStore(t,
new(model.Repo),
new(model.Redirection))
defer closer()
repo := model.Repo{
UserID: 1,
RemoteID: "1",
FullName: "bradrydzewski/test",
Owner: "bradrydzewski",
Name: "test",
}
assert.NoError(t, store.CreateRepo(&repo))
repoUpdated := model.Repo{
RemoteID: "1",
FullName: "bradrydzewski/test-renamed",
Owner: "bradrydzewski",
Name: "test-renamed",
}
assert.NoError(t, store.RepoBatch([]*model.Repo{&repoUpdated}))
// test redirection from old repo name
repoFromStore, err := store.GetRepoNameFallback("1", "bradrydzewski/test")
assert.NoError(t, err)
assert.Equal(t, repoFromStore.FullName, repoUpdated.FullName)
// test getting repo without remote ID (use name fallback)
repo = model.Repo{
UserID: 1,
FullName: "bradrydzewski/test-no-remote-id",
Owner: "bradrydzewski",
Name: "test-no-remote-id",
}
assert.NoError(t, store.CreateRepo(&repo))
repoFromStore, err = store.GetRepoNameFallback("", "bradrydzewski/test-no-remote-id")
assert.NoError(t, err)
assert.Equal(t, repoFromStore.FullName, repo.FullName)
}