TEZ-4389: TestSecureShuffle fails if localhost name is different than…

… "localhost" (#191) (Mark Bathori reviewed by Laszlo Bodor)
apache · Feb 18, 2022 · a495fb6 · a495fb6
1 parent f62bf12
commit a495fb6
Showing 1 changed file with 7 additions and 2 deletions.
diff --git a/tez-tests/src/test/java/org/apache/tez/test/TestSecureShuffle.java b/tez-tests/src/test/java/org/apache/tez/test/TestSecureShuffle.java
@@ -302,10 +302,15 @@ public static X509Certificate generateCertificate(String dn, KeyPair pair, int d
     KeyPair keyPair = pair;
     X509V3CertificateGenerator certGen = new X509V3CertificateGenerator();
 
+    String hostName = InetAddress.getLocalHost().getHostName();
     String hostAddress = InetAddress.getLocalHost().getHostAddress();
     certGen.addExtension(X509Extensions.SubjectAlternativeName, false,
-        new GeneralNames(new GeneralName[] { new GeneralName(GeneralName.iPAddress, hostAddress),
-            new GeneralName(GeneralName.dNSName, "localhost") }));
+        new GeneralNames(new GeneralName[] {
+            new GeneralName(GeneralName.iPAddress, hostAddress),
+            new GeneralName(GeneralName.dNSName, hostName),
+            new GeneralName(GeneralName.dNSName, "localhost")
+        })
+    );
     X500Principal dnName = new X500Principal(dn);
 
     certGen.setSerialNumber(sn);