/sigmah/src/test/java/org/sigmah/server/endpoint/file/FileManagerTestConfigModule.java

http://sigma-h.googlecode.com/ · Java · 49 lines · 33 code · 16 blank · 0 comment · 0 complexity · a5f12d78114285e137d429276c014aae MD5 · raw file

  1. package org.sigmah.server.endpoint.file;
  2. import java.io.ByteArrayInputStream;
  3. import java.io.ByteArrayOutputStream;
  4. import java.io.File;
  5. import java.io.IOException;
  6. import java.io.InputStream;
  7. import java.io.OutputStream;
  8. import java.util.Date;
  9. import java.util.HashMap;
  10. import java.util.Map;
  11. import java.util.Properties;
  12. import com.google.inject.AbstractModule;
  13. import com.google.inject.Provides;
  14. import com.google.inject.Singleton;
  15. public class FileManagerTestConfigModule extends AbstractModule {
  16. @Override
  17. protected void configure() {
  18. bind(FileStorageProvider.class).toInstance(new StorageProviderStub());
  19. }
  20. private static class StorageProviderStub implements FileStorageProvider {
  21. public Map<String, ByteArrayOutputStream> files = new HashMap<String, ByteArrayOutputStream>();
  22. @Override
  23. public OutputStream create(String storageId) throws IOException {
  24. ByteArrayOutputStream baos = new ByteArrayOutputStream();
  25. files.put(storageId, baos);
  26. return baos;
  27. }
  28. @Override
  29. public InputStream open(String storageId) throws IOException {
  30. return new ByteArrayInputStream(files.get(storageId).toByteArray());
  31. }
  32. }
  33. }