test

Signed-off-by: Alan Protasio <approtas@amazon.com>
thanos-io · Jul 6, 2022 · 2d5c73a · 2d5c73a
1 parent 55df13d
commit 2d5c73a
Showing 1 changed file with 15 additions and 0 deletions.
diff --git a/pkg/objstore/objstore_test.go b/pkg/objstore/objstore_test.go
@@ -7,6 +7,7 @@ import (
 	"bytes"
 	"context"
 	"io"
+	"io/ioutil"
 	"os"
 	"testing"
 
@@ -92,6 +93,20 @@ func TestTracingReader(t *testing.T) {
 	testutil.Equals(t, int64(11), size)
 }
 
+func TestDownloadDirConcurrency(t *testing.T) {
+	m := BucketWithMetrics("", NewInMemBucket(), nil)
+	tempDir := t.TempDir()
+
+	testutil.Ok(t, m.Upload(context.Background(), "dir/obj1", bytes.NewReader([]byte("1"))))
+	testutil.Ok(t, m.Upload(context.Background(), "dir/obj2", bytes.NewReader([]byte("2"))))
+	testutil.Ok(t, m.Upload(context.Background(), "dir/obj3", bytes.NewReader([]byte("3"))))
+
+	testutil.Ok(t, DownloadDir(context.Background(), log.NewNopLogger(), m, "dir/", "dir/", tempDir, WithFetchConcurrency(10)))
+	i, err := ioutil.ReadDir(tempDir)
+	testutil.Ok(t, err)
+	testutil.Assert(t, len(i) == 3)
+}
+
 func TestTimingTracingReader(t *testing.T) {
 	m := BucketWithMetrics("", NewInMemBucket(), nil)
 	r := bytes.NewReader([]byte("hello world"))