update package

update
This commit is contained in:
wangyan 2017-08-04 18:02:48 +08:00
parent b4a81db15c
commit bcc1a5c41d
4 changed files with 38 additions and 3 deletions

View File

@ -49,6 +49,6 @@ func SetClairVulnTimestamp(namespace string, timestamp time.Time) error {
func ListClairVulnTimestamps() ([]*models.ClairVulnTimestamp, error) {
var res []*models.ClairVulnTimestamp
o := GetOrmer()
_, err := o.QueryTable(models.ClairVulnTimestampTable).All(&res)
_, err := o.QueryTable(models.ClairVulnTimestampTable).Limit(-1).All(&res)
return res, err
}

View File

@ -51,7 +51,7 @@ func GetRepositoryByName(name string) (*models.RepoRecord, error) {
func GetAllRepositories() ([]*models.RepoRecord, error) {
o := GetOrmer()
var repos []*models.RepoRecord
_, err := o.QueryTable("repository").
_, err := o.QueryTable("repository").Limit(-1).
OrderBy("Name").All(&repos)
return repos, err
}

View File

@ -16,6 +16,7 @@ package dao
import (
"fmt"
"strconv"
"testing"
"github.com/stretchr/testify/require"
@ -220,6 +221,40 @@ func TestGetRepositoriesByProject(t *testing.T) {
t.Errorf("repository %s not found", repoName)
}
func TestGetAllRepositories(t *testing.T) {
var reposBefore []*models.RepoRecord
reposBefore, err := GetAllRepositories()
if err != nil {
t.Fatalf("error occurred while getting all registories. %v", err)
return
}
allBefore := len(reposBefore)
for i := 0; i < 1200; i++ {
t := strconv.Itoa(i)
repoRecord := models.RepoRecord{
Name: "test" + t,
ProjectID: 1,
}
if err := AddRepository(repoRecord); err != nil {
t.Fatalf("Error happens when adding the repository: %v", err)
return
}
}
var reposAfter []*models.RepoRecord
reposAfter, err2 := GetAllRepositories()
if err2 != nil {
t.Fatalf("error occurred while getting all registories. %v", err2)
return
}
allAfter := len(reposAfter)
if allAfter != allBefore+1200 {
t.Errorf("unexpected total: %d != %d", allAfter, allBefore+1200)
}
}
func addRepository(repository *models.RepoRecord) error {
return AddRepository(*repository)
}

View File

@ -102,7 +102,7 @@ func GetTotalOfUsers(query *models.UserQuery) (int64, error) {
// ListUsers lists all users according to different conditions.
func ListUsers(query *models.UserQuery) ([]models.User, error) {
users := []models.User{}
_, err := userQueryConditions(query).
_, err := userQueryConditions(query).Limit(-1).
OrderBy("username").
All(&users)
return users, err