commit 81416d0b177aa78af57ecc34a3c95ed4a80c3b8d
author Sho SHIMIZU <sshimizu@us.fujitsu.com> Wed Jul 06 14:53:28 2016 -0700
committer Gerrit Code Review <gerrit@onlab.us> Thu Jul 07 17:27:49 2016 +0000
tree c7f15b628aa41d44d988500d24550b7a5aba5d48
parent 797692374e6b01fadb00c6328c46910c7017dbc0

Improve test coverage

Change-Id: I12d6827c644f45189f3a5929a8c1837aa112bdb9

core/store/dist/src/test/java/org/onosproject/store/resource/impl/ContinuousResourceAllocationTest.java

diff --git a/core/store/dist/src/test/java/org/onosproject/store/resource/impl/ContinuousResourceAllocationTest.java b/core/store/dist/src/test/java/org/onosproject/store/resource/impl/ContinuousResourceAllocationTest.java
index f12e5b8..56d005a 100644
--- a/core/store/dist/src/test/java/org/onosproject/store/resource/impl/ContinuousResourceAllocationTest.java
+++ b/core/store/dist/src/test/java/org/onosproject/store/resource/impl/ContinuousResourceAllocationTest.java
@@ -27,6 +27,7 @@
 import org.onosproject.net.resource.ResourceConsumer;
 import org.onosproject.net.resource.Resources;
 
+import static org.hamcrest.Matchers.hasItem;
 import static org.hamcrest.Matchers.is;
 import static org.junit.Assert.assertThat;
 
@@ -138,4 +139,47 @@
         assertThat(allocations.size(), is(1));
         assertThat(allocations.get(0).resource().equals(allocated), is(true));
     }
+
+    @Test
+    public void testAllocateDifferentValue() {
+        ContinuousResource original =
+                Resources.continuous(DID, PN1, Bandwidth.class).resource(Bandwidth.gbps(1).bps());
+        ContinuousResource allocated =
+                Resources.continuous(DID, PN1, Bandwidth.class).resource(Bandwidth.mbps(500).bps());
+        ResourceConsumer consumer = IntentId.valueOf(1);
+
+        ContinuousResourceAllocation sut = new ContinuousResourceAllocation(original,
+                ImmutableList.of(new ResourceAllocation(allocated, consumer)));
+
+        ContinuousResource request =
+                Resources.continuous(DID, PN1, Bandwidth.class).resource(Bandwidth.mbps(200).bps());
+
+        ContinuousResourceAllocation newValue = sut.allocate(new ResourceAllocation(request, consumer));
+
+        assertThat(newValue.allocations().size(), is(2));
+        assertThat(newValue.allocations(), hasItem(new ResourceAllocation(allocated, consumer)));
+        assertThat(newValue.allocations(), hasItem(new ResourceAllocation(request, consumer)));
+    }
+
+    @Test
+    public void testAllocateSameValue() {
+        ContinuousResource original =
+                Resources.continuous(DID, PN1, Bandwidth.class).resource(Bandwidth.gbps(1).bps());
+        ContinuousResource allocated =
+                Resources.continuous(DID, PN1, Bandwidth.class).resource(Bandwidth.mbps(300).bps());
+        ResourceConsumer consumer = IntentId.valueOf(1);
+
+        ContinuousResourceAllocation sut = new ContinuousResourceAllocation(original,
+                ImmutableList.of(new ResourceAllocation(allocated, consumer)));
+
+        ContinuousResource request =
+                Resources.continuous(DID, PN1, Bandwidth.class).resource(Bandwidth.mbps(300).bps());
+
+        ContinuousResourceAllocation newValue = sut.allocate(new ResourceAllocation(request, consumer));
+
+        assertThat(newValue.allocations().size(), is(2));
+        assertThat(newValue.allocations()
+                .stream()
+                .allMatch(x -> x.equals(new ResourceAllocation(allocated, consumer))), is(true));
+    }
 }