unit test

Signed-off-by: thorfour <[email protected]>
thanos-io · Oct 29, 2024 · fe7aa2d · fe7aa2d
1 parent 88a1fdc
commit fe7aa2d
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/objstore_test.go b/objstore_test.go
@@ -595,3 +595,11 @@ func (b *mockBucket) GetRange(ctx context.Context, name string, off, length int6
 	}
 	return nil, errors.New("GetRange has not been mocked")
 }
+
+func Test_TryToGetSizeLimitedReader(t *testing.T) {
+	b := &bytes.Buffer{}
+	r := io.LimitReader(b, 1024)
+	size, err := TryToGetSize(r)
+	testutil.Ok(t, err)
+	testutil.Equals(t, int64(1024), size)
+}