storagetesting: verify progress callbacks

kopia · Oct 27, 2018 · 4237bbc · 4237bbc
1 parent 0eded38
commit 4237bbc
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 1 deletion.
diff --git a/internal/storagetesting/verify.go b/internal/storagetesting/verify.go
@@ -27,9 +27,13 @@ func VerifyStorage(ctx context.Context, t *testing.T, r storage.Storage) {
 		AssertGetBlockNotFound(ctx, t, r, b.blk)
 	}
 
+	ctx2 := storage.WithUploadProgressCallback(ctx, func(desc string, completed, total int64) {
+		log.Infof("progress %v: %v/%v", desc, completed, total)
+	})
+
 	// Now add blocks.
 	for _, b := range blocks {
-		if err := r.PutBlock(ctx, b.blk, b.contents); err != nil {
+		if err := r.PutBlock(ctx2, b.blk, b.contents); err != nil {
 			t.Errorf("can't put block: %v", err)
 		}
 

diff --git a/storage/gcs/gcs_storage_test.go b/storage/gcs/gcs_storage_test.go
@@ -34,4 +34,6 @@ func TestGCSStorage(t *testing.T) {
 	}
 
 	storagetesting.VerifyStorage(ctx, t, st)
+	storagetesting.AssertConnectionInfoRoundTrips(ctx, t, st)
+
 }