Example usage for org.springframework.data.mongodb.core MongoTemplate find

List of usage examples for org.springframework.data.mongodb.core MongoTemplate find

Introduction

In this page you can find the example usage for org.springframework.data.mongodb.core MongoTemplate find.

Prototype

@Override
    public <T> List<T> find(Query query, Class<T> entityClass, String collectionName) 

Source Link

Usage

From source file:com.comcast.video.dawg.service.pound.DawgPoundMongoServiceTest.java

@SuppressWarnings("unchecked")
@Test/*  w w  w.j  a v a 2  s.  c o m*/
public void testGetReserverWithNoDeviceAssoicatedWithToken() {
    List<PersistableDevice> devices = new ArrayList<PersistableDevice>();

    MongoTemplate mockMongoTemplate = EasyMock.createMock(MongoTemplate.class);
    DawgPoundMongoService dawgPoundService = new DawgPoundMongoService();

    ReflectionTestUtils.setField(dawgPoundService, "mongoTemplate", mockMongoTemplate);
    EasyMock.expect(mockMongoTemplate.find((Query) EasyMock.anyObject(),
            (Class<PersistableDevice>) EasyMock.anyObject(), (String) EasyMock.anyObject())).andReturn(devices);

    EasyMock.replay(mockMongoTemplate);

    Map<String, Object>[] reservers = dawgPoundService.getByReserver(TestConstants.TOKEN);
    Assert.assertEquals(reservers.length, 0);
    EasyMock.verify(mockMongoTemplate);
}

From source file:com.comcast.video.dawg.service.pound.DawgPoundMongoServiceTest.java

@SuppressWarnings("unchecked")
@Test//  w w w.ja  v  a2  s . c om
public void testGetByReserver() {
    Map<String, Object> deviceIds = new HashMap<String, Object>();
    deviceIds.put("id", TestConstants.DEVICE_ID);

    List<PersistableDevice> devices = new ArrayList<PersistableDevice>();

    MongoTemplate mockMongoTemplate = EasyMock.createMock(MongoTemplate.class);
    PersistableDevice persistableDev = EasyMock.createMock(PersistableDevice.class);
    devices.add(persistableDev);

    DawgPoundMongoService dawgPoundService = new DawgPoundMongoService();

    ReflectionTestUtils.setField(dawgPoundService, "mongoTemplate", mockMongoTemplate);
    EasyMock.expect(mockMongoTemplate.find((Query) EasyMock.anyObject(),
            (Class<PersistableDevice>) EasyMock.anyObject(), (String) EasyMock.anyObject())).andReturn(devices);
    EasyMock.expect(persistableDev.getData()).andReturn(deviceIds);

    EasyMock.replay(mockMongoTemplate, persistableDev);

    Map<String, Object>[] reservers = dawgPoundService.getByReserver(TestConstants.TOKEN);

    Assert.assertEquals(reservers.length, 1);
    Assert.assertEquals(reservers[0].get("id"), TestConstants.DEVICE_ID);
    EasyMock.verify(mockMongoTemplate, persistableDev);
}