test: add test to service data handler

internal/api/v1beta1/servicedata_test.go

@@ -140,6 +140,31 @@ func TestHandler_CreateKey(t *testing.T) {
 			want:    nil,
 			wantErr: grpcBadBodyError,
 		},
+		{
+			name: "should return created key if no error",
+			setup: func(ctx context.Context, ss *mocks.ServiceDataService) context.Context {
+				ss.EXPECT().CreateKey(mock.AnythingOfType("*context.valueCtx"), servicedata.Key{
+					ProjectID:   testKey.ProjectID,
+					Key:         testKey.Key,
+					Description: testKey.Description,
+				}).Return(servicedata.Key{
+					ID:  testKey.ID,
+					URN: testKey.URN,
+				}, nil)
+				return user.SetContextWithEmail(ctx, email)
+			},
+			request: &shieldv1beta1.CreateServiceDataKeyRequest{
+				Body: &shieldv1beta1.ServiceDataKeyRequestBody{
+					Project:     testKey.ProjectID,
+					Key:         testKey.Key,
+					Description: testKey.Description,
+				},
+			},
+			want: &shieldv1beta1.CreateServiceDataKeyResponse{
+				ServiceDataKey: testKeyPB,
+			},
+			wantErr: nil,
+		},
 	}
 	for _, tt := range tests {
 		t.Run(tt.name, func(t *testing.T) {