/**
 * Licensed under the Apache License, Version 2.0 (the "License"); you may not use this file except in compliance with
 * the License. You may obtain a copy of the License at
 *
 * http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software distributed under the License is distributed on
 * an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the License for the
 * specific language governing permissions and limitations under the License.
 *
 * Copyright 2012-2015 the original author or authors.
 */
package org.assertj.swing.testng.listener;

import static org.assertj.core.api.Assertions.assertThat;
import static org.assertj.core.util.Files.temporaryFolderPath;
import static org.assertj.core.util.Strings.concat;
import static org.assertj.core.util.Strings.join;
import static org.assertj.swing.assertions.Assertions.assertThat;
import static org.assertj.swing.util.ImageReader.readImageFrom;

import java.awt.Toolkit;
import java.text.SimpleDateFormat;
import java.util.Date;
import java.util.GregorianCalendar;
import java.util.List;

import org.assertj.swing.annotation.GUITest;
import org.junit.BeforeClass;
import org.junit.Test;
import org.testng.Reporter;

/**
 * Tests for <code>{@link ScreenshotOnFailureListener}</code>.
 * 
 * @author Yvonne Wang
 */
public class ScreenshotOnFailureListener_Test {

  @GUITest
  public static class SomeGUITestClass {
    @GUITest
    public void someGUITestMethod() {
    }

    public void someNonGUITestMethod() {
    }
  }

  static TestContextStub testContext;
  static TestResultStub testResult;
  static ScreenshotOnFailureListener listener;

  @BeforeClass
  public static void setUpOnce() {
    testContext = new TestContextStub();
    testResult = new TestResultStub();
    listener = new ScreenshotOnFailureListener();
  }

  @Test
  public void should_Get_Output_Folder_On_Start() {
    String outputFolder = temporaryFolderPath();
    testContext.setOutputDirectory(outputFolder);
    listener.onStart(testContext);
    assertThat(listener.output()).isEqualTo(outputFolder);
  }

  @Test
  public void should_Take_Screenshot_On_Test_Failure() throws Exception {
    setUpStubsForScreenshot();
    listener.onTestFailure(testResult);
    String imageFileName = screenshotFileName();
    String screenshotPath = concat(testContext.getOutputDirectory(), imageFileName);
    assertThat(readImageFrom(screenshotPath)).hasSize(Toolkit.getDefaultToolkit().getScreenSize());
    assertThatScreenshotHyperlinkWasAddedToReport(imageFileName);
  }

  private void setUpStubsForScreenshot() {
    Date now = new GregorianCalendar().getTime();
    ClassStub testClass = testResult.getTestClass();
    testClass.setName(new SimpleDateFormat("yyMMdd").format(now));
    testClass.setRealClass(SomeGUITestClass.class);
    testResult.getMethod().setMethodName(new SimpleDateFormat("hhmmss").format(now));
  }

  private String screenshotFileName() {
    String className = testResult.getTestClass().getName();
    String methodName = testResult.getMethod().getMethodName();
    return join(className, methodName, "png").with(".");
  }

  private void assertThatScreenshotHyperlinkWasAddedToReport(String imageFileName) {
    List<String> reporterOutput = Reporter.getOutput();
    assertThat(reporterOutput).hasSize(1);
    assertThat(reporterOutput.get(0)).isEqualTo(concat("<a href=\"", imageFileName, "\">Screenshot</a>"));
  }
}