ING-1174: Allow time for index to build in test case

Author: Westwooo

gateway/dataimpl/server_v1/errorhandler.go

@@ -488,6 +488,17 @@ func (e ErrorHandler) NewOnlyBucketOrScopeSetStatus(baseErr error, indexName str
 	return st
 }
 
+func (e ErrorHandler) NewSearchIndexNotReadyStatus(baseErr error, indexName string) *status.Status {
+	st := status.New(codes.Unavailable, "Search index is still being built, try again later.")
+	st = e.tryAttachStatusDetails(st, &epb.ResourceInfo{
+		ResourceType: "searchindex",
+		ResourceName: indexName,
+		Description:  "",
+	})
+	st = e.tryAttachExtraContext(st, baseErr)
+	return st
+}
+
 func (e ErrorHandler) NewIncorrectSearchSourceTypeStatus(baseErr error, indexName string, sourceType *string) *status.Status {
 	var sT string
 	if sourceType != nil {

gateway/dataimpl/server_v1/searchadminserver.go

@@ -391,6 +391,8 @@ func (s *SearchIndexAdminServer) AnalyzeDocument(ctx context.Context, in *admin_
 			return nil, s.errorHandler.NewSearchIndexNameEmptyStatus(err).Err()
 		} else if errors.Is(err, cbsearchx.ErrOnlyBucketOrScopeSet) {
 			return nil, s.errorHandler.NewOnlyBucketOrScopeSetStatus(err, in.Name).Err()
+		} else if errors.Is(err, cbsearchx.ErrNoIndexPartitionsFound) {
+			return nil, s.errorHandler.NewSearchIndexNotReadyStatus(err, in.Name).Err()
 		}
 		return nil, s.errorHandler.NewGenericStatus(err).Err()
 	}

gateway/test/search_mgmt_test.go

@@ -1078,14 +1078,13 @@ func (s *GatewayOpsTestSuite) TestAnalyzeDocument() {
 	}
 
 	analyzeTests := []analyzeTest{
-		// TODO - ING-1174
-		// {
-		// 	description: "Basic",
-		// 	modifyDefault: func(def *admin_search_v1.AnalyzeDocumentRequest) *admin_search_v1.AnalyzeDocumentRequest {
-		// 		return def
-		// 	},
-		// 	expect: codes.OK,
-		// },
+		{
+			description: "Basic",
+			modifyDefault: func(def *admin_search_v1.AnalyzeDocumentRequest) *admin_search_v1.AnalyzeDocumentRequest {
+				return def
+			},
+			expect: codes.OK,
+		},
 		{
 			description: "IndexNotFound",
 			modifyDefault: func(def *admin_search_v1.AnalyzeDocumentRequest) *admin_search_v1.AnalyzeDocumentRequest {
@@ -1108,6 +1107,8 @@ func (s *GatewayOpsTestSuite) TestAnalyzeDocument() {
 
 	for i := range analyzeTests {
 		t := analyzeTests[i]
+		ctx, cancel := context.WithTimeout(context.Background(), 90*time.Second)
+		defer cancel()
 		s.Run(t.description, func() {
 			defaultAnalyzeRequest := admin_search_v1.AnalyzeDocumentRequest{
 				Name:       indexName,
@@ -1121,13 +1122,23 @@ func (s *GatewayOpsTestSuite) TestAnalyzeDocument() {
 				creds = *t.creds
 			}
 
-			resp, err := searchAdminClient.AnalyzeDocument(ctx, req, grpc.PerRPCCredentials(creds))
 			if t.expect == codes.OK {
-				require.NoError(s.T(), err)
-				requireRpcSuccess(s.T(), resp, err)
+				require.Eventually(s.T(), func() bool {
+					_, err := searchAdminClient.AnalyzeDocument(ctx, req, grpc.PerRPCCredentials(creds))
+
+					if err != nil {
+						assertRpcStatus(s.T(), err, codes.Unavailable)
+						assertRpcErrorDetails(s.T(), err, func(d *epb.ResourceInfo) {
+							assert.Equal(s.T(), "searchindex", d.ResourceType)
+						})
+					}
+
+					return err == nil
+				}, time.Second*90, time.Second*5)
 				return
 			}
 
+			_, err := searchAdminClient.AnalyzeDocument(ctx, req, grpc.PerRPCCredentials(creds))
 			assertRpcStatus(s.T(), err, t.expect)
 			assertRpcErrorDetails(s.T(), err, func(d *epb.ResourceInfo) {
 				assert.Equal(s.T(), t.resourceDetails, d.ResourceType)