pekko/akka-docs/java/code/akka/docs/transactor/TransactorDocTest.java

/**
 * Copyright (C) 2009-2011 Typesafe Inc. <http://www.typesafe.com>
 */

package akka.docs.transactor;

import static org.junit.Assert.*;
import org.junit.Test;

//#imports
import akka.actor.*;
import akka.dispatch.Await;
import akka.transactor.Coordinated;
import akka.util.Duration;
import akka.util.Timeout;
import java.util.concurrent.TimeUnit;
//#imports

public class TransactorDocTest {

    @Test
    public void coordinatedExample() {
        //#coordinated-example
        ActorSystem system = ActorSystem.create("CoordinatedExample");

        ActorRef counter1 = system.actorOf(new Props().withCreator(CoordinatedCounter.class));
        ActorRef counter2 = system.actorOf(new Props().withCreator(CoordinatedCounter.class));

        Timeout timeout = new Timeout(Duration.create(5, TimeUnit.SECONDS));

        counter1.tell(new Coordinated(new Increment(counter2), timeout));

        Integer count = (Integer) Await.result(counter1.ask("GetCount", timeout), timeout.duration());
        //#coordinated-example

        assertEquals(count, new Integer(1));

        system.shutdown();
    }

    @Test
    public void coordinatedApi() {
        //#create-coordinated
        Timeout timeout = new Timeout(Duration.create(5, TimeUnit.SECONDS));
        Coordinated coordinated = new Coordinated(timeout);
        //#create-coordinated

        ActorSystem system = ActorSystem.create("CoordinatedApi");
        ActorRef actor = system.actorOf(new Props().withCreator(Coordinator.class));

        //#send-coordinated
        actor.tell(new Coordinated(new Message(), timeout));
        //#send-coordinated

        //#include-coordinated
        actor.tell(coordinated.coordinate(new Message()));
        //#include-coordinated

        coordinated.await();

        system.shutdown();
    }

    @Test
    public void counterTransactor() {
        ActorSystem system = ActorSystem.create("CounterTransactor");
        ActorRef counter = system.actorOf(new Props().withCreator(Counter.class));

        Timeout timeout = new Timeout(Duration.create(5, TimeUnit.SECONDS));
        Coordinated coordinated = new Coordinated(timeout);
        counter.tell(coordinated.coordinate(new Increment()));
        coordinated.await();

        Integer count = (Integer) Await.result(counter.ask("GetCount", timeout), timeout.duration());
        assertEquals(count, new Integer(1));

        system.shutdown();
    }

    @Test
    public void friendlyCounterTransactor() {
        ActorSystem system = ActorSystem.create("FriendlyCounterTransactor");
        ActorRef friend = system.actorOf(new Props().withCreator(Counter.class));
        ActorRef friendlyCounter = system.actorOf(new Props().withCreator(FriendlyCounter.class));

        Timeout timeout = new Timeout(Duration.create(5, TimeUnit.SECONDS));
        Coordinated coordinated = new Coordinated(timeout);
        friendlyCounter.tell(coordinated.coordinate(new Increment(friend)));
        coordinated.await();

        Integer count1 = (Integer) Await.result(friendlyCounter.ask("GetCount", timeout), timeout.duration());
        assertEquals(count1, new Integer(1));

        Integer count2 = (Integer) Await.result(friend.ask("GetCount", timeout), timeout.duration());
        assertEquals(count2, new Integer(1));

        system.shutdown();
    }
}