Add ComparableSubject#isAtLeast to Kruth

androidx · Apr 14, 2023 · 222d549 · 222d549
1 parent beccf74
commit 222d549
Show file tree

Hide file tree

Showing 2 changed files with 31 additions and 0 deletions.
diff --git a/testutils/testutils-kmp/src/commonMain/kotlin/androidx/kruth/ComparableSubject.kt b/testutils/testutils-kmp/src/commonMain/kotlin/androidx/kruth/ComparableSubject.kt
@@ -37,4 +37,17 @@ open class ComparableSubject<T : Comparable<T>> constructor(actual: T?) : Subjec
             fail("Expected to be less than $other, but was $actual")
         }
     }
+
+    /**
+     * Checks that the subject is greater than or equal to [other].
+     *
+     * @throws NullPointerException if [actual] or [other] is `null`.
+     */
+    fun isAtLeast(other: T?) {
+        if (actual == null || other == null) {
+            throw NullPointerException("Expected to be at least $other, but was $actual")
+        } else if (actual < other) {
+            fail("Expected to be at least $other, but was $actual")
+        }
+    }
 }
diff --git a/testutils/testutils-kmp/src/commonTest/kotlin/androidx/kruth/ComparableSubjectTest.kt b/testutils/testutils-kmp/src/commonTest/kotlin/androidx/kruth/ComparableSubjectTest.kt
@@ -35,6 +35,9 @@ class ComparableSubjectTest {
         assertFailsWith<NullPointerException> {
             assertThat(6).isLessThan(null)
         }
+        assertFailsWith<NullPointerException> {
+            assertThat(6).isAtLeast(null)
+        }
     }
 
     @Test
@@ -52,16 +55,31 @@ class ComparableSubjectTest {
         }
     }
 
+    @Test
+    fun isAtLeast() {
+        assertThat(4).isAtLeast(3)
+        assertThat(4).isAtLeast(4)
+        assertFailsWith<AssertionError> {
+            assertThat(4).isAtLeast(5)
+        }
+    }
+
     // Brief tests with other comparable types (no negative test cases)
 
     @Test
     fun longs() {
         assertThat(4L).isLessThan(5L)
+
+        assertThat(4L).isAtLeast(4L)
+        assertThat(4L).isAtLeast(3L)
     }
 
     @Test
     fun strings() {
         assertThat("gak").isLessThan("kak")
+
+        assertThat("kak").isAtLeast("kak")
+        assertThat("kak").isAtLeast("gak")
     }
 
     @Test