Added one template.

PicnicSupermarket · Sep 6, 2022 · 2346fee · 2346fee
1 parent 23a34ad
commit 2346fee
Show file tree

Hide file tree

Showing 3 changed files with 22 additions and 0 deletions.
diff --git a/...ib/src/main/java/tech/picnic/errorprone/refastertemplates/AssertJComparableTemplates.java b/...ib/src/main/java/tech/picnic/errorprone/refastertemplates/AssertJComparableTemplates.java
@@ -12,6 +12,20 @@
 final class AssertJComparableTemplates {
   private AssertJComparableTemplates() {}
 
+  static final class AbstractComparableAssertActualIsEqualByComparingToExpected<
+      T extends Comparable<? super T>> {
+    @BeforeTemplate
+    AbstractIntegerAssert<?> before(T actual, T expected) {
+      return assertThat(actual.compareTo(expected)).isEqualTo(0);
+    }
+
+    @AfterTemplate
+    @UseImportPolicy(STATIC_IMPORT_ALWAYS)
+    AbstractComparableAssert<?, ?> after(T actual, T expected) {
+      return assertThat(actual).isEqualByComparingTo(expected);
+    }
+  }
+
   static final class AbstractComparableAssertActualIsLessThanExpected<
       T extends Comparable<? super T>> {
     @BeforeTemplate

diff --git a/...sources/tech/picnic/errorprone/refastertemplates/AssertJComparableTemplatesTestInput.java b/...sources/tech/picnic/errorprone/refastertemplates/AssertJComparableTemplatesTestInput.java
@@ -7,6 +7,10 @@
 import tech.picnic.errorprone.refaster.test.RefasterTemplateTestCase;
 
 final class AssertJComparableTemplatesTest implements RefasterTemplateTestCase {
+  AbstractAssert<?, ?> testAbstractComparableAssertActualIsEqualByComparingToExpected() {
+    return assertThat(BigDecimal.ZERO.compareTo(BigDecimal.ONE)).isEqualTo(0);
+  }
+
   AbstractAssert<?, ?> testAbstractComparableAssertActualIsLessThanExpected() {
     return assertThat(BigDecimal.ZERO.compareTo(BigDecimal.ONE)).isNegative();
   }

diff --git a/...ources/tech/picnic/errorprone/refastertemplates/AssertJComparableTemplatesTestOutput.java b/...ources/tech/picnic/errorprone/refastertemplates/AssertJComparableTemplatesTestOutput.java
@@ -7,6 +7,10 @@
 import tech.picnic.errorprone.refaster.test.RefasterTemplateTestCase;
 
 final class AssertJComparableTemplatesTest implements RefasterTemplateTestCase {
+  AbstractAssert<?, ?> testAbstractComparableAssertActualIsEqualByComparingToExpected() {
+    return assertThat(BigDecimal.ZERO).isEqualByComparingTo(BigDecimal.ONE);
+  }
+
   AbstractAssert<?, ?> testAbstractComparableAssertActualIsLessThanExpected() {
     return assertThat(BigDecimal.ZERO).isLessThan(BigDecimal.ONE);
   }