Tests MeasurementDescriptor and MeasurementUnit equals() and hashCode().

1 files changed, 39 insertions, 0 deletions

diff --git a/core/src/test/java/com/google/instrumentation/stats/MeasurementDescriptorTest.java b/core/src/test/java/com/google/instrumentation/stats/MeasurementDescriptorTest.java
index 9fa16f37..e362b473 100644
--- a/core/src/test/java/com/google/instrumentation/stats/MeasurementDescriptorTest.java
+++ b/core/src/test/java/com/google/instrumentation/stats/MeasurementDescriptorTest.java
@@ -15,6 +15,7 @@ package com.google.instrumentation.stats;
 
 import static com.google.common.truth.Truth.assertThat;
 
+import com.google.common.testing.EqualsTester;
 import com.google.instrumentation.stats.MeasurementDescriptor.BasicUnit;
 import com.google.instrumentation.stats.MeasurementDescriptor.MeasurementUnit;
 import java.util.Arrays;
@@ -60,6 +61,44 @@ public final class MeasurementDescriptorTest {
     assertThat(measurement.getUnit().getDenominators().get(0)).isEqualTo(BasicUnit.SECONDS);
   }
 
+  @Test
+  public void testMeasurementUnitEquals() {
+    new EqualsTester()
+        .addEqualityGroup(
+            MeasurementUnit.create(
+                1, Arrays.asList(BasicUnit.BYTES), Arrays.asList(BasicUnit.SECONDS)),
+            MeasurementUnit.create(
+                1, Arrays.asList(BasicUnit.BYTES), Arrays.asList(BasicUnit.SECONDS)))
+        .addEqualityGroup(
+            MeasurementUnit.create(
+                2, Arrays.asList(BasicUnit.BYTES), Arrays.asList(BasicUnit.SECONDS)))
+        .addEqualityGroup(MeasurementUnit.create(1, Arrays.asList(BasicUnit.BYTES)))
+        .testEquals();
+  }
+
+  @Test
+  public void testMeasurementDescriptorEquals() {
+    new EqualsTester()
+        .addEqualityGroup(
+            MeasurementDescriptor.create(
+                "name",
+                "description",
+                MeasurementUnit.create(
+                    1, Arrays.asList(BasicUnit.BITS), Arrays.asList(BasicUnit.SECONDS))),
+            MeasurementDescriptor.create(
+                "name",
+                "description",
+                MeasurementUnit.create(
+                    1, Arrays.asList(BasicUnit.BITS), Arrays.asList(BasicUnit.SECONDS))))
+        .addEqualityGroup(
+            MeasurementDescriptor.create(
+                "name",
+                "description 2",
+                MeasurementUnit.create(
+                    1, Arrays.asList(BasicUnit.BYTES), Arrays.asList(BasicUnit.SECONDS))))
+        .testEquals();
+  }
+
   private static final MeasurementDescriptor makeSimpleDescriptor(String name) {
     return MeasurementDescriptor.create(
         name,