Rewrite assert: (a closeTo: b) into assert: a closeTo: b

jecisc · Mar 13, 2023 · 344964f · 344964f
1 parent ae96563
commit 344964f
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 0 deletions.
diff --git a/src/Chanel-Tests/ChanelTestEqualityCleanerTest.class.st b/src/Chanel-Tests/ChanelTestEqualityCleanerTest.class.st
@@ -28,6 +28,11 @@ ChanelTestEqualityCleanerTest >> setUp [
 	class := self createDefaultTestClass
 ]
 
+{ #category : #tests }
+ChanelTestEqualityCleanerTest >> testAssertCloseTo [
+	self assert: 'self assert: (3 closeTo: 2)' isRewrittenAs: 'self assert: 3 closeTo: 2'
+]
+
 { #category : #tests }
 ChanelTestEqualityCleanerTest >> testAssertEquals [
 	self assert: 'self assert: 3 = 2' isRewrittenAs: 'self assert: 3 equals: 2'
@@ -83,6 +88,11 @@ ChanelTestEqualityCleanerTest >> testAssertIdenticalToTrue2 [
 	self assert: 'self assert: 3 identicalTo: true' isRewrittenAs: 'self assert: 3'
 ]
 
+{ #category : #tests }
+ChanelTestEqualityCleanerTest >> testDenyCloseTo [
+	self assert: 'self deny: (3 closeTo: 2)' isRewrittenAs: 'self deny: 3 closeTo: 2'
+]
+
 { #category : #tests }
 ChanelTestEqualityCleanerTest >> testDenyEquals [
 	self assert: 'self deny: 3 = 2' isRewrittenAs: 'self deny: 3 equals: 2'

diff --git a/src/Chanel/ChanelTestEqualityCleaner.class.st b/src/Chanel/ChanelTestEqualityCleaner.class.st
@@ -68,6 +68,9 @@ ChanelTestEqualityCleaner >> rewriter [
 		replace: '`@receiver deny: `@arg = `@arg2' with: '`@receiver deny: `@arg equals: `@arg2' when: [ :n | self minimalPharoVersion >= 7 ];
 		replace: '`@receiver assert: `@arg == `@arg2' with: '`@receiver assert: `@arg identicalTo: `@arg2' when: [ :n | self minimalPharoVersion >= 7 ];
 		replace: '`@receiver deny: `@arg == `@arg2' with: '`@receiver deny: `@arg identicalTo: `@arg2' when: [ :n | self minimalPharoVersion >= 7 ];
+
+		replace: '`@receiver assert: (`@arg closeTo: `@arg2)' with: '`@receiver assert: `@arg closeTo: `@arg2';
+		replace: '`@receiver deny: (`@arg closeTo: `@arg2)' with: '`@receiver deny: `@arg closeTo: `@arg2';
 		yourself
 ]