How to use

org.apache.james.dnsservice.api.InMemoryDNSService

constructor

/**
 * Return a fake DNSServer.
 */
protected DNSService getDNSServer(final String hostName) throws UnknownHostException {
  return new InMemoryDNSService()
    .registerMxRecord(hostName, "127.0.0.1")
    .registerMxRecord("127.0.0.1", "127.0.0.1");
}

  @Test
  public void testRejectLoopbackMX() throws Exception {
    String bannedAddress = "172.53.64.2";

    DNSService dns = new InMemoryDNSService()
      .registerMxRecord(INVALID_HOST, bannedAddress)
      .registerMxRecord("255.255.255.255", "255.255.255.255")
      .registerMxRecord(bannedAddress, bannedAddress);
    MailAddress mailAddress = new MailAddress("test@" + INVALID_HOST);
    SMTPSession session = setupMockedSMTPSession(mailAddress);

    ValidRcptMX handler = new ValidRcptMX();
    handler.setDNSService(dns);
    handler.setBannedNetworks(ImmutableList.of(bannedAddress), dns);
    HookReturnCode rCode = handler.doRcpt(session, null, mailAddress).getResult();

    assertThat(HookReturnCode.deny()).describedAs("Reject").isEqualTo(rCode);
  }
}

@Before
public void setup() throws Exception {
  DNSService dnsServer = new InMemoryDNSService()
    .registerMxRecord("192.168.0.1", "192.168.0.1")
    .registerMxRecord("192.168.200.1", "192.168.200.1")
    .registerMxRecord("192.168.200.0", "192.168.200.0")
    .registerMxRecord("255.255.255.0", "255.255.255.0");
  matcherConfig = FakeMatcherConfig.builder()
      .matcherName("AllowedNetworkIs")
      .condition("192.168.200.0/24")
      .build();
  matcher = new RemoteAddrNotInNetwork();
  matcher.setDNSService(dnsServer);
  matcher.init(matcherConfig);
  testRecipient = new MailAddress("test@james.apache.org");
}

@Before
public void setup() throws Exception {
  DNSService dnsServer = new InMemoryDNSService()
    .registerMxRecord("192.168.0.1", "192.168.0.1")
    .registerMxRecord("192.168.200.1", "192.168.200.1")
    .registerMxRecord("192.168.200.0", "192.168.200.0")
    .registerMxRecord("255.255.255.0", "255.255.255.0");
  FakeMatcherConfig matcherConfig = FakeMatcherConfig.builder()
      .matcherName("AllowedNetworkIs")
      .condition("192.168.200.0/24")
      .build();
  matcher = new RemoteAddrInNetwork();
  matcher.setDNSService(dnsServer);
  matcher.init(matcherConfig);
  testRecipient = new MailAddress("test@james.apache.org");
}