/** * 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"); }