Publish

1 files changed, 34 insertions, 0 deletions

diff --git a/jdisc_core/src/test/java/com/yahoo/jdisc/ReferencedResourceTestCase.java b/jdisc_core/src/test/java/com/yahoo/jdisc/ReferencedResourceTestCase.java
new file mode 100644
index 00000000000..4337d8b8c6c
--- /dev/null
+++ b/jdisc_core/src/test/java/com/yahoo/jdisc/ReferencedResourceTestCase.java
@@ -0,0 +1,34 @@
+// Copyright 2016 Yahoo Inc. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root.
+package com.yahoo.jdisc;
+
+import org.junit.Test;
+
+import static org.hamcrest.CoreMatchers.is;
+import static org.hamcrest.CoreMatchers.sameInstance;
+import static org.hamcrest.MatcherAssert.assertThat;
+import static org.mockito.Mockito.mock;
+import static org.mockito.Mockito.times;
+import static org.mockito.Mockito.verify;
+
+/**
+ * @author <a href="mailto:bakksjo@yahoo-inc.com">Oyvind Bakksjo</a>
+ */
+public class ReferencedResourceTestCase {
+    @Test
+    public void requireThatGettersMatchConstructor() {
+        final SharedResource resource = mock(SharedResource.class);
+        final ResourceReference reference = mock(ResourceReference.class);
+        final ReferencedResource<SharedResource> referencedResource = new ReferencedResource<>(resource, reference);
+        assertThat(referencedResource.getResource(), is(sameInstance(resource)));
+        assertThat(referencedResource.getReference(), is(sameInstance(reference)));
+    }
+
+    @Test
+    public void requireThatCloseCallsReferenceClose() {
+        final SharedResource resource = mock(SharedResource.class);
+        final ResourceReference reference = mock(ResourceReference.class);
+        final ReferencedResource<SharedResource> referencedResource = new ReferencedResource<>(resource, reference);
+        referencedResource.close();
+        verify(reference, times(1)).close();
+    }
+}