core/api/src/test/java/org/apache/polygene/api/unitofwork/ToValueConversionTest.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.polygene.api.unitofwork;

import java.util.Map;
import java.util.Set;
import java.util.function.Function;
import java.util.stream.Stream;
import org.apache.polygene.api.association.Association;
import org.apache.polygene.api.association.AssociationDescriptor;
import org.apache.polygene.api.association.ManyAssociation;
import org.apache.polygene.api.association.NamedAssociation;
import org.apache.polygene.api.common.Optional;
import org.apache.polygene.api.common.QualifiedName;
import org.apache.polygene.api.common.UseDefaults;
import org.apache.polygene.api.entity.EntityBuilder;
import org.apache.polygene.api.entity.EntityReference;
import org.apache.polygene.api.identity.HasIdentity;
import org.apache.polygene.api.identity.Identity;
import org.apache.polygene.api.identity.StringIdentity;
import org.apache.polygene.api.injection.scope.Structure;
import org.apache.polygene.api.property.Property;
import org.apache.polygene.api.property.PropertyDescriptor;
import org.apache.polygene.api.usecase.Usecase;
import org.apache.polygene.api.usecase.UsecaseBuilder;
import org.apache.polygene.bootstrap.AssemblyException;
import org.apache.polygene.bootstrap.ModuleAssembly;
import org.apache.polygene.spi.PolygeneSPI;
import org.apache.polygene.test.AbstractPolygeneTest;
import org.apache.polygene.test.EntityTestAssembler;
import org.junit.Test;

import static org.hamcrest.core.AnyOf.anyOf;
import static org.hamcrest.core.IsEqual.equalTo;
import static org.junit.Assert.assertThat;

public class ToValueConversionTest
    extends AbstractPolygeneTest
{
    @Override
    public void assemble( ModuleAssembly module )
        throws AssemblyException
    {
        new EntityTestAssembler().assemble( module );
        module.entities( SomeType.class );
        module.values( SomeType.class );
    }

    @Test
    public void testConversionToValue()
        throws Exception
    {
        Usecase usecase = UsecaseBuilder.buildUsecase( "test case" )
                                        .withMetaInfo( new SomeValueConverter() )
                                        .newUsecase();
        SomeType value;
        try( UnitOfWork uow = unitOfWorkFactory.newUnitOfWork(usecase) )
        {
            SomeType entity1 = createEntity( uow, StringIdentity.identity( "Niclas" ) );
            SomeType entity2 = createEntity( uow, StringIdentity.identity( "Paul" ) );
            SomeType entity3 = createEntity( uow, StringIdentity.identity( "Jiri" ) );
            SomeType entity4 = createEntity( uow, StringIdentity.identity( "Kent" ) );
            SomeType entity5 = createEntity( uow, StringIdentity.identity( "Stan" ) );
            entity1.assoc().set( entity2 );
            entity1.many().add( entity3 );
            entity1.named().put( "kent", entity4 );
            entity1.named().put( "stan", entity5 );

            value = uow.toValue( SomeType.class, entity1 );
            uow.complete();

        }
        try( UnitOfWork uow = unitOfWorkFactory.newUnitOfWork(usecase) )
        {
            assertThat( value.identity().get(), equalTo( StringIdentity.identity( "Niclas" ) ) );
            assertThat( value.name().get(), equalTo( "[Niclas]" ) );

            assertThat( uow.toValue( SomeType.class, value.assoc().get()).name().get(), equalTo( "[Paul]" ));
            assertThat( uow.toValueList( value.many() ).get(0).name().get(), equalTo( "[Jiri]" ));
            assertThat( uow.toValueSet( value.many() ).iterator().next().name().get(), equalTo( "[Jiri]" ));
            Set<Map.Entry<String, SomeType>> actual = uow.toValueMap( value.named() ).entrySet();
            assertThat( actual.iterator().next().getKey(), anyOf(equalTo( "stan" ), equalTo( "kent" )) );
            assertThat( actual.iterator().next().getValue().name().get(), anyOf(equalTo( "[Stan]" ), equalTo( "[Kent]" )) );
        }

    }

    private SomeType createEntity( UnitOfWork uow, Identity identity )
    {
        EntityBuilder<SomeType> builder = uow.newEntityBuilder( SomeType.class, identity );
        builder.instance().name().set( identity.toString() );
        return builder.newInstance();
    }

    interface SomeType extends HasIdentity
    {
        Property<String> name();

        @Optional
        Association<SomeType> assoc();

        @UseDefaults
        ManyAssociation<SomeType> many();

        @UseDefaults
        NamedAssociation<SomeType> named();
    }

    private static class SomeValueConverter
        implements ToValueConverter
    {
        @Structure
        private PolygeneSPI spi;

        @Override
        public Function<PropertyDescriptor, Object> properties( Object entityComposite, Function<PropertyDescriptor, Object> defaultFn )
        {
            return descriptor ->
            {
                Object value = defaultFn.apply( descriptor );
                QualifiedName name = QualifiedName.fromClass( SomeType.class, "name" );
                if( name.equals( descriptor.qualifiedName() ) )
                {
                    return "[" + value + "]";
                }
                return value;
            };
        }

        @Override
        public Function<AssociationDescriptor, EntityReference> associations( Object entityComposite, Function<AssociationDescriptor, EntityReference> defaultFn )
        {
            return defaultFn;
        }

        @Override
        public Function<AssociationDescriptor, Stream<EntityReference>> manyAssociations( Object entityComposite, Function<AssociationDescriptor, Stream<EntityReference>> defaultFn )
        {
            return defaultFn;
        }

        @Override
        public Function<AssociationDescriptor, Stream<Map.Entry<String, EntityReference>>> namedAssociations( Object entityComposite, Function<AssociationDescriptor, Stream<Map.Entry<String, EntityReference>>> defaultFn )
        {
            return defaultFn;
        }
    }
}