fix the tests

davidkyle · Sep 18, 2024 · 764d2ff · 764d2ff
1 parent 88eb0f0
commit 764d2ff
Show file tree

Hide file tree

Showing 2 changed files with 25 additions and 1 deletion.
diff --git a/docs/reference/rest-api/usage.asciidoc b/docs/reference/rest-api/usage.asciidoc
@@ -195,7 +195,13 @@ GET /_xpack/usage
         }
       }
     },
-    "node_count" : 1
+    "node_count" : 1,
+    "memory": {
+      anomaly_detectors_memory_bytes: 0,
+      data_frame_analytics_memory_bytes: 0,
+      pytorch_inference_memory_bytes: 0,
+      total_used_memory_bytes: 0
+    }
   },
   "inference": {
     "available" : true,

diff --git a/...in/core/src/main/java/org/elasticsearch/xpack/core/ml/MachineLearningFeatureSetUsage.java b/...in/core/src/main/java/org/elasticsearch/xpack/core/ml/MachineLearningFeatureSetUsage.java
@@ -127,4 +127,22 @@ public Map<String, Object> getMemoryUsage() {
     public int getNodeCount() {
         return nodeCount;
     }
+
+    @Override
+    public boolean equals(Object o) {
+        if (this == o) return true;
+        if (o == null || getClass() != o.getClass()) return false;
+        MachineLearningFeatureSetUsage that = (MachineLearningFeatureSetUsage) o;
+        return nodeCount == that.nodeCount
+            && Objects.equals(jobsUsage, that.jobsUsage)
+            && Objects.equals(datafeedsUsage, that.datafeedsUsage)
+            && Objects.equals(analyticsUsage, that.analyticsUsage)
+            && Objects.equals(inferenceUsage, that.inferenceUsage)
+            && Objects.equals(memoryUsage, that.memoryUsage);
+    }
+
+    @Override
+    public int hashCode() {
+        return Objects.hash(jobsUsage, datafeedsUsage, analyticsUsage, inferenceUsage, memoryUsage, nodeCount);
+    }
 }