Skip to content

Commit 10315fa

Browse files
committed
Fix unit test
1 parent c471e28 commit 10315fa

1 file changed

Lines changed: 14 additions & 7 deletions

File tree

proxy/src/test/java/org/apache/rocketmq/proxy/grpc/ProxyAndTlsProtocolNegotiatorTest.java

Lines changed: 14 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -20,9 +20,12 @@
2020
import io.grpc.netty.shaded.io.netty.buffer.ByteBuf;
2121
import io.grpc.netty.shaded.io.netty.buffer.Unpooled;
2222
import io.grpc.netty.shaded.io.netty.handler.codec.haproxy.HAProxyTLV;
23-
import java.net.URISyntaxException;
23+
import java.io.File;
24+
import java.io.IOException;
25+
import java.io.InputStream;
2426
import java.nio.charset.StandardCharsets;
25-
import java.nio.file.Paths;
27+
import java.nio.file.Files;
28+
import java.nio.file.StandardCopyOption;
2629
import org.apache.rocketmq.proxy.config.ConfigurationManager;
2730
import org.apache.rocketmq.proxy.config.ProxyConfig;
2831
import org.junit.After;
@@ -78,17 +81,21 @@ public void testLoadSslContextWithWrongPassword() throws Exception {
7881
ProxyAndTlsProtocolNegotiator.loadSslContext();
7982
}
8083

81-
private void configureTls(String keyFile, String certFile, String password) throws URISyntaxException {
84+
private void configureTls(String keyFile, String certFile, String password) throws IOException {
8285
ProxyConfig proxyConfig = ConfigurationManager.getProxyConfig();
8386
proxyConfig.setTlsTestModeEnable(false);
8487
proxyConfig.setTlsKeyPath(getCertsPath(keyFile));
8588
proxyConfig.setTlsCertPath(getCertsPath(certFile));
8689
proxyConfig.setTlsKeyPassword(password);
8790
}
8891

89-
private static String getCertsPath(String fileName) throws URISyntaxException {
90-
return Paths.get(ProxyAndTlsProtocolNegotiatorTest.class
91-
.getClassLoader().getResource("certs/" + fileName).toURI())
92-
.toAbsolutePath().toString();
92+
private static String getCertsPath(String fileName) throws IOException {
93+
File tempFile = File.createTempFile(fileName, null);
94+
tempFile.deleteOnExit();
95+
try (InputStream is = ProxyAndTlsProtocolNegotiatorTest.class
96+
.getClassLoader().getResourceAsStream("certs/" + fileName)) {
97+
Files.copy(is, tempFile.toPath(), StandardCopyOption.REPLACE_EXISTING);
98+
}
99+
return tempFile.getAbsolutePath();
93100
}
94101
}

0 commit comments

Comments
 (0)