Fix sporadic test failure

Author: trask

test/smoke/testApps/HeartBeat/src/smokeTest/java/com/microsoft/applicationinsights/smoketest/HeartBeatTest.java

@@ -10,6 +10,7 @@
 
 import com.google.common.base.Preconditions;
 import com.google.common.base.Predicate;
+import com.microsoft.applicationinsights.internal.schemav2.Data;
 import com.microsoft.applicationinsights.internal.schemav2.Envelope;
 import com.microsoft.applicationinsights.internal.schemav2.MetricData;
 
@@ -23,7 +24,7 @@ public void testHeartBeat() throws Exception {
         List<Envelope> metrics = mockedIngestion.waitForItems(getMetricPredicate("HeartbeatState"), 2, 70, TimeUnit.SECONDS);
         assertEquals(2, metrics.size());
 
-        MetricData data = getTelemetryDataForType(0, "MetricData");
+        MetricData data = (MetricData) ((Data) metrics.get(0).getData()).getBaseData();
         assertNotNull(data.getProperties().get("jreVersion"));
         assertNotNull(data.getProperties().get("sdkVersion"));
         assertNotNull(data.getProperties().get("osVersion"));