unit test

thorfour · thorfour · commit fe7aa2db1fb6 · 2024-10-29T10:44:35.000-05:00
Signed-off-by: thorfour &lt;me@thor-hansen.com&gt;
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)
+}

Original file line number	Diff line number	Diff line change
`@@ -595,3 +595,11 @@ func (b *mockBucket) GetRange(ctx context.Context, name string, off, length int6`
`595`	`595`	`}`
`596`	`596`	`return nil, errors.New("GetRange has not been mocked")`
`597`	`597`	`}`
	`598`	`+`
	`599`	`+func Test_TryToGetSizeLimitedReader(t *testing.T) {`
	`600`	`+ b := &bytes.Buffer{}`
	`601`	`+ r := io.LimitReader(b, 1024)`
	`602`	`+ size, err := TryToGetSize(r)`
	`603`	`+ testutil.Ok(t, err)`
	`604`	`+ testutil.Equals(t, int64(1024), size)`
	`605`	`+}`