Fix allocation profile test

runtime/observatory/tests/observatory_ui/allocation_profile/element_test.dart

 // Copyright (c) 2016, the Dart project authors.  Please see the AUTHORS file
 // for details. All rights reserved. Use of this source code is governed by a
 // BSD-style license that can be found in the LICENSE file.
 
 import 'dart:html';
 import 'dart:async';
 import 'package:unittest/unittest.dart';
 import 'package:observatory/models.dart' as M;
 import 'package:observatory/src/elements/allocation_profile.dart';
 import 'package:observatory/src/elements/class_ref.dart';
@@ skipping 110 matching lines @@
       await (() async {}());
       e.remove();
       await e.onRendered.first;
       expect(e.children.length, isZero, reason: 'is empty');
     });
     test('to sort change', () async {
       const clazz1 = const ClassRefMock(name: 'class1');
       const clazz2 = const ClassRefMock(name: 'class2');
       const clazz3 = const ClassRefMock(name: 'class3');
       const profile = const AllocationProfileMock(members: const [
+        const ClassHeapStatsMock(clazz: clazz1),
         const ClassHeapStatsMock(
             clazz: clazz2,
             newSpace: const AllocationsMock(
-                accumulated: const AllocationCountMock(bytes: 10))),
-        const ClassHeapStatsMock(clazz: clazz3),
+                current: const AllocationCountMock(bytes: 10)),
+            oldSpace: const AllocationsMock(
+                current: const AllocationCountMock(bytes: 2))),
         const ClassHeapStatsMock(
-            clazz: clazz1,
+            clazz: clazz3,
             newSpace: const AllocationsMock(
-                accumulated: const AllocationCountMock(bytes: 5)))
+                current: const AllocationCountMock(bytes: 5)),
+            oldSpace: const AllocationsMock(
+                current: const AllocationCountMock(bytes: 3)))
       ]);
       final completer = new Completer<AllocationProfileMock>();
       final repo = new AllocationProfileRepositoryMock(
           getter: expectAsync((M.IsolateRef i, bool gc, bool reset) {
         expect(i, equals(isolate));
         expect(gc, isFalse);
         expect(reset, isFalse);
         return completer.future;
       }, count: 1));
       final e = new AllocationProfileElement(vm, isolate, events, notif, repo);
       document.body.append(e);
       await e.onRendered.first;
       completer.complete(profile);
       await e.onRendered.first;
+      expect((e.querySelector(cTag) as ClassRefElement).cls, equals(clazz2));
+      e.querySelector('button.name').click();
+      await e.onRendered.first;
       expect((e.querySelector(cTag) as ClassRefElement).cls, equals(clazz1));
-      e.querySelector('button.bytes').click();
-      await e.onRendered.first;
-      expect((e.querySelector(cTag) as ClassRefElement).cls, equals(clazz2));
-      e.querySelector('button.bytes').click();
+      e.querySelector('button.name').click();
       await e.onRendered.first;
       expect((e.querySelector(cTag) as ClassRefElement).cls, equals(clazz3));
+      e.querySelectorAll('button.bytes').last.click();
+      await e.onRendered.first;
+      expect((e.querySelector(cTag) as ClassRefElement).cls, equals(clazz3));
+      e.querySelectorAll('button.bytes').last.click();
+      await e.onRendered.first;
+      expect((e.querySelector(cTag) as ClassRefElement).cls, equals(clazz1));
       e.remove();
       await e.onRendered.first;
       expect(e.children.length, isZero, reason: 'is empty');
     });
   });
 }