Class AddressFilterTest

xds/src/test/java/io/grpc/xds/AddressFilterTest.java:32–86

Tests for AddressFilter.

/** Tests for {@link AddressFilter}. */
@RunWith(JUnit4.class)
public class AddressFilterTest {
  @Test
  public void filterAddresses() {
    Attributes.Key<String> key1 = Attributes.Key.create("key1");
    Attributes attributes1 = Attributes.newBuilder().set(key1, "value1").build();
    EquivalentAddressGroup eag0 = new EquivalentAddressGroup(new InetSocketAddress(8000));
    EquivalentAddressGroup eag1 =
        new EquivalentAddressGroup(new InetSocketAddress(8001), attributes1);
    EquivalentAddressGroup eag2 = new EquivalentAddressGroup(new InetSocketAddress(8002));
    EquivalentAddressGroup eag3 =
        new EquivalentAddressGroup(
            Arrays.<SocketAddress>asList(new InetSocketAddress(8003), new InetSocketAddress(8083)));
    eag0 = AddressFilter.setPathFilter(eag0, Arrays.asList("A", "C"));
    eag1 = AddressFilter.setPathFilter(eag1, Arrays.asList("A", "B"));
    eag2 = AddressFilter.setPathFilter(eag2, Arrays.asList("D", "C"));
    eag3 = AddressFilter.setPathFilter(eag3, Arrays.asList("A", "B"));

    List<EquivalentAddressGroup> addresses =
        AddressFilter.filter(Arrays.asList(eag0, eag1, eag2, eag3), "A");
    assertThat(addresses).hasSize(3);
    addresses = AddressFilter.filter(addresses, "B");
    assertThat(addresses).hasSize(2);
    EquivalentAddressGroup filteredAddress0 = addresses.get(0);
    EquivalentAddressGroup filteredAddress1 = addresses.get(1);
    assertThat(filteredAddress0.getAddresses()).containsExactlyElementsIn(eag1.getAddresses());
    assertThat(filteredAddress0.getAttributes().get(key1)).isEqualTo("value1");
    assertThat(filteredAddress1.getAddresses()).containsExactlyElementsIn(eag3.getAddresses());
  }

  @Test
  public void longerPathChain() {
    List<EquivalentAddressGroup> addresses = Arrays.asList(
        newEag(new InetSocketAddress(8000), Arrays.asList("A", "B", "C")),
        newEag(new InetSocketAddress(8001), Arrays.asList("Z", "B", "C")),
        newEag(new InetSocketAddress(8002), Arrays.asList("A", "Z", "C")),
        newEag(new InetSocketAddress(8003), Arrays.asList("A", "B", "Z")));
    addresses = AddressFilter.filter(addresses, "A");
    assertThat(addresses).hasSize(3);

    addresses = AddressFilter.filter(addresses, "B");
    assertThat(addresses).hasSize(2);

    addresses = AddressFilter.filter(addresses, "C");
    assertThat(addresses).hasSize(1);
    assertThat(addresses.get(0).getAddresses()).containsExactly(new InetSocketAddress(8000));

    addresses = AddressFilter.filter(addresses, "D");
    assertThat(addresses).hasSize(0);
  }

  private static EquivalentAddressGroup newEag(InetSocketAddress address, List<String> names) {
    return AddressFilter.setPathFilter(new EquivalentAddressGroup(address), names);
  }
}