List of usage examples for org.springframework.data.mongodb.core MongoTemplate find
@Override
public <T> List<T> find(Query query, Class<T> entityClass, String collectionName)
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); }