unit test

thanos-io · Oct 28, 2024 · b444a20 · b444a20
1 parent de41e1e
commit b444a20
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)
+}