junit单元测试代码

Question

java怎么针对main函数进行单元测试：需要测试的代码如下
package Part1;

import java.util.Scanner;

public class LeapYear1 {

public static void main(String[] args) {
    Scanner input = new Scanner(System.in);
    String yearStr;
    int year;
    while (true) {
        System.out.print("请输入4位年份（输入q退出程序）：");
        yearStr = input.nextLine();
        if (yearStr.equals("q")) {
            break;
        } else if (!yearStr.matches("\\d{4}")) {
           System.out.println("输入有误，请重新输入4位数字年份");
           continue;
        }
     year = Integer.parseInt(yearStr);
     if ((year % 4 == 0 && year % 100 != 0) || year % 400 == 0) {
         System.out.println(year+"是闰年");
     } else {
         System.out.println(year+"不是闰年");
     }
   }
}    

}

Answer 1

main方法可以直接执行测试的，如果你非要调用 LeapYear1.main（）方法测试，可以通过下面代码测试：
1 创建一个测试类，使用JUnit的@Test注解标记测试方法。

import org.junit.Test;
import java.io.ByteArrayInputStream;
import java.util.Scanner;

public class LeapYear1Test {

    @Test
    public void testLeapYear() {
        String input = "2020\nq\n"; // 模拟用户输入
        System.setIn(new ByteArrayInputStream(input.getBytes())); // 重定向标准输入流
        LeapYear1.main(new String[]{}); // 调用待测试的main函数
    }
}

2 在测试方法中，使用ByteArrayInputStream模拟用户输入，重定向标准输入流，然后调用待测试的main函数。

3 运行测试类，观察测试结果。