package org.jboss.resteasy.test.stream.resource;

import java.util.HashMap;
import java.util.Map;

import javax.ws.rs.core.MediaType;

import org.jboss.resteasy.annotations.Stream;
import org.junit.Assert;

public class StreamRawMediaTypes {

   public static final MediaType applicationOctetStreamDefault;
   public static final MediaType applicationOctetStreamTrue;
   public static final MediaType textPlainDefault;
   public static final MediaType textPlainTrue;


   static {
      Map<String, String> map = new HashMap<String, String>();
      map.put("x", "y");
      applicationOctetStreamDefault = new MediaType("application", "octet-stream", map);

      map.put(Stream.INCLUDE_STREAMING_PARAMETER, "true");
      applicationOctetStreamTrue = new MediaType("application", "octet-stream", map);

      map.clear();
      map.put("charset", "UTF-8");
      textPlainDefault = new MediaType("text", "plain", map);

      map.put(Stream.INCLUDE_STREAMING_PARAMETER, "true");
      textPlainTrue = new MediaType("text", "plain", map);
   }

   public static void testMediaType(String type, String include, MediaType actual) {

      outer:
         switch (type) {

            case "byte":
               switch (include) {
                  case "default":
                  case "false":
                     Assert.assertEquals(StreamRawMediaTypes.applicationOctetStreamDefault, actual);
                     break outer;

                  case "true":
                     Assert.assertEquals(StreamRawMediaTypes.applicationOctetStreamTrue, actual);
                     break outer;
               }

            case "char":
               switch (include) {
                  case "default":
                  case "false":
                     Assert.assertEquals(StreamRawMediaTypes.textPlainDefault, actual);
                     break outer;

                  case "true":
                     Assert.assertEquals(StreamRawMediaTypes.textPlainTrue, actual);
                     break outer;
               }
         }
   }
}
