added PollingTransactionReceiptProcessorTest (#237)

core/src/test/java/org/web3j/tx/response/PollingTransactionReceiptProcessorTest.java

@@ -0,0 +1,76 @@
+package org.web3j.tx.response;
+
+import java.io.IOException;
+
+import org.junit.Before;
+import org.junit.Test;
+
+import org.web3j.protocol.Web3j;
+import org.web3j.protocol.core.Request;
+import org.web3j.protocol.core.Response;
+import org.web3j.protocol.core.methods.response.EthGetTransactionReceipt;
+import org.web3j.protocol.core.methods.response.TransactionReceipt;
+import org.web3j.protocol.exceptions.TransactionException;
+
+import static org.hamcrest.CoreMatchers.sameInstance;
+import static org.junit.Assert.assertThat;
+import static org.junit.Assert.fail;
+import static org.mockito.Mockito.doReturn;
+import static org.mockito.Mockito.mock;
+import static org.mockito.Mockito.when;
+
+public class PollingTransactionReceiptProcessorTest {
+    private static final String TRANSACTION_HASH = "0x00";
+    private Web3j web3j;
+    private long sleepDuration;
+    private int attemps;
+    private PollingTransactionReceiptProcessor processor;
+
+    @Before
+    public void setUp() {
+        web3j = mock(Web3j.class);
+        sleepDuration = 100;
+        attemps = 3;
+        processor = new PollingTransactionReceiptProcessor(web3j, sleepDuration, attemps);
+    }
+
+    @Test
+    public void returnsTransactionReceiptWhenItIsAvailableInstantly() throws Exception {
+        TransactionReceipt transactionReceipt = new TransactionReceipt();
+        doReturn(requestReturning(response(transactionReceipt)))
+                .when(web3j).ethGetTransactionReceipt(TRANSACTION_HASH);
+
+        TransactionReceipt receipt = processor.waitForTransactionReceipt(TRANSACTION_HASH);
+
+        assertThat(receipt, sameInstance(transactionReceipt));
+    }
+
+    @Test
+    public void throwsTransactionExceptionWhenReceiptIsNotAvailableInTime() throws Exception {
+        doReturn(requestReturning(response(null)))
+                .when(web3j).ethGetTransactionReceipt(TRANSACTION_HASH);
+
+        try {
+            processor.waitForTransactionReceipt(TRANSACTION_HASH);
+            fail("call should fail with TransactionException");
+        } catch (TransactionException e) {
+            // this is expected
+        }
+    }
+
+    private static <T extends Response<?>> Request<String, T> requestReturning(T response) {
+        Request<String, T> request = mock(Request.class);
+        try {
+            when(request.send()).thenReturn(response);
+        } catch (IOException e) {
+            // this will never happen
+        }
+        return request;
+    }
+
+    private static EthGetTransactionReceipt response(TransactionReceipt transactionReceipt) {
+        EthGetTransactionReceipt response = new EthGetTransactionReceipt();
+        response.setResult(transactionReceipt);
+        return response;
+    }
+}