@Test
  public void testPtOrderMemoryHandler() {
    PutThroughTransaction pushThroughTransactionInfo = new PutThroughTransaction();
    pushThroughTransactionInfo.setSymbol("SAM");
    pushThroughTransactionInfo.setVolume(4120.0);
    pushThroughTransactionInfo.setPrice(14000.0);
    pushThroughTransactionInfo.setFloorCode("10");
    pushThroughTransactionInfo.setTradingDate("2015019");

    PutThroughTransaction pushThroughTransactionInfo1 = new PutThroughTransaction();
    pushThroughTransactionInfo1.setSymbol("SSI");
    pushThroughTransactionInfo1.setVolume(6120.0);
    pushThroughTransactionInfo1.setPrice(24000.0);
    pushThroughTransactionInfo1.setFloorCode("10");
    pushThroughTransactionInfo1.setTradingDate("2015019");

    List<PutThroughTransaction> listOfPt = new ArrayList<PutThroughTransaction>();

    listOfPt.add(pushThroughTransactionInfo);
    listOfPt.add(pushThroughTransactionInfo1);

    SecInfo stock2 = new SecInfo();
    stock2.setCode("SAM");
    stock2.setBasicPrice(21.0);
    stock2.setCeilingPrice(21.9);
    stock2.setFloorPrice(19.5);

    SecInfo stock1 = new SecInfo();
    stock1.setCode("SSI");
    stock1.setBasicPrice(8.0);
    stock1.setCeilingPrice(8.8);
    stock1.setFloorPrice(7.5);

    memory.put("STOCK", "SSI", stock1);
    memory.put("STOCK", "SAM", stock2);

    memoryHandler.handle(pushThroughTransactionInfo);
    memoryHandler.handle(pushThroughTransactionInfo1);

    List<PutThroughTransaction> putThroughTransactions =
        (List<PutThroughTransaction>) memory.get("PutThroughTransaction", "10");
    Assert.assertEquals(listOfPt, putThroughTransactions);
  }
Exemplo n.º 2
0
  @Test
  public void testCompanyNameLoaderLoadData() throws Exception {
    List<Object> expectedListCompany = new ArrayList<Object>();
    Company company1 = new Company();
    company1.setCode("VND");
    company1.setCompanyName("CT chung khoan VNDIRECT");
    Company company2 = new Company();
    company2.setCode("SSI");
    company2.setCompanyName("CT chung khoan SSI");
    expectedListCompany.add(company1);
    expectedListCompany.add(company2);

    Mockito.when(
            elasticSearchClient.getDataByIndex(
                Mockito.anyString(), Mockito.anyString(), Mockito.any(), Mockito.any()))
        .thenReturn(expectedListCompany);
    loader.load();
    List<Object> listCompany = (List<Object>) memory.get("COMPANY_LIST", "COMPANY_LIST");
    Assert.assertEquals(expectedListCompany, listCompany);
  }