[Java] WebアプリにLoginしてDownloadするTest

DownloadTest.java

import static org.junit.Assert.*;
import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.io.PrintStream;
import java.net.CookieHandler;
import java.net.CookieManager;
import java.net.CookiePolicy;
import java.net.HttpURLConnection;
import java.net.MalformedURLException;
import java.net.URL;
import org.junit.After;
import org.junit.Before;
import org.junit.Test;

public class DownloadTest {
  private static final String BASE_URL = "http://localhost:8080/";
  private CookieManager cookie;

  @Before
  public void setUp() throws Exception {
    cookie = new CookieManager();
    cookie.setCookiePolicy(CookiePolicy.ACCEPT_ALL);
    CookieHandler.setDefault(cookie);
    login();
  }

  @After
  public void tearDown() throws Exception {
    logout();
  }

  @Test
  public void testDownload() throws Exception {
    HttpURLConnection con = openHttpURLConnection("download");
    assertEquals(200, con.getResponseCode());

    assertEquals("filename=\"test.csv\"",
        con.getHeaderField("Content-Disposition"));

    BufferedReader reader = new BufferedReader(new InputStreamReader(
        con.getInputStream()));
    assertEquals("Test download", reader.readLine());
    reader.close();
  }

  private void login() throws IOException, MalformedURLException {
    HttpURLConnection con = openHttpURLConnection("login");
    con.setDoOutput(true);
    PrintStream out = new PrintStream(con.getOutputStream());
    out.print("user=testuser&password=testpass");
    out.close();
    assertEquals(200, con.getResponseCode());
    // System.out.println(cookie.getCookieStore().getCookies());
  }

  private void logout() throws IOException, MalformedURLException {
    HttpURLConnection con = openHttpURLConnection("logout");
    assertEquals(200, con.getResponseCode());
  }

  private HttpURLConnection openHttpURLConnection(String url)
      throws IOException, MalformedURLException {
    return (HttpURLConnection) new URL(BASE_URL + url).openConnection();
  }
}