Skip to content

Commit dd7cc26

Browse files
Fix equals method and add tests for it.
1 parent 0da3c83 commit dd7cc26

File tree

2 files changed

+24
-21
lines changed

2 files changed

+24
-21
lines changed

src/main/java/com/amazonaws/encryptionsdk/keyrings/KeyringTraceEntry.java

+9-21
Original file line numberDiff line numberDiff line change
@@ -13,7 +13,6 @@
1313

1414
package com.amazonaws.encryptionsdk.keyrings;
1515

16-
import org.apache.commons.lang3.builder.EqualsBuilder;
1716
import org.apache.commons.lang3.builder.ToStringBuilder;
1817
import org.apache.commons.lang3.builder.ToStringStyle;
1918

@@ -80,29 +79,18 @@ public Set<KeyringTraceFlag> getFlags() {
8079
}
8180

8281
@Override
83-
public int hashCode() {
84-
return Objects.hash(keyNamespace, keyName, flags);
82+
public boolean equals(Object o) {
83+
if (this == o) return true;
84+
if (o == null || getClass() != o.getClass()) return false;
85+
KeyringTraceEntry that = (KeyringTraceEntry) o;
86+
return Objects.equals(keyNamespace, that.keyNamespace) &&
87+
Objects.equals(keyName, that.keyName) &&
88+
Objects.equals(flags, that.flags);
8589
}
8690

8791
@Override
88-
public boolean equals(Object obj) {
89-
if (obj == null) {
90-
return false;
91-
}
92-
if (obj == this) {
93-
return true;
94-
}
95-
if (obj.getClass() != getClass()) {
96-
return false;
97-
}
98-
99-
KeyringTraceEntry rhs = (KeyringTraceEntry) obj;
100-
101-
return new EqualsBuilder()
102-
.append(keyNamespace, rhs.keyNamespace)
103-
.append(keyName, this.keyName)
104-
.append(flags, this.flags)
105-
.isEquals();
92+
public int hashCode() {
93+
return Objects.hash(keyNamespace, keyName, flags);
10694
}
10795

10896
@Override

src/test/java/com/amazonaws/encryptionsdk/keyrings/KeyringTraceTest.java

+15
Original file line numberDiff line numberDiff line change
@@ -17,6 +17,7 @@
1717

1818
import static java.util.Collections.singleton;
1919
import static org.junit.Assert.assertEquals;
20+
import static org.junit.Assert.assertNotEquals;
2021

2122
public class KeyringTraceTest {
2223

@@ -47,4 +48,18 @@ public void testImmutable() {
4748
trace.getEntries().add(new KeyringTraceEntry("ns1", "name1",
4849
singleton(KeyringTraceFlag.WRAPPING_KEY_GENERATED_DATA_KEY)));
4950
}
51+
52+
@Test
53+
public void testKeyringTraceEntryEquals() {
54+
KeyringTraceEntry entry1 = new KeyringTraceEntry("namespace", "name",
55+
singleton(KeyringTraceFlag.WRAPPING_KEY_GENERATED_DATA_KEY));
56+
KeyringTraceEntry entry2 = new KeyringTraceEntry(entry1.getKeyNamespace(), entry1.getKeyName(),
57+
entry1.getFlags());
58+
KeyringTraceEntry entry3 = new KeyringTraceEntry("othernamespace", "name",
59+
singleton(KeyringTraceFlag.WRAPPING_KEY_GENERATED_DATA_KEY));
60+
61+
assertEquals(entry1, entry1);
62+
assertEquals(entry1, entry2);
63+
assertNotEquals(entry2, entry3);
64+
}
5065
}

0 commit comments

Comments
 (0)