Add unit test for PrivateDnsConfig
The unit test is added to ensure the conversion between PrivateDnsConfig
and PrivateDnsConfigParcel.
Bug: 240259333
Bug: 261404136
Test: atest NetworkStackTests:PrivateDnsConfigTest
Change-Id: I9a03d3b4c07fd3e15720c9fc3f47b3389c7ec029
diff --git a/tests/unit/src/android/net/shared/PrivateDnsConfigTest.java b/tests/unit/src/android/net/shared/PrivateDnsConfigTest.java
new file mode 100644
index 0000000..c9b66ab
--- /dev/null
+++ b/tests/unit/src/android/net/shared/PrivateDnsConfigTest.java
@@ -0,0 +1,126 @@
+/*
+ * Copyright (C) 2023 The Android Open Source Project
+ *
+ * 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 android.net.shared;
+
+import static android.net.ConnectivitySettingsManager.PRIVATE_DNS_MODE_OFF;
+import static android.net.ConnectivitySettingsManager.PRIVATE_DNS_MODE_OPPORTUNISTIC;
+import static android.net.ConnectivitySettingsManager.PRIVATE_DNS_MODE_PROVIDER_HOSTNAME;
+
+import static org.junit.Assert.assertArrayEquals;
+import static org.junit.Assert.assertEquals;
+import static org.junit.Assert.assertFalse;
+import static org.junit.Assert.assertTrue;
+import static org.junit.Assert.fail;
+
+import android.net.PrivateDnsConfigParcel;
+
+import org.junit.Test;
+import org.junit.runner.RunWith;
+import org.junit.runners.JUnit4;
+
+import java.net.InetAddress;
+
+@RunWith(JUnit4.class)
+public final class PrivateDnsConfigTest {
+ private static final InetAddress[] TEST_ADDRS = new InetAddress[] {
+ InetAddress.parseNumericAddress("1.2.3.4"),
+ InetAddress.parseNumericAddress("2001:db8::2"),
+ };
+
+ private String[] toStringArray(InetAddress[] ips) {
+ String[] out = new String[ips.length];
+ int i = 0;
+ for (InetAddress ip : ips) {
+ out[i++] = ip.getHostAddress();
+ }
+ return out;
+ }
+
+ private void assertPrivateDnsConfigEquals(PrivateDnsConfig a, PrivateDnsConfig b) {
+ assertEquals(a.useTls, b.useTls);
+ assertEquals(a.hostname, b.hostname);
+ assertArrayEquals(a.ips, b.ips);
+ assertEquals(a.dohName, b.dohName);
+ assertArrayEquals(a.dohIps, b.dohIps);
+ assertEquals(a.dohPath, b.dohPath);
+ assertEquals(a.dohPort, b.dohPort);
+ }
+
+ private void assertParcelEquals(PrivateDnsConfig cfg, PrivateDnsConfigParcel parcel) {
+ assertEquals(parcel.hostname, cfg.hostname);
+ assertArrayEquals(parcel.ips, toStringArray(cfg.ips));
+ switch (parcel.privateDnsMode) {
+ case PRIVATE_DNS_MODE_OFF:
+ assertFalse(cfg.useTls);
+ assertTrue(cfg.hostname.isEmpty());
+ break;
+ case PRIVATE_DNS_MODE_OPPORTUNISTIC:
+ assertTrue(cfg.useTls);
+ assertTrue(cfg.hostname.isEmpty());
+ break;
+ case PRIVATE_DNS_MODE_PROVIDER_HOSTNAME:
+ assertTrue(cfg.useTls);
+ assertFalse(cfg.hostname.isEmpty());
+ break;
+ default:
+ fail("Unexpected private DNS mode");
+ }
+ assertEquals(parcel.dohName, cfg.dohName);
+ assertEquals(parcel.dohPath, cfg.dohPath);
+ assertEquals(parcel.dohPort, cfg.dohPort);
+ assertArrayEquals(parcel.dohIps, toStringArray(cfg.dohIps));
+ }
+
+ // Tests both toParcel() and fromParcel() together.
+ private void testPrivateDnsConfigConversion(PrivateDnsConfig cfg) {
+ final PrivateDnsConfigParcel parcel = cfg.toParcel();
+ assertParcelEquals(cfg, parcel);
+
+ final PrivateDnsConfig convertedCfg = PrivateDnsConfig.fromParcel(parcel);
+ assertPrivateDnsConfigEquals(cfg, convertedCfg);
+ }
+
+ @Test
+ public void testParcelableConversion() {
+ // Test the constructor: PrivateDnsConfig()
+ testPrivateDnsConfigConversion(new PrivateDnsConfig());
+
+ // Test the constructor: PrivateDnsConfig(boolean useTls)
+ testPrivateDnsConfigConversion(new PrivateDnsConfig(true));
+ testPrivateDnsConfigConversion(new PrivateDnsConfig(false));
+
+ // Test the constructor: PrivateDnsConfig(String hostname, InetAddress[] ips)
+ testPrivateDnsConfigConversion(new PrivateDnsConfig(null, null));
+ testPrivateDnsConfigConversion(new PrivateDnsConfig(null, TEST_ADDRS));
+ testPrivateDnsConfigConversion(new PrivateDnsConfig("dns.com", null));
+ testPrivateDnsConfigConversion(new PrivateDnsConfig("dns.com", TEST_ADDRS));
+
+ // Test the constructor:
+ // PrivateDnsConfig(boolean useTls, String hostname, InetAddress[] ips, String dohName,
+ // InetAddress[] dohIps, String dohPath, int dohPort)
+ testPrivateDnsConfigConversion(new PrivateDnsConfig(true, "dns.com", TEST_ADDRS, null,
+ null, null, -1));
+ testPrivateDnsConfigConversion(new PrivateDnsConfig(true, "dns.com", TEST_ADDRS, "doh.com",
+ null, null, -1));
+ testPrivateDnsConfigConversion(new PrivateDnsConfig(true, "dns.com", TEST_ADDRS, "doh.com",
+ TEST_ADDRS, null, -1));
+ testPrivateDnsConfigConversion(new PrivateDnsConfig(true, "dns.com", TEST_ADDRS, "doh.com",
+ TEST_ADDRS, "dohpath=/some-path{?dns}", -1));
+ testPrivateDnsConfigConversion(new PrivateDnsConfig(true, "dns.com", TEST_ADDRS, "doh.com",
+ TEST_ADDRS, "dohpath=/some-path{?dns}", 443));
+ }
+}
