Fix unit tests

droberts195 · Oct 2, 2023 · 7c564c7 · 7c564c7
1 parent 8244779
commit 7c564c7
Showing 1 changed file with 16 additions and 2 deletions.
diff --git a/...test/java/org/elasticsearch/xpack/ml/inference/ingest/InferenceProcessorFactoryTests.java b/...test/java/org/elasticsearch/xpack/ml/inference/ingest/InferenceProcessorFactoryTests.java
@@ -475,8 +475,22 @@ private static ClusterState builderClusterStateWithModelReferences(MlConfigVersi
  Set.of(DiscoveryNodeRole.MASTER_ROLE, DiscoveryNodeRole.ML_ROLE, DiscoveryNodeRole.DATA_ROLE)
  )
  )
- .add(DiscoveryNodeUtils.create("current_node", new TransportAddress(InetAddress.getLoopbackAddress(), 9302)))
- .add(DiscoveryNodeUtils.create("_node_id", new TransportAddress(InetAddress.getLoopbackAddress(), 9304)))
+ .add(
+ DiscoveryNodeUtils.create(
+ "current_node",
+ new TransportAddress(InetAddress.getLoopbackAddress(), 9302),
+ Map.of(MachineLearning.ML_CONFIG_VERSION_NODE_ATTR, MlConfigVersion.CURRENT.toString()),
+ Set.of(DiscoveryNodeRole.MASTER_ROLE, DiscoveryNodeRole.ML_ROLE, DiscoveryNodeRole.DATA_ROLE)
+ )
+ )
+ .add(
+ DiscoveryNodeUtils.create(
+ "_node_id",
+ new TransportAddress(InetAddress.getLoopbackAddress(), 9304),
+ Map.of(MachineLearning.ML_CONFIG_VERSION_NODE_ATTR, MlConfigVersion.CURRENT.toString()),
+ Set.of(DiscoveryNodeRole.MASTER_ROLE, DiscoveryNodeRole.ML_ROLE, DiscoveryNodeRole.DATA_ROLE)
+ )
+ )
  .localNodeId("_node_id")
  .masterNodeId("_node_id")
  )