如何覆蓋@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文件，以激活該選擇。