Add per project rate limit

This commit is contained in:
simon987
2019-02-24 21:18:55 -05:00
parent 9acf6e27c1
commit 3415f95337
13 changed files with 347 additions and 267 deletions

View File

@@ -57,7 +57,7 @@ func TestCreateTaskInvalidProject(t *testing.T) {
func TestGetTaskInvalidWid(t *testing.T) {
resp := getTask(nil)
resp := getTaskFromProject(testProject, genWid())
if resp.Ok != false {
t.Error()
@@ -70,7 +70,7 @@ func TestGetTaskInvalidWid(t *testing.T) {
func TestGetTaskInvalidWorker(t *testing.T) {
resp := getTask(&storage.Worker{
resp := getTaskFromProject(testProject, &storage.Worker{
Id: -1,
})
@@ -138,12 +138,14 @@ func TestCreateTaskInvalidRecipe(t *testing.T) {
func TestCreateGetTask(t *testing.T) {
pid := createProjectAsAdmin(api.CreateProjectRequest{
Name: "My project",
Version: "1.0",
CloneUrl: "http://github.com/test/test",
GitRepo: "myrepo",
Priority: 999,
Public: true,
Name: "My project",
Version: "1.0",
CloneUrl: "http://github.com/test/test",
GitRepo: "myrepo",
Priority: 999,
Public: true,
AssignRate: 2,
SubmitRate: 2,
}).Content.Id
worker := genWid()
@@ -197,6 +199,12 @@ func TestCreateGetTask(t *testing.T) {
if taskResp.Task.Project.Public != true {
t.Error()
}
if taskResp.Task.Project.AssignRate == 1 {
t.Error()
}
if taskResp.Task.Project.SubmitRate != 2 {
t.Error()
}
}
func createTasks(prefix string) (int64, int64) {
@@ -279,36 +287,6 @@ func TestTaskProjectPriority(t *testing.T) {
}
}
func TestTaskPriority(t *testing.T) {
wid := genWid()
// Clean other tasks
for i := 0; i < 20; i++ {
getTask(wid)
}
createTasks("")
t1 := getTask(wid).Content
t2 := getTask(wid).Content
t3 := getTask(wid).Content
t4 := getTask(wid).Content
if t1.Task.Recipe != "high2" {
t.Error()
}
if t2.Task.Recipe != "high1" {
t.Error()
}
if t3.Task.Recipe != "low2" {
t.Error()
}
if t4.Task.Recipe != "low1" {
t.Error()
}
}
func TestTaskNoAccess(t *testing.T) {
worker := genWid()
@@ -400,15 +378,6 @@ func TestTaskHasAccess(t *testing.T) {
}
}
func TestNoMoreTasks(t *testing.T) {
worker := genWid()
for i := 0; i < 15; i++ {
getTask(worker)
}
}
func TestReleaseTaskSuccess(t *testing.T) {
worker := genWid()