如何在 JUnit 中测试代码抛出特定异常及自定义消息

import org.junit.Test;
import static org.junit.Assert.fail; // 用于演示如果异常未抛出

// 假设这是要测试的业务逻辑类
class B3_E2 {
    public static int getMultiplesOfGivenNumber(int number, int[] array) {
        if (number == 0) {
            throw new ArithmeticException("Number cannot be zero");
        }
        // ... 其他业务逻辑 ...
        return 0; // 简化处理
    }
}

public class JUnit4ExpectedExceptionTest {

    @Test(expected = ArithmeticException.class)
    public void testDivideByZeroWithExpectedAnnotation() {
        // 调用可能抛出异常的方法
        B3_E2.getMultiplesOfGivenNumber(0, new int[]{1, 2, 3});
        // 如果代码执行到这里，说明预期的 ArithmeticException 没有被抛出，测试将失败。
        // @Test(expected=...) 会自动处理这种情况，无需手动调用 fail()。
    }
}

import org.junit.Test;
import static org.junit.Assert.assertEquals;
import static org.junit.Assert.fail; // 用于明确指示测试失败

// 假设这是要测试的业务逻辑类 (与上文相同)
class B3_E2 {
    public static int getMultiplesOfGivenNumber(int number, int[] array) {
        if (number == 0) {
            throw new ArithmeticException("Number cannot be zero");
        }
        // ... 其他业务逻辑 ...
        return 0; // 简化处理
    }
}

public class TryCatchExceptionTest {

    // 辅助方法，用于模拟数据
    private int[] intervalFromOneToTen() {
        return new int[]{1, 2, 3, 4, 5, 6, 7, 8, 9, 10};
    }

    @Test
    public void testDivideByZeroWithTryCatch() {
        try {
            // 调用可能抛出异常的方法
            B3_E2.getMultiplesOfGivenNumber(0, intervalFromOneToTen());
            // 如果代码执行到这里，说明预期的异常没有被抛出，测试应该失败。
            fail("Expected ArithmeticException was not thrown.");
        } catch (ArithmeticException e) {
            // 捕获到预期的 ArithmeticException，现在可以检查其消息或其他属性
            assertEquals("Number cannot be zero", e.getMessage());
            // 对于更复杂的异常，可以在这里检查更多的属性，例如 cause、自定义字段等。
        } catch (Exception e) {
            // 捕获到其他类型的异常，这不是我们预期的异常类型，测试也应失败。
            fail("Thrown an unexpected exception type: " + e.getClass().getName());
        }
    }
}

import org.junit.jupiter.api.Test; // JUnit 5 的 @Test 注解
import static org.junit.jupiter.api.Assertions.assertThrows;
import static org.junit.jupiter.api.Assertions.assertEquals;

// 假设这是要测试的业务逻辑类 (与上文相同)
class B3_E2 {
    public static int getMultiplesOfGivenNumber(int number, int[] array) {
        if (number == 0) {
            throw new ArithmeticException("Number cannot be zero");
        }
        // ... 其他业务逻辑 ...
        return 0; // 简化处理
    }
}

public class JUnit5AssertThrowsTest {

    // 辅助方法，用于模拟数据
    private int[] intervalFromOneToTen() {
        return new int[]{1, 2, 3, 4, 5, 6, 7, 8, 9, 10};
    }

    @Test
    void testDivideByZeroWithAssertThrows() { // JUnit 5 的 @Test 方法可以不声明为 public
        // 使用 assertThrows 捕获异常
        ArithmeticException thrown = assertThrows(
            ArithmeticException.class, // 期望的异常类型
            () -> B3_E2.getMultiplesOfGivenNumber(0, intervalFromOneToTen()), // 包含可能抛出异常的代码
            "Expected ArithmeticException to be thrown, but it wasn't" // 可选的失败消息
        );

        // 验证异常消息
        assertEquals("Number cannot be zero", thrown.getMessage());
        // 也可以进一步检查异常的其他属性
    }
}