|
@@ -28,6 +28,7 @@ import org.apache.hadoop.fs.azurebfs.AbfsConfiguration;
|
|
|
import org.apache.hadoop.fs.azurebfs.utils.SSLSocketFactoryEx;
|
|
|
import org.apache.hadoop.conf.Configuration;
|
|
|
import org.apache.hadoop.fs.azurebfs.constants.ConfigurationKeys;
|
|
|
+import org.apache.hadoop.util.VersionInfo;
|
|
|
|
|
|
/**
|
|
|
* Test useragent of abfs client.
|
|
@@ -49,12 +50,15 @@ public final class TestAbfsClient {
|
|
|
}
|
|
|
String userAgent = client.initializeUserAgent(config, sslProviderName);
|
|
|
Pattern pattern = Pattern.compile(expectedPattern);
|
|
|
- Assert.assertTrue(pattern.matcher(userAgent).matches());
|
|
|
+ Assert.assertTrue("Incorrect User Agent String",
|
|
|
+ pattern.matcher(userAgent).matches());
|
|
|
}
|
|
|
|
|
|
@Test
|
|
|
public void verifyUnknownUserAgent() throws Exception {
|
|
|
- String expectedUserAgentPattern = "Azure Blob FS\\/1.0 \\(JavaJRE ([^\\)]+)\\)";
|
|
|
+ String clientVersion = "Azure Blob FS/" + VersionInfo.getVersion();
|
|
|
+ String expectedUserAgentPattern = String.format(clientVersion
|
|
|
+ + " %s", "\\(JavaJRE ([^\\)]+)\\)");
|
|
|
final Configuration configuration = new Configuration();
|
|
|
configuration.unset(ConfigurationKeys.FS_AZURE_USER_AGENT_PREFIX_KEY);
|
|
|
AbfsConfiguration abfsConfiguration = new AbfsConfiguration(configuration, accountName);
|
|
@@ -64,7 +68,9 @@ public final class TestAbfsClient {
|
|
|
|
|
|
@Test
|
|
|
public void verifyUserAgent() throws Exception {
|
|
|
- String expectedUserAgentPattern = "Azure Blob FS\\/1.0 \\(JavaJRE ([^\\)]+)\\) Partner Service";
|
|
|
+ String clientVersion = "Azure Blob FS/" + VersionInfo.getVersion();
|
|
|
+ String expectedUserAgentPattern = String.format(clientVersion
|
|
|
+ + " %s", "\\(JavaJRE ([^\\)]+)\\) Partner Service");
|
|
|
final Configuration configuration = new Configuration();
|
|
|
configuration.set(ConfigurationKeys.FS_AZURE_USER_AGENT_PREFIX_KEY, "Partner Service");
|
|
|
AbfsConfiguration abfsConfiguration = new AbfsConfiguration(configuration, accountName);
|
|
@@ -74,7 +80,9 @@ public final class TestAbfsClient {
|
|
|
|
|
|
@Test
|
|
|
public void verifyUserAgentWithSSLProvider() throws Exception {
|
|
|
- String expectedUserAgentPattern = "Azure Blob FS\\/1.0 \\(JavaJRE ([^\\)]+) SunJSSE-1.8\\) Partner Service";
|
|
|
+ String clientVersion = "Azure Blob FS/" + VersionInfo.getVersion();
|
|
|
+ String expectedUserAgentPattern = String.format(clientVersion
|
|
|
+ + " %s", "\\(JavaJRE ([^\\)]+)\\) Partner Service");
|
|
|
final Configuration configuration = new Configuration();
|
|
|
configuration.set(ConfigurationKeys.FS_AZURE_USER_AGENT_PREFIX_KEY, "Partner Service");
|
|
|
configuration.set(ConfigurationKeys.FS_AZURE_SSL_CHANNEL_MODE_KEY,
|