pekko/akka-actor-tests/src/test/java/akka/japi/MatchBuilderTest.java

/*
 * Copyright (C) 2009-2018 Lightbend Inc. <https://www.lightbend.com>
 */

package akka.japi;

import akka.japi.pf.FI;
import akka.japi.pf.Match;
import org.junit.Rule;
import org.junit.rules.ExpectedException;
import org.junit.Test;
import org.scalatest.junit.JUnitSuite;
import scala.MatchError;

import static org.junit.Assert.*;

public class MatchBuilderTest extends JUnitSuite {

  @Rule
  public ExpectedException exception = ExpectedException.none();

  @Test
  public void shouldPassBasicMatchTest() {
    Match<Object, Double> pf = Match.create(Match.match(Integer.class, new FI.Apply<Integer, Double>() {
      @Override
      public Double apply(Integer integer) {
        return integer * 10.0;
      }
    }).match(Number.class, new FI.Apply<Number, Double>() {
      @Override
      public Double apply(Number number) {
        return number.doubleValue() * (-10.0);
      }
    }));

    assertTrue("An integer should be multiplied by 10", new Double(47110).equals(pf.match(new Integer(4711))));
    assertTrue("A double should be multiplied by -10", new Double(-47110).equals(pf.match(new Double(4711))));

    exception.expect(MatchError.class);
    assertFalse("A string should throw a MatchError", new Double(4711).equals(pf.match("4711")));
  }


  static class GenericClass<T> {
    T val;

    public GenericClass(T val) {
      this.val = val;
    }
  }

  @Test
  public void shouldHandleMatchOnGenericClass() {
    Match<Object, String> pf = Match.create(Match.matchUnchecked(GenericClass.class, new FI.Apply<GenericClass<String>, String>() {
      @Override
      public String apply(GenericClass<String> stringGenericClass) {
        return stringGenericClass.val;
      }
    }));

    assertTrue("String value should be extract from GenericMessage", "A".equals(pf.match(new GenericClass<String>("A"))));
  }


  @Test
  public void shouldHandleMatchWithPredicateOnGenericClass() {
    Match<Object, String> pf = Match.create(Match.matchUnchecked(GenericClass.class, new FI.TypedPredicate<GenericClass<String>>() {
      @Override
      public boolean defined(GenericClass<String> genericClass) {
        return !genericClass.val.isEmpty();
      }
    }, new FI.Apply<GenericClass<String>, String>() {
      @Override
      public String apply(GenericClass<String> stringGenericClass) {
        return stringGenericClass.val;
      }
    }));

    exception.expect(MatchError.class);
    assertTrue("empty GenericMessage should throw match error", "".equals(pf.match(new GenericClass<String>(""))));
  }
}