Merge branch 'jws305-master'

Author: russcam

src/Nest/Aggregations/AggregateDictionary.cs

@@ -209,8 +209,7 @@ public CompositeBucketAggregate Composite(string key)
 
 		public ValueAggregate MedianAbsoluteDeviation(string key) => TryGet<ValueAggregate>(key);
 
-		private TAggregate TryGet<TAggregate>(string key)
-			where TAggregate : class, IAggregate =>
+		private TAggregate TryGet<TAggregate>(string key) where TAggregate : class, IAggregate =>
 			BackingDictionary.TryGetValue(key, out var agg) ? agg as TAggregate : null;
 
 		private MultiBucketAggregate<TBucket> GetMultiBucketAggregate<TBucket>(string key)
@@ -238,19 +237,23 @@ private MultiBucketAggregate<KeyedBucket<TKey>> GetMultiKeyedBucketAggregate<TKe
 			};
 		}
 
-
 		private IEnumerable<KeyedBucket<TKey>> GetKeyedBuckets<TKey>(IEnumerable<IBucket> items)
 		{
 			var buckets = items.Cast<KeyedBucket<object>>();
 
 			foreach (var bucket in buckets)
 				yield return new KeyedBucket<TKey>(bucket.BackingDictionary)
 				{
-					Key = (TKey)Convert.ChangeType(bucket.Key, typeof(TKey)),
+					Key = GetKeyFromBucketKey<TKey>(bucket.Key),
 					KeyAsString = bucket.KeyAsString,
 					DocCount = bucket.DocCount,
 					DocCountErrorUpperBound = bucket.DocCountErrorUpperBound
 				};
 		}
+
+		private static TKey GetKeyFromBucketKey<TKey>(object key) =>
+			typeof(TKey).IsEnum
+				? (TKey)Enum.Parse(typeof(TKey), key.ToString(), true)
+				: (TKey)Convert.ChangeType(key, typeof(TKey));
 	}
 }

src/Tests/Tests/Aggregations/Bucket/Terms/TermsAggregationUsageTests.cs

@@ -188,15 +188,15 @@ public TermsAggregationIncludePatternUsageTests(ReadOnlyCluster i, EndpointUsage
 		protected override void ExpectResponse(ISearchResponse<Project> response)
 		{
 			response.ShouldBeValid();
-			var states = response.Aggregations.Terms("states");
+			var states = response.Aggregations.Terms<StateOfBeing>("states");
 			states.Should().NotBeNull();
 			states.DocCountErrorUpperBound.Should().HaveValue();
 			states.SumOtherDocCount.Should().HaveValue();
 			states.Buckets.Should().NotBeNull();
 			states.Buckets.Count.Should().BeGreaterThan(0);
 			foreach (var item in states.Buckets)
 			{
-				item.Key.Should().NotBeNullOrEmpty();
+				item.Key.Should().BeOfType<StateOfBeing>();
 				item.DocCount.Should().BeGreaterOrEqualTo(1);
 			}
 			states.Meta.Should().NotBeNull().And.HaveCount(1);