不启动项目的情况下编写一下代码的单元测试 void restoreAlarmsSetString idSet throws IOException SearchSourceBuilder ssb = new SearchSourceBuilderquerynew IdsQueryBuilderaddIdsidSettoArraynew Stringsize10000;

@Test public void testRestoreAlarms() throws IOException { // Create a mock Set of alarm IDs Set idSet = new HashSet<>(); idSet.add("123"); idSet.add("456");

// Create a mock SearchResponse
SearchHit hit1 = new SearchHit(1, "device_screened_alarm", "123", new HashMap<>());
SearchHit hit2 = new SearchHit(1, "device_screened_alarm", "456", new HashMap<>());
SearchHit[] searchHits = new SearchHit[]{hit1, hit2};
SearchHits hits = new SearchHits(searchHits, searchHits.length, 1);
SearchResponse searchResponse = new SearchResponse(hits, new Aggregations(new ArrayList<>()), 1, 1, 1, new ShardSearchFailure[]{}, new SearchProfileShardResults(new ArrayList<>()));

// Mock the ESHighLevelClient and set it to return the mock SearchResponse
ESHighLevelClient esHighLevelClient = mock(ESHighLevelClient.class);
when(esHighLevelClient.search(any(SearchRequest.class), any(RequestOptions.class))).thenReturn(searchResponse);

// Mock the BulkRequest and BulkResponse classes
BulkRequest bulkRequest = mock(BulkRequest.class);
BulkResponse bulkResponse = mock(BulkResponse.class);
when(esHighLevelClient.bulk(any(BulkRequest.class), any(RequestOptions.class))).thenReturn(bulkResponse);

// Call the method and verify that the correct requests were added to the BulkRequest
restoreAlarms(idSet);
verify(bulkRequest).add(any(DeleteRequest.class));
verify(bulkRequest).add(any(IndexRequest.class));