|
@@ -0,0 +1,464 @@
|
|
|
+/*
|
|
|
+ *
|
|
|
+ * Licensed to the Apache Software Foundation (ASF) under one
|
|
|
+ * or more contributor license agreements. See the NOTICE file
|
|
|
+ * distributed with this work for additional information
|
|
|
+ * regarding copyright ownership. The ASF licenses this file
|
|
|
+ * to you under the Apache License, Version 2.0 (the
|
|
|
+ * "License"); you may not use this file except in compliance
|
|
|
+ * with the License. You may obtain a copy of the License at
|
|
|
+ *
|
|
|
+ * http://www.apache.org/licenses/LICENSE-2.0
|
|
|
+ *
|
|
|
+ * Unless required by applicable law or agreed to in writing, software
|
|
|
+ * distributed under the License is distributed on an "AS IS" BASIS,
|
|
|
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
|
+ * See the License for the specific language governing permissions and
|
|
|
+ * limitations under the License.
|
|
|
+ *
|
|
|
+ */
|
|
|
+
|
|
|
+package org.apache.hadoop.resourceestimator.skylinestore.impl;
|
|
|
+
|
|
|
+import java.util.ArrayList;
|
|
|
+import java.util.List;
|
|
|
+import java.util.Map;
|
|
|
+import java.util.TreeMap;
|
|
|
+
|
|
|
+import org.apache.hadoop.resourceestimator.common.api.RecurrenceId;
|
|
|
+import org.apache.hadoop.resourceestimator.common.api.ResourceSkyline;
|
|
|
+import org.apache.hadoop.resourceestimator.skylinestore.api.SkylineStore;
|
|
|
+import org.apache.hadoop.resourceestimator.skylinestore.exceptions.DuplicateRecurrenceIdException;
|
|
|
+import org.apache.hadoop.resourceestimator.skylinestore.exceptions.EmptyResourceSkylineException;
|
|
|
+import org.apache.hadoop.resourceestimator.skylinestore.exceptions.NullPipelineIdException;
|
|
|
+import org.apache.hadoop.resourceestimator.skylinestore.exceptions.NullRLESparseResourceAllocationException;
|
|
|
+import org.apache.hadoop.resourceestimator.skylinestore.exceptions.NullRecurrenceIdException;
|
|
|
+import org.apache.hadoop.resourceestimator.skylinestore.exceptions.NullResourceSkylineException;
|
|
|
+import org.apache.hadoop.resourceestimator.skylinestore.exceptions.RecurrenceIdNotFoundException;
|
|
|
+import org.apache.hadoop.resourceestimator.skylinestore.exceptions.SkylineStoreException;
|
|
|
+import org.apache.hadoop.yarn.api.records.Resource;
|
|
|
+import org.apache.hadoop.yarn.server.resourcemanager.reservation.RLESparseResourceAllocation;
|
|
|
+import org.apache.hadoop.yarn.server.resourcemanager.reservation.ReservationInterval;
|
|
|
+import org.apache.hadoop.yarn.util.resource.DefaultResourceCalculator;
|
|
|
+import org.junit.After;
|
|
|
+import org.junit.Assert;
|
|
|
+import org.junit.Before;
|
|
|
+import org.junit.Test;
|
|
|
+
|
|
|
+/**
|
|
|
+ * Test {@link SkylineStore} class.
|
|
|
+ */
|
|
|
+public abstract class TestSkylineStore {
|
|
|
+ /**
|
|
|
+ * Testing variables.
|
|
|
+ */
|
|
|
+ private SkylineStore skylineStore;
|
|
|
+
|
|
|
+ private TreeMap<Long, Resource> resourceOverTime;
|
|
|
+ private RLESparseResourceAllocation skylineList;
|
|
|
+ private ReservationInterval riAdd;
|
|
|
+ private Resource resource;
|
|
|
+
|
|
|
+ protected abstract SkylineStore createSkylineStore();
|
|
|
+
|
|
|
+ @Before public final void setup() {
|
|
|
+ skylineStore = createSkylineStore();
|
|
|
+ resourceOverTime = new TreeMap<>();
|
|
|
+ resource = Resource.newInstance(1024 * 100, 100);
|
|
|
+ }
|
|
|
+
|
|
|
+ private void compare(final ResourceSkyline skyline1,
|
|
|
+ final ResourceSkyline skyline2) {
|
|
|
+ Assert.assertEquals(skyline1.getJobId(), skyline2.getJobId());
|
|
|
+ Assert.assertEquals(skyline1.getJobInputDataSize(),
|
|
|
+ skyline2.getJobInputDataSize(), 0);
|
|
|
+ Assert.assertEquals(skyline1.getJobSubmissionTime(),
|
|
|
+ skyline2.getJobSubmissionTime());
|
|
|
+ Assert
|
|
|
+ .assertEquals(skyline1.getJobFinishTime(), skyline2.getJobFinishTime());
|
|
|
+ Assert.assertEquals(skyline1.getContainerSpec().getMemorySize(),
|
|
|
+ skyline2.getContainerSpec().getMemorySize());
|
|
|
+ Assert.assertEquals(skyline1.getContainerSpec().getVirtualCores(),
|
|
|
+ skyline2.getContainerSpec().getVirtualCores());
|
|
|
+ Assert.assertEquals(true,
|
|
|
+ skyline2.getSkylineList().equals(skyline1.getSkylineList()));
|
|
|
+ }
|
|
|
+
|
|
|
+ private void addToStore(final RecurrenceId recurrenceId,
|
|
|
+ final ResourceSkyline resourceSkyline) throws SkylineStoreException {
|
|
|
+ final List<ResourceSkyline> resourceSkylines = new ArrayList<>();
|
|
|
+ resourceSkylines.add(resourceSkyline);
|
|
|
+ skylineStore.addHistory(recurrenceId, resourceSkylines);
|
|
|
+ final List<ResourceSkyline> resourceSkylinesGet =
|
|
|
+ skylineStore.getHistory(recurrenceId).get(recurrenceId);
|
|
|
+ Assert.assertTrue(resourceSkylinesGet.contains(resourceSkyline));
|
|
|
+ }
|
|
|
+
|
|
|
+ private ResourceSkyline getSkyline(final int n) {
|
|
|
+ skylineList = new RLESparseResourceAllocation(resourceOverTime,
|
|
|
+ new DefaultResourceCalculator());
|
|
|
+ for (int i = 0; i < n; i++) {
|
|
|
+ riAdd = new ReservationInterval(i * 10, (i + 1) * 10);
|
|
|
+ skylineList.addInterval(riAdd, resource);
|
|
|
+ }
|
|
|
+ final ResourceSkyline resourceSkyline =
|
|
|
+ new ResourceSkyline(Integer.toString(n), 1024.5, 0, 20, resource,
|
|
|
+ skylineList);
|
|
|
+
|
|
|
+ return resourceSkyline;
|
|
|
+ }
|
|
|
+
|
|
|
+ @Test public final void testGetHistory() throws SkylineStoreException {
|
|
|
+ // addHistory first recurring pipeline
|
|
|
+ final RecurrenceId recurrenceId1 =
|
|
|
+ new RecurrenceId("FraudDetection", "17/06/20 00:00:00");
|
|
|
+ final ResourceSkyline resourceSkyline1 = getSkyline(1);
|
|
|
+ addToStore(recurrenceId1, resourceSkyline1);
|
|
|
+ final ResourceSkyline resourceSkyline2 = getSkyline(2);
|
|
|
+ addToStore(recurrenceId1, resourceSkyline2);
|
|
|
+ final RecurrenceId recurrenceId2 =
|
|
|
+ new RecurrenceId("FraudDetection", "17/06/21 00:00:00");
|
|
|
+ final ResourceSkyline resourceSkyline3 = getSkyline(3);
|
|
|
+ addToStore(recurrenceId2, resourceSkyline3);
|
|
|
+ final ResourceSkyline resourceSkyline4 = getSkyline(4);
|
|
|
+ addToStore(recurrenceId2, resourceSkyline4);
|
|
|
+ // addHistory second recurring pipeline
|
|
|
+ final RecurrenceId recurrenceId3 =
|
|
|
+ new RecurrenceId("Random", "17/06/20 00:00:00");
|
|
|
+ addToStore(recurrenceId3, resourceSkyline1);
|
|
|
+ addToStore(recurrenceId3, resourceSkyline2);
|
|
|
+ // test getHistory {pipelineId, runId}
|
|
|
+ Map<RecurrenceId, List<ResourceSkyline>> jobHistory =
|
|
|
+ skylineStore.getHistory(recurrenceId1);
|
|
|
+ Assert.assertEquals(1, jobHistory.size());
|
|
|
+ for (final Map.Entry<RecurrenceId, List<ResourceSkyline>> entry : jobHistory
|
|
|
+ .entrySet()) {
|
|
|
+ Assert.assertEquals(recurrenceId1, entry.getKey());
|
|
|
+ final List<ResourceSkyline> getSkylines = entry.getValue();
|
|
|
+ Assert.assertEquals(2, getSkylines.size());
|
|
|
+ compare(resourceSkyline1, getSkylines.get(0));
|
|
|
+ compare(resourceSkyline2, getSkylines.get(1));
|
|
|
+ }
|
|
|
+ // test getHistory {pipelineId, *}
|
|
|
+ RecurrenceId recurrenceIdTest = new RecurrenceId("FraudDetection", "*");
|
|
|
+ jobHistory = skylineStore.getHistory(recurrenceIdTest);
|
|
|
+ Assert.assertEquals(2, jobHistory.size());
|
|
|
+ for (final Map.Entry<RecurrenceId, List<ResourceSkyline>> entry : jobHistory
|
|
|
+ .entrySet()) {
|
|
|
+ Assert.assertEquals(recurrenceId1.getPipelineId(),
|
|
|
+ entry.getKey().getPipelineId());
|
|
|
+ final List<ResourceSkyline> getSkylines = entry.getValue();
|
|
|
+ if (entry.getKey().getRunId().equals("17/06/20 00:00:00")) {
|
|
|
+ Assert.assertEquals(2, getSkylines.size());
|
|
|
+ compare(resourceSkyline1, getSkylines.get(0));
|
|
|
+ compare(resourceSkyline2, getSkylines.get(1));
|
|
|
+ } else {
|
|
|
+ Assert.assertEquals(entry.getKey().getRunId(), "17/06/21 00:00:00");
|
|
|
+ Assert.assertEquals(2, getSkylines.size());
|
|
|
+ compare(resourceSkyline3, getSkylines.get(0));
|
|
|
+ compare(resourceSkyline4, getSkylines.get(1));
|
|
|
+ }
|
|
|
+ }
|
|
|
+ // test getHistory {*, runId}
|
|
|
+ recurrenceIdTest = new RecurrenceId("*", "some random runId");
|
|
|
+ jobHistory = skylineStore.getHistory(recurrenceIdTest);
|
|
|
+ Assert.assertEquals(3, jobHistory.size());
|
|
|
+ for (final Map.Entry<RecurrenceId, List<ResourceSkyline>> entry : jobHistory
|
|
|
+ .entrySet()) {
|
|
|
+ if (entry.getKey().getPipelineId().equals("FraudDetection")) {
|
|
|
+ final List<ResourceSkyline> getSkylines = entry.getValue();
|
|
|
+ if (entry.getKey().getRunId().equals("17/06/20 00:00:00")) {
|
|
|
+ Assert.assertEquals(2, getSkylines.size());
|
|
|
+ compare(resourceSkyline1, getSkylines.get(0));
|
|
|
+ compare(resourceSkyline2, getSkylines.get(1));
|
|
|
+ } else {
|
|
|
+ Assert.assertEquals(entry.getKey().getRunId(), "17/06/21 00:00:00");
|
|
|
+ Assert.assertEquals(2, getSkylines.size());
|
|
|
+ compare(resourceSkyline3, getSkylines.get(0));
|
|
|
+ compare(resourceSkyline4, getSkylines.get(1));
|
|
|
+ }
|
|
|
+ } else {
|
|
|
+ Assert.assertEquals("Random", entry.getKey().getPipelineId());
|
|
|
+ Assert.assertEquals(entry.getKey().getRunId(), "17/06/20 00:00:00");
|
|
|
+ final List<ResourceSkyline> getSkylines = entry.getValue();
|
|
|
+ Assert.assertEquals(2, getSkylines.size());
|
|
|
+ compare(resourceSkyline1, getSkylines.get(0));
|
|
|
+ compare(resourceSkyline2, getSkylines.get(1));
|
|
|
+ }
|
|
|
+ }
|
|
|
+ // test getHistory with wrong RecurrenceId
|
|
|
+ recurrenceIdTest =
|
|
|
+ new RecurrenceId("some random pipelineId", "some random runId");
|
|
|
+ Assert.assertNull(skylineStore.getHistory(recurrenceIdTest));
|
|
|
+ }
|
|
|
+
|
|
|
+ @Test public final void testGetEstimation() throws SkylineStoreException {
|
|
|
+ // first, add estimation to the skyline store
|
|
|
+ final RLESparseResourceAllocation skylineList2 =
|
|
|
+ new RLESparseResourceAllocation(resourceOverTime,
|
|
|
+ new DefaultResourceCalculator());
|
|
|
+ for (int i = 0; i < 5; i++) {
|
|
|
+ riAdd = new ReservationInterval(i * 10, (i + 1) * 10);
|
|
|
+ skylineList2.addInterval(riAdd, resource);
|
|
|
+ }
|
|
|
+ skylineStore.addEstimation("FraudDetection", skylineList2);
|
|
|
+ // then, try to get the estimation
|
|
|
+ final RLESparseResourceAllocation estimation =
|
|
|
+ skylineStore.getEstimation("FraudDetection");
|
|
|
+ for (int i = 0; i < 50; i++) {
|
|
|
+ Assert.assertEquals(skylineList2.getCapacityAtTime(i),
|
|
|
+ estimation.getCapacityAtTime(i));
|
|
|
+ }
|
|
|
+ }
|
|
|
+
|
|
|
+ @Test(expected = NullRecurrenceIdException.class)
|
|
|
+ public final void testGetNullRecurrenceId()
|
|
|
+ throws SkylineStoreException {
|
|
|
+ // addHistory first recurring pipeline
|
|
|
+ final RecurrenceId recurrenceId1 =
|
|
|
+ new RecurrenceId("FraudDetection", "17/06/20 00:00:00");
|
|
|
+ final ResourceSkyline resourceSkyline1 = getSkyline(1);
|
|
|
+ addToStore(recurrenceId1, resourceSkyline1);
|
|
|
+ final ResourceSkyline resourceSkyline2 = getSkyline(2);
|
|
|
+ addToStore(recurrenceId1, resourceSkyline2);
|
|
|
+ final RecurrenceId recurrenceId2 =
|
|
|
+ new RecurrenceId("FraudDetection", "17/06/21 00:00:00");
|
|
|
+ final ResourceSkyline resourceSkyline3 = getSkyline(3);
|
|
|
+ addToStore(recurrenceId2, resourceSkyline3);
|
|
|
+ final ResourceSkyline resourceSkyline4 = getSkyline(4);
|
|
|
+ addToStore(recurrenceId2, resourceSkyline4);
|
|
|
+ // addHistory second recurring pipeline
|
|
|
+ final RecurrenceId recurrenceId3 =
|
|
|
+ new RecurrenceId("Random", "17/06/20 00:00:00");
|
|
|
+ addToStore(recurrenceId3, resourceSkyline1);
|
|
|
+ addToStore(recurrenceId3, resourceSkyline2);
|
|
|
+ // try to getHistory with null recurringId
|
|
|
+ skylineStore.getHistory(null);
|
|
|
+ }
|
|
|
+
|
|
|
+ @Test(expected = NullPipelineIdException.class)
|
|
|
+ public final void testGetNullPipelineIdException()
|
|
|
+ throws SkylineStoreException {
|
|
|
+ skylineStore.getEstimation(null);
|
|
|
+ }
|
|
|
+
|
|
|
+ @Test public final void testAddNormal() throws SkylineStoreException {
|
|
|
+ // addHistory resource skylines to the in-memory store
|
|
|
+ final RecurrenceId recurrenceId =
|
|
|
+ new RecurrenceId("FraudDetection", "17/06/20 00:00:00");
|
|
|
+ final ResourceSkyline resourceSkyline1 = getSkyline(1);
|
|
|
+ addToStore(recurrenceId, resourceSkyline1);
|
|
|
+ final ArrayList<ResourceSkyline> resourceSkylines =
|
|
|
+ new ArrayList<ResourceSkyline>();
|
|
|
+ // the resource skylines to be added contain null
|
|
|
+ resourceSkylines.add(null);
|
|
|
+ final ResourceSkyline resourceSkyline2 = getSkyline(2);
|
|
|
+ resourceSkylines.add(resourceSkyline2);
|
|
|
+ skylineStore.addHistory(recurrenceId, resourceSkylines);
|
|
|
+ // query the in-memory store
|
|
|
+ final Map<RecurrenceId, List<ResourceSkyline>> jobHistory =
|
|
|
+ skylineStore.getHistory(recurrenceId);
|
|
|
+ Assert.assertEquals(1, jobHistory.size());
|
|
|
+ for (final Map.Entry<RecurrenceId, List<ResourceSkyline>> entry : jobHistory
|
|
|
+ .entrySet()) {
|
|
|
+ Assert.assertEquals(recurrenceId, entry.getKey());
|
|
|
+ final List<ResourceSkyline> getSkylines = entry.getValue();
|
|
|
+ Assert.assertEquals(2, getSkylines.size());
|
|
|
+ compare(resourceSkyline1, getSkylines.get(0));
|
|
|
+ compare(resourceSkyline2, getSkylines.get(1));
|
|
|
+ }
|
|
|
+ }
|
|
|
+
|
|
|
+ @Test(expected = NullRecurrenceIdException.class)
|
|
|
+ public final void testAddNullRecurrenceId()
|
|
|
+ throws SkylineStoreException {
|
|
|
+ // recurrenceId is null
|
|
|
+ final RecurrenceId recurrenceIdNull = null;
|
|
|
+ final ArrayList<ResourceSkyline> resourceSkylines =
|
|
|
+ new ArrayList<ResourceSkyline>();
|
|
|
+ final ResourceSkyline resourceSkyline1 = getSkyline(1);
|
|
|
+ resourceSkylines.add(resourceSkyline1);
|
|
|
+ skylineStore.addHistory(recurrenceIdNull, resourceSkylines);
|
|
|
+ }
|
|
|
+
|
|
|
+ @Test(expected = NullResourceSkylineException.class)
|
|
|
+ public final void testAddNullResourceSkyline()
|
|
|
+ throws SkylineStoreException {
|
|
|
+ final RecurrenceId recurrenceId =
|
|
|
+ new RecurrenceId("FraudDetection", "17/06/20 00:00:00");
|
|
|
+ final ArrayList<ResourceSkyline> resourceSkylines =
|
|
|
+ new ArrayList<ResourceSkyline>();
|
|
|
+ final ResourceSkyline resourceSkyline1 = getSkyline(1);
|
|
|
+ resourceSkylines.add(resourceSkyline1);
|
|
|
+ // resourceSkylines is null
|
|
|
+ skylineStore.addHistory(recurrenceId, null);
|
|
|
+ }
|
|
|
+
|
|
|
+ @Test(expected = DuplicateRecurrenceIdException.class)
|
|
|
+ public final void testAddDuplicateRecurrenceId()
|
|
|
+ throws SkylineStoreException {
|
|
|
+ final RecurrenceId recurrenceId =
|
|
|
+ new RecurrenceId("FraudDetection", "17/06/20 00:00:00");
|
|
|
+ final ArrayList<ResourceSkyline> resourceSkylines =
|
|
|
+ new ArrayList<ResourceSkyline>();
|
|
|
+ final ResourceSkyline resourceSkyline1 = getSkyline(1);
|
|
|
+ resourceSkylines.add(resourceSkyline1);
|
|
|
+ // trying to addHistory duplicate resource skylines
|
|
|
+ skylineStore.addHistory(recurrenceId, resourceSkylines);
|
|
|
+ skylineStore.addHistory(recurrenceId, resourceSkylines);
|
|
|
+ }
|
|
|
+
|
|
|
+ @Test(expected = NullPipelineIdException.class)
|
|
|
+ public final void testAddNullPipelineIdException()
|
|
|
+ throws SkylineStoreException {
|
|
|
+ final RLESparseResourceAllocation skylineList2 =
|
|
|
+ new RLESparseResourceAllocation(resourceOverTime,
|
|
|
+ new DefaultResourceCalculator());
|
|
|
+ for (int i = 0; i < 5; i++) {
|
|
|
+ riAdd = new ReservationInterval(i * 10, (i + 1) * 10);
|
|
|
+ skylineList2.addInterval(riAdd, resource);
|
|
|
+ }
|
|
|
+ skylineStore.addEstimation(null, skylineList2);
|
|
|
+ }
|
|
|
+
|
|
|
+ @Test(expected = NullRLESparseResourceAllocationException.class)
|
|
|
+ public final void testAddNullRLESparseResourceAllocationExceptionException()
|
|
|
+ throws SkylineStoreException {
|
|
|
+ skylineStore.addEstimation("FraudDetection", null);
|
|
|
+ }
|
|
|
+
|
|
|
+ @Test public final void testDeleteNormal() throws SkylineStoreException {
|
|
|
+ // addHistory first recurring pipeline
|
|
|
+ final RecurrenceId recurrenceId1 =
|
|
|
+ new RecurrenceId("FraudDetection", "17/06/20 00:00:00");
|
|
|
+ final ResourceSkyline resourceSkyline1 = getSkyline(1);
|
|
|
+ addToStore(recurrenceId1, resourceSkyline1);
|
|
|
+ final ResourceSkyline resourceSkyline2 = getSkyline(2);
|
|
|
+ addToStore(recurrenceId1, resourceSkyline2);
|
|
|
+ // test deleteHistory function of the in-memory store
|
|
|
+ skylineStore.deleteHistory(recurrenceId1);
|
|
|
+ }
|
|
|
+
|
|
|
+ @Test(expected = NullRecurrenceIdException.class)
|
|
|
+ public final void testDeleteNullRecurrenceId()
|
|
|
+ throws SkylineStoreException {
|
|
|
+ final RecurrenceId recurrenceId1 =
|
|
|
+ new RecurrenceId("FraudDetection", "17/06/20 00:00:00");
|
|
|
+ final ResourceSkyline resourceSkyline1 = getSkyline(1);
|
|
|
+ addToStore(recurrenceId1, resourceSkyline1);
|
|
|
+ // try to deleteHistory with null recurringId
|
|
|
+ skylineStore.deleteHistory(null);
|
|
|
+ }
|
|
|
+
|
|
|
+ @Test(expected = RecurrenceIdNotFoundException.class)
|
|
|
+ public final void testDeleteRecurrenceIdNotFound()
|
|
|
+ throws SkylineStoreException {
|
|
|
+ final RecurrenceId recurrenceId1 =
|
|
|
+ new RecurrenceId("FraudDetection", "17/06/20 00:00:00");
|
|
|
+ final ResourceSkyline resourceSkyline1 = getSkyline(1);
|
|
|
+ addToStore(recurrenceId1, resourceSkyline1);
|
|
|
+ final RecurrenceId recurrenceIdInvalid =
|
|
|
+ new RecurrenceId("Some random pipelineId", "Some random runId");
|
|
|
+ // try to deleteHistory non-existing recurringId
|
|
|
+ skylineStore.deleteHistory(recurrenceIdInvalid);
|
|
|
+ }
|
|
|
+
|
|
|
+ @Test public final void testUpdateNormal() throws SkylineStoreException {
|
|
|
+ // addHistory first recurring pipeline
|
|
|
+ final RecurrenceId recurrenceId1 =
|
|
|
+ new RecurrenceId("FraudDetection", "17/06/20 00:00:00");
|
|
|
+ final ResourceSkyline resourceSkyline1 = getSkyline(1);
|
|
|
+ addToStore(recurrenceId1, resourceSkyline1);
|
|
|
+ final ArrayList<ResourceSkyline> resourceSkylines =
|
|
|
+ new ArrayList<ResourceSkyline>();
|
|
|
+ final ResourceSkyline resourceSkyline2 = getSkyline(2);
|
|
|
+ resourceSkylines.add(resourceSkyline1);
|
|
|
+ resourceSkylines.add(resourceSkyline2);
|
|
|
+ skylineStore.updateHistory(recurrenceId1, resourceSkylines);
|
|
|
+ // query the in-memory store
|
|
|
+ final Map<RecurrenceId, List<ResourceSkyline>> jobHistory =
|
|
|
+ skylineStore.getHistory(recurrenceId1);
|
|
|
+ Assert.assertEquals(1, jobHistory.size());
|
|
|
+ for (final Map.Entry<RecurrenceId, List<ResourceSkyline>> entry : jobHistory
|
|
|
+ .entrySet()) {
|
|
|
+ Assert.assertEquals(recurrenceId1, entry.getKey());
|
|
|
+ final List<ResourceSkyline> getSkylines = entry.getValue();
|
|
|
+ Assert.assertEquals(2, getSkylines.size());
|
|
|
+ compare(resourceSkyline1, getSkylines.get(0));
|
|
|
+ compare(resourceSkyline2, getSkylines.get(1));
|
|
|
+ }
|
|
|
+ }
|
|
|
+
|
|
|
+ @Test(expected = NullRecurrenceIdException.class)
|
|
|
+ public final void testUpdateNullRecurrenceId()
|
|
|
+ throws SkylineStoreException {
|
|
|
+ final ArrayList<ResourceSkyline> resourceSkylines =
|
|
|
+ new ArrayList<ResourceSkyline>();
|
|
|
+ final ResourceSkyline resourceSkyline1 = getSkyline(1);
|
|
|
+ resourceSkylines.add(resourceSkyline1);
|
|
|
+ final ArrayList<ResourceSkyline> resourceSkylinesInvalid =
|
|
|
+ new ArrayList<ResourceSkyline>();
|
|
|
+ resourceSkylinesInvalid.add(null);
|
|
|
+ // try to updateHistory with null recurringId
|
|
|
+ skylineStore.updateHistory(null, resourceSkylines);
|
|
|
+ }
|
|
|
+
|
|
|
+ @Test(expected = NullResourceSkylineException.class)
|
|
|
+ public final void testUpdateNullResourceSkyline()
|
|
|
+ throws SkylineStoreException {
|
|
|
+ final RecurrenceId recurrenceId =
|
|
|
+ new RecurrenceId("FraudDetection", "17/06/20 00:00:00");
|
|
|
+ final ArrayList<ResourceSkyline> resourceSkylines =
|
|
|
+ new ArrayList<ResourceSkyline>();
|
|
|
+ final ResourceSkyline resourceSkyline1 = getSkyline(1);
|
|
|
+ resourceSkylines.add(resourceSkyline1);
|
|
|
+ final ArrayList<ResourceSkyline> resourceSkylinesInvalid =
|
|
|
+ new ArrayList<ResourceSkyline>();
|
|
|
+ resourceSkylinesInvalid.add(null);
|
|
|
+ // try to updateHistory with null resourceSkylines
|
|
|
+ skylineStore.addHistory(recurrenceId, resourceSkylines);
|
|
|
+ skylineStore.updateHistory(recurrenceId, null);
|
|
|
+ }
|
|
|
+
|
|
|
+ @Test(expected = EmptyResourceSkylineException.class)
|
|
|
+ public final void testUpdateEmptyRecurrenceId()
|
|
|
+ throws SkylineStoreException {
|
|
|
+ final RecurrenceId recurrenceId =
|
|
|
+ new RecurrenceId("FraudDetection", "17/06/20 00:00:00");
|
|
|
+ final ArrayList<ResourceSkyline> resourceSkylines =
|
|
|
+ new ArrayList<ResourceSkyline>();
|
|
|
+ final ResourceSkyline resourceSkyline1 = getSkyline(1);
|
|
|
+ resourceSkylines.add(resourceSkyline1);
|
|
|
+ final ArrayList<ResourceSkyline> resourceSkylinesInvalid =
|
|
|
+ new ArrayList<ResourceSkyline>();
|
|
|
+ resourceSkylinesInvalid.add(null);
|
|
|
+ skylineStore.addHistory(recurrenceId, resourceSkylines);
|
|
|
+ // try to updateHistory with empty resourceSkyline
|
|
|
+ skylineStore.updateHistory(recurrenceId, resourceSkylinesInvalid);
|
|
|
+ }
|
|
|
+
|
|
|
+ @Test(expected = RecurrenceIdNotFoundException.class)
|
|
|
+ public final void testUpdateRecurrenceIdNotFound()
|
|
|
+ throws SkylineStoreException {
|
|
|
+ final ArrayList<ResourceSkyline> resourceSkylines =
|
|
|
+ new ArrayList<ResourceSkyline>();
|
|
|
+ final ResourceSkyline resourceSkyline1 = getSkyline(1);
|
|
|
+ resourceSkylines.add(resourceSkyline1);
|
|
|
+ final RecurrenceId recurrenceIdInvalid =
|
|
|
+ new RecurrenceId("Some random pipelineId", "Some random runId");
|
|
|
+ final ArrayList<ResourceSkyline> resourceSkylinesInvalid =
|
|
|
+ new ArrayList<ResourceSkyline>();
|
|
|
+ resourceSkylinesInvalid.add(null);
|
|
|
+ // try to updateHistory with non-existing recurringId
|
|
|
+ skylineStore.updateHistory(recurrenceIdInvalid, resourceSkylines);
|
|
|
+ }
|
|
|
+
|
|
|
+ @After public final void cleanUp() {
|
|
|
+ skylineStore = null;
|
|
|
+ resourceOverTime.clear();
|
|
|
+ resourceOverTime = null;
|
|
|
+ skylineList = null;
|
|
|
+ riAdd = null;
|
|
|
+ resource = null;
|
|
|
+ }
|
|
|
+}
|