io.beanmapper.spring.web.mockmvc.MockIdToEntityBeanConverter Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of beanmapper-spring Show documentation

Spring support for the Bean Mapper

There is a newer version: 6.0.0

package io.beanmapper.spring.web.mockmvc;

import io.beanmapper.BeanMapper;
import io.beanmapper.core.BeanFieldMatch;
import io.beanmapper.core.converter.BeanConverter;

import org.springframework.data.domain.Persistable;
import org.springframework.data.repository.CrudRepository;

public class MockIdToEntityBeanConverter implements BeanConverter {

    private Class targetClass;

    private CrudRepository repository;

    public MockIdToEntityBeanConverter(CrudRepository repository, Class targetClass) {
        this.repository = repository;
        this.targetClass = targetClass;
    }

    @Override
    public Object convert(BeanMapper beanMapper, Object source, Class targetClass, BeanFieldMatch beanFieldMatch) {
        return repository.findOne((Long)source);
    }

    @Override
    public boolean match(Class sourceClass, Class targetClass) {
        return sourceClass.equals(Long.class) && targetClass.equals(this.targetClass);
    }

}