src/test/java/net/jaekl/squelch/util/ConsoleInputMock.java

   1 package net.jaekl.squelch.util;
   2
   3 import java.io.IOException;
   4 import java.io.PrintWriter;
   5
   6 public class ConsoleInputMock implements ConsoleInput {
   7         private int m_linePos;
   8         private String[] m_lines;
   9         private PrintWriter m_pw;
  10
  11         public ConsoleInputMock(PrintWriter pw, String[] lines) throws IOException {
  12                 m_linePos = 0;
  13                 m_lines = lines;
  14                 m_pw = pw;
  15         }
  16
  17         @Override
  18         public String readLine(String prompt)
  19         {
  20                 if (m_linePos >= m_lines.length) {
  21                         return null;
  22                 }
  23                 m_pw.print(prompt);
  24                 m_pw.flush();
  25                 String line = m_lines[m_linePos];
  26                 m_linePos++;
  27                 m_pw.println(line);
  28                 return line;
  29         }
  30 }