/*
    * Licensed to the Apache Software Foundation (ASF) under one
    * or more contributor license agreements.  See the NOTICE file
    * distributed with this work for additional information
    * regarding copyright ownership.  The ASF licenses this file
    * to you 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.
   */
  package org.apache.maven.surefire.its;
  
  import java.util.HashSet;
  import java.util.List;
  import java.util.Set;
  
  import org.apache.maven.plugins.surefire.report.ReportTestSuite;
  import org.apache.maven.surefire.its.fixture.HelperAssertions;
  import org.apache.maven.surefire.its.fixture.IntegrationTestSuiteResults;
  import org.apache.maven.surefire.its.fixture.OutputValidator;
  import org.apache.maven.surefire.its.fixture.SurefireJUnit4IntegrationTestCase;
  import org.junit.Test;
  
  import static org.junit.Assert.assertEquals;
  import static org.junit.Assert.fail;
  
  /**
   * Test running two test cases; confirms reporting works correctly
   *
   * @author <a href="mailto:dfabulich@apache.org">Dan Fabulich</a>
   */
  public class TwoTestCasesIT extends SurefireJUnit4IntegrationTestCase {
      @Test
      public void testTwoTestCases() {
          unpack("junit-twoTestCases")
                  .sysProp("testNgVersion", "5.7")
                  .sysProp("testNgClassifier", "jdk15")
                  .executeTest()
                  .verifyErrorFreeLog()
                  .assertTestSuiteResults(2, 0, 0, 0);
      }
  
      /**
       * Runs two tests encapsulated in a suite
       */
      @Test
      public void testTwoTestCaseSuite() {
          final OutputValidator outputValidator = unpack("junit-twoTestCaseSuite")
                  .sysProp("testNgVersion", "5.7")
                  .sysProp("testNgClassifier", "jdk15")
                  .executeTest();
          outputValidator.verifyErrorFreeLog().assertTestSuiteResults(2, 0, 0, 0);
          List<ReportTestSuite> reports = HelperAssertions.extractReports(outputValidator.getBaseDir());
          Set<String> classNames = extractClassNames(reports);
          assertContains(classNames, "junit.twoTestCaseSuite.BasicTest");
          assertContains(classNames, "junit.twoTestCaseSuite.TestTwo");
          assertEquals("wrong number of classes", 2, classNames.size());
          IntegrationTestSuiteResults results = HelperAssertions.parseReportList(reports);
          HelperAssertions.assertTestSuiteResults(2, 0, 0, 0, results);
      }
  
      private void assertContains(Set<String> set, String expected) {
          if (set.contains(expected)) {
              return;
          }
          fail("Set didn't contain " + expected);
      }
  
      private Set<String> extractClassNames(List<ReportTestSuite> reports) {
          HashSet<String> classNames = new HashSet<>();
          for (ReportTestSuite suite : reports) {
              classNames.add(suite.getFullClassName());
          }
          return classNames;
      }
  
      @Test
      public void testJunit4Suite() {
          final OutputValidator outputValidator = unpack("junit4-twoTestCaseSuite")
                  .sysProp("testNgVersion", "5.7")
                  .sysProp("testNgClassifier", "jdk15")
                  .executeTest();
          outputValidator.verifyErrorFreeLog().assertTestSuiteResults(2, 0, 0, 0);
  
          List<ReportTestSuite> reports = HelperAssertions.extractReports(outputValidator.getBaseDir());
          Set<String> classNames = extractClassNames(reports);
          assertContains(classNames, "twoTestCaseSuite.BasicTest");
          assertContains(classNames, "twoTestCaseSuite.Junit4TestTwo");
          assertEquals("wrong number of classes", 2, classNames.size());
          IntegrationTestSuiteResults results = HelperAssertions.parseReportList(reports);
          HelperAssertions.assertTestSuiteResults(2, 0, 0, 0, results);
     }
 
     @Test
     public void testTestNGSuite() {
         final OutputValidator outputValidator = unpack("testng-twoTestCaseSuite")
                 .sysProp("testNgVersion", "5.7")
                 .sysProp("testNgClassifier", "jdk15")
                 .executeTest();
         outputValidator.assertTestSuiteResults(2, 0, 0, 0);
         List<ReportTestSuite> reports = HelperAssertions.extractReports(outputValidator.getBaseDir());
         Set<String> classNames = extractClassNames(reports);
         assertContains(classNames, "testng.two.TestNGTestTwo");
         assertContains(classNames, "testng.two.TestNGSuiteTest");
         assertEquals("wrong number of classes", 2, classNames.size());
         IntegrationTestSuiteResults results = HelperAssertions.parseReportList(reports);
         HelperAssertions.assertTestSuiteResults(2, 0, 0, 0, results);
     }
 }