private IEnumerable<DataSample> BucketedDataQuery(Counter counter, DimensionSpecification queryParams, int expectedCount = 1) { if (queryParams != null) { queryParams[ReservedDimensions.AggregateSamplesDimension] = "false"; } counter.DataSet.Flush(); var queryResult = counter.Query(queryParams); if (queryResult == null) { return null; } var bucketedSamplesList = counter.Query(queryParams).ToList(); Assert.AreEqual(expectedCount, bucketedSamplesList.Count); queryParams.Remove(ReservedDimensions.AggregateSamplesDimension); return bucketedSamplesList; }
private DataSample CombinedDataQuery(Counter counter, DimensionSpecification queryParams) { if (queryParams != null) { queryParams[ReservedDimensions.AggregateSamplesDimension] = "true"; } counter.DataSet.Flush(); var queryResults = counter.Query(queryParams).ToList(); queryParams.Remove(ReservedDimensions.AggregateSamplesDimension); return queryResults.Count > 0 ? queryResults[0] : null; }