diff options
Diffstat (limited to 'internal/store/buckets_web_test.go')
| -rw-r--r-- | internal/store/buckets_web_test.go | 62 |
1 files changed, 62 insertions, 0 deletions
diff --git a/internal/store/buckets_web_test.go b/internal/store/buckets_web_test.go new file mode 100644 index 0000000..564b09a --- /dev/null +++ b/internal/store/buckets_web_test.go @@ -0,0 +1,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) + } +} |
