update test

Signed-off-by: Ben Ye <yb532204897@gmail.com>
thanos-io · Oct 31, 2020 · 5e3289c · 5e3289c
1 parent 7be1882
commit 5e3289c
Showing 1 changed file with 32 additions and 1 deletion.
diff --git a/pkg/replicate/scheme_test.go b/pkg/replicate/scheme_test.go
@@ -62,9 +62,11 @@ func testMeta(ulid ulid.ULID) *metadata.Meta {
 }
 
 func TestReplicationSchemeAll(t *testing.T) {
+	testBlockID := testULID(0)
 	var cases = []struct {
 		name     string
 		selector labels.Selector
+		blockIDs []string
 		prepare  func(ctx context.Context, t *testing.T, originBucket, targetBucket *objstore.InMemBucket)
 		assert   func(ctx context.Context, t *testing.T, originBucket, targetBucket *objstore.InMemBucket)
 	}{
@@ -291,6 +293,35 @@ func TestReplicationSchemeAll(t *testing.T) {
 				testutil.Equals(t, expected, got)
 			},
 		},
+		{
+			name:     "BlockIDs",
+			blockIDs: []string{testBlockID.String()},
+			prepare: func(ctx context.Context, t *testing.T, originBucket, targetBucket *objstore.InMemBucket) {
+				meta := testMeta(testBlockID)
+
+				b, err := json.Marshal(meta)
+				testutil.Ok(t, err)
+				_ = originBucket.Upload(ctx, path.Join(testBlockID.String(), "meta.json"), bytes.NewReader(b))
+				_ = originBucket.Upload(ctx, path.Join(testBlockID.String(), "chunks", "000001"), bytes.NewReader(nil))
+				_ = originBucket.Upload(ctx, path.Join(testBlockID.String(), "index"), bytes.NewReader(nil))
+
+				ulid := testULID(1)
+				meta = testMeta(ulid)
+
+				b, err = json.Marshal(meta)
+				testutil.Ok(t, err)
+				_ = originBucket.Upload(ctx, path.Join(ulid.String(), "meta.json"), bytes.NewReader(b))
+				_ = originBucket.Upload(ctx, path.Join(ulid.String(), "chunks", "000001"), bytes.NewReader(nil))
+				_ = originBucket.Upload(ctx, path.Join(ulid.String(), "index"), bytes.NewReader(nil))
+			},
+			assert: func(ctx context.Context, t *testing.T, originBucket, targetBucket *objstore.InMemBucket) {
+				expected := 3
+				got := len(targetBucket.Objects())
+				if got != expected {
+					t.Fatalf("TargetBucket should have one block made up of three objects replicated. Got %d but expected %d objects.", got, expected)
+				}
+			},
+		},
 	}
 
 	for _, c := range cases {
@@ -311,7 +342,7 @@ func TestReplicationSchemeAll(t *testing.T) {
 			selector = c.selector
 		}
 
-		filter := NewBlockFilter(logger, selector, []compact.ResolutionLevel{compact.ResolutionLevelRaw}, []int{1}).Filter
+		filter := NewBlockFilter(logger, selector, []compact.ResolutionLevel{compact.ResolutionLevelRaw}, []int{1}, c.blockIDs).Filter
 		fetcher, err := block.NewMetaFetcher(logger, 32, objstore.WithNoopInstr(originBucket), "", nil, nil, nil)
 		testutil.Ok(t, err)