Move CostReportConsumerMock to controller-api

1 files changed, 36 insertions, 0 deletions

diff --git a/controller-api/src/main/java/com/yahoo/vespa/hosted/controller/api/integration/resource/CostReportConsumerMock.java b/controller-api/src/main/java/com/yahoo/vespa/hosted/controller/api/integration/resource/CostReportConsumerMock.java
new file mode 100644
index 00000000000..c59e13246cd
--- /dev/null
+++ b/controller-api/src/main/java/com/yahoo/vespa/hosted/controller/api/integration/resource/CostReportConsumerMock.java
@@ -0,0 +1,36 @@
+package com.yahoo.vespa.hosted.controller.api.integration.resource;
+
+import com.yahoo.vespa.hosted.controller.api.identifiers.Property;
+
+import java.util.Map;
+import java.util.function.Consumer;
+
+/**
+ * @author ldalves
+ */
+public class CostReportConsumerMock implements CostReportConsumer {
+
+    private final Consumer<String> csvConsumer;
+    private final Map<Property, ResourceAllocation> fixedAllocations;
+
+    public CostReportConsumerMock() {
+        this.csvConsumer = (ignored) -> {};
+        this.fixedAllocations = Map.of();
+    }
+
+    public CostReportConsumerMock(Consumer<String> csvConsumer, Map<Property, ResourceAllocation> fixedAllocations) {
+        this.csvConsumer = csvConsumer;
+        this.fixedAllocations = Map.copyOf(fixedAllocations);
+    }
+
+    @Override
+    public void consume(String csv) {
+        csvConsumer.accept(csv);
+    }
+
+    @Override
+    public Map<Property, ResourceAllocation> fixedAllocations() {
+        return fixedAllocations;
+    }
+
+}