content: /*
   * 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.camel.quarkus.component.file.it;
  
  import java.io.IOException;
  import java.nio.charset.Charset;
  import java.nio.charset.StandardCharsets;
  import java.nio.file.Path;
  import java.nio.file.Paths;
  import java.util.Arrays;
  import java.util.HashMap;
  import java.util.Map;
  import java.util.Map.Entry;
  
  import io.quarkus.test.common.QuarkusTestResourceLifecycleManager;
  import org.apache.commons.io.FileUtils;
  
  /**
   * Ensures that all test files are created with expected content before camel routes starts to consume them.
   */
  public class FileTestResource implements QuarkusTestResourceLifecycleManager {
  
      static final Path TEST_FILES_FOLDER = Paths.get("target", "test-files");
  
      static final String FILTER_NON_SKIPPED_FILE_NAME = "non-skipped-file";
      static final String FILTER_NON_SKIPPED_FILE_CONTENT = FILTER_NON_SKIPPED_FILE_NAME + "-CONTENT";
      static final String IDEMPOTENT_FILE_NAME = "moved-back-read-once";
      static final String IDEMPOTENT_FILE_CONTENT = IDEMPOTENT_FILE_NAME + "-CONTENT";
      static final String POLL_ENRICH_FILE_NAME = "poll-enrich-file";
      static final String POLL_ENRICH_FILE_CONTENT = POLL_ENRICH_FILE_NAME + "-CONTENT";
      static final String QUARTZ_SCHEDULED_FILE_NAME = "quartz-schedule-file";
      static final String QUARTZ_SCHEDULED_FILE_CONTENT = QUARTZ_SCHEDULED_FILE_NAME + "-CONTENT";
      static final String CHARSET_READ_FILE_NAME = "charset-read-file";
      static final String CHARSET_READ_FILE_CONTENT = "A string with \u00F0 char to be read";
      static final String CHARSET_WRITE_FILE_CREATION_FOLDER = "charset-write";
      static final String CHARSET_WRITE_FILE_NAME = "charset-write-file";
      static final String CHARSET_WRITE_FILE_CONTENT = "A string with \u00F0 char to be written";
      static final String FILE_CREATION_FOLDER = "file-creation";
      static final String FILE_CREATION_FILE_NAME = "should-be-created";
      static final String FILE_CREATION_FILE_CONTENT = FILE_CREATION_FILE_NAME + "-CONTENT";
      static final String READ_LOCK_FOLDER_IN = "read-lock-in";
      static final String READ_LOCK_FOLDER_OUT = "read-lock-out";
      static final String READ_LOCK_FILE_NAME = "read-lock-file";
      static final String BATCH_FOLDER_IN = "batch-in";
      static final String BATCH_FILE_NAME_1 = "batch-file-1";
      static final String BATCH_FILE_NAME_1_CONTENT = BATCH_FILE_NAME_1 + "-CONTENT";
      static final String BATCH_FILE_NAME_2 = "batch-file-2";
      static final String BATCH_FILE_NAME_2_CONTENT = BATCH_FILE_NAME_2 + "-CONTENT";
      static final String SORT_BY_FOLDER_IN = "sortby-in";
      static final String SORT_BY_NAME_1 = "sortby-file-1";
      static final String SORT_BY_NAME_1_CONTENT = SORT_BY_NAME_1 + "-CONTENT";
      static final String SORT_BY_NAME_2 = "sortby-file-2";
      static final String SORT_BY_NAME_2_CONTENT = SORT_BY_NAME_2 + "-CONTENT";
      static final String SORT_BY_NAME_3 = "sortby-file-3";
      static final String SORT_BY_NAME_3_CONTENT = SORT_BY_NAME_3 + "-CONTENT";
  
      private final Map<Path, byte[]> createdTestFilesExpectedContent = new HashMap<>();
  
      @Override
      public Map<String, String> start() {
          try {
              FileUtils.deleteDirectory(TEST_FILES_FOLDER.toFile());
  
              createTestFile("filter", FILTER_NON_SKIPPED_FILE_NAME);
              createTestFile("filter", "skipped_file");
  
              createTestFile("idempotent", IDEMPOTENT_FILE_NAME);
  
              createTestFile("poll-enrich", POLL_ENRICH_FILE_NAME);
  
              createTestFile("quartz-scheduled", QUARTZ_SCHEDULED_FILE_NAME);
  
              createTestFile("charset-read", CHARSET_READ_FILE_NAME, CHARSET_READ_FILE_CONTENT, StandardCharsets.ISO_8859_1);
  
              createTestFile("read-lock-in", READ_LOCK_FILE_NAME);
  
              createTestFile("batch-in", BATCH_FILE_NAME_1);
              createTestFile("batch-in", BATCH_FILE_NAME_2);
  
              createTestFile("sortby-in", SORT_BY_NAME_3);
              createTestFile("sortby-in", SORT_BY_NAME_1);
              createTestFile("sortby-in", SORT_BY_NAME_2);
  
              // Do not use  createTestFile("file-creation"... as it is already reserved by the createFileShouldSucceed test
              // Do not use  createTestFile("charset-write"... as it is already reserved by the writeFileWithIso8859_1CharsetShouldSucceed test
              // Do not use  createTestFile("read-lock-out"... as it is already reserved by the readLockCantBeAcquiredOnFileSmallerThanReadLockMinLength test
  
              ensureAllTestFilesCreatedWithExpectedContent();
          } catch (Exception ex) {
              throw new RuntimeException("Problem while initializing test files", ex);
          }
          return null;
      }
  
      private void createTestFile(String testName, String testFileName, String content, Charset charset) throws IOException {
          Path testFilePath = TEST_FILES_FOLDER.resolve(Paths.get(testName, testFileName));
          byte[] contentBytes = content.getBytes(charset);
          FileUtils.writeByteArrayToFile(testFilePath.toFile(), contentBytes);
          createdTestFilesExpectedContent.put(testFilePath, contentBytes);
      }
  
      private void createTestFile(String testName, String testFileName) throws IOException {
          createTestFile(testName, testFileName, testFileName + "-CONTENT", StandardCharsets.UTF_8);
      }
  
      private void ensureAllTestFilesCreatedWithExpectedContent() {
          for (Entry<Path, byte[]> createdTestFileExpectedContent : createdTestFilesExpectedContent.entrySet()) {
              Path path = createdTestFileExpectedContent.getKey();
              try {
                  byte[] content = FileUtils.readFileToByteArray(path.toFile());
                  byte[] expectedContent = createdTestFileExpectedContent.getValue();
  
                  if (!Arrays.equals(expectedContent, content)) {
                      String format = "The NonFlakyFileTestResource failed to create the test file %s with expected content";
                      throw new RuntimeException(String.format(format, path));
                  }
              } catch (IOException ioex) {
                  String format = "The NonFlakyFileTestResource has exception while creating the test file %s with expected content";
                  throw new RuntimeException(String.format(format, path), ioex);
              }
          }
      }
  
      @Override
      public void stop() {
      }
  
  }
  