You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
108 lines
4.6 KiB
108 lines
4.6 KiB
package com.sigmaflare.binancej; |
|
|
|
import com.codepoetics.ambivalence.Either; |
|
import com.fasterxml.jackson.databind.ObjectMapper; |
|
import com.sigmaflare.binancej.entities.ServiceError; |
|
import com.sigmaflare.binancej.entities.Time; |
|
import com.sigmaflare.binancej.exceptions.BinanceServiceUnreachableException; |
|
import com.sigmaflare.binancej.matchers.GetMatcher; |
|
import org.apache.http.ProtocolVersion; |
|
import org.apache.http.client.methods.CloseableHttpResponse; |
|
import org.apache.http.entity.BasicHttpEntity; |
|
import org.apache.http.impl.client.CloseableHttpClient; |
|
import org.apache.http.message.BasicStatusLine; |
|
import org.junit.Test; |
|
|
|
import java.io.ByteArrayInputStream; |
|
import java.io.IOException; |
|
import java.nio.charset.StandardCharsets; |
|
|
|
import static com.sigmaflare.binancej.Constant.BASE_ENDPOINT; |
|
import static org.junit.Assert.assertEquals; |
|
import static org.junit.Assert.assertTrue; |
|
import static org.mockito.ArgumentMatchers.argThat; |
|
import static org.mockito.Mockito.mock; |
|
import static org.mockito.Mockito.when; |
|
|
|
public class TimeMethodTests { |
|
private static ObjectMapper mapper = new ObjectMapper().findAndRegisterModules(); |
|
private final String url = String.format("%s%s", BASE_ENDPOINT, "/api/v1/time"); |
|
|
|
@Test |
|
public void testTimeReturnsSuccessfully() throws IOException, BinanceServiceUnreachableException { |
|
CloseableHttpClient mockedCloseableHttpClient = mock(CloseableHttpClient.class); |
|
CloseableHttpResponse mockedCloseableHttpResponse = mock(CloseableHttpResponse.class); |
|
|
|
when(mockedCloseableHttpResponse.getStatusLine()) |
|
.thenReturn(new BasicStatusLine(new ProtocolVersion("TEST", 1, 0), |
|
200, "test")); |
|
|
|
when(mockedCloseableHttpClient.execute(argThat(new GetMatcher(url, "1234")))) |
|
.thenReturn(mockedCloseableHttpResponse); |
|
|
|
Time time = Time.builder().serverTime(System.currentTimeMillis() / 1000L).build(); |
|
|
|
BasicHttpEntity httpEntity = new BasicHttpEntity(); |
|
httpEntity.setContent(new ByteArrayInputStream( |
|
mapper.writeValueAsString(time).getBytes(StandardCharsets.UTF_8))); |
|
|
|
|
|
when(mockedCloseableHttpResponse.getEntity()).thenReturn(httpEntity); |
|
|
|
GeneralUtilities generalUtilities = |
|
new GeneralUtilities("1234", "abcd", mockedCloseableHttpClient); |
|
|
|
Either<ServiceError, Time> res = generalUtilities.getServerTime(); |
|
|
|
assertTrue(res.isRight()); |
|
assertEquals(time, Helpers.extractEitherValueSafely(res.right())); |
|
} |
|
|
|
@Test(expected = BinanceServiceUnreachableException.class) |
|
public void testTimeNoEntityReturnedFails() throws IOException, BinanceServiceUnreachableException { |
|
CloseableHttpClient mockedCloseableHttpClient = mock(CloseableHttpClient.class); |
|
CloseableHttpResponse mockedCloseableHttpResponse = mock(CloseableHttpResponse.class); |
|
|
|
when(mockedCloseableHttpResponse.getStatusLine()) |
|
.thenReturn(new BasicStatusLine(new ProtocolVersion("TEST", 1, 0), |
|
200, "test")); |
|
|
|
when(mockedCloseableHttpClient.execute(argThat(new GetMatcher(url, "1234")))) |
|
.thenReturn(mockedCloseableHttpResponse); |
|
|
|
GeneralUtilities generalUtilities = |
|
new GeneralUtilities("1234", "abcd", mockedCloseableHttpClient); |
|
|
|
generalUtilities.getServerTime(); |
|
} |
|
|
|
@Test |
|
public void testTimeBadApiKeyReturns400() throws IOException, BinanceServiceUnreachableException { |
|
CloseableHttpClient mockedCloseableHttpClient = mock(CloseableHttpClient.class); |
|
CloseableHttpResponse mockedCloseableHttpResponse = mock(CloseableHttpResponse.class); |
|
|
|
when(mockedCloseableHttpResponse.getStatusLine()) |
|
.thenReturn(new BasicStatusLine(new ProtocolVersion("TEST", 1, 0), |
|
400, "Bad API Key")); |
|
|
|
when(mockedCloseableHttpClient.execute(argThat(new GetMatcher(url, "1234")))) |
|
.thenReturn(mockedCloseableHttpResponse); |
|
|
|
ServiceError serviceError = ServiceError.builder().code(400).message("Bad API Key").build(); |
|
|
|
BasicHttpEntity httpEntity = new BasicHttpEntity(); |
|
httpEntity.setContent(new ByteArrayInputStream( |
|
mapper.writeValueAsString(serviceError).getBytes(StandardCharsets.UTF_8))); |
|
|
|
|
|
when(mockedCloseableHttpResponse.getEntity()).thenReturn(httpEntity); |
|
|
|
GeneralUtilities generalUtilities = |
|
new GeneralUtilities("1234", "abcd", mockedCloseableHttpClient); |
|
|
|
Either<ServiceError, Time> res = generalUtilities.getServerTime(); |
|
|
|
assertTrue(res.isLeft()); |
|
assertEquals(serviceError, Helpers.extractEitherValueSafely(res.left())); |
|
} |
|
}
|
|
|