You are viewing a plain text version of this content. The canonical link for it is here.
Posted to commits@servicecomb.apache.org by se...@apache.org on 2018/01/10 01:00:27 UTC

[incubator-servicecomb-saga] 01/04: SCB-173 mark start of saga

This is an automated email from the ASF dual-hosted git repository.

seanyinx pushed a commit to branch master
in repository https://gitbox.apache.org/repos/asf/incubator-servicecomb-saga.git

commit 541360c4361f69cde406672fc552a196fc21ace6
Author: Eric Lee <da...@huawei.com>
AuthorDate: Mon Jan 8 20:48:09 2018 +0800

    SCB-173 mark start of saga
    
    Signed-off-by: Eric Lee <da...@huawei.com>
---
 .../integration/pack/tests/GreetingController.java |  2 +
 .../saga/integration/pack/tests/PackIT.java        | 91 +++++++++++-----------
 .../integration/pack/tests/TxEventEnvelope.java    |  4 +
 .../pack/tests/TxEventEnvelopeRepository.java      |  4 +
 .../connector/grpc/GrpcClientMessageSender.java    |  4 +-
 .../saga/omega/context/annotations/SagaStart.java  | 13 ++--
 .../transaction/SagaStartAnnotationProcessor.java  | 20 +++--
 .../saga/omega/transaction/SagaStartedEvent.java   | 12 +--
 .../saga/omega/transaction/TransactionAspect.java  | 27 ++++++-
 .../SagaStartAnnotationProcessorTest.java          | 71 +++++++++++++++++
 10 files changed, 179 insertions(+), 69 deletions(-)

diff --git a/integration-tests/pack-tests/src/test/java/org/apache/servicecomb/saga/integration/pack/tests/GreetingController.java b/integration-tests/pack-tests/src/test/java/org/apache/servicecomb/saga/integration/pack/tests/GreetingController.java
index 12356ee..9d8d2e8 100644
--- a/integration-tests/pack-tests/src/test/java/org/apache/servicecomb/saga/integration/pack/tests/GreetingController.java
+++ b/integration-tests/pack-tests/src/test/java/org/apache/servicecomb/saga/integration/pack/tests/GreetingController.java
@@ -17,6 +17,7 @@
 
 package org.apache.servicecomb.saga.integration.pack.tests;
 
+import org.apache.servicecomb.saga.omega.context.annotations.SagaStart;
 import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.http.ResponseEntity;
 import org.springframework.stereotype.Controller;
@@ -39,6 +40,7 @@ public class GreetingController {
   }
 
 
+  @SagaStart
   @GetMapping("/greet")
   ResponseEntity<String> greet(@RequestParam String name) {
     String greetings = greetingService.greet(name);
diff --git a/integration-tests/pack-tests/src/test/java/org/apache/servicecomb/saga/integration/pack/tests/PackIT.java b/integration-tests/pack-tests/src/test/java/org/apache/servicecomb/saga/integration/pack/tests/PackIT.java
index 66454b7..de9e48f 100644
--- a/integration-tests/pack-tests/src/test/java/org/apache/servicecomb/saga/integration/pack/tests/PackIT.java
+++ b/integration-tests/pack-tests/src/test/java/org/apache/servicecomb/saga/integration/pack/tests/PackIT.java
@@ -25,7 +25,6 @@ import static org.hamcrest.CoreMatchers.nullValue;
 import static org.hamcrest.Matchers.contains;
 import static org.hamcrest.core.Is.is;
 import static org.junit.Assert.assertThat;
-import static org.springframework.http.HttpMethod.GET;
 import static org.springframework.http.HttpStatus.INTERNAL_SERVER_ERROR;
 import static org.springframework.http.HttpStatus.OK;
 
@@ -41,8 +40,6 @@ import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.boot.test.context.SpringBootTest;
 import org.springframework.boot.test.context.SpringBootTest.WebEnvironment;
 import org.springframework.boot.test.web.client.TestRestTemplate;
-import org.springframework.http.HttpEntity;
-import org.springframework.http.HttpHeaders;
 import org.springframework.http.ResponseEntity;
 import org.springframework.test.context.junit4.SpringRunner;
 
@@ -72,90 +69,92 @@ public class PackIT {
 
   @Test
   public void updatesTxStateToAlpha() throws Exception {
-    HttpHeaders headers = new HttpHeaders();
-
-    headers.set(OmegaContext.GLOBAL_TX_ID_KEY, globalTxId);
-
-    ResponseEntity<String> entity = restTemplate.exchange("/greet?name={name}",
-        GET,
-        new HttpEntity<>(headers),
+    ResponseEntity<String> entity = restTemplate.getForEntity("/greet?name={name}",
         String.class,
         "mike");
 
     assertThat(entity.getStatusCode(), is(OK));
     assertThat(entity.getBody(), is("Greetings, mike; Bonjour, mike"));
 
+    List<String> distinctGlobalTxIds = repository.findDistinctGlobalTxId();
+    assertThat(distinctGlobalTxIds.size(), is(1));
+
+    String globalTxId = distinctGlobalTxIds.get(0);
     List<TxEventEnvelope> envelopes = repository.findByGlobalTxIdOrderByCreationTime(globalTxId);
 
-    assertThat(envelopes.size(), is(4));
-    assertThat(envelopes.get(0).type(), is("TxStartedEvent"));
+    assertThat(envelopes.size(), is(5));
+    assertThat(envelopes.get(0).type(), is("SagaStartedEvent"));
     assertThat(envelopes.get(0).localTxId(), is(notNullValue()));
     assertThat(envelopes.get(0).parentTxId(), is(nullValue()));
     assertThat(envelopes.get(0).serviceName(), is(serviceName));
     assertThat(envelopes.get(0).instanceId(), is(notNullValue()));
 
-    assertThat(envelopes.get(1).type(), is("TxEndedEvent"));
+    assertThat(envelopes.get(1).type(), is("TxStartedEvent"));
     assertThat(envelopes.get(1).localTxId(), is(envelopes.get(0).localTxId()));
     assertThat(envelopes.get(1).parentTxId(), is(nullValue()));
     assertThat(envelopes.get(1).serviceName(), is(serviceName));
     assertThat(envelopes.get(1).instanceId(), is(envelopes.get(0).instanceId()));
 
-
-    assertThat(envelopes.get(2).type(), is("TxStartedEvent"));
-    assertThat(envelopes.get(2).localTxId(), is(notNullValue()));
-    assertThat(envelopes.get(2).parentTxId(), is(envelopes.get(0).localTxId()));
+    assertThat(envelopes.get(2).type(), is("TxEndedEvent"));
+    assertThat(envelopes.get(2).localTxId(), is(envelopes.get(0).localTxId()));
+    assertThat(envelopes.get(2).parentTxId(), is(nullValue()));
     assertThat(envelopes.get(2).serviceName(), is(serviceName));
-    assertThat(envelopes.get(2).instanceId(), is(notNullValue()));
+    assertThat(envelopes.get(2).instanceId(), is(envelopes.get(0).instanceId()));
 
-    assertThat(envelopes.get(3).type(), is("TxEndedEvent"));
-    assertThat(envelopes.get(3).localTxId(), is(envelopes.get(2).localTxId()));
+    assertThat(envelopes.get(3).type(), is("TxStartedEvent"));
+    assertThat(envelopes.get(3).localTxId(), is(notNullValue()));
     assertThat(envelopes.get(3).parentTxId(), is(envelopes.get(0).localTxId()));
     assertThat(envelopes.get(3).serviceName(), is(serviceName));
-    assertThat(envelopes.get(3).instanceId(), is(envelopes.get(2).instanceId()));
+    assertThat(envelopes.get(3).instanceId(), is(notNullValue()));
+
+    assertThat(envelopes.get(4).type(), is("TxEndedEvent"));
+    assertThat(envelopes.get(4).localTxId(), is(envelopes.get(3).localTxId()));
+    assertThat(envelopes.get(4).parentTxId(), is(envelopes.get(0).localTxId()));
+    assertThat(envelopes.get(4).serviceName(), is(serviceName));
+    assertThat(envelopes.get(4).instanceId(), is(envelopes.get(3).instanceId()));
 
     assertThat(compensatedMessages.isEmpty(), is(true));
   }
 
   @Test
   public void compensatesFailedGlobalTransaction() throws Exception {
-    HttpHeaders headers = new HttpHeaders();
-
-    headers.set(OmegaContext.GLOBAL_TX_ID_KEY, globalTxId);
-
-    ResponseEntity<String> entity = restTemplate.exchange("/greet?name={name}",
-        GET,
-        new HttpEntity<>(headers),
+    ResponseEntity<String> entity = restTemplate.getForEntity("/greet?name={name}",
         String.class,
         TRESPASSER);
 
     assertThat(entity.getStatusCode(), is(INTERNAL_SERVER_ERROR));
 
-    await().atMost(2, SECONDS).until(() -> repository.count() == 6);
+    await().atMost(2, SECONDS).until(() -> repository.count() == 7);
 
-    List<TxEventEnvelope> envelopes = repository.findByGlobalTxIdOrderByCreationTime(globalTxId);
-    assertThat(envelopes.size(), is(6));
+    List<String> distinctGlobalTxIds = repository.findDistinctGlobalTxId();
+    assertThat(distinctGlobalTxIds.size(), is(1));
 
-    assertThat(envelopes.get(0).type(), is("TxStartedEvent"));
-    assertThat(envelopes.get(1).type(), is("TxEndedEvent"));
-    assertThat(envelopes.get(2).type(), is("TxStartedEvent"));
+    String globalTxId = distinctGlobalTxIds.get(0);
+    List<TxEventEnvelope> envelopes = repository.findByGlobalTxIdOrderByCreationTime(globalTxId);
+    assertThat(envelopes.size(), is(7));
 
-    assertThat(envelopes.get(3).type(), is("TxAbortedEvent"));
-    assertThat(envelopes.get(3).localTxId(), is(envelopes.get(2).localTxId()));
-    assertThat(envelopes.get(3).parentTxId(), is(envelopes.get(0).localTxId()));
-    assertThat(envelopes.get(3).serviceName(), is(serviceName));
-    assertThat(envelopes.get(3).instanceId(), is(envelopes.get(2).instanceId()));
+    assertThat(envelopes.get(0).type(), is("SagaStartedEvent"));
+    assertThat(envelopes.get(1).type(), is("TxStartedEvent"));
+    assertThat(envelopes.get(2).type(), is("TxEndedEvent"));
+    assertThat(envelopes.get(3).type(), is("TxStartedEvent"));
 
-    assertThat(envelopes.get(4).type(), is("TxCompensatedEvent"));
-    assertThat(envelopes.get(4).localTxId(), is(envelopes.get(0).localTxId()));
-    assertThat(envelopes.get(4).parentTxId(), is(nullValue()));
+    assertThat(envelopes.get(4).type(), is("TxAbortedEvent"));
+    assertThat(envelopes.get(4).localTxId(), is(envelopes.get(3).localTxId()));
+    assertThat(envelopes.get(4).parentTxId(), is(envelopes.get(1).localTxId()));
     assertThat(envelopes.get(4).serviceName(), is(serviceName));
-    assertThat(envelopes.get(4).instanceId(), is(envelopes.get(0).instanceId()));
+    assertThat(envelopes.get(4).instanceId(), is(envelopes.get(3).instanceId()));
 
     assertThat(envelopes.get(5).type(), is("TxCompensatedEvent"));
-    assertThat(envelopes.get(5).localTxId(), is(envelopes.get(2).localTxId()));
-    assertThat(envelopes.get(5).parentTxId(), is(envelopes.get(0).localTxId()));
+    assertThat(envelopes.get(5).localTxId(), is(envelopes.get(1).localTxId()));
+    assertThat(envelopes.get(5).parentTxId(), is(nullValue()));
     assertThat(envelopes.get(5).serviceName(), is(serviceName));
-    assertThat(envelopes.get(5).instanceId(), is(envelopes.get(2).instanceId()));
+    assertThat(envelopes.get(5).instanceId(), is(envelopes.get(1).instanceId()));
+
+    assertThat(envelopes.get(6).type(), is("TxCompensatedEvent"));
+    assertThat(envelopes.get(6).localTxId(), is(envelopes.get(3).localTxId()));
+    assertThat(envelopes.get(6).parentTxId(), is(envelopes.get(1).localTxId()));
+    assertThat(envelopes.get(6).serviceName(), is(serviceName));
+    assertThat(envelopes.get(6).instanceId(), is(envelopes.get(3).instanceId()));
 
     assertThat(compensatedMessages, contains(
         "Goodbye, " + TRESPASSER,
diff --git a/integration-tests/pack-tests/src/test/java/org/apache/servicecomb/saga/integration/pack/tests/TxEventEnvelope.java b/integration-tests/pack-tests/src/test/java/org/apache/servicecomb/saga/integration/pack/tests/TxEventEnvelope.java
index d99c2b6..0087dd7 100644
--- a/integration-tests/pack-tests/src/test/java/org/apache/servicecomb/saga/integration/pack/tests/TxEventEnvelope.java
+++ b/integration-tests/pack-tests/src/test/java/org/apache/servicecomb/saga/integration/pack/tests/TxEventEnvelope.java
@@ -50,6 +50,10 @@ class TxEventEnvelope {
     return instanceId;
   }
 
+  String globalTxId() {
+    return globalTxId;
+  }
+
   String localTxId() {
     return localTxId;
   }
diff --git a/integration-tests/pack-tests/src/test/java/org/apache/servicecomb/saga/integration/pack/tests/TxEventEnvelopeRepository.java b/integration-tests/pack-tests/src/test/java/org/apache/servicecomb/saga/integration/pack/tests/TxEventEnvelopeRepository.java
index 7e4b7e9..5400d7c 100644
--- a/integration-tests/pack-tests/src/test/java/org/apache/servicecomb/saga/integration/pack/tests/TxEventEnvelopeRepository.java
+++ b/integration-tests/pack-tests/src/test/java/org/apache/servicecomb/saga/integration/pack/tests/TxEventEnvelopeRepository.java
@@ -19,8 +19,12 @@ package org.apache.servicecomb.saga.integration.pack.tests;
 
 import java.util.List;
 
+import org.springframework.data.jpa.repository.Query;
 import org.springframework.data.repository.CrudRepository;
 
 interface TxEventEnvelopeRepository extends CrudRepository<TxEventEnvelope, Long> {
   List<TxEventEnvelope> findByGlobalTxIdOrderByCreationTime(String globalTxId);
+
+  @Query("SELECT DISTINCT(e.globalTxId) from TxEventEnvelope e")
+  List<String> findDistinctGlobalTxId();
 }
diff --git a/omega/omega-connector/omega-connector-grpc/src/main/java/org/apache/servicecomb/saga/omega/connector/grpc/GrpcClientMessageSender.java b/omega/omega-connector/omega-connector-grpc/src/main/java/org/apache/servicecomb/saga/omega/connector/grpc/GrpcClientMessageSender.java
index b2e837c..59fbce1 100644
--- a/omega/omega-connector/omega-connector-grpc/src/main/java/org/apache/servicecomb/saga/omega/connector/grpc/GrpcClientMessageSender.java
+++ b/omega/omega-connector/omega-connector-grpc/src/main/java/org/apache/servicecomb/saga/omega/connector/grpc/GrpcClientMessageSender.java
@@ -84,13 +84,11 @@ public class GrpcClientMessageSender implements MessageSender {
         .setTimestamp(event.timestamp())
         .setGlobalTxId(event.globalTxId())
         .setLocalTxId(event.localTxId())
+        .setParentTxId(event.parentTxId() == null ? "" : event.parentTxId())
         .setType(event.type())
         .setCompensationMethod(event.compensationMethod())
         .setPayloads(payloads);
 
-    if (event.parentTxId() != null) {
-      builder.setParentTxId(event.parentTxId());
-    }
     return builder.build();
   }
 
diff --git a/integration-tests/pack-tests/src/test/java/org/apache/servicecomb/saga/integration/pack/tests/TxEventEnvelopeRepository.java b/omega/omega-context/src/main/java/org/apache/servicecomb/saga/omega/context/annotations/SagaStart.java
similarity index 71%
copy from integration-tests/pack-tests/src/test/java/org/apache/servicecomb/saga/integration/pack/tests/TxEventEnvelopeRepository.java
copy to omega/omega-context/src/main/java/org/apache/servicecomb/saga/omega/context/annotations/SagaStart.java
index 7e4b7e9..435d72f 100644
--- a/integration-tests/pack-tests/src/test/java/org/apache/servicecomb/saga/integration/pack/tests/TxEventEnvelopeRepository.java
+++ b/omega/omega-context/src/main/java/org/apache/servicecomb/saga/omega/context/annotations/SagaStart.java
@@ -15,12 +15,15 @@
  * limitations under the License.
  */
 
-package org.apache.servicecomb.saga.integration.pack.tests;
+package org.apache.servicecomb.saga.omega.context.annotations;
 
-import java.util.List;
+import static java.lang.annotation.ElementType.METHOD;
+import static java.lang.annotation.RetentionPolicy.RUNTIME;
 
-import org.springframework.data.repository.CrudRepository;
+import java.lang.annotation.Retention;
+import java.lang.annotation.Target;
 
-interface TxEventEnvelopeRepository extends CrudRepository<TxEventEnvelope, Long> {
-  List<TxEventEnvelope> findByGlobalTxIdOrderByCreationTime(String globalTxId);
+@Retention(RUNTIME)
+@Target(METHOD)
+public @interface SagaStart {
 }
diff --git a/integration-tests/pack-tests/src/test/java/org/apache/servicecomb/saga/integration/pack/tests/TxEventEnvelopeRepository.java b/omega/omega-transaction/src/main/java/org/apache/servicecomb/saga/omega/transaction/SagaStartAnnotationProcessor.java
similarity index 60%
copy from integration-tests/pack-tests/src/test/java/org/apache/servicecomb/saga/integration/pack/tests/TxEventEnvelopeRepository.java
copy to omega/omega-transaction/src/main/java/org/apache/servicecomb/saga/omega/transaction/SagaStartAnnotationProcessor.java
index 7e4b7e9..66fadda 100644
--- a/integration-tests/pack-tests/src/test/java/org/apache/servicecomb/saga/integration/pack/tests/TxEventEnvelopeRepository.java
+++ b/omega/omega-transaction/src/main/java/org/apache/servicecomb/saga/omega/transaction/SagaStartAnnotationProcessor.java
@@ -15,12 +15,22 @@
  * limitations under the License.
  */
 
-package org.apache.servicecomb.saga.integration.pack.tests;
+package org.apache.servicecomb.saga.omega.transaction;
 
-import java.util.List;
+import org.apache.servicecomb.saga.omega.context.OmegaContext;
 
-import org.springframework.data.repository.CrudRepository;
+public class SagaStartAnnotationProcessor {
 
-interface TxEventEnvelopeRepository extends CrudRepository<TxEventEnvelope, Long> {
-  List<TxEventEnvelope> findByGlobalTxIdOrderByCreationTime(String globalTxId);
+  private final OmegaContext omegaContext;
+
+  private final MessageSender sender;
+
+  SagaStartAnnotationProcessor(OmegaContext omegaContext, MessageSender sender) {
+    this.omegaContext = omegaContext;
+    this.sender = sender;
+  }
+
+  void intercept() {
+    sender.send(new SagaStartedEvent(omegaContext.newGlobalTxId(), omegaContext.newLocalTxId()));
+  }
 }
diff --git a/integration-tests/pack-tests/src/test/java/org/apache/servicecomb/saga/integration/pack/tests/TxEventEnvelopeRepository.java b/omega/omega-transaction/src/main/java/org/apache/servicecomb/saga/omega/transaction/SagaStartedEvent.java
similarity index 72%
copy from integration-tests/pack-tests/src/test/java/org/apache/servicecomb/saga/integration/pack/tests/TxEventEnvelopeRepository.java
copy to omega/omega-transaction/src/main/java/org/apache/servicecomb/saga/omega/transaction/SagaStartedEvent.java
index 7e4b7e9..0985f59 100644
--- a/integration-tests/pack-tests/src/test/java/org/apache/servicecomb/saga/integration/pack/tests/TxEventEnvelopeRepository.java
+++ b/omega/omega-transaction/src/main/java/org/apache/servicecomb/saga/omega/transaction/SagaStartedEvent.java
@@ -15,12 +15,12 @@
  * limitations under the License.
  */
 
-package org.apache.servicecomb.saga.integration.pack.tests;
+package org.apache.servicecomb.saga.omega.transaction;
 
-import java.util.List;
+public class SagaStartedEvent extends TxEvent {
 
-import org.springframework.data.repository.CrudRepository;
-
-interface TxEventEnvelopeRepository extends CrudRepository<TxEventEnvelope, Long> {
-  List<TxEventEnvelope> findByGlobalTxIdOrderByCreationTime(String globalTxId);
+  public SagaStartedEvent(String globalTxId, String localTxId) {
+    // use "" instead of null as compensationMethod requires not null in sql
+    super(globalTxId, localTxId, null, "");
+  }
 }
diff --git a/omega/omega-transaction/src/main/java/org/apache/servicecomb/saga/omega/transaction/TransactionAspect.java b/omega/omega-transaction/src/main/java/org/apache/servicecomb/saga/omega/transaction/TransactionAspect.java
index 2b68da5..3ad52a8 100644
--- a/omega/omega-transaction/src/main/java/org/apache/servicecomb/saga/omega/transaction/TransactionAspect.java
+++ b/omega/omega-transaction/src/main/java/org/apache/servicecomb/saga/omega/transaction/TransactionAspect.java
@@ -20,6 +20,8 @@ package org.apache.servicecomb.saga.omega.transaction;
 import java.lang.invoke.MethodHandles;
 import java.lang.reflect.Method;
 
+import org.apache.servicecomb.saga.omega.context.OmegaContext;
+import org.apache.servicecomb.saga.omega.transaction.annotations.Compensable;
 import org.aspectj.lang.ProceedingJoinPoint;
 import org.aspectj.lang.annotation.Around;
 import org.aspectj.lang.annotation.Aspect;
@@ -27,22 +29,24 @@ import org.aspectj.lang.reflect.MethodSignature;
 import org.slf4j.Logger;
 import org.slf4j.LoggerFactory;
 
-import org.apache.servicecomb.saga.omega.context.OmegaContext;
-import org.apache.servicecomb.saga.omega.transaction.annotations.Compensable;
-
 @Aspect
 public class TransactionAspect {
   private static final Logger LOG = LoggerFactory.getLogger(MethodHandles.lookup().lookupClass());
   private final PreTransactionInterceptor preTransactionInterceptor;
   private final PostTransactionInterceptor postTransactionInterceptor;
   private final FailedTransactionInterceptor failedTransactionInterceptor;
+
+  private final SagaStartAnnotationProcessor sagaStartAnnotationProcessor;
   private final OmegaContext context;
 
-  public TransactionAspect(MessageSender sender, OmegaContext context) {
+  public TransactionAspect(MessageSender sender,
+      OmegaContext context) {
     this.context = context;
     this.preTransactionInterceptor = new PreTransactionInterceptor(sender);
     this.postTransactionInterceptor = new PostTransactionInterceptor(sender);
     this.failedTransactionInterceptor = new FailedTransactionInterceptor(sender);
+    this.sagaStartAnnotationProcessor = new SagaStartAnnotationProcessor(this.context, sender);
+
   }
 
   @Around("execution(@org.apache.servicecomb.saga.omega.transaction.annotations.Compensable * *(..)) && @annotation(compensable)")
@@ -65,6 +69,21 @@ public class TransactionAspect {
     }
   }
 
+  @Around("execution(@org.apache.servicecomb.saga.omega.context.annotations.SagaStart * *(..))")
+  Object advise(ProceedingJoinPoint joinPoint) throws Throwable {
+    Method method = ((MethodSignature) joinPoint.getSignature()).getMethod();
+
+    LOG.debug("Initializing global tx id before execution of method {}", method.toString());
+    sagaStartAnnotationProcessor.intercept();
+
+    try {
+      return joinPoint.proceed();
+    } catch (Throwable throwable) {
+      LOG.error("Failed to process SagaStart method: {}", method.toString());
+      throw throwable;
+    }
+  }
+
   private String compensationMethodSignature(ProceedingJoinPoint joinPoint, Compensable compensable, Method method)
       throws NoSuchMethodException {
 
diff --git a/omega/omega-transaction/src/test/java/org/apache/servicecomb/saga/omega/transaction/SagaStartAnnotationProcessorTest.java b/omega/omega-transaction/src/test/java/org/apache/servicecomb/saga/omega/transaction/SagaStartAnnotationProcessorTest.java
new file mode 100644
index 0000000..6ef5aff
--- /dev/null
+++ b/omega/omega-transaction/src/test/java/org/apache/servicecomb/saga/omega/transaction/SagaStartAnnotationProcessorTest.java
@@ -0,0 +1,71 @@
+/*
+ * 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.servicecomb.saga.omega.transaction;
+
+import static org.hamcrest.core.Is.is;
+import static org.hamcrest.core.IsNull.nullValue;
+import static org.junit.Assert.assertThat;
+import static org.mockito.Mockito.mock;
+import static org.mockito.Mockito.when;
+
+import java.util.ArrayList;
+import java.util.List;
+import java.util.UUID;
+
+import org.apache.servicecomb.saga.omega.context.IdGenerator;
+import org.apache.servicecomb.saga.omega.context.OmegaContext;
+import org.junit.Test;
+
+public class SagaStartAnnotationProcessorTest {
+
+  private final List<TxEvent> messages = new ArrayList<>();
+
+  private final MessageSender sender = messages::add;
+
+  private final String globalTxId = UUID.randomUUID().toString();
+
+  private final String localTxId = UUID.randomUUID().toString();
+
+
+  private final IdGenerator generator = mock(IdGenerator.class);
+
+  @SuppressWarnings("unchecked")
+  private final OmegaContext context = new OmegaContext(generator);
+
+  private final SagaStartAnnotationProcessor sagaStartAnnotationProcessor = new SagaStartAnnotationProcessor(context,
+      sender);
+
+  @Test
+  public void sendsSagaStartedEvent() {
+    when(generator.nextId()).thenReturn(globalTxId, localTxId);
+
+    sagaStartAnnotationProcessor.intercept();
+
+    assertThat(context.globalTxId(), is(globalTxId));
+    assertThat(context.localTxId(), is(localTxId));
+
+    TxEvent event = messages.get(0);
+
+    assertThat(event.globalTxId(), is(globalTxId));
+    assertThat(event.localTxId(), is(localTxId));
+    assertThat(event.parentTxId(), is(nullValue()));
+    assertThat(event.compensationMethod().isEmpty(), is(true));
+    assertThat(event.type(), is("SagaStartedEvent"));
+    assertThat(event.parentTxId(), is(nullValue()));
+  }
+}
\ No newline at end of file

-- 
To stop receiving notification emails like this one, please contact
"commits@servicecomb.apache.org" <co...@servicecomb.apache.org>.