mirror of
https://github.com/woodpecker-ci/woodpecker.git
synced 2024-11-30 08:06:52 +02:00
177 lines
4.1 KiB
Go
177 lines
4.1 KiB
Go
package datastore
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/drone/drone/model"
|
|
"github.com/franela/goblin"
|
|
)
|
|
|
|
func TestRepos(t *testing.T) {
|
|
db := openTest()
|
|
defer db.Close()
|
|
|
|
s := From(db)
|
|
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() {
|
|
db.Exec("DELETE FROM builds")
|
|
db.Exec("DELETE FROM repos")
|
|
db.Exec("DELETE FROM users")
|
|
})
|
|
|
|
g.It("Should Set a Repo", func() {
|
|
repo := model.Repo{
|
|
UserID: 1,
|
|
FullName: "bradrydzewski/drone",
|
|
Owner: "bradrydzewski",
|
|
Name: "drone",
|
|
}
|
|
err1 := s.CreateRepo(&repo)
|
|
err2 := s.UpdateRepo(&repo)
|
|
getrepo, err3 := s.GetRepo(repo.ID)
|
|
|
|
g.Assert(err1 == nil).IsTrue()
|
|
g.Assert(err2 == nil).IsTrue()
|
|
g.Assert(err3 == nil).IsTrue()
|
|
g.Assert(repo.ID).Equal(getrepo.ID)
|
|
})
|
|
|
|
g.It("Should Add a Repo", func() {
|
|
repo := model.Repo{
|
|
UserID: 1,
|
|
FullName: "bradrydzewski/drone",
|
|
Owner: "bradrydzewski",
|
|
Name: "drone",
|
|
}
|
|
err := s.CreateRepo(&repo)
|
|
g.Assert(err == nil).IsTrue()
|
|
g.Assert(repo.ID != 0).IsTrue()
|
|
})
|
|
|
|
g.It("Should Get a Repo by ID", func() {
|
|
repo := model.Repo{
|
|
UserID: 1,
|
|
FullName: "bradrydzewski/drone",
|
|
Owner: "bradrydzewski",
|
|
Name: "drone",
|
|
}
|
|
s.CreateRepo(&repo)
|
|
getrepo, err := s.GetRepo(repo.ID)
|
|
g.Assert(err == nil).IsTrue()
|
|
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{
|
|
UserID: 1,
|
|
FullName: "bradrydzewski/drone",
|
|
Owner: "bradrydzewski",
|
|
Name: "drone",
|
|
}
|
|
s.CreateRepo(&repo)
|
|
getrepo, err := s.GetRepoName(repo.FullName)
|
|
g.Assert(err == nil).IsTrue()
|
|
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 List", func() {
|
|
repo1 := &model.Repo{
|
|
UserID: 1,
|
|
Owner: "bradrydzewski",
|
|
Name: "drone",
|
|
FullName: "bradrydzewski/drone",
|
|
}
|
|
repo2 := &model.Repo{
|
|
UserID: 2,
|
|
Owner: "drone",
|
|
Name: "drone",
|
|
FullName: "drone/drone",
|
|
}
|
|
repo3 := &model.Repo{
|
|
UserID: 2,
|
|
Owner: "octocat",
|
|
Name: "hello-world",
|
|
FullName: "octocat/hello-world",
|
|
}
|
|
s.CreateRepo(repo1)
|
|
s.CreateRepo(repo2)
|
|
s.CreateRepo(repo3)
|
|
|
|
repos, err := s.GetRepoListOf([]*model.RepoLite{
|
|
{FullName: "bradrydzewski/drone"},
|
|
{FullName: "drone/drone"},
|
|
})
|
|
g.Assert(err == nil).IsTrue()
|
|
g.Assert(len(repos)).Equal(2)
|
|
g.Assert(repos[0].ID).Equal(repo1.ID)
|
|
g.Assert(repos[1].ID).Equal(repo2.ID)
|
|
})
|
|
|
|
g.It("Should Get a Repo List", func() {
|
|
repo1 := &model.Repo{
|
|
UserID: 1,
|
|
Owner: "bradrydzewski",
|
|
Name: "drone",
|
|
FullName: "bradrydzewski/drone",
|
|
}
|
|
repo2 := &model.Repo{
|
|
UserID: 2,
|
|
Owner: "drone",
|
|
Name: "drone",
|
|
FullName: "drone/drone",
|
|
}
|
|
s.CreateRepo(repo1)
|
|
s.CreateRepo(repo2)
|
|
|
|
count, err := s.GetRepoCount()
|
|
g.Assert(err == nil).IsTrue()
|
|
g.Assert(count).Equal(2)
|
|
})
|
|
|
|
g.It("Should Delete a Repo", func() {
|
|
repo := model.Repo{
|
|
UserID: 1,
|
|
FullName: "bradrydzewski/drone",
|
|
Owner: "bradrydzewski",
|
|
Name: "drone",
|
|
}
|
|
s.CreateRepo(&repo)
|
|
_, err1 := s.GetRepo(repo.ID)
|
|
err2 := s.DeleteRepo(&repo)
|
|
_, err3 := s.GetRepo(repo.ID)
|
|
g.Assert(err1 == nil).IsTrue()
|
|
g.Assert(err2 == nil).IsTrue()
|
|
g.Assert(err3 == nil).IsFalse()
|
|
})
|
|
|
|
g.It("Should Enforce Unique Repo Name", func() {
|
|
repo1 := model.Repo{
|
|
UserID: 1,
|
|
FullName: "bradrydzewski/drone",
|
|
Owner: "bradrydzewski",
|
|
Name: "drone",
|
|
}
|
|
repo2 := model.Repo{
|
|
UserID: 2,
|
|
FullName: "bradrydzewski/drone",
|
|
Owner: "bradrydzewski",
|
|
Name: "drone",
|
|
}
|
|
err1 := s.CreateRepo(&repo1)
|
|
err2 := s.CreateRepo(&repo2)
|
|
g.Assert(err1 == nil).IsTrue()
|
|
g.Assert(err2 == nil).IsFalse()
|
|
})
|
|
})
|
|
}
|