graphenej/graphenej/src/test/java/cy/agorise/graphenej/api/GetLimitOrdersTest.java

package cy.agorise.graphenej.api;

import com.google.common.primitives.UnsignedLong;
import com.neovisionaries.ws.client.WebSocketException;

import org.junit.After;
import org.junit.Assert;
import org.junit.Before;
import org.junit.Test;

import java.util.List;

import cy.agorise.graphenej.Asset;
import cy.agorise.graphenej.AssetAmount;
import cy.agorise.graphenej.Converter;
import cy.agorise.graphenej.LimitOrder;
import cy.agorise.graphenej.OrderBook;
import cy.agorise.graphenej.UserAccount;
import cy.agorise.graphenej.interfaces.WitnessResponseListener;
import cy.agorise.graphenej.models.BaseResponse;
import cy.agorise.graphenej.models.WitnessResponse;
import cy.agorise.graphenej.operations.LimitOrderCreateOperation;

import static org.hamcrest.CoreMatchers.is;

/**
 * Class used to test the 'get_limit_order' API wrapper
 */
public class GetLimitOrdersTest extends BaseApiTest {
    private UserAccount seller = new UserAccount("1.2.143563");
    private final Asset base = new Asset("1.3.121", "USD", 4);
    private final Asset quote = new Asset("1.3.0", "BTS", 5);

    @Before
    public void setup(){
        System.out.println("Connecting to node: "+NODE_URL);
    }

    @Test
    public void testGetLimitOrders(){
        try {
            mWebSocket.addListener(new GetLimitOrders(base.getObjectId(), quote.getObjectId(), 100, new WitnessResponseListener() {
                @Override
                public void onSuccess(WitnessResponse response) {
                    List<LimitOrder> orders = (List<LimitOrder>) response.result;
                    Assert.assertThat("Checking that we have orders", orders.isEmpty(), is(false));
                    Converter converter = new Converter();
                    double baseToQuoteExchange, quoteToBaseExchange;

                    for(LimitOrder order : orders){
                        if(order.getSellPrice().base.getAsset().getObjectId().equals(base.getObjectId())){
                            order.getSellPrice().base.getAsset().setPrecision(base.getPrecision());
                            order.getSellPrice().quote.getAsset().setPrecision(quote.getPrecision());

                            baseToQuoteExchange = converter.getConversionRate(order.getSellPrice(), Converter.BASE_TO_QUOTE);
                            quoteToBaseExchange = converter.getConversionRate(order.getSellPrice(), Converter.QUOTE_TO_BASE);
                            System.out.println(String.format("> id: %s, base to quote: %.5f, quote to base: %.5f", order.getObjectId(), baseToQuoteExchange, quoteToBaseExchange));
                        }else{
                            order.getSellPrice().base.getAsset().setPrecision(quote.getPrecision());
                            order.getSellPrice().quote.getAsset().setPrecision(base.getPrecision());

                            baseToQuoteExchange = converter.getConversionRate(order.getSellPrice(), Converter.BASE_TO_QUOTE);
                            quoteToBaseExchange = converter.getConversionRate(order.getSellPrice(), Converter.QUOTE_TO_BASE);
                            System.out.println(String.format("< id: %s, base to quote: %.5f, quote to base: %.5f", order.getObjectId(), baseToQuoteExchange, quoteToBaseExchange));
                        }
                    }

                    synchronized (GetLimitOrdersTest.this){
                        GetLimitOrdersTest.this.notifyAll();
                    }
                }

                @Override
                public void onError(BaseResponse.Error error) {
                    System.out.println("onError. Msg: "+error.message);
                    synchronized (GetLimitOrdersTest.this){
                        GetLimitOrdersTest.this.notifyAll();
                    }
                }
            }));

            mWebSocket.connect();

            synchronized (this){
                wait();
            }
        } catch (WebSocketException e) {
            System.out.println("WebSocketException. Msg: " + e.getMessage());
        } catch (InterruptedException e) {
            System.out.println("InterruptedException. Msg: "+e.getMessage());
        }
    }

    /**
     * This method is designed to test how the OrderBook class handles data obtained
     * from the GetLimitOrders API call.
     */
    @Test
    public void testOrderBook(){
        LimitOrderCreateOperation limitOrderOperation;

        try {
            mWebSocket.addListener(new GetLimitOrders(base.getObjectId(), quote.getObjectId(), 100, new WitnessResponseListener() {
                @Override
                public void onSuccess(WitnessResponse response) {
                    List<LimitOrder> orders = (List<LimitOrder>) response.result;
                    OrderBook orderBook = new OrderBook(orders);

                    AssetAmount toBuy = new AssetAmount(UnsignedLong.valueOf(100000), quote);
                    int expiration = (int) ((System.currentTimeMillis() + 60000) / 1000);
                    LimitOrderCreateOperation operation = orderBook.exchange(seller, base, toBuy, expiration);

                    // Testing the successful creation of a limit order create operation
                    Assert.assertTrue(operation != null);
                    double price = (double) Math.pow(10, base.getPrecision() - quote.getPrecision()) * operation.getMinToReceive().getAmount().longValue() / operation.getAmountToSell().getAmount().longValue();
                    System.out.println("price: "+price);
                    synchronized (GetLimitOrdersTest.this){
                        GetLimitOrdersTest.this.notifyAll();
                    }
                }

                @Override
                public void onError(BaseResponse.Error error) {
                    System.out.println("onError. Msg: "+error.message);
                    synchronized (GetLimitOrdersTest.this){
                        GetLimitOrdersTest.this.notifyAll();
                    }
                }
            }));

            mWebSocket.connect();

            synchronized (this){
                wait();
            }

        } catch (WebSocketException e) {
            System.out.println("WebSocketException. Msg: " + e.getMessage());
        } catch (InterruptedException e) {
            System.out.println("InterruptedException. Msg: "+e.getMessage());
        }
    }

    @Test
    public void testRequiredBase(){
        try {
            final Asset _quote = new Asset("1.3.121", "USD", 4);
            final Asset _base = new Asset("1.3.0", "BTS", 5);
            mWebSocket.addListener(new GetLimitOrders(_base.getObjectId(), _quote.getObjectId(), 100, new WitnessResponseListener() {
                @Override
                public void onSuccess(WitnessResponse response) {
                    List<LimitOrder> orders = (List<LimitOrder>) response.result;
                    OrderBook orderBook = new OrderBook(orders);

                    long _totalQuote = 1000;
                    long _totalBase = orderBook.calculateRequiredBase(new AssetAmount(UnsignedLong.valueOf(_totalQuote), _quote));

                    System.out.println(String.format("Base: %s, Quote: %s", _base.getObjectId(), _quote.getObjectId()));
                    System.out.println(String.format("_totalQuote: %d, _totalBase: %d", _totalQuote, _totalBase));

                    synchronized (GetLimitOrdersTest.this){
                        GetLimitOrdersTest.this.notifyAll();
                    }
                }

                @Override
                public void onError(BaseResponse.Error error) {
                    System.out.println("onError. Msg: "+error.message);
                    synchronized (GetLimitOrdersTest.this){
                        GetLimitOrdersTest.this.notifyAll();
                    }
                }
            }));

            mWebSocket.connect();

            synchronized (this){
                wait();
            }

        } catch (WebSocketException e) {
            System.out.println("WebSocketException. Msg: " + e.getMessage());
        } catch (InterruptedException e) {
            System.out.println("InterruptedException. Msg: "+e.getMessage());
        }
    }

    @After
    public void tearDown() throws Exception {
        mWebSocket.disconnect();
    }
}