fix TestSslContextFactory

instaclustr · Jun 27, 2024 · 5177e1c · 5177e1c
1 parent 0f0161c
commit 5177e1c
Showing 1 changed file with 3 additions and 4 deletions.
diff --git a/common/src/test/java/com/zegelin/cassandra/exporter/netty/ssl/TestSslContextFactory.java b/common/src/test/java/com/zegelin/cassandra/exporter/netty/ssl/TestSslContextFactory.java
@@ -7,7 +7,6 @@
 import io.netty.handler.ssl.util.SelfSignedCertificate;
 import org.testng.annotations.BeforeMethod;
 import org.testng.annotations.Test;
-import sun.security.ssl.SSLEngineImpl;
 
 import java.io.File;
 import java.io.IOException;
@@ -46,7 +45,7 @@ public void testCreateJdkSslContext() {
 
         SslContext context = contextFactory.createSslContext();
 
-        assertThat(context.newEngine(ByteBufAllocator.DEFAULT)).isInstanceOf(SSLEngineImpl.class);
+        assertThat(context.newEngine(ByteBufAllocator.DEFAULT).getClass().getSimpleName()).isEqualTo("SSLEngineImpl");
     }
 
     @Test
@@ -163,7 +162,7 @@ public void testCreateSslContextWithServerKeyAndCert() {
 
         SslContext context = contextFactory.createSslContext();
 
-        assertThat(context.newEngine(ByteBufAllocator.DEFAULT)).isInstanceOf(SSLEngineImpl.class);
+        assertThat(context.newEngine(ByteBufAllocator.DEFAULT).getClass().getSimpleName()).isEqualTo("SSLEngineImpl");
     }
 
     @Test
@@ -175,7 +174,7 @@ public void testCreateSslContextWithServerKeyAndCertWithPassword() {
 
         SslContext context = contextFactory.createSslContext();
 
-        assertThat(context.newEngine(ByteBufAllocator.DEFAULT)).isInstanceOf(SSLEngineImpl.class);
+        assertThat(context.newEngine(ByteBufAllocator.DEFAULT).getClass().getSimpleName()).isEqualTo("SSLEngineImpl");
     }
 
     @Test