/*
 * Copyright (c) 2012-2018 Red Hat, Inc.
 * This program and the accompanying materials are made
 * available under the terms of the Eclipse Public License 2.0
 * which is available at https://www.eclipse.org/legal/epl-2.0/
 *
 * SPDX-License-Identifier: EPL-2.0
 *
 * Contributors:
 *   Red Hat, Inc. - initial API and implementation
 */
package org.eclipse.che.commons.lang;

import static org.mockito.ArgumentMatchers.any;
import static org.mockito.Mockito.mock;
import static org.mockito.Mockito.times;
import static org.mockito.Mockito.verify;

import java.io.File;
import java.io.FileOutputStream;
import java.io.IOException;
import java.io.InputStream;
import java.net.URL;
import java.nio.file.Files;
import java.nio.file.Path;
import java.nio.file.Paths;
import java.util.Random;
import java.util.function.Consumer;
import java.util.regex.Pattern;
import java.util.zip.ZipEntry;
import java.util.zip.ZipFile;
import java.util.zip.ZipOutputStream;
import org.testng.Assert;
import org.testng.annotations.BeforeMethod;
import org.testng.annotations.Test;

public class ZipUtilsTest {

  public static final String FILE_NAME = "test";
  private File zipFile;
  private byte[] testData;

  @BeforeMethod
  public void setUp() throws IOException {
    zipFile = File.createTempFile("test", "zip");
    zipFile.deleteOnExit();

    testData = new byte[2048];
    Random random = new Random();
    random.nextBytes(testData);

    try (ZipOutputStream zos = new ZipOutputStream(new FileOutputStream(zipFile))) {
      ZipEntry entry = new ZipEntry(FILE_NAME);
      entry.setSize(testData.length);
      zos.putNextEntry(entry);
      zos.write(testData);
      zos.closeEntry();
      zos.close();
    }
  }

  @Test
  public void shouldBeAbleToDetectZipFile() throws IOException {
    Assert.assertTrue(ZipUtils.isZipFile(zipFile));
  }

  @Test
  public void testGetResources() throws Exception {
    URL testJar = ZipUtilsTest.class.getResource("/che/che.jar");
    @SuppressWarnings("unchecked")
    Consumer<InputStream> consumer = mock(Consumer.class);

    ZipUtils.getResources(
        new ZipFile(testJar.getFile()), Pattern.compile(".*[//]?codenvy/[^//]+[.]json"), consumer);

    verify(consumer, times(2)).accept(any(InputStream.class));
  }

  @Test
  public void testUnzipFile() throws Exception {
    Path targetDir = Paths.get(zipFile.getParent());
    ZipUtils.unzip(zipFile, targetDir.toFile());
    Path unzippedFile = Paths.get(targetDir.toString(), FILE_NAME);
    Assert.assertEquals(testData, Files.readAllBytes(unzippedFile));
    Files.delete(unzippedFile);
  }
}
