Fix failing test (ManyShardsIT)

Author: GalLalouche

x-pack/plugin/esql/src/internalClusterTest/java/org/elasticsearch/xpack/esql/action/ManyShardsIT.java

@@ -247,15 +247,15 @@ public void testLimitConcurrentShards() {
                 for (SearchService searchService : searchServices) {
                     SearchContextCounter counter = new SearchContextCounter(pragmas.maxConcurrentShardsPerNode());
                     var mockSearchService = (MockSearchService) searchService;
-                    mockSearchService.setOnPutContext(r -> counter.onNewContext());
+                    mockSearchService.setOnCreateSearchContext(r -> counter.onNewContext());
                     mockSearchService.setOnRemoveContext(r -> counter.onContextReleased());
                 }
                 run(q, pragmas).close();
             }
         } finally {
             for (SearchService searchService : searchServices) {
                 var mockSearchService = (MockSearchService) searchService;
-                mockSearchService.setOnPutContext(r -> {});
+                mockSearchService.setOnCreateSearchContext(r -> {});
                 mockSearchService.setOnRemoveContext(r -> {});
             }
         }