test: cover CachingInboundEventSourceTest

Author: scrocquesel

operator-framework-core/src/test/java/io/javaoperatorsdk/operator/processing/event/source/inbound/CachingInboundEventSourceTest.java

@@ -0,0 +1,87 @@
+package io.javaoperatorsdk.operator.processing.event.source.inbound;
+
+import static org.assertj.core.api.Assertions.assertThat;
+import static org.mockito.ArgumentMatchers.any;
+import static org.mockito.ArgumentMatchers.eq;
+import static org.mockito.Mockito.atLeast;
+import static org.mockito.Mockito.mock;
+import static org.mockito.Mockito.never;
+import static org.mockito.Mockito.times;
+import static org.mockito.Mockito.verify;
+import static org.mockito.Mockito.when;
+
+import java.util.Collections;
+import java.util.Set;
+
+import org.junit.jupiter.api.BeforeEach;
+import org.junit.jupiter.api.Test;
+
+import io.javaoperatorsdk.operator.TestUtils;
+import io.javaoperatorsdk.operator.processing.event.EventHandler;
+import io.javaoperatorsdk.operator.processing.event.ResourceID;
+import io.javaoperatorsdk.operator.processing.event.source.AbstractEventSourceTestBase;
+import io.javaoperatorsdk.operator.processing.event.source.CacheKeyMapper;
+import io.javaoperatorsdk.operator.processing.event.source.SampleExternalResource;
+import io.javaoperatorsdk.operator.sample.simple.TestCustomResource;
+
+public class CachingInboundEventSourceTest extends
+        AbstractEventSourceTestBase<CachingInboundEventSource<SampleExternalResource, TestCustomResource>, EventHandler> {
+
+    public static final int PERIOD = 150;
+    private CachingInboundEventSource.ResourceFetcher<SampleExternalResource, TestCustomResource> supplier = mock(
+            CachingInboundEventSource.ResourceFetcher.class);
+    private TestCustomResource testCustomResource = TestUtils.testCustomResource();
+    private CacheKeyMapper<SampleExternalResource> cacheKeyMapper = r -> r.getName() + "#" + r.getValue();
+
+    @BeforeEach
+    public void setup() {
+        when(supplier.fetchResources(any()))
+                .thenReturn(Set.of(SampleExternalResource.testResource1()));
+
+        setUpSource(new CachingInboundEventSource<>(supplier,
+                SampleExternalResource.class, cacheKeyMapper));
+    }
+
+    @Test
+    void getSecondaryResourceFromCacheOrSupplier() throws InterruptedException {
+        when(supplier.fetchResources(any()))
+                .thenReturn(Set.of(SampleExternalResource.testResource1()));
+
+        var value = source.getSecondaryResources(testCustomResource);
+
+        verify(supplier, times(1)).fetchResources(eq(testCustomResource));
+        verify(eventHandler, never()).handleEvent(any());
+        assertThat(value).hasSize(1);
+
+        value = source.getSecondaryResources(testCustomResource);
+
+        assertThat(value).hasSize(1);
+        verify(supplier, times(1)).fetchResources(eq(testCustomResource));
+        verify(eventHandler, never()).handleEvent(any());
+
+        source.handleResourceEvent(ResourceID.fromResource(testCustomResource), Set.of(SampleExternalResource.testResource1(), SampleExternalResource.testResource2()));
+
+        verify(supplier, times(1)).fetchResources(eq(testCustomResource));
+        value = source.getSecondaryResources(testCustomResource);
+        assertThat(value).hasSize(2);
+    }
+
+    @Test
+    void propagateEventOnDeletedResource() throws InterruptedException {
+        source.handleResourceEvent(ResourceID.fromResource(testCustomResource), SampleExternalResource.testResource1());
+        source.handleResourceDeleteEvent(ResourceID.fromResource(testCustomResource),
+                cacheKeyMapper.keyFor(SampleExternalResource.testResource1()));
+                source.handleResourceDeleteEvent(ResourceID.fromResource(testCustomResource),
+                cacheKeyMapper.keyFor(SampleExternalResource.testResource2()));
+
+        verify(eventHandler, times(2)).handleEvent(any());
+    }
+
+    @Test
+    void propagateEventOnUpdateResources() throws InterruptedException {
+        source.handleResourceEvent(ResourceID.fromResource(testCustomResource),
+                Set.of(SampleExternalResource.testResource1(), SampleExternalResource.testResource2()));
+
+        verify(eventHandler, times(1)).handleEvent(any());
+    }
+}