/*
 * Licensed to the Apache Software Foundation (ASF) under one or more
 *  contributor license agreements.  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.  For additional information regarding
 * copyright in this work, please see the NOTICE file in the top level
 * directory of this distribution.
 */
package org.apache.usergrid.persistence.collection.util;


import java.lang.annotation.Retention;
import java.lang.annotation.RetentionPolicy;
import java.util.ArrayList;
import java.util.List;
import java.util.UUID;

import org.junit.experimental.theories.ParameterSignature;
import org.junit.experimental.theories.ParameterSupplier;
import org.junit.experimental.theories.ParametersSuppliedBy;
import org.junit.experimental.theories.PotentialAssignment;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import org.apache.usergrid.persistence.model.entity.Id;
import org.apache.usergrid.persistence.model.util.UUIDGenerator;

import static org.mockito.Mockito.mock;
import static org.mockito.Mockito.when;

/**
 * Generates a list of invalid entities for input verification. To be used with Theory testing.
 *
 * @author tnine
 */
public class InvalidIdGenerator {

    private static final Logger LOG = LoggerFactory.getLogger( InvalidIdGenerator.class );

    @Retention( RetentionPolicy.RUNTIME )
    @ParametersSuppliedBy( NullFieldsSupplier.class )
    public @interface NullFields {

    }

    /**
     * Supplies all possible combination of null fields on ids
     */
    public static class NullFieldsSupplier extends ParameterSupplier {

        @Override
        public List<PotentialAssignment> getValueSources( final ParameterSignature sig ) {
            final List<PotentialAssignment> result = new ArrayList<PotentialAssignment>();

            result.add( PotentialAssignment.forValue( "nullId", null ) );
            result.add( PotentialAssignment.forValue( "nullEntityId", nullEntityId() ) );
            result.add( PotentialAssignment.forValue( "nullEntityType", nullEntityType() ) );

            return result;
        }

        /**
         * Missing fields
         */
        private static Id nullEntityId() {

            final Id entityId = mock( Id.class );

            when( entityId.getUuid() ).thenReturn( null );
            when( entityId.getType() ).thenReturn( "test" );

            return entityId;
        }

        /**
         * Null entity type
         */
        private static Id nullEntityType() {

            final Id entityId = mock( Id.class );

            when( entityId.getUuid() ).thenReturn( UUIDGenerator.newTimeUUID() );
            when( entityId.getType() ).thenReturn( null );

            return entityId;
        }
    }

    @Retention( RetentionPolicy.RUNTIME )
    @ParametersSuppliedBy( IllegalFieldsSupplier.class )
    public @interface IllegalFields {

    }

    /**
     * Supplies all possible combination of null fields on ids
     */
    public static class IllegalFieldsSupplier extends ParameterSupplier {

        @Override
        public List<PotentialAssignment> getValueSources( final ParameterSignature sig ) {

            final List<PotentialAssignment> result = new ArrayList<PotentialAssignment>();
            result.add( PotentialAssignment.forValue( "wrongEntityTypeLength", wrongEntityTypeLength() ) );

            return result;
        }


        private static Id wrongEntityTypeLength() {

            final Id entityId = mock( Id.class );

            //set this to a non time uuid
            when( entityId.getUuid() ).thenReturn( UUID.randomUUID() );
            when( entityId.getType() ).thenReturn( "" );

            return entityId;
        }
    }
}