summaryrefslogtreecommitdiff
path: root/internal/store/buckets_web_test.go
blob: 564b09af54c9f98347b2e354ded1e95c9a8482d1 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
package store

import (
	"testing"
	"time"
)

func TestGetBucketItems_ReturnsActiveFirst(t *testing.T) {
	s := newNativeTasksTestStore(t)
	bucket, err := s.CreateBucket("Gutters", 30, 5)
	if err != nil {
		t.Fatal(err)
	}
	createDormantTask(t, s, "dormant-1", bucket.ID, 1, nil)
	createDormantTask(t, s, "dormant-2", bucket.ID, 1, nil)
	if _, err := s.selectBucketCycle(bucket.ID, time.Now()); err != nil {
		t.Fatal(err)
	}

	items, err := s.GetBucketItems(bucket.ID)
	if err != nil {
		t.Fatalf("GetBucketItems: %v", err)
	}
	if len(items) != 2 {
		t.Fatalf("len(items) = %d, want 2", len(items))
	}
	if items[0].BucketState != "active" {
		t.Errorf("items[0].BucketState = %q, want active (active items sort first)", items[0].BucketState)
	}
}

func TestDeleteBucket_UnbucketsItemsAndRemovesBucket(t *testing.T) {
	s := newNativeTasksTestStore(t)
	bucket, err := s.CreateBucket("Gutters", 30, 1)
	if err != nil {
		t.Fatal(err)
	}
	createDormantTask(t, s, "item-1", bucket.ID, 1, nil)

	if err := s.DeleteBucket(bucket.ID); err != nil {
		t.Fatalf("DeleteBucket: %v", err)
	}

	if _, err := s.GetBucketByID(bucket.ID); err != ErrNativeTaskNotFound {
		t.Errorf("bucket should be gone, err = %v", err)
	}
	task, err := s.GetNativeTaskByID("item-1")
	if err != nil {
		t.Fatal(err)
	}
	if task.BucketID != "" || task.BucketState != "" {
		t.Errorf("task = %+v, want unbucketed (task itself kept, not deleted)", task)
	}
}

func TestDeleteBucket_UnknownID_ReturnsErrNotFound(t *testing.T) {
	s := newNativeTasksTestStore(t)

	if err := s.DeleteBucket("does-not-exist"); err != ErrNativeTaskNotFound {
		t.Errorf("err = %v, want ErrNativeTaskNotFound", err)
	}
}