package com.atlassian.plugin.cache.filecache.impl;
   
   import com.atlassian.plugin.cache.filecache.FileCacheStreamProvider;
   import com.atlassian.plugin.servlet.DownloadException;
   
   import java.io.IOException;
   import java.io.OutputStream;
   
   public class MockFileCacheStreamProvider implements FileCacheStreamProvider {
      final String contents;
      final RuntimeException re;
      int callCount = 0;
  
      public MockFileCacheStreamProvider(String contents) {
          this(contents, null);
      }
      public MockFileCacheStreamProvider(String contents, RuntimeException re) {
          this.contents = contents;
          this.re = re;
      }
  
      @Override
      public void writeStream(OutputStream dest) throws DownloadException {
          callCount++;
          try {
              dest.write(contents.getBytes());
              if (re != null) {
                  throw re;
              }
          } catch (IOException e) {
              throw new DownloadException(e);
          }
      }
  }