Thử nghiệm JUnit với đầu vào của người dùng được mô phỏng

Tôi đang cố gắng tạo một số thử nghiệm JUnit cho một phương pháp yêu cầu người dùng nhập. Phương pháp đang thử nghiệm trông giống như phương pháp sau:

public static int testUserInput() {
    Scanner keyboard = new Scanner(System.in);
    System.out.println("Give a number between 1 and 10");
    int input = keyboard.nextInt();

    while (input < 1 || input > 10) {
        System.out.println("Wrong number, try again.");
        input = keyboard.nextInt();
    }

    return input;
}

Có cách nào có thể để tự động chuyển chương trình một int thay vì tôi hoặc người khác thực hiện việc này theo cách thủ công trong phương pháp kiểm tra JUnit không? Giống như mô phỏng đầu vào của người dùng?

Cảm ơn trước.

Bạn có thể thay thế System.in bằng luồng của riêng bạn bằng cách gọi System.setIn (InputStream in) . InputStream có thể là một mảng byte:

InputStream sysInBackup = System.in; // backup System.in to restore it later
ByteArrayInputStream in = new ByteArrayInputStream("My string".getBytes());
System.setIn(in);

// do your thing

// optionally, reset System.in to its original
System.setIn(sysInBackup);

Có thể làm cho phương pháp này dễ kiểm tra hơn bằng cách chuyển IN và OUT dưới dạng các tham số:

public static int testUserInput(InputStream in,PrintStream out) {
    Scanner keyboard = new Scanner(in);
    out.println("Give a number between 1 and 10");
    int input = keyboard.nextInt();

    while (input < 1 || input > 10) {
        out.println("Wrong number, try again.");
        input = keyboard.nextInt();
    }

    return input;
}

Để kiểm tra mã của bạn, bạn nên tạo một trình bao bọc cho các chức năng nhập / xuất hệ thống. Bạn có thể thực hiện việc này bằng cách sử dụng phương pháp tiêm phụ thuộc, cung cấp cho chúng tôi một lớp có thể yêu cầu các số nguyên mới:

public static class IntegerAsker {
    private final Scanner scanner;
    private final PrintStream out;

    public IntegerAsker(InputStream in, PrintStream out) {
        scanner = new Scanner(in);
        this.out = out;
    }

    public int ask(String message) {
        out.println(message);
        return scanner.nextInt();
    }
}

Sau đó, bạn có thể tạo các bài kiểm tra cho chức năng của mình, bằng cách sử dụng một khuôn khổ giả (tôi sử dụng Mockito):

@Test
public void getsIntegerWhenWithinBoundsOfOneToTen() throws Exception {
    IntegerAsker asker = mock(IntegerAsker.class);
    when(asker.ask(anyString())).thenReturn(3);

    assertEquals(getBoundIntegerFromUser(asker), 3);
}

@Test
public void asksForNewIntegerWhenOutsideBoundsOfOneToTen() throws Exception {
    IntegerAsker asker = mock(IntegerAsker.class);
    when(asker.ask("Give a number between 1 and 10")).thenReturn(99);
    when(asker.ask("Wrong number, try again.")).thenReturn(3);

    getBoundIntegerFromUser(asker);

    verify(asker).ask("Wrong number, try again.");
}

Sau đó, viết hàm của bạn vượt qua các bài kiểm tra. Hàm này gọn gàng hơn nhiều vì bạn có thể loại bỏ sự trùng lặp yêu cầu / nhận số nguyên và các lệnh gọi hệ thống thực tế được đóng gói.

public static void main(String[] args) {
    getBoundIntegerFromUser(new IntegerAsker(System.in, System.out));
}

public static int getBoundIntegerFromUser(IntegerAsker asker) {
    int input = asker.ask("Give a number between 1 and 10");
    while (input < 1 || input > 10)
        input = asker.ask("Wrong number, try again.");
    return input;
}

Điều này có vẻ như quá mức cần thiết đối với ví dụ nhỏ của bạn, nhưng nếu bạn đang xây dựng một ứng dụng lớn hơn đang phát triển như thế này thì có thể thu được lợi nhuận khá nhanh.

Một cách phổ biến để kiểm tra mã tương tự là trích xuất một phương pháp có trong Máy quét và Máy in, tương tự như câu trả lời StackOverflow này và kiểm tra rằng:

public void processUserInput() {
  processUserInput(new Scanner(System.in), System.out);
}

/** For testing. Package-private if possible. */
public void processUserInput(Scanner scanner, PrintWriter output) {
  output.println("Give a number between 1 and 10");
  int input = scanner.nextInt();

  while (input < 1 || input > 10) {
    output.println("Wrong number, try again.");
    input = scanner.nextInt();
  }

  return input;
}

Xin lưu ý rằng bạn sẽ không thể đọc đầu ra của mình cho đến khi kết thúc và bạn sẽ phải chỉ định trước tất cả đầu vào của mình:

@Test
public void shouldProcessUserInput() {
  StringWriter output = new StringWriter();
  String input = "11\n"       // "Wrong number, try again."
               + "10\n";

  assertEquals(10, systemUnderTest.processUserInput(
      new Scanner(input), new PrintWriter(output)));

  assertThat(output.toString(), contains("Wrong number, try again.")););
}

Tất nhiên, thay vì tạo một phương thức quá tải, bạn cũng có thể giữ "máy quét" và "đầu ra" dưới dạng các trường có thể thay đổi trong hệ thống của bạn đang được thử nghiệm. Tôi có xu hướng muốn giữ các lớp học càng không có trạng thái càng tốt, nhưng đó không phải là một nhượng bộ quá lớn nếu nó quan trọng với bạn hoặc đồng nghiệp / người hướng dẫn của bạn.

Bạn cũng có thể chọn đặt mã thử nghiệm của mình trong cùng một gói Java với mã đang thử nghiệm (ngay cả khi nó nằm trong một thư mục nguồn khác), điều này cho phép bạn giảm khả năng hiển thị của quá tải hai tham số thành gói riêng tư.

Tôi đã tìm ra một cách đơn giản hơn. Tuy nhiên, bạn phải sử dụng thư viện bên ngoài System.rules của @Stefan Birkner

Tôi chỉ lấy ví dụ được cung cấp ở đó, tôi nghĩ nó không thể trở nên đơn giản hơn:

import java.util.Scanner;

public class Summarize {
  public static int sumOfNumbersFromSystemIn() {
    Scanner scanner = new Scanner(System.in);
    int firstSummand = scanner.nextInt();
    int secondSummand = scanner.nextInt();
    return firstSummand + secondSummand;
  }
}

Kiểm tra

import static org.junit.Assert.*;
import static org.junit.contrib.java.lang.system.TextFromStandardInputStream.*;

import org.junit.Rule;
import org.junit.Test;
import org.junit.contrib.java.lang.system.TextFromStandardInputStream;

public class SummarizeTest {
  @Rule
  public final TextFromStandardInputStream systemInMock
    = emptyStandardInputStream();

  @Test
  public void summarizesTwoNumbers() {
    systemInMock.provideLines("1", "2");
    assertEquals(3, Summarize.sumOfNumbersFromSystemIn());
  }
}

Tuy nhiên, vấn đề trong trường hợp đầu vào thứ hai của tôi có khoảng trắng và điều này làm cho toàn bộ luồng đầu vào rỗng!

Bạn có thể bắt đầu bằng cách trích xuất logic truy xuất số từ bàn phím thành phương pháp riêng của nó. Sau đó, bạn có thể kiểm tra logic xác thực mà không cần lo lắng về bàn phím. Để kiểm tra lệnh gọi keyboard.nextInt (), bạn có thể muốn xem xét sử dụng một đối tượng giả.

Tôi đã khắc phục sự cố về việc đọc từ stdin để mô phỏng bảng điều khiển ...

Vấn đề của tôi là tôi muốn thử viết trong JUnit, kiểm tra bảng điều khiển để tạo một đối tượng nhất định ...

Vấn đề giống như tất cả những gì bạn nói: Làm thế nào tôi có thể viết trong bài kiểm tra Stdin từ JUnit?

Sau đó, ở trường đại học, tôi học về chuyển hướng giống như bạn nói System.setIn (InputStream) thay đổi bộ ký hiệu stdin và sau đó bạn có thể viết ...

Nhưng có một vấn đề nữa cần khắc phục ... khối kiểm tra JUnit đang đợi đọc từ InputStream mới của bạn, vì vậy bạn cần tạo một chuỗi để đọc từ InputStream và từ kiểm tra JUnit. Viết chuỗi trong Stdin mới ... Đầu tiên bạn phải viết trong Stdin vì nếu bạn viết sau khi tạo Chủ đề để đọc từ stdin, bạn có thể sẽ có Điều kiện chủng tộc ... bạn có thể viết trong InputStream trước để đọc hoặc bạn có thể đọc từ InputStream trước khi ghi ...

Đây là mã của tôi, kỹ năng tiếng Anh của tôi kém, tôi hy vọng tất cả các bạn có thể hiểu được vấn đề và giải pháp để mô phỏng viết bằng stdin từ bài kiểm tra JUnit.

private void readFromConsole(String data) throws InterruptedException {
    System.setIn(new ByteArrayInputStream(data.getBytes()));

    Thread rC = new Thread() {
        @Override
        public void run() {
            study = new Study();
            study.read(System.in);
        }
    };
    rC.start();
    rC.join();      
}

Tôi thấy hữu ích khi tạo một giao diện xác định các phương thức tương tự như java.io.Console và sau đó sử dụng nó để đọc hoặc ghi vào System.out. Việc triển khai thực sự sẽ ủy quyền cho System.console () trong khi phiên bản JUnit của bạn có thể là một đối tượng giả với đầu vào soạn sẵn và các phản hồi mong đợi.

Ví dụ: bạn sẽ xây dựng MockConsole chứa dữ liệu đầu vào được soạn sẵn từ người dùng. Việc triển khai mô hình sẽ bật một chuỗi đầu vào ra khỏi danh sách mỗi khi readLine được gọi. Nó cũng sẽ tập hợp tất cả đầu ra được ghi vào danh sách các phản hồi. Vào cuối bài kiểm tra, nếu mọi thứ diễn ra tốt đẹp, thì tất cả đầu vào của bạn sẽ được đọc và bạn có thể khẳng định trên đầu ra.