如何覆盖@Produces使用模拟

Question

我已经与@Inject

@Produces 
@ApplicationScoped 
public Client createClient() { 
    return new TransportClient().addTransportAddress(new InetSocketTransportAddress(IP, 9300)); 
} 

我想嘲笑这个客户端在我的单元测试的东西，如以下

使用产生的ElasticSearch客户端类

@Produces 
@Mock 
private Client client; 

这会导致AmbiguousResolutionException，因为同一个bean有两个提供程序。

我该如何优先考虑我的模拟生产的类只进行单元测试？

---

从下面的评论，我做了以下更改。我在替代方案中设定了一个突破点，但没有受到打击。

ElasticSearchProducer.java

@ApplicationScoped 
public class ElasticSearchProducer { 

    public static final String IP = "10.9.215.28"; 

    private Client client; 

    @PostConstruct 
    public void init() { 
     client = createClient(); 
    } 

    protected Client createClient() { 
     return new TransportClient().addTransportAddress(new InetSocketTransportAddress(IP, 9300)); 
    } 

    @Produces 
    @ApplicationScoped 
    public Client getClient() { 
     return client; 
    } 
} 

ElasticSearchProducerAlternative.java

@Alternative 
@ApplicationScoped 
public class ElasticSearchProducerAlternative extends ElasticSearchProducer { 

    @Override 
    public Client createClient() { 
     return Mockito.mock(Client.class); 
    } 
} 

测试-beans.xml文件

<?xml version="1.0" encoding="UTF-8"?> 
<beans xmlns="http://java.sun.com/xml/ns/javaee" 
     xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance" 
     xsi:schemaLocation="http://java.sun.com/xml/ns/javaee http://java.sun.com/xml/ns/javaee/beans_1_0.xsd"> 

    <alternatives> 
     <class>com.walmart.platform.pv.cdi.ElasticSearchProducerAlternative</class> 
    </alternatives> 

</beans> 

ArquillianTest.java

@RunWith(Arquillian.class) 
public abstract class ArquillianTest { 
    @Deployment(testable = true) 
    public static WebArchive createDeployment() { 


     WebArchive war = ShrinkWrap.create(WebArchive.class, "quintessence.war") 
       .addPackages(true, "com.w.platform"); 

     war.merge(ShrinkWrap.create(GenericArchive.class).as(ExplodedImporter.class) 
         .importDirectory("src/main/webapp").as(GenericArchive.class), 
       "/", Filters.includeAll()); 

     war.addAsManifestResource("etc/conf/test/import.sql"); 
     war.addAsManifestResource("test-persistence.xml", "persistence.xml"); 
     war.addAsWebInfResource("test-beans.xml", "beans.xml"); 
     war.addAsWebInfResource("test-resources.xml", "resources.xml"); 

     System.out.println(war.toString(true)); 
     return war; 
    } 
} 

在我的测试中，不使用替代类

Answer 1

您需要记住总是编写可测试的代码。假设你的班级是这样写的

@ApplicationScoped 
public class ESClient { 
    @Produces 
    @ApplicationScoped 
    public Client createClient() { 
     return new TransportClient().addTransportAddress(new InetSocketTransportAddress(IP, 9300)); 
    } 
} 

这个工程，但不是很可测试。你可能想使用模拟进行单元测试，没有活动连接。我们可以将此重构为更易于测试。考虑这个替代实施方案：

@ApplicationScoped 
public class ESClient { 
    private Client client; 
    @PostConstruct 
    public void init() { 
     this.client = createClient(); 
    } 
    @Produces 
    @ApplicationScoped 
    public Client getClient() { 
     return this.client; 
    } 
    protected Client createClient() { 
     return new TransportClient().addTransportAddress(new InetSocketTransportAddress(IP, 9300)); 
    } 
} 

是的，它有点冗长，但您的担忧分开得更好。然后，您可以提供另一种实现在你的测试

@ApplicationScoped 
@Alternative 
public class MockESClient extends ESClient { 
    @Override 
    protected Client createClient() { 
     // return your mock here 
    } 
} 

现在你只需要在你的beans.xml文件，以激活该选择。