tests: Add tests for DefaultCollectionMutator.replace(E e)

Test happy path. Test null handling. Test empty handling. Test missing handling.
jwtk · lhy-hoyin · Oct 16, 2024 · Oct 16, 2024 · Oct 17, 2024 · Oct 19, 2024
commit 508227cf4d8b59a80df2f7062406b0c47bf624e1
diff --git a/impl/src/test/groovy/io/jsonwebtoken/impl/lang/DefaultCollectionMutatorTest.groovy b/impl/src/test/groovy/io/jsonwebtoken/impl/lang/DefaultCollectionMutatorTest.groovy
@@ -16,10 +16,14 @@
 package io.jsonwebtoken.impl.lang

 import io.jsonwebtoken.Identifiable
+import io.jsonwebtoken.Jwts
 import io.jsonwebtoken.lang.Strings
+import io.jsonwebtoken.security.MacAlgorithm
 import org.junit.Before
 import org.junit.Test

+import java.lang.reflect.Constructor
+
 import static org.junit.Assert.*

 /**
@@ -128,6 +132,35 @@ class DefaultCollectionMutatorTest {
        assertEquals 0, changeCount
    }

+    @Test
+    void replace() {
+        Class<?> c = Class.forName("io.jsonwebtoken.impl.security.DefaultMacAlgorithm")
+        Constructor<?> ctor = c.getDeclaredConstructor(String.class, String.class, int.class)
+        ctor.setAccessible(true)
+        MacAlgorithm custom = (MacAlgorithm) ctor.newInstance('HS512', 'HmacSHA512', 80)
+
+        m.add(Jwts.SIG.HS512)
+        m.replace(custom)
+        assertEquals 2, changeCount // replace is count as one change
+        assertEquals 1, m.getCollection().size() // existing is removed as part of replacement
+        assertEquals 80, ((MacAlgorithm) m.getCollection().toArray()[0]).getKeyBitLength()
+    }
+
+    @Test(expected = NoSuchElementException)
+    void replaceMissing() {
+        m.replace('foo')
+    }
+
+    @Test(expected = NullPointerException)
+    void replaceNull() {
+        m.replace(null)
+    }
+
+    @Test(expected = IllegalArgumentException)
+    void replaceEmpty() {
+        m.replace(Strings.EMPTY)
+    }
+
    @Test
    void clear() {
        m.add('one').add('two').add(['three', 'four'])