StorageTestUtils.java

package de.is24.infrastructure.gridfs.http.storage;

import de.is24.infrastructure.gridfs.http.gridfs.StorageService;

import java.io.IOException;

import static de.is24.infrastructure.gridfs.http.utils.RepositoryUtils.simpleInputStream;
import static de.is24.infrastructure.gridfs.http.utils.RepositoryUtils.uniqueRepoName;

public class StorageTestUtils {

  private final StorageService storageService;
  private final FileStorageService fileStorageService;

  public StorageTestUtils(StorageService storageService, FileStorageService fileStorageService) {
    this.storageService = storageService;
    this.fileStorageService = fileStorageService;
  }

  public static final String REPOMD_PATH = "/repodata/repomd.xml";
  public static final String PRIMARY_XMl_PATH = "/repodata/primary.xml.gz";
  public static final String METADATA_PATH = "/repodate/generation-metadata.yaml";

  public String givenFullRepository() throws Exception {
    String repoName = uniqueRepoName();
    storageService.storeRpm(repoName, getClass().getResourceAsStream("/rpms/valid.noarch.rpm"));
    storeFile(repoName, REPOMD_PATH);
    storeFile(repoName, PRIMARY_XMl_PATH);
    storeFile(repoName, METADATA_PATH);
    return repoName;
  }

  public void storeFile(String repoName, String path) {
    fileStorageService.storeFile(simpleInputStream(), new FileDescriptor(repoName + path));
  }

  public FileStorageItem givenFileWithDescriptor(FileDescriptor descriptor) throws IOException {
    return fileStorageService.storeFile(simpleInputStream(), descriptor);
  }
}