如何模拟从JUnit测试用例到程序的多个输入？

Question

我写了一个程序，看起来像这样：

import java.util.Scanner;
import java.util.ArrayList;

public class Triangles {

    public static void main(String[] args) {

        Scanner user_input = new Scanner(System.in);

        ArrayList<String> triangleLengths = new ArrayList<String>();

        for (int i=0; i < 3; i++) {

            System.out.print("Triangle length #" + i + ": ");
            triangleLengths.add(i,user_input.next());   
        }

        if (triangleLengths.get(0) == triangleLengths.get(1) && triangleLengths.get(1) == triangleLengths.get(2)) {

            System.out.println("This triangle is an equilateral triangle");

        } else if (triangleLengths.get(0) == triangleLengths.get(1) || triangleLengths.get(0) == triangleLengths.get(2) || triangleLengths.get(1) == triangleLengths.get(2)) {

            System.out.println("This triangle is an isosceles triangle");

        } else if (triangleLengths.get(0) != triangleLengths.get(1) && triangleLengths.get(1) != triangleLengths.get(2)) {

            System.out.println("This triangle is a scalene triangle");

        } else {

            System.out.println("The input does not make a triangle!");

        }


    }

}

我的任务是编写一个JUnit测试用例，以通过各种输入进行测试，从本质上尝试并“破坏”我的程序。 作为一个Java新手，我无法终生思考如何做到这一点-有人能指出我正确的方向吗？

Answer 1

因此，我对如何解决提出了建议。

您想要制作它，以便可以自动使用不同的参数进行测试，而无需手动输入它，因此我隔离了三角形部分，如下所示。

编辑：我有点重做代码

正常运行类src / main.java

import java.util.ArrayList;
import java.util.List;
import java.util.Scanner;

public class Main {

    public static void main(String[] args) {
        Scanner user_input = new Scanner(System.in);
        List<String> triangleLengths = new ArrayList<>();

        for (int i = 0; i < 3; i++) {
            System.out.print("Triangle length #" + i + ": ");
            triangleLengths.add(i, user_input.next());
        }

        // Result output will be here
        Triangle subject = new Triangle(triangleLengths);
        if (subject.getTriangleType() == Triangle.Type.INVALID) {
            System.out.println("Triangle is invalid");
        } else {
            System.out.println("Triangle is: " + subject.getTriangleType());
        }
    }
}

JUnit类test / TrianglesTest.java

import org.junit.Test;

import java.util.Arrays;
import java.util.List;

import static org.junit.Assert.assertEquals;

public class TrianglesTest {

    /**
     * Testing with String inputs (what you'd enter manually)
     */
    @Test
    public void testWithStrings() {
        List<String> triangleLengths = Arrays.asList("len1", "len2", "len3");

        Triangle subject = new Triangle(triangleLengths);
        // Example of checking if expected type
        assertEquals(Triangle.Type.ISOSCELES, subject.getTriangleType());
    }

    /**
     * Testing with numbers as what I'd expect the triangle to be made of
     *
     * Here you test with a triangle object
     *
     * Haven't tested what the 3 types is sorry :O
     */
    @Test
    public void testWithNumbersAsObject() {
        Triangle subject = new Triangle(4, 5.32, 7);
        assertEquals(Triangle.Type.SCALENE, subject);
    }

    /**
     * This piece you check the static method but have no object
     */
    @Test
    public void testWithNumbersStaticMethod() {
        assertEquals(Triangle.Type.ISOSCELES, Triangle.getTriangleType(3.4d, 4d, 1.111d));
    }
}

最后，您要测试src / Triangles.java的实际代码

import java.util.List;

/**
 * I created so you can both have an object of the triangle or make the check purely static
 * maybe you need an object type for later?
 */
public class Triangle {

    final double side0;
    final double side1;
    final double side2;

    public Triangle(List<String> triangleLengths) {
        this(Double.parseDouble(triangleLengths.get(0)),
                Double.parseDouble(triangleLengths.get(1)),
                Double.parseDouble(triangleLengths.get(2)));
    }

    public Triangle(double side0, double side1, double side2) {
        this.side0 = side0;
        this.side1 = side1;
        this.side2 = side2;
    }

    public Triangle.Type getTriangleType() {
        return Triangle.getTriangleType(side0, side1, side2);
    }

    public static Triangle.Type getTriangleType(double side0, double side1, double side2) {
        if (isEquilateral(side0, side1, side2)) {
            return Type.EQUILATERAL;
        } else if (isIsosceles(side0, side1, side2)) {
            return Type.ISOSCELES;
        } else if (isScalene(side0, side1, side2)) {
            return Type.SCALENE;
        } else {
            return Type.INVALID;
        }
    }

    private static boolean isScalene(double side0, double side1, double side2) {
        return side0 != side1 && side1 != side2;
    }

    private static boolean isIsosceles(double side0, double side1, double side2) {
        return side0 == side1 || side0 == side2 || side1 == side2;
    }

    private static boolean isEquilateral(double side0, double side1, double side2) {
        return side0 == side1 && side1 == side2;
    }

    public enum Type {
        EQUILATERAL,
        ISOSCELES,
        SCALENE,
        INVALID
    }
}

希望这可以帮助。

请注意，我从Triangles类返回答案，而不是立即将其写出来。 而且只有在手动运行时，我才从main方法中写出来。

Answer 2

Wisienkas的答案就在现场。 但我认为更多的概念背景会有所帮助。

您会发现，当您开始编码时，使用扫描仪从stdin进行“用户输入”，就像“大事”。 您获取一些价值，对此进行一些处理； 并打印一些结果。

问题是：这不是“现实世界”的工作方式； 因此，您遇到了代码无法测试的问题。 这里要理解的关键是：您只需一点点抽象就可以使事情变得更好。

那些是：

1. 您的“处理”代码将进入其自己的类和方法。 理想情况下，那些方法（除数据外）在调用时作为参数工作。
2. 您的“处理”代码应返回其结果。 当您仅打印到System.out时，几乎不可能验证某些代码在做什么。

Answer 3

最好的办法是重新设计代码，以便

• 三角逻辑属于自己的类，
• 用户输入逻辑采用自己的方法，
• main方法绝对是最小的。

然后，您可以分别对三角形逻辑和用户输入逻辑进行单元测试。 如果main足够小，则无需编写单元测试即可摆脱困境。

但是，这不是您要的。 如果要完全按照编写的代码进行编码，可以通过将System.in和System.out设置为控制输入和输出的流来对其进行测试。

您的测试课程可能看起来像这样。 这会将System.in重定向为从您在每次测试中指定的String中读取，然后将System.out重定向为可随后对其内容进行验证的流中进行写入。

public class TrianglesTest {
    @Test
    public void identifiesEquilateralTriangle() {
        ByteArrayOutputStream output = setStreams("5 5 5 ");
        Triangles.main(new String[0]);
        assertEquals("This triangle is an equilateral triangle", output.toString().trim());
    }

    // Plus a whole lot more tests for other cases

    private ByteArrayOutputStream setStreams(String input) {
        System.setIn(new ByteArrayInputStream(input.getBytes()));
        ByteArrayOutputStream toReturn = new ByteArrayOutputStream();
        System.setOut(new PrintStream(toReturn));
        return toReturn;
    }
}