MockTypedInput.java

// Copyright 2013 Square, Inc.
package retrofit;

import retrofit.converter.Converter;
import retrofit.mime.TypedInput;

import java.io.ByteArrayInputStream;
import java.io.ByteArrayOutputStream;
import java.io.IOException;
import java.io.InputStream;

class MockTypedInput implements TypedInput
{
  private final Converter converter;
  private final Object    body;

  private byte[] bytes;

  MockTypedInput(Converter converter, Object body)
  {
    this.converter = converter;
    this.body = body;
  }

  @Override
  public String mimeType()
  {
    return "application/unknown";
  }

  @Override
  public long length()
  {
    try
    {
      initBytes();
    }
    catch (IOException e)
    {
      throw new RuntimeException(e);
    }
    return bytes.length;
  }

  @Override
  public InputStream in() throws IOException
  {
    initBytes();
    return new ByteArrayInputStream(bytes);
  }

  private synchronized void initBytes() throws IOException
  {
    if (bytes == null)
    {
      ByteArrayOutputStream out = new ByteArrayOutputStream();
      converter.toBody(body).writeTo(out);
      bytes = out.toByteArray();
    }
  }
}