ConventionsTest.java

/*
 * Copyright 2017 MongoDB, Inc.
 *
 * Licensed 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.bson.codecs.pojo;

import org.bson.codecs.pojo.entities.SimpleModel;
import org.bson.codecs.pojo.entities.conventions.AnnotationDefaultsModel;
import org.bson.codecs.pojo.entities.conventions.AnnotationModel;
import org.junit.Test;

import static java.util.Collections.singletonList;
import static junit.framework.TestCase.assertFalse;
import static junit.framework.TestCase.assertTrue;
import static org.bson.codecs.pojo.Conventions.ANNOTATION_CONVENTION;
import static org.bson.codecs.pojo.Conventions.CLASS_AND_FIELD_CONVENTION;
import static org.bson.codecs.pojo.Conventions.DEFAULT_CONVENTIONS;
import static org.junit.Assert.assertEquals;
import static org.junit.Assert.assertNotNull;
import static org.junit.Assert.assertNull;

public final class ConventionsTest {

    @Test
    public void testDefaultConventions() {
        ClassModel<AnnotationModel> classModel = ClassModel.builder(AnnotationModel.class)
                .conventions(DEFAULT_CONVENTIONS).build();

        assertTrue(classModel.useDiscriminator());
        assertEquals("_cls", classModel.getDiscriminatorKey());
        assertEquals("MyAnnotationModel", classModel.getDiscriminator());

        assertEquals(3, classModel.getFieldModels().size());
        FieldModel<?> idFieldModel = classModel.getIdFieldModel();
        assertNotNull(idFieldModel);
        assertEquals("customId", idFieldModel.getFieldName());
        assertEquals("_id", idFieldModel.getDocumentFieldName());

        FieldModel<?> childFieldModel = classModel.getFieldModel("child");
        assertNotNull(childFieldModel);
        assertFalse(childFieldModel.useDiscriminator());

        FieldModel<?> renamedFieldModel = classModel.getFieldModel("renamed");
        assertNotNull(renamedFieldModel);
    }

    @Test
    public void testAnnotationDefaults() {
        ClassModel<AnnotationDefaultsModel> classModel = ClassModel.builder(AnnotationDefaultsModel.class)
                .conventions(singletonList(ANNOTATION_CONVENTION)).build();

        assertTrue(classModel.useDiscriminator());
        assertEquals("_t", classModel.getDiscriminatorKey());
        assertEquals("AnnotationDefaultsModel", classModel.getDiscriminator());

        assertEquals(2, classModel.getFieldModels().size());
        FieldModel<?> idFieldModel = classModel.getIdFieldModel();
        assertNotNull(idFieldModel);
        assertEquals("customId", idFieldModel.getFieldName());
        assertEquals("_id", idFieldModel.getDocumentFieldName());

        FieldModel<?> childFieldModel = classModel.getFieldModel("child");
        assertNotNull(childFieldModel);
        assertFalse(childFieldModel.useDiscriminator());
    }

    @Test
    @SuppressWarnings("unchecked")
    public void testClassAndFieldConventionDoesNotOverwrite() {
        ClassModelBuilder<SimpleModel> builder = ClassModel.builder(SimpleModel.class)
                .enableDiscriminator(true)
                .discriminatorKey("_cls")
                .discriminator("Simples")
                .conventions(singletonList(CLASS_AND_FIELD_CONVENTION))
                .instanceCreatorFactory(new InstanceCreatorFactory<SimpleModel>() {
                    @Override
                    public InstanceCreator<SimpleModel> create() {
                        return null;
                    }
                });

        FieldModelBuilder<Integer> fieldModelBuilder = (FieldModelBuilder<Integer>) builder.getField("integerField");
        fieldModelBuilder.documentFieldName("id")
                .fieldSerialization(new FieldModelSerializationImpl<Integer>())
                .fieldAccessor(new FieldAccessorTest<Integer>());

        FieldModelBuilder<String> fieldModelBuilder2 = (FieldModelBuilder<String>) builder.getField("stringField");
        fieldModelBuilder2.documentFieldName("_id")
                .fieldSerialization(new FieldModelSerializationImpl<String>())
                .fieldAccessor(new FieldAccessorTest<String>());

        ClassModel<SimpleModel> classModel  = builder.idField("stringField").build();

        assertTrue(classModel.useDiscriminator());
        assertEquals("_cls", classModel.getDiscriminatorKey());
        assertEquals("Simples", classModel.getDiscriminator());

        assertEquals(2, classModel.getFieldModels().size());
        FieldModel<?> idFieldModel = classModel.getIdFieldModel();
        assertEquals("stringField", idFieldModel.getFieldName());
        assertEquals("_id", idFieldModel.getDocumentFieldName());

        FieldModel<?> childFieldModel = classModel.getFieldModel("id");
        assertNull(childFieldModel.useDiscriminator());
    }

    private class FieldAccessorTest<T> implements FieldAccessor<T> {

        @Override
        public <S> T get(final S instance) {
            return null;
        }

        @Override
        public <S> void set(final S instance, final T value) {

        }
    }
}