/*
    * 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.shared.dependency.analyzer;
  
  import java.io.ByteArrayOutputStream;
  import java.io.FileOutputStream;
  import java.io.IOException;
  import java.nio.charset.StandardCharsets;
  import java.nio.file.Files;
  import java.nio.file.Path;
  import java.util.HashSet;
  import java.util.Set;
  import java.util.jar.JarOutputStream;
  import java.util.zip.ZipEntry;
  import java.util.zip.ZipException;
  
  import org.junit.jupiter.api.BeforeEach;
  import org.junit.jupiter.api.Test;
  import org.junit.jupiter.api.io.TempDir;
  
  import static org.assertj.core.api.Assertions.assertThat;
  import static org.assertj.core.api.Assertions.fail;
  
  /**
   * Tests <code>DefaultClassAnalyzer</code>.
   *
   * @author <a href="mailto:markhobson@gmail.com">Mark Hobson</a>
   * @see DefaultClassAnalyzer
   */
  class DefaultClassAnalyzerTest {
  
      @TempDir
      private Path tempDir;
  
      private Path file;
  
      @BeforeEach
      void setUp() throws IOException {
          file = Files.createTempFile(tempDir, "test", ".jar");
          try (JarOutputStream out = new JarOutputStream(new FileOutputStream(file.toFile()))) {
              addZipEntry(out, "a/b/c.class", "class a.b.c");
              addZipEntry(out, "x/y/z.class", "class x.y.z");
          }
      }
  
      @Test
      void testAnalyzeWithJar() throws IOException {
          Set<String> expectedClasses = new HashSet<>();
          expectedClasses.add("a.b.c");
          expectedClasses.add("x.y.z");
  
          DefaultClassAnalyzer analyzer = new DefaultClassAnalyzer();
          Set<String> actualClasses = analyzer.analyze(file.toUri().toURL());
  
          assertThat(actualClasses).isEqualTo(expectedClasses);
      }
  
      @Test
      void testAnalyzeBadJar() throws IOException {
          // to reproduce MDEP-143
          // corrupt the jar file by altering its contents
          ByteArrayOutputStream baos = new ByteArrayOutputStream(100);
          Files.copy(file, baos);
          byte[] ba = baos.toByteArray();
          ba[50] = 1;
          Files.write(file, ba);
  
          ClassAnalyzer analyzer = new DefaultClassAnalyzer();
  
          try {
              analyzer.analyze(file.toUri().toURL());
              fail("Exception expected");
          } catch (ZipException e) {
              assertThat(e).hasMessageStartingWith("Cannot process Jar entry on URL:");
          }
      }
  
      private void addZipEntry(JarOutputStream out, String fileName, String content) throws IOException {
          out.putNextEntry(new ZipEntry(fileName));
          byte[] bytes = content.getBytes(StandardCharsets.UTF_8);
          out.write(bytes, 0, bytes.length);
      }
  }