| Index: build_scheduler/go/db/modified_tasks_test.go
|
| diff --git a/build_scheduler/go/db/modified_tasks_test.go b/build_scheduler/go/db/modified_tasks_test.go
|
| deleted file mode 100644
|
| index f0a26ca86453930702b83572081fd13bbb41c9a6..0000000000000000000000000000000000000000
|
| --- a/build_scheduler/go/db/modified_tasks_test.go
|
| +++ /dev/null
|
| @@ -1,97 +0,0 @@
|
| -package db
|
| -
|
| -import (
|
| - "testing"
|
| - "time"
|
| -
|
| - assert "github.com/stretchr/testify/require"
|
| -
|
| - "go.skia.org/infra/go/testutils"
|
| -)
|
| -
|
| -func TestModifiedTasks(t *testing.T) {
|
| - m := ModifiedTasks{}
|
| -
|
| - _, err := m.GetModifiedTasks("dummy-id")
|
| - assert.True(t, IsUnknownId(err))
|
| -
|
| - id, err := m.StartTrackingModifiedTasks()
|
| - assert.NoError(t, err)
|
| -
|
| - tasks, err := m.GetModifiedTasks(id)
|
| - assert.NoError(t, err)
|
| - assert.Equal(t, 0, len(tasks))
|
| -
|
| - t1 := makeTask(time.Unix(0, 1470674132000000), []string{"a", "b", "c", "d"})
|
| - t1.Id = "1"
|
| -
|
| - // Insert the task.
|
| - m.TrackModifiedTask(t1)
|
| -
|
| - // Ensure that the task shows up in the modified list.
|
| - tasks, err = m.GetModifiedTasks(id)
|
| - assert.NoError(t, err)
|
| - testutils.AssertDeepEqual(t, []*Task{t1}, tasks)
|
| -
|
| - // Insert two more tasks.
|
| - t2 := makeTask(time.Unix(0, 1470674376000000), []string{"e", "f"})
|
| - t2.Id = "2"
|
| - m.TrackModifiedTask(t2)
|
| - t3 := makeTask(time.Unix(0, 1470674884000000), []string{"g", "h"})
|
| - t3.Id = "3"
|
| - m.TrackModifiedTask(t3)
|
| -
|
| - // Ensure that both tasks show up in the modified list.
|
| - tasks, err = m.GetModifiedTasks(id)
|
| - assert.NoError(t, err)
|
| - testutils.AssertDeepEqual(t, []*Task{t2, t3}, tasks)
|
| -
|
| - // Check StopTrackingModifiedTasks.
|
| - m.StopTrackingModifiedTasks(id)
|
| - _, err = m.GetModifiedTasks(id)
|
| - assert.True(t, IsUnknownId(err))
|
| -}
|
| -
|
| -// Test that if a Task is modified multiple times, it only appears once in the
|
| -// result of GetModifiedTasks.
|
| -func TestMultipleModifications(t *testing.T) {
|
| - m := ModifiedTasks{}
|
| -
|
| - id, err := m.StartTrackingModifiedTasks()
|
| - assert.NoError(t, err)
|
| -
|
| - t1 := makeTask(time.Unix(0, 1470674132000000), []string{"a", "b", "c", "d"})
|
| - t1.Id = "1"
|
| -
|
| - // Insert the task.
|
| - m.TrackModifiedTask(t1)
|
| -
|
| - // Make several more modifications.
|
| - t1.Status = TASK_STATUS_RUNNING
|
| - m.TrackModifiedTask(t1)
|
| - t1.Status = TASK_STATUS_SUCCESS
|
| - m.TrackModifiedTask(t1)
|
| -
|
| - // Ensure that the task shows up only once in the modified list.
|
| - tasks, err := m.GetModifiedTasks(id)
|
| - assert.NoError(t, err)
|
| - testutils.AssertDeepEqual(t, []*Task{t1}, tasks)
|
| -}
|
| -
|
| -func TestModifiedTasksTooManyUsers(t *testing.T) {
|
| - m := ModifiedTasks{}
|
| -
|
| - var oneId string
|
| - // Max out the number of modified-tasks users; ensure that we error out.
|
| - for i := 0; i < MAX_MODIFIED_TASKS_USERS; i++ {
|
| - id, err := m.StartTrackingModifiedTasks()
|
| - assert.NoError(t, err)
|
| - oneId = id
|
| - }
|
| - _, err := m.StartTrackingModifiedTasks()
|
| - assert.True(t, IsTooManyUsers(err))
|
| -
|
| - m.StopTrackingModifiedTasks(oneId)
|
| - _, err = m.StartTrackingModifiedTasks()
|
| - assert.NoError(t, err)
|
| -}
|
|
|