Fix test

CJSCommonPlatform · Oct 25, 2018 · 70cf7b5 · 70cf7b5
1 parent 75eb75d
commit 70cf7b5
Showing 1 changed file with 17 additions and 3 deletions.
diff --git a/...gov/justice/tools/eventsourcing/transformation/EventTransformationStreamIdFilterTest.java b/...gov/justice/tools/eventsourcing/transformation/EventTransformationStreamIdFilterTest.java
@@ -4,11 +4,15 @@
 import static java.util.UUID.randomUUID;
 import static javax.json.Json.createObjectBuilder;
 import static org.junit.Assert.assertFalse;
+import static org.junit.Assert.assertTrue;
 import static uk.gov.justice.services.messaging.JsonEnvelope.envelopeFrom;
 import static uk.gov.justice.services.messaging.JsonEnvelope.metadataBuilder;
+import static uk.gov.justice.tools.eventsourcing.transformation.api.Action.NO_ACTION;
+import static uk.gov.justice.tools.eventsourcing.transformation.api.Action.TRANSFORM;
 
 import uk.gov.justice.services.core.enveloper.Enveloper;
 import uk.gov.justice.services.messaging.JsonEnvelope;
+import uk.gov.justice.tools.eventsourcing.transformation.api.Action;
 import uk.gov.justice.tools.eventsourcing.transformation.api.EventTransformation;
 import uk.gov.justice.tools.eventsourcing.transformation.api.annotation.Transformation;
 
@@ -30,13 +34,13 @@ public void shouldReturnStreamId() throws Exception {
         final JsonEnvelope event1 = buildEnvelope(SOURCE_EVENT_NAME);
         final JsonEnvelope event2 = buildEnvelope(SOURCE_EVENT_NAME_2);
 
-        final TestTransformation transformation1 = new TestTransformation();
-        final TestTransformation transformation2 = new TestTransformation();
+        final TestTransformation1 transformation1 = new TestTransformation1();
+        final TestTransformation1 transformation2 = new TestTransformation1();
 
         final EventTransformationStreamIdFilter eventTransformationStreamIdFilter = new EventTransformationStreamIdFilter();
         final Optional<UUID> eventTransformationStreamId = eventTransformationStreamIdFilter.getEventTransformationStreamId(newHashSet(transformation1, transformation2), Arrays.asList(event1, event2));
 
-        assertFalse(eventTransformationStreamId.isPresent());
+        assertTrue(eventTransformationStreamId.isPresent());
     }
 
     @Test
@@ -71,8 +75,18 @@ public void setEnveloper(Enveloper enveloper) {
     @Transformation
     public static class TestTransformation1 implements EventTransformation {
 
+        @Override
+        public Action actionFor(final JsonEnvelope event) {
+            if (event.metadata().name().equalsIgnoreCase("eventSourceName") ||
+                    event.metadata().name().equalsIgnoreCase("eventSourceName2")) {
+                return TRANSFORM;
+            }
+            return NO_ACTION;
+        }
+
         @Override
         public Stream<JsonEnvelope> apply(JsonEnvelope event) {
+
             return Stream.of(event);
         }