pekko/akka-stream-tests/src/test/java/akka/stream/javadsl/AttributesTest.java

79 lines
2 KiB
Java

/**
* Copyright (C) 2015-2016 Lightbend Inc. <http://www.lightbend.com>
*/
package akka.stream.javadsl;
import static org.junit.Assert.assertEquals;
import java.util.Arrays;
import java.util.Collections;
import java.util.Optional;
import org.junit.ClassRule;
import org.junit.Test;
import akka.stream.Attributes;
import akka.stream.StreamTest;
import akka.testkit.AkkaSpec;
public class AttributesTest extends StreamTest {
public AttributesTest() {
super(actorSystemResource);
}
@ClassRule
public static AkkaJUnitActorSystemResource actorSystemResource = new AkkaJUnitActorSystemResource("AttributesTest",
AkkaSpec.testConf());
final Attributes attributes =
Attributes.name("a")
.and(Attributes.name("b"))
.and(Attributes.inputBuffer(1, 2));
@Test
public void mustGetAttributesByClass() {
assertEquals(
Arrays.asList(new Attributes.Name("a"), new Attributes.Name("b")),
attributes.getAttributeList(Attributes.Name.class));
assertEquals(
Collections.singletonList(new Attributes.InputBuffer(1, 2)),
attributes.getAttributeList(Attributes.InputBuffer.class));
}
@Test
public void mustGetAttributeByClass() {
assertEquals(
new Attributes.Name("b"),
attributes.getAttribute(Attributes.Name.class, new Attributes.Name("default")));
}
@Test
public void mustGetMissingAttributeByClass() {
assertEquals(
Optional.empty(),
attributes.getAttribute(Attributes.LogLevels.class));
}
@Test
public void mustGetPossiblyMissingAttributeByClass() {
assertEquals(
Optional.of(new Attributes.Name("b")),
attributes.getAttribute(Attributes.Name.class));
}
@Test
public void mustGetPossiblyMissingFirstAttributeByClass() {
assertEquals(
Optional.of(new Attributes.Name("a")),
attributes.getFirstAttribute(Attributes.Name.class));
}
@Test
public void mustGetMissingFirstAttributeByClass() {
assertEquals(
Optional.empty(),
attributes.getFirstAttribute(Attributes.LogLevels.class));
}
}