Skipping setting user agent info if it is empty (#4686)

* Skipping setting user agent info if it is empty * fixing assemblyinfo
Azure · Aug 29, 2018 · 804e6f7 · 804e6f7
1 parent 67b2534
commit 804e6f7
Show file tree

Hide file tree

Showing 4 changed files with 23 additions and 5 deletions.
diff --git a/src/SdkCommon/ClientRuntime/ClientRuntime.Tests/ServiceClientTests.cs b/src/SdkCommon/ClientRuntime/ClientRuntime.Tests/ServiceClientTests.cs
@@ -24,6 +24,23 @@ public void ClientAddHandlerToPipelineAddsHandler()
             Assert.Equal(HttpStatusCode.InternalServerError, result2.StatusCode);
         }
 
+        [Fact]
+        public void ClientDefaultHeaderValuesTest()
+        {
+            var fakeClient = new FakeServiceClient(new HttpClientHandler(), new BadResponseDelegatingHandler());
+            var arr = fakeClient.HttpClient.DefaultRequestHeaders.UserAgent.Where(pihv => String.IsNullOrWhiteSpace(pihv.Product.Version)).ToArray();
+            Assert.Empty(arr);
+        }
+
+        [Fact]
+        public void ClientEmptyProductHeaderValuesTest()
+        {
+            var fakeClient = new FakeServiceClient(new HttpClientHandler(), new BadResponseDelegatingHandler());
+            fakeClient.SetUserAgent("MySpecialHeader", string.Empty);
+            var arr = fakeClient.HttpClient.DefaultRequestHeaders.UserAgent.Where(pihv => (pihv.Product.Name == "MySpecialHeader")).ToArray();
+            Assert.Empty(arr);
+        }
+
         [Fact]
         public void ClientAddHandlersToPipelineAddSingleHandler()
         {

diff --git a/src/SdkCommon/ClientRuntime/ClientRuntime/Microsoft.Rest.ClientRuntime.csproj b/src/SdkCommon/ClientRuntime/ClientRuntime/Microsoft.Rest.ClientRuntime.csproj
@@ -4,12 +4,12 @@
     <Description>Infrastructure for error handling, tracing, and HttpClient pipeline configuration. Required by client libraries generated using AutoRest.</Description>
     <AssemblyName>Microsoft.Rest.ClientRuntime</AssemblyName>
     <AssemblyTitle>Client Runtime Library for Microsoft AutoRest Generated Clients</AssemblyTitle>
-    <Version>2.3.13</Version>
+    <Version>2.3.14</Version>
     <PackageId>Microsoft.Rest.ClientRuntime</PackageId>
     <PackageTags>Microsoft AutoRest ClientRuntime $(NugetCommonTags) $(NugetCommonProfileTags)</PackageTags>
     <PackageReleaseNotes>
       <![CDATA[
-        1) Added a basic support around retrying requests on getting 429 from resource provider
+        1) Skipping empty product header values when sending requests
       ]]>
     </PackageReleaseNotes>
   </PropertyGroup>

diff --git a/src/SdkCommon/ClientRuntime/ClientRuntime/Properties/AssemblyInfo.cs b/src/SdkCommon/ClientRuntime/ClientRuntime/Properties/AssemblyInfo.cs
@@ -9,7 +9,7 @@
 [assembly: AssemblyTitle("Microsoft Rest Client Runtime")]
 [assembly: AssemblyDescription("Client infrastructure for client libraries generated by AutoRest.")]
 [assembly: AssemblyVersion("2.0.0.0")]
-[assembly: AssemblyFileVersion("2.3.13.0")]
+[assembly: AssemblyFileVersion("2.3.14.0")]
 
 [assembly: AssemblyConfiguration("")]
 [assembly: AssemblyCompany("Microsoft Corporation")]

diff --git a/src/SdkCommon/ClientRuntime/ClientRuntime/ServiceClient.cs b/src/SdkCommon/ClientRuntime/ClientRuntime/ServiceClient.cs
@@ -484,7 +484,7 @@ public bool SetUserAgent(string productName)
         /// <param name="version">Version of the product to be used in the user agent</param>
         public bool SetUserAgent(string productName, string version)
         {
-            if (!_disposed && HttpClient != null)
+            if (!_disposed && HttpClient != null && !string.IsNullOrWhiteSpace(version))
             {
                 string cleanedProductName = CleanUserAgentInfoEntry(productName);
                 AddUserAgentEntry(new ProductInfoHeaderValue(cleanedProductName, version));
@@ -527,7 +527,8 @@ private void MergeUserAgentInfo(List<ProductInfoHeaderValue> defaultUserAgentInf
             // If you want to log ProductName in userAgent, it has to be without spaces
             foreach(ProductInfoHeaderValue piHv in defaultUserAgentInfoList)
             {
-                if(!HttpClient.DefaultRequestHeaders.UserAgent.Any<ProductInfoHeaderValue>((hv) => hv.Product.Name.Equals(piHv.Product.Name, StringComparison.OrdinalIgnoreCase)))
+                if(!HttpClient.DefaultRequestHeaders.UserAgent.Any<ProductInfoHeaderValue>((hv) => hv.Product.Name.Equals(piHv.Product.Name, StringComparison.OrdinalIgnoreCase)) 
+                    && !string.IsNullOrWhiteSpace(piHv.Product.Version))
                 {
                     HttpClient.DefaultRequestHeaders.UserAgent.Add(piHv);
                 }