mirror of
https://github.com/vanitasvitae/Smack.git
synced 2024-12-24 03:27:59 +01:00
Add unit test for smack-resolver-minidns-dox
This commit is contained in:
parent
a1c88f1fad
commit
7fd0676ff4
3 changed files with 174 additions and 4 deletions
|
@ -4,4 +4,5 @@ DNS over XMPP (DoX) support using MiniDNS."""
|
|||
dependencies {
|
||||
compile project(path: ':smack-resolver-minidns')
|
||||
compile project(path: ':smack-experimental')
|
||||
testCompile project(path: ":smack-core", configuration: "testRuntime")
|
||||
}
|
||||
|
|
|
@ -29,12 +29,14 @@ import org.minidns.dnssec.DnssecQueryResult;
|
|||
|
||||
public final class DnsOverXmppMiniDnsResolver implements DnsOverXmppResolver {
|
||||
|
||||
public static final DnsOverXmppMiniDnsResolver INSTANCE = new DnsOverXmppMiniDnsResolver();
|
||||
public static final DnsOverXmppMiniDnsResolver INSTANCE = new DnsOverXmppMiniDnsResolver(new DnsClient(), new DnssecClient());
|
||||
|
||||
private final DnsClient dnsClient = new DnsClient();
|
||||
private final DnssecClient dnssecClient = new DnssecClient();
|
||||
private final DnsClient dnsClient;
|
||||
private final DnssecClient dnssecClient;
|
||||
|
||||
private DnsOverXmppMiniDnsResolver() {
|
||||
DnsOverXmppMiniDnsResolver(DnsClient dnsClient, DnssecClient dnssecClient) {
|
||||
this.dnsClient = dnsClient;
|
||||
this.dnssecClient = dnssecClient;
|
||||
}
|
||||
|
||||
@Override
|
||||
|
|
|
@ -0,0 +1,167 @@
|
|||
/**
|
||||
*
|
||||
* Copyright 2019 Florian Schmaus
|
||||
*
|
||||
* Licensed under the Apache License, Version 2.0 (the "License");
|
||||
* you may not use this file except in compliance with the License.
|
||||
* You may obtain a copy of the License at
|
||||
*
|
||||
* http://www.apache.org/licenses/LICENSE-2.0
|
||||
*
|
||||
* Unless required by applicable law or agreed to in writing, software
|
||||
* distributed under the License is distributed on an "AS IS" BASIS,
|
||||
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||
* See the License for the specific language governing permissions and
|
||||
* limitations under the License.
|
||||
*/
|
||||
package org.jivesoftware.smackx.dox.resolver.minidns;
|
||||
|
||||
import static org.junit.Assert.assertFalse;
|
||||
import static org.junit.Assert.assertNotNull;
|
||||
import static org.junit.Assert.assertTrue;
|
||||
|
||||
import java.io.IOException;
|
||||
import java.net.InetAddress;
|
||||
import java.util.concurrent.atomic.AtomicBoolean;
|
||||
|
||||
import org.junit.Test;
|
||||
import org.minidns.DnsCache;
|
||||
import org.minidns.DnsClient;
|
||||
import org.minidns.MiniDnsFuture;
|
||||
import org.minidns.dnsmessage.DnsMessage;
|
||||
import org.minidns.dnsmessage.DnsMessage.RESPONSE_CODE;
|
||||
import org.minidns.dnsmessage.Question;
|
||||
import org.minidns.dnsname.DnsName;
|
||||
import org.minidns.dnsqueryresult.CachedDnsQueryResult;
|
||||
import org.minidns.dnsqueryresult.DnsQueryResult;
|
||||
import org.minidns.dnssec.DnssecClient;
|
||||
import org.minidns.dnssec.DnssecValidationFailedException;
|
||||
import org.minidns.record.Record.TYPE;
|
||||
import org.minidns.source.DnsDataSource;
|
||||
|
||||
public final class DnsOverXmppMiniDnsResolverTest {
|
||||
|
||||
@Test
|
||||
public void dnsOverXmppMiniDnsResolverTest() throws IOException {
|
||||
TestDnsDataSource dnsSource = new TestDnsDataSource();
|
||||
TestDnsDataSource dnssecSource = new TestDnsDataSource();
|
||||
|
||||
DnsClient dnsClient = new DnsClient(NoopDnsCache.INSTANCE);
|
||||
dnsClient.setDataSource(dnsSource);
|
||||
|
||||
DnssecClient dnssecClient = new DnssecClient(NoopDnsCache.INSTANCE);
|
||||
dnssecClient.setDataSource(dnssecSource);
|
||||
|
||||
DnsOverXmppMiniDnsResolver doxResolver = new DnsOverXmppMiniDnsResolver(dnsClient, dnssecClient);
|
||||
|
||||
Question question = new Question("example.org", TYPE.A);
|
||||
|
||||
{
|
||||
DnsMessage nondnssecQuery = question.asQueryMessage();
|
||||
|
||||
doxResolver.resolve(nondnssecQuery);
|
||||
|
||||
assertTrue(dnsSource.getAndResetWasQueried());
|
||||
assertFalse(dnssecSource.getAndResetWasQueried());
|
||||
}
|
||||
|
||||
{
|
||||
DnsMessage.Builder dnssecQueryBuilder = question.asMessageBuilder();
|
||||
dnssecQueryBuilder.getEdnsBuilder().setDnssecOk();
|
||||
DnsMessage dnssecQuery = dnssecQueryBuilder.build();
|
||||
|
||||
DnssecValidationFailedException dnssecValidationFailedException = null;
|
||||
try {
|
||||
doxResolver.resolve(dnssecQuery);
|
||||
} catch (DnssecValidationFailedException e) {
|
||||
dnssecValidationFailedException = e;
|
||||
}
|
||||
// This exception is expected since we don't have a realy DNS source.
|
||||
assertNotNull(dnssecValidationFailedException);
|
||||
|
||||
assertFalse(dnsSource.getAndResetWasQueried());
|
||||
assertTrue(dnssecSource.getAndResetWasQueried());
|
||||
}
|
||||
}
|
||||
|
||||
public static class TestDnsDataSource implements DnsDataSource {
|
||||
|
||||
private final AtomicBoolean wasQueried = new AtomicBoolean();
|
||||
|
||||
public boolean getAndResetWasQueried() {
|
||||
return wasQueried.getAndSet(false);
|
||||
}
|
||||
|
||||
private void setWasQueried() {
|
||||
wasQueried.set(true);
|
||||
}
|
||||
|
||||
@Override
|
||||
public DnsQueryResult query(DnsMessage query, InetAddress address, int port) throws IOException {
|
||||
setWasQueried();
|
||||
return new TestDnsQueryResult(query);
|
||||
}
|
||||
|
||||
@Override
|
||||
public MiniDnsFuture<DnsQueryResult, IOException> queryAsync(DnsMessage query, InetAddress address, int port,
|
||||
OnResponseCallback onResponseCallback) {
|
||||
setWasQueried();
|
||||
DnsQueryResult result = new TestDnsQueryResult(query);
|
||||
return MiniDnsFuture.from(result);
|
||||
}
|
||||
|
||||
@Override
|
||||
public int getUdpPayloadSize() {
|
||||
return 0;
|
||||
}
|
||||
|
||||
@Override
|
||||
public int getTimeout() {
|
||||
return 0;
|
||||
}
|
||||
|
||||
@Override
|
||||
public void setTimeout(int timeout) {
|
||||
}
|
||||
|
||||
private static class TestDnsQueryResult extends DnsQueryResult {
|
||||
|
||||
protected TestDnsQueryResult(DnsMessage query) {
|
||||
super(QueryMethod.testWorld, query, createNxDomainAnswerFor(query));
|
||||
}
|
||||
|
||||
private static DnsMessage createNxDomainAnswerFor(DnsMessage query) {
|
||||
Question question = query.getQuestion();
|
||||
|
||||
DnsMessage response = DnsMessage.builder()
|
||||
.setQuestion(question)
|
||||
.setRecursionAvailable(true)
|
||||
.setResponseCode(RESPONSE_CODE.NX_DOMAIN)
|
||||
.build();
|
||||
|
||||
return response;
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
// TODO: Workaround for NPE-if-no-cache-set bug in MiniDNS. Remove we use a MiniDNS version where this is fixed,
|
||||
// i.e. one that has 864fbb5 ("Fix NPE in AbstractDnsClient if cache is 'null'")
|
||||
private static class NoopDnsCache extends DnsCache {
|
||||
|
||||
private static final NoopDnsCache INSTANCE = new NoopDnsCache();
|
||||
|
||||
@Override
|
||||
protected void putNormalized(DnsMessage normalizedQuery, DnsQueryResult result) {
|
||||
}
|
||||
|
||||
@Override
|
||||
public void offer(DnsMessage query, DnsQueryResult result, DnsName authoritativeZone) {
|
||||
}
|
||||
|
||||
@Override
|
||||
protected CachedDnsQueryResult getNormalized(DnsMessage normalizedQuery) {
|
||||
return null;
|
||||
}
|
||||
|
||||
}
|
||||
}
|
Loading…
Reference in a new issue