dataSource na classe de teste

Olá estou com um problema com do DataSource dos testes porque o retorno do método é um DataSource e dentro do método é criado um DriverManagerDataSource e quando configuro tudo e mando o retorno como dataSource ele pede pra fazer um (cast) ou mudar o tipo de retorno do método, como posso resolver isso?

package br.com.casadocodigo.loja.conf;

import javax.activation.DataSource;

import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Profile;
import org.springframework.jdbc.datasource.DriverManagerDataSource;

public class DataSourceConfigurationTest {

    @Bean
    @Profile("test")
    public DataSource dataSource() {

    DriverManagerDataSource dataSource = new DriverManagerDataSource();
    dataSource.setUrl("jdbc:sqlserver://localhost:1433;databaseName=casadocodigo_test");
    dataSource.setDriverClassName("com.microsoft.sqlserver.jdbc.SQLServerDriver");
    dataSource.setUsername("sa");
    dataSource.setPassword("root");

    return dataSource;
}
}