test non-exist in middle of list632-blob-service-get-blobs-by-shas

1 files changed, 29 insertions, 10 deletions

diff --git a/internal/service/blob/get_blobs_test.go b/internal/service/blob/get_blobs_test.go
index e5c4e93d0..f35ed0a16 100644
--- a/internal/service/blob/get_blobs_test.go
+++ b/internal/service/blob/get_blobs_test.go
@@ -121,21 +121,40 @@ func TestGetBlobsNotFound(t *testing.T) {
 	client, conn := newBlobClient(t, serverSocketPath)
 	defer conn.Close()
 
-	request := &pb.GetBlobsRequest{
-		Repository: testRepo,
-		Oids:       []string{"doesnotexist", "95d9f0a5e7bb054e9dd3975589b8dfc689e20e88"}, // Second exist
+	tests := []struct {
+		desc string
+		req  pb.GetBlobsRequest
+	}{
+		{
+			desc: "first of two is non-exist",
+			req: pb.GetBlobsRequest{
+				Repository: testRepo,
+				Oids:       []string{"doesnotexist", "95d9f0a5e7bb054e9dd3975589b8dfc689e20e88"}, // Second exist
+			},
+		},
+		{
+			desc: "second of three is non-exist",
+			req: pb.GetBlobsRequest{
+				Repository: testRepo,
+				Oids:       []string{"95d9f0a5e7bb054e9dd3975589b8dfc689e20e88", "doesnotexist", "95d9f0a5e7bb054e9dd3975589b8dfc689e20e88"}, // Second exist
+			},
+		},
 	}
 
-	ctx, cancel := testhelper.Context()
-	defer cancel()
+	for _, tc := range tests {
+		t.Run(tc.desc, func(t *testing.T) {
+			ctx, cancel := testhelper.Context()
+			defer cancel()
 
-	stream, err := client.GetBlobs(ctx, request)
-	require.NoError(t, err)
+			stream, err := client.GetBlobs(ctx, &tc.req)
+			require.NoError(t, err)
 
-	blobs, err := getAllBlobs(stream)
-	require.NoError(t, err)
+			blobs, err := getAllBlobs(stream)
+			require.NoError(t, err)
 
-	require.Nil(t, blobs)
+			require.Nil(t, blobs)
+		})
+	}
 }
 
 func TestFailedGetBlobsRequestDueToValidationError(t *testing.T) {