1 files changed, 181 insertions, 0 deletions
diff --git a/tools/ahat/test/DiffFieldsTest.java b/tools/ahat/test/DiffFieldsTest.java
new file mode 100644
index 0000000000..6abdd47ef6
--- /dev/null
+++ b/tools/ahat/test/DiffFieldsTest.java
@@ -0,0 +1,181 @@
+/*
+ * Copyright (C) 2016 The Android Open Source Project
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *      http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+package com.android.ahat;
+
+import com.android.ahat.heapdump.DiffFields;
+import com.android.ahat.heapdump.DiffedFieldValue;
+import com.android.ahat.heapdump.FieldValue;
+import com.android.ahat.heapdump.Value;
+import java.util.ArrayList;
+import java.util.List;
+import org.junit.Test;
+
+import static org.junit.Assert.assertEquals;
+import static org.junit.Assert.assertNull;
+
+public class DiffFieldsTest {
+  @Test
+  public void normalMatchedDiffedFieldValues() {
+    FieldValue normal1 = new FieldValue("name", "type", new Value(1));
+    FieldValue normal2 = new FieldValue("name", "type", new Value(2));
+
+    DiffedFieldValue x = DiffedFieldValue.matched(normal1, normal2);
+    assertEquals("name", x.name);
+    assertEquals("type", x.type);
+    assertEquals(new Value(1), x.current);
+    assertEquals(new Value(2), x.baseline);
+    assertEquals(DiffedFieldValue.Status.MATCHED, x.status);
+  }
+
+  @Test
+  public void nulledMatchedDiffedFieldValues() {
+    FieldValue normal = new FieldValue("name", "type", new Value(1));
+    FieldValue nulled = new FieldValue("name", "type", null);
+
+    DiffedFieldValue x = DiffedFieldValue.matched(normal, nulled);
+    assertEquals("name", x.name);
+    assertEquals("type", x.type);
+    assertEquals(new Value(1), x.current);
+    assertNull(x.baseline);
+    assertEquals(DiffedFieldValue.Status.MATCHED, x.status);
+
+    DiffedFieldValue y = DiffedFieldValue.matched(nulled, normal);
+    assertEquals("name", y.name);
+    assertEquals("type", y.type);
+    assertNull(y.current);
+    assertEquals(new Value(1), y.baseline);
+    assertEquals(DiffedFieldValue.Status.MATCHED, y.status);
+  }
+
+  @Test
+  public void normalAddedDiffedFieldValues() {
+    FieldValue normal = new FieldValue("name", "type", new Value(1));
+
+    DiffedFieldValue x = DiffedFieldValue.added(normal);
+    assertEquals("name", x.name);
+    assertEquals("type", x.type);
+    assertEquals(new Value(1), x.current);
+    assertEquals(DiffedFieldValue.Status.ADDED, x.status);
+  }
+
+  @Test
+  public void nulledAddedDiffedFieldValues() {
+    FieldValue nulled = new FieldValue("name", "type", null);
+
+    DiffedFieldValue x = DiffedFieldValue.added(nulled);
+    assertEquals("name", x.name);
+    assertEquals("type", x.type);
+    assertNull(x.current);
+    assertEquals(DiffedFieldValue.Status.ADDED, x.status);
+  }
+
+  @Test
+  public void normalDeletedDiffedFieldValues() {
+    FieldValue normal = new FieldValue("name", "type", new Value(1));
+
+    DiffedFieldValue x = DiffedFieldValue.deleted(normal);
+    assertEquals("name", x.name);
+    assertEquals("type", x.type);
+    assertEquals(new Value(1), x.baseline);
+    assertEquals(DiffedFieldValue.Status.DELETED, x.status);
+  }
+
+  @Test
+  public void nulledDeletedDiffedFieldValues() {
+    FieldValue nulled = new FieldValue("name", "type", null);
+
+    DiffedFieldValue x = DiffedFieldValue.deleted(nulled);
+    assertEquals("name", x.name);
+    assertEquals("type", x.type);
+    assertNull(x.baseline);
+    assertEquals(DiffedFieldValue.Status.DELETED, x.status);
+  }
+
+  @Test
+  public void basicDiff() {
+    List<FieldValue> a = new ArrayList<FieldValue>();
+    a.add(new FieldValue("n0", "t0", null));
+    a.add(new FieldValue("n2", "t2", null));
+    a.add(new FieldValue("n3", "t3", null));
+    a.add(new FieldValue("n4", "t4", null));
+    a.add(new FieldValue("n5", "t5", null));
+    a.add(new FieldValue("n6", "t6", null));
+
+    List<FieldValue> b = new ArrayList<FieldValue>();
+    b.add(new FieldValue("n0", "t0", null));
+    b.add(new FieldValue("n1", "t1", null));
+    b.add(new FieldValue("n2", "t2", null));
+    b.add(new FieldValue("n3", "t3", null));
+    b.add(new FieldValue("n5", "t5", null));
+    b.add(new FieldValue("n6", "t6", null));
+    b.add(new FieldValue("n7", "t7", null));
+
+    // Note: The expected result makes assumptions about the implementation of
+    // field diff to match the order of the returned fields. If the
+    // implementation changes, this test may need to be generalized to accept
+    // the new implementation.
+    List<DiffedFieldValue> expected = new ArrayList<DiffedFieldValue>();
+    expected.add(DiffedFieldValue.matched(a.get(0), b.get(0)));
+    expected.add(DiffedFieldValue.deleted(b.get(1)));
+    expected.add(DiffedFieldValue.matched(a.get(1), b.get(2)));
+    expected.add(DiffedFieldValue.matched(a.get(2), b.get(3)));
+    expected.add(DiffedFieldValue.added(a.get(3)));
+    expected.add(DiffedFieldValue.matched(a.get(4), b.get(4)));
+    expected.add(DiffedFieldValue.matched(a.get(5), b.get(5)));
+    expected.add(DiffedFieldValue.deleted(b.get(6)));
+
+    List<DiffedFieldValue> diffed = DiffFields.diff(a, b);
+    assertEquals(expected, diffed);
+  }
+
+  @Test
+  public void reorderedDiff() {
+    List<FieldValue> a = new ArrayList<FieldValue>();
+    a.add(new FieldValue("n0", "t0", null));
+    a.add(new FieldValue("n1", "t1", null));
+    a.add(new FieldValue("n2", "t2", null));
+    a.add(new FieldValue("n3", "t3", null));
+    a.add(new FieldValue("n4", "t4", null));
+    a.add(new FieldValue("n5", "t5", null));
+    a.add(new FieldValue("n6", "t6", null));
+
+    List<FieldValue> b = new ArrayList<FieldValue>();
+    b.add(new FieldValue("n4", "t4", null));
+    b.add(new FieldValue("n1", "t1", null));
+    b.add(new FieldValue("n3", "t3", null));
+    b.add(new FieldValue("n0", "t0", null));
+    b.add(new FieldValue("n5", "t5", null));
+    b.add(new FieldValue("n2", "t2", null));
+    b.add(new FieldValue("n6", "t6", null));
+
+    // Note: The expected result makes assumptions about the implementation of
+    // field diff to match the order of the returned fields. If the
+    // implementation changes, this test may need to be generalized to accept
+    // the new implementation.
+    List<DiffedFieldValue> expected = new ArrayList<DiffedFieldValue>();
+    expected.add(DiffedFieldValue.matched(a.get(0), b.get(3)));
+    expected.add(DiffedFieldValue.matched(a.get(1), b.get(1)));
+    expected.add(DiffedFieldValue.matched(a.get(2), b.get(5)));
+    expected.add(DiffedFieldValue.matched(a.get(3), b.get(2)));
+    expected.add(DiffedFieldValue.matched(a.get(4), b.get(0)));
+    expected.add(DiffedFieldValue.matched(a.get(5), b.get(4)));
+    expected.add(DiffedFieldValue.matched(a.get(6), b.get(6)));
+
+    List<DiffedFieldValue> diffed = DiffFields.diff(a, b);
+    assertEquals(expected, diffed);
+  }
+}