fix final feature disablement test. whoops

Signed-off-by: HenryL27 <[email protected]>
HenryL27 · Feb 13, 2024 · b3936d9 · b3936d9
1 parent 66ae5ac
commit b3936d9
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/...a/org/opensearch/ml/memory/action/conversation/CreateInteractionTransportActionTests.java b/...a/org/opensearch/ml/memory/action/conversation/CreateInteractionTransportActionTests.java
@@ -227,8 +227,10 @@ public void testDoExecuteFails_thenFail() {
  }
 
  public void testFeatureDisabled_ThenFail() {
- when(this.clusterService.getSettings()).thenReturn(Settings.EMPTY);
- when(this.clusterService.getClusterSettings()).thenReturn(new ClusterSettings(Settings.EMPTY, Set.of(ConversationalIndexConstants.ML_COMMONS_MEMORY_FEATURE_ENABLED)));
+ Settings settings = Settings.builder().put(ConversationalIndexConstants.ML_COMMONS_MEMORY_FEATURE_ENABLED.getKey(), false).build();
+ when(this.clusterService.getSettings()).thenReturn(settings);
+ when(this.clusterService.getClusterSettings())
+ .thenReturn(new ClusterSettings(settings, Set.of(ConversationalIndexConstants.ML_COMMONS_MEMORY_FEATURE_ENABLED)));
  this.action = spy(new CreateInteractionTransportAction(transportService, actionFilters, cmHandler, client, clusterService));
 
  action.doExecute(null, request, actionListener);