adideshpande commented on code in PR #7379:
URL: https://github.com/apache/hadoop/pull/7379#discussion_r2098515721
##########
hadoop-tools/hadoop-aws/src/test/java/org/apache/hadoop/fs/s3a/impl/TestAwsClientConfig.java:
##########
@@ -201,4 +208,64 @@ public void testCreateApiConnectionSettingsDefault() {
private void setOptionsToValue(String value, Configuration conf, String...
keys) {
Arrays.stream(keys).forEach(key -> conf.set(key, value));
}
+
+ /**
+ * if {@link org.apache.hadoop.fs.s3a.Constants#CUSTOM_HEADERS_STS} is set,
+ * verify that returned client configuration has desired headers set.
+ */
+ @Test
+ public void testInitRequestHeadersForSTS() throws IOException {
+ final Configuration conf = new Configuration();
+ conf.set(CUSTOM_HEADERS_STS, "foo=bar;baz,qux=quux");
+ Assertions.assertThat(conf.get(CUSTOM_HEADERS_S3))
+ .describedAs("Custom client headers for s3 %s", CUSTOM_HEADERS_S3)
+ .isNull();
+
+ Assertions.assertThat(createClientConfigBuilder(conf,
AWS_SERVICE_IDENTIFIER_S3)
+ .headers().size())
+ .describedAs("Count of S3 client headers")
+ .isEqualTo(0);
+ Assertions.assertThat(createClientConfigBuilder(conf,
AWS_SERVICE_IDENTIFIER_STS)
+ .headers().size())
+ .describedAs("Count of STS client headers")
+ .isEqualTo(2);
+ Assertions.assertThat(createClientConfigBuilder(conf,
AWS_SERVICE_IDENTIFIER_STS)
+ .headers().get("foo"))
+ .describedAs("STS client 'foo' header value")
+ .isEqualTo(Lists.newArrayList("bar", "baz"));
+ Assertions.assertThat(createClientConfigBuilder(conf,
AWS_SERVICE_IDENTIFIER_STS)
+ .headers().get("qux"))
+ .describedAs("STS client 'qux' header value")
+ .isEqualTo(Lists.newArrayList("quux"));
+ }
+
+ /**
+ * if {@link org.apache.hadoop.fs.s3a.Constants#CUSTOM_HEADERS_S3} is set,
+ * verify that returned client configuration has desired headers set.
+ */
+ @Test
+ public void testInitRequestHeadersForS3() throws IOException {
+ final Configuration conf = new Configuration();
+ conf.set(CUSTOM_HEADERS_S3, "foo=bar;baz,qux=quux");
Review Comment:
Thanks for the suggestion. It's a good idea to have the header values
trimmed and headers without any values should be skipped. So I have updated the
logic and unit tests accordingly.
--
This is an automated message from the Apache Git Service.
To respond to the message, please log on to GitHub and use the
URL above to go to the specific comment.
To unsubscribe, e-mail: common-issues-unsubscr...@hadoop.apache.org
For queries about this service, please contact Infrastructure at:
us...@infra.apache.org
---------------------------------------------------------------------
To unsubscribe, e-mail: common-issues-unsubscr...@hadoop.apache.org
For additional commands, e-mail: common-issues-h...@hadoop.apache.org
