VariantListBuilderImplTest.java

/*******************************************************************************
 * Licensed to the Apache Software Foundation (ASF) under one
 * or more contributor license agreements.  See the NOTICE file
 * distributed with this work for additional information
 * regarding copyright ownership.  The ASF licenses this file
 * to you under the Apache License, Version 2.0 (the
 * "License"); you may not use this file except in compliance
 * with the License.  You may obtain a copy of the License at
 *  
 *   http://www.apache.org/licenses/LICENSE-2.0
 *  
 *  Unless required by applicable law or agreed to in writing,
 *  software distributed under the License is distributed on an
 *  "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
 *  KIND, either express or implied.  See the License for the
 *  specific language governing permissions and limitations
 *  under the License.
 *  
 *******************************************************************************/

package org.apache.wink.common.internal;

import static org.junit.Assert.assertArrayEquals;

import java.util.List;
import java.util.Locale;

import javax.ws.rs.core.MediaType;
import javax.ws.rs.core.Variant;
import javax.ws.rs.core.Variant.VariantListBuilder;

import junit.framework.TestCase;

public class VariantListBuilderImplTest extends TestCase {

    public void testBuildMediaTypes() {
        Variant variant1 = new Variant(MediaType.APPLICATION_ATOM_XML_TYPE, null, null);
        Variant variant2 = new Variant(MediaType.APPLICATION_JSON_TYPE, null, null);

        VariantListBuilder builder = new VariantListBuilderImpl();
        List<Variant> list = builder.mediaTypes(MediaType.APPLICATION_ATOM_XML_TYPE).add().build();
        assertList(new Variant[] {variant1}, list);

        list =
            builder
                .mediaTypes(MediaType.APPLICATION_ATOM_XML_TYPE, MediaType.APPLICATION_JSON_TYPE)
                .add().build();
        assertList(new Variant[] {variant1, variant2}, list);
    }

    public void testBuildLanguages() {
        Variant variant1 = new Variant(null, Locale.ENGLISH, null);
        Variant variant2 = new Variant(null, Locale.CHINESE, null);

        VariantListBuilder builder = new VariantListBuilderImpl();
        List<Variant> list = builder.languages(Locale.ENGLISH).add().build();
        assertList(new Variant[] {variant1}, list);

        list = builder.languages(Locale.ENGLISH, Locale.CHINESE).add().build();
        assertList(new Variant[] {variant1, variant2}, list);
    }

    public void testBuildEncodings() {
        Variant variant1 = new Variant(null, null, "UTF-8");
        Variant variant2 = new Variant(null, null, "UTF-16");

        VariantListBuilder builder = new VariantListBuilderImpl();
        List<Variant> list = builder.encodings("UTF-8").add().build();
        assertList(new Variant[] {variant1}, list);

        list = builder.encodings("UTF-8", "UTF-16").add().build();
        assertList(new Variant[] {variant1, variant2}, list);
    }

    public void testBuildMediaTypesAndLanguages() {
        Variant variant1 = new Variant(MediaType.APPLICATION_ATOM_XML_TYPE, Locale.ENGLISH, null);
        Variant variant2 = new Variant(MediaType.APPLICATION_ATOM_XML_TYPE, Locale.CHINESE, null);
        Variant variant3 = new Variant(MediaType.APPLICATION_JSON_TYPE, Locale.ENGLISH, null);
        Variant variant4 = new Variant(MediaType.APPLICATION_JSON_TYPE, Locale.CHINESE, null);

        VariantListBuilder builder = new VariantListBuilderImpl();
        List<Variant> list = null;

        list =
            builder.mediaTypes(MediaType.APPLICATION_ATOM_XML_TYPE).languages(Locale.ENGLISH).add()
                .build();
        assertList(new Variant[] {variant1}, list);

        list =
            builder.mediaTypes(MediaType.APPLICATION_ATOM_XML_TYPE).languages(Locale.CHINESE).add()
                .build();
        assertList(new Variant[] {variant2}, list);

        list =
            builder.mediaTypes(MediaType.APPLICATION_ATOM_XML_TYPE).languages(Locale.ENGLISH,
                                                                              Locale.CHINESE).add()
                .build();
        assertList(new Variant[] {variant1, variant2}, list);

        list =
            builder
                .mediaTypes(MediaType.APPLICATION_ATOM_XML_TYPE, MediaType.APPLICATION_JSON_TYPE)
                .languages(Locale.ENGLISH).add().build();
        assertList(new Variant[] {variant1, variant3}, list);

        list =
            builder
                .mediaTypes(MediaType.APPLICATION_ATOM_XML_TYPE, MediaType.APPLICATION_JSON_TYPE)
                .languages(Locale.ENGLISH, Locale.CHINESE).add().build();
        assertList(new Variant[] {variant1, variant2, variant3, variant4}, list);
    }

    public void testBuildMediaTypesAndEncodings() {
        Variant variant1 = new Variant(MediaType.APPLICATION_ATOM_XML_TYPE, null, "UTF-8");
        Variant variant2 = new Variant(MediaType.APPLICATION_ATOM_XML_TYPE, null, "UTF-16");
        Variant variant3 = new Variant(MediaType.APPLICATION_JSON_TYPE, null, "UTF-8");
        Variant variant4 = new Variant(MediaType.APPLICATION_JSON_TYPE, null, "UTF-16");

        VariantListBuilder builder = new VariantListBuilderImpl();
        List<Variant> list = null;

        list =
            builder.mediaTypes(MediaType.APPLICATION_ATOM_XML_TYPE).encodings("UTF-8").add()
                .build();
        assertList(new Variant[] {variant1}, list);

        list =
            builder.mediaTypes(MediaType.APPLICATION_ATOM_XML_TYPE).encodings("UTF-16").add()
                .build();
        assertList(new Variant[] {variant2}, list);

        list =
            builder.mediaTypes(MediaType.APPLICATION_ATOM_XML_TYPE).encodings("UTF-8", "UTF-16")
                .add().build();
        assertList(new Variant[] {variant1, variant2}, list);

        list =
            builder
                .mediaTypes(MediaType.APPLICATION_ATOM_XML_TYPE, MediaType.APPLICATION_JSON_TYPE)
                .encodings("UTF-8").add().build();
        assertList(new Variant[] {variant1, variant3}, list);

        list =
            builder
                .mediaTypes(MediaType.APPLICATION_ATOM_XML_TYPE, MediaType.APPLICATION_JSON_TYPE)
                .encodings("UTF-8", "UTF-16").add().build();
        assertList(new Variant[] {variant1, variant2, variant3, variant4}, list);
    }

    public void testBuildLanguagesAndEncodings() {
        Variant variant1 = new Variant(null, Locale.ENGLISH, "UTF-8");
        Variant variant2 = new Variant(null, Locale.ENGLISH, "UTF-16");
        Variant variant3 = new Variant(null, Locale.CHINESE, "UTF-8");
        Variant variant4 = new Variant(null, Locale.CHINESE, "UTF-16");

        VariantListBuilder builder = new VariantListBuilderImpl();
        List<Variant> list = null;

        list = builder.languages(Locale.ENGLISH).encodings("UTF-8").add().build();
        assertList(new Variant[] {variant1}, list);

        list = builder.languages(Locale.ENGLISH).encodings("UTF-16").add().build();
        assertList(new Variant[] {variant2}, list);

        list = builder.languages(Locale.ENGLISH).encodings("UTF-8", "UTF-16").add().build();
        assertList(new Variant[] {variant1, variant2}, list);

        list = builder.languages(Locale.ENGLISH, Locale.CHINESE).encodings("UTF-8").add().build();
        assertList(new Variant[] {variant1, variant3}, list);

        list =
            builder.languages(Locale.ENGLISH, Locale.CHINESE).encodings("UTF-8", "UTF-16").add()
                .build();
        assertList(new Variant[] {variant1, variant2, variant3, variant4}, list);
    }

    public void testBuildMediaTypesAndLanguagesAndEncodings() {
        Variant variant1 =
            new Variant(MediaType.APPLICATION_ATOM_XML_TYPE, Locale.ENGLISH, "UTF-8");
        Variant variant2 =
            new Variant(MediaType.APPLICATION_ATOM_XML_TYPE, Locale.ENGLISH, "UTF-16");
        Variant variant3 =
            new Variant(MediaType.APPLICATION_ATOM_XML_TYPE, Locale.CHINESE, "UTF-8");
        Variant variant4 =
            new Variant(MediaType.APPLICATION_ATOM_XML_TYPE, Locale.CHINESE, "UTF-16");
        Variant variant5 = new Variant(MediaType.APPLICATION_JSON_TYPE, Locale.ENGLISH, "UTF-8");
        Variant variant6 = new Variant(MediaType.APPLICATION_JSON_TYPE, Locale.ENGLISH, "UTF-16");
        Variant variant7 = new Variant(MediaType.APPLICATION_JSON_TYPE, Locale.CHINESE, "UTF-8");
        Variant variant8 = new Variant(MediaType.APPLICATION_JSON_TYPE, Locale.CHINESE, "UTF-16");

        VariantListBuilder builder = new VariantListBuilderImpl();
        List<Variant> list = null;

        list =
            builder.mediaTypes(MediaType.APPLICATION_ATOM_XML_TYPE).languages(Locale.ENGLISH)
                .encodings("UTF-8").add().build();
        assertList(new Variant[] {variant1}, list);

        list =
            builder.mediaTypes(MediaType.APPLICATION_ATOM_XML_TYPE).languages(Locale.ENGLISH)
                .encodings("UTF-16").add().build();
        assertList(new Variant[] {variant2}, list);

        list =
            builder.mediaTypes(MediaType.APPLICATION_ATOM_XML_TYPE).languages(Locale.ENGLISH)
                .encodings("UTF-8", "UTF-16").add().build();
        assertList(new Variant[] {variant1, variant2}, list);

        list =
            builder.mediaTypes(MediaType.APPLICATION_ATOM_XML_TYPE).languages(Locale.ENGLISH,
                                                                              Locale.CHINESE)
                .encodings("UTF-8", "UTF-16").add().build();
        assertList(new Variant[] {variant1, variant2, variant3, variant4}, list);

        list =
            builder
                .mediaTypes(MediaType.APPLICATION_ATOM_XML_TYPE, MediaType.APPLICATION_JSON_TYPE)
                .languages(Locale.ENGLISH, Locale.CHINESE).encodings("UTF-8", "UTF-16").add()
                .build();
        assertList(new Variant[] {variant1, variant2, variant3, variant4, variant5, variant6,
            variant7, variant8}, list);
    }

    public void testBuildAllNull() {
        VariantListBuilder builder = new VariantListBuilderImpl();
        List<Variant> list = builder.build();
        assertEquals(0, list.size());

        try {
            builder.add();
            fail("expected IllegalArgumentException");
        } catch (IllegalArgumentException e) {
        }
    }

    private void assertList(Variant[] expected, List<Variant> actual) {
        assertArrayEquals(expected, actual.toArray());
    }
}