/* * 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.Entity; import org.apache.usergrid.persistence.model.util.EntityUtils; 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. In this case, we just * inject all invalid types of Ids from the Id generator into the Entity * * @author tnine */ public class InvalidEntityGenerator { 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 entities */ public static class NullFieldsSupplier extends ParameterSupplier { @Override public List<PotentialAssignment> getValueSources( final ParameterSignature sig ) { final List<PotentialAssignment> result = new ArrayList<PotentialAssignment>(); final Entity entity = mock( Entity.class ); when( entity.getId() ).thenReturn( null ); when( entity.getVersion() ).thenReturn( null ); result.add( PotentialAssignment.forValue( "nullEntity", null ) ); result.add( PotentialAssignment.forValue( "nullIdsEntity", entity ) ); return result; } } @Retention( RetentionPolicy.RUNTIME ) @ParametersSuppliedBy( IllegalFieldsSupplier.class ) public @interface IllegalFields { } /** Supplies all possible combination of null fields on entities */ 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( "invalidVersion", invalidVersion() ) ); return result; } private Entity invalidVersion(){ Entity entity = new Entity("test"); EntityUtils.setVersion( entity, UUID.randomUUID() ); return entity; } } }