@@ -102,43 +102,6 @@ public static ResultAssertion assertAskedInStdIn() {
102102 };
103103 }
104104
105- public static ResultAssertion assertReturnValueEquals (final int expected ) {
106- return (p , res ) -> {
107- if (res .executionResult == null ) {
108- p .println (format (EXPECTED_TO_RETURN_BUT_RETURNED_NULL , Formats .Red , code (res .resultExpressionSourceCode ), code (expected )));
109- return false ;
110- } else if (!(res .executionResult instanceof Integer )) {
111- p .println (format (EXPECTED_TO_RETURN_INT_BUT_RETURNED_OTHER_TYPE , Formats .Red , code (res .resultExpressionSourceCode ), code (res .executionResult .getClass ().getSimpleName ())));
112- return false ;
113- } else if (((Integer )res .executionResult ).intValue () != expected ) {
114- p .println (format (EXPECTED_TO_RETURN_BUT_RETURNED , Formats .Red , code (res .resultExpressionSourceCode ), code (Integer .toString (expected )), code (res .executionResult .toString ())));
115- return false ;
116- }
117-
118- p .println (format (OK_RETURNED , Formats .Green , code (res .resultExpressionSourceCode ), code (expected )));
119- return true ;
120- };
121- }
122-
123- public static ResultAssertion assertReturnValueEquals (final int [] expected ) {
124- return (p , res ) -> {
125- final Formatted <String > expectedFmted = code (Expression .formatLiteralSourceCode (expected ));
126- if (res .executionResult == null ) {
127- p .println (format (EXPECTED_TO_RETURN_BUT_RETURNED_NULL , Formats .Red , code (res .resultExpressionSourceCode ), expectedFmted ));
128- return false ;
129- } else if (!(res .executionResult instanceof int [])) {
130- p .println (format (EXPECTED_TO_RETURN_INT_ARRAY_BUT_RETURNED_OTHER_TYPE , Formats .Red , code (res .resultExpressionSourceCode ), code (res .executionResult .getClass ().getSimpleName ())));
131- return false ;
132- } else if (!Arrays .equals (expected , (int [])res .executionResult )) {
133- p .println (format (EXPECTED_TO_RETURN_BUT_RETURNED , Formats .Red , code (res .resultExpressionSourceCode ), expectedFmted , code (Expression .formatLiteralSourceCode ((int [])res .executionResult ))));
134- return false ;
135- }
136-
137- p .println (format (OK_RETURNED , Formats .Green , code (res .resultExpressionSourceCode ), expectedFmted ));
138- return true ;
139- };
140- }
141-
142105 private static final double EPSILON = 0.0000000001 ;
143106 private static boolean eq (Object expected , Object actual ) {
144107 if (actual == null ) {
@@ -157,14 +120,15 @@ public static ResultAssertion assertVariableEquals(final String variableName, fi
157120 final Object val = res .executionContext .getVariableValue (variableName );
158121 final Formatted <String > expectedFmted = code (Expression .formatLiteralSourceCode (expected ));
159122 final Formatted <String > actualFmted = code (Expression .formatLiteralSourceCode (val ));
123+ final Formatted <String > expressionFmted = code (res .resultExpressionSourceCode );
160124 if (val == null ) {
161- p .println (format (EXPECTED_VARIABLE_TO_EQUAL_BUT_IS_NULL , Formats .Red , code ( res . resultExpressionSourceCode ) , code (variableName ), expectedFmted ));
125+ p .println (format (EXPECTED_VARIABLE_TO_EQUAL_BUT_IS_NULL , Formats .Red , expressionFmted , code (variableName ), expectedFmted ));
162126 return false ;
163127 } else if (val .getClass () != expected .getClass ()) {
164- p .println (format (EXPECTED_VARIABLE_TO_BE_BUT_WAS_OTHER_TYPE , Formats .Red , code ( res . resultExpressionSourceCode ) , code (variableName ), code (expected .getClass ().getSimpleName ()), code (val .getClass ().getSimpleName ())));
128+ p .println (format (EXPECTED_VARIABLE_TO_BE_BUT_WAS_OTHER_TYPE , Formats .Red , expressionFmted , code (variableName ), code (expected .getClass ().getSimpleName ()), code (val .getClass ().getSimpleName ())));
165129 return false ;
166130 } else if (!eq (expected , val )) {
167- p .println (format (EXPECTED_VARIABLE_TO_EQUAL_BUT_EQUAL , Formats .Red , code ( res . resultExpressionSourceCode ) , code (variableName ), expectedFmted , actualFmted ));
131+ p .println (format (EXPECTED_VARIABLE_TO_EQUAL_BUT_EQUAL , Formats .Red , expressionFmted , code (variableName ), expectedFmted , actualFmted ));
168132 return false ;
169133 }
170134
@@ -173,38 +137,23 @@ public static ResultAssertion assertVariableEquals(final String variableName, fi
173137 };
174138 }
175139
176- public static ResultAssertion assertReturnValueEquals (final double expected ) {
177- return (p , res ) -> {
178- if (res .executionResult == null ) {
179- p .println (format (EXPECTED_TO_RETURN_BUT_RETURNED_NULL , Formats .Red , res .resultExpressionSourceCode , expected ));
180- return false ;
181- } else if (!(res .executionResult instanceof Double )) {
182- p .println (format (EXPECTED_TO_RETURN_DOUBLE_BUT_RETURNED_OTHER_TYPE , Formats .Red , code (res .resultExpressionSourceCode ), res .executionResult .getClass ().getSimpleName ()));
183- return false ;
184- } else if (!eq (res .executionResult , expected )) {
185- p .println (format (EXPECTED_TO_RETURN_BUT_RETURNED , Formats .Red , res .resultExpressionSourceCode , expected , ((Double )res .executionResult ).doubleValue ()));
186- return false ;
187- }
188-
189- p .println (format (OK_RETURNED , Formats .Green , code (res .resultExpressionSourceCode ), code (expected )));
190- return true ;
191- };
192- }
193-
194- public static ResultAssertion assertReturnValueEquals (final boolean expected ) {
140+ public static ResultAssertion assertReturnValueEquals (final Object expected ) {
195141 return (p , res ) -> {
142+ final Formatted <String > expectedFmted = code (Expression .formatLiteralSourceCode (expected ));
143+ final Formatted <String > actualFmted = code (Expression .formatLiteralSourceCode (res .executionResult ));
144+ final Formatted <String > expressionFmted = code (res .resultExpressionSourceCode );
196145 if (res .executionResult == null ) {
197- p .println (format (EXPECTED_TO_RETURN_BUT_RETURNED_NULL , Formats .Red , res . resultExpressionSourceCode , expected ));
146+ p .println (format (EXPECTED_TO_RETURN_BUT_RETURNED_NULL , Formats .Red , expressionFmted , expectedFmted ));
198147 return false ;
199- } else if (!( res .executionResult instanceof Boolean )) {
200- p .println (format (EXPECTED_TO_RETURN_BOOLEAN_BUT_RETURNED_OTHER_TYPE , Formats .Red , code (res . resultExpressionSourceCode ), res .executionResult .getClass ().getSimpleName ()));
148+ } else if (res .executionResult . getClass () != expected . getClass ( )) {
149+ p .println (format (EXPECTED_TO_RETURN_BUT_RETURNED_OTHER_TYPE , Formats .Red , expressionFmted , code (expected . getClass (). getSimpleName ()), code ( res .executionResult .getClass ().getSimpleName () )));
201150 return false ;
202- } else if ((( Boolean ) res .executionResult ). booleanValue () != expected ) {
203- p .println (format (EXPECTED_TO_RETURN_BUT_RETURNED , Formats .Red , res . resultExpressionSourceCode , expected , (( Boolean ) res . executionResult ). booleanValue () ));
151+ } else if (! eq ( expected , res .executionResult )) {
152+ p .println (format (EXPECTED_TO_RETURN_BUT_RETURNED , Formats .Red , expressionFmted , expectedFmted , actualFmted ));
204153 return false ;
205154 }
206155
207- p .println (format (OK_RETURNED , Formats .Green , code ( res . resultExpressionSourceCode ), code ( expected ) ));
156+ p .println (format (OK_RETURNED , Formats .Green , expressionFmted , expectedFmted ));
208157 return true ;
209158 };
210159 }
0 commit comments