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

Introduction

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

Prototype

@Nullable
    @Override
    public <T> T findById(Object id, Class<T> entityClass, String collectionName) 

Source Link

Usage

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

@Test
public void testIsReservedWithInvalidDeviceId() {
    MongoTemplate mockMongoTemplate = EasyMock.createMock(MongoTemplate.class);
    DawgPoundMongoService dawgPoundService = new DawgPoundMongoService();
    ReflectionTestUtils.setField(dawgPoundService, "mongoTemplate", mockMongoTemplate);

    EasyMock.expect(mockMongoTemplate.findById(TestConstants.DEVICE_ID, PersistableDevice.class,
            TestConstants.COLLECTION_NAME)).andReturn(null);
    EasyMock.replay(mockMongoTemplate);/* w  ww. jav  a2s . com*/

    Assert.assertNull(dawgPoundService.isReserved(TestConstants.DEVICE_ID));
    EasyMock.verify(mockMongoTemplate);
}

---

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

@Test
public void testReserveFeatureWithNoDevice() {
    MongoTemplate mockMongoTemplate = EasyMock.createMock(MongoTemplate.class);
    DawgPoundMongoService dawgPoundService = new DawgPoundMongoService();
    ReflectionTestUtils.setField(dawgPoundService, "mongoTemplate", mockMongoTemplate);

    EasyMock.expect(mockMongoTemplate.findById(TestConstants.DEVICE_ID, PersistableDevice.class,
            TestConstants.COLLECTION_NAME)).andReturn(null);
    EasyMock.replay(mockMongoTemplate);//  w w  w.j  ava  2  s.co  m

    Assert.assertNull(dawgPoundService.reserve(TestConstants.DEVICE_ID, TestConstants.TOKEN,
            (System.currentTimeMillis() + TestConstants.ONE_DAY_IN_MILLIS)));
    EasyMock.verify(mockMongoTemplate);
}

---

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

@Test
public void testUnReserveWithInvalidDevice() {
    MongoTemplate mockMongoTemplate = EasyMock.createMock(MongoTemplate.class);

    DawgPoundMongoService dawgPoundService = new DawgPoundMongoService();
    ReflectionTestUtils.setField(dawgPoundService, "mongoTemplate", mockMongoTemplate);

    EasyMock.expect(mockMongoTemplate.findById(TestConstants.DEVICE_ID, PersistableDevice.class,
            TestConstants.COLLECTION_NAME)).andReturn(null);

    EasyMock.replay(mockMongoTemplate);//from w  w w .j  a v  a 2s  . co m
    Assert.assertNull(dawgPoundService.unreserve(TestConstants.DEVICE_ID));
    EasyMock.verify(mockMongoTemplate);
}

---

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

@Test
public void testOverrideReserveWithInvalidDevice() {
    MongoTemplate mockMongoTemplate = EasyMock.createMock(MongoTemplate.class);

    DawgPoundMongoService dawgPoundService = new DawgPoundMongoService();
    ReflectionTestUtils.setField(dawgPoundService, "mongoTemplate", mockMongoTemplate);
    EasyMock.expect(mockMongoTemplate.findById(TestConstants.DEVICE_ID, PersistableDevice.class,
            TestConstants.COLLECTION_NAME)).andReturn(null);

    EasyMock.replay(mockMongoTemplate);// w  ww.j av a 2 s  .  c om
    long expiryTime = System.currentTimeMillis() + TestConstants.ONE_DAY_IN_MILLIS;
    Assert.assertFalse(
            dawgPoundService.overrideReserve(TestConstants.DEVICE_ID, TestConstants.TOKEN, expiryTime));
    EasyMock.verify(mockMongoTemplate);
}

---

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

@Test
public void testUnReserveWithValidDevice() {
    MongoTemplate mockMongoTemplate = EasyMock.createMock(MongoTemplate.class);
    PersistableDevice persistableDev = EasyMock.createMock(PersistableDevice.class);

    DawgPoundMongoService dawgPoundService = new DawgPoundMongoService();
    ReflectionTestUtils.setField(dawgPoundService, "mongoTemplate", mockMongoTemplate);

    EasyMock.expect(mockMongoTemplate.findById(TestConstants.DEVICE_ID, PersistableDevice.class,
            TestConstants.COLLECTION_NAME)).andReturn(persistableDev);
    EasyMock.expect(persistableDev.unreserve()).andReturn(TestConstants.TOKEN);
    mockMongoTemplate.save(persistableDev, TestConstants.COLLECTION_NAME);

    EasyMock.replay(mockMongoTemplate, persistableDev);
    Assert.assertEquals(dawgPoundService.unreserve(TestConstants.DEVICE_ID), TestConstants.TOKEN);
    EasyMock.verify(mockMongoTemplate, persistableDev);
}

---

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

@Test
public void testIsReservedWithDeviceNotReserved() {
    MongoTemplate mockMongoTemplate = EasyMock.createMock(MongoTemplate.class);
    PersistableDevice persistableDev = EasyMock.createMock(PersistableDevice.class);

    DawgPoundMongoService dawgPoundService = new DawgPoundMongoService();
    ReflectionTestUtils.setField(dawgPoundService, "mongoTemplate", mockMongoTemplate);

    EasyMock.expect(mockMongoTemplate.findById(TestConstants.DEVICE_ID, PersistableDevice.class,
            TestConstants.COLLECTION_NAME)).andReturn(persistableDev);
    EasyMock.expect(persistableDev.isReserved()).andReturn(false);

    EasyMock.replay(mockMongoTemplate, persistableDev);

    Assert.assertNull(dawgPoundService.isReserved(TestConstants.DEVICE_ID));
    EasyMock.verify(mockMongoTemplate, persistableDev);
}

---

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

@Test
public void testReserveWithDeviceAlreadyReserved() {
    String prevReserveToken = "8e6488ed-60f9-487f-b4bf-83c74fa0b06a";
    MongoTemplate mockMongoTemplate = EasyMock.createMock(MongoTemplate.class);
    PersistableDevice persistableDev = EasyMock.createMock(PersistableDevice.class);

    DawgPoundMongoService dawgPoundService = new DawgPoundMongoService();
    ReflectionTestUtils.setField(dawgPoundService, "mongoTemplate", mockMongoTemplate);

    EasyMock.expect(mockMongoTemplate.findById(TestConstants.DEVICE_ID, PersistableDevice.class,
            TestConstants.COLLECTION_NAME)).andReturn(persistableDev);
    EasyMock.expect(persistableDev.isReserved()).andReturn(true);
    EasyMock.expect(persistableDev.getReserver()).andReturn(prevReserveToken);
    EasyMock.replay(mockMongoTemplate, persistableDev);

    Assert.assertEquals(dawgPoundService.reserve(TestConstants.DEVICE_ID, TestConstants.TOKEN,
            (System.currentTimeMillis() + TestConstants.ONE_DAY_IN_MILLIS)), prevReserveToken);
    EasyMock.verify(mockMongoTemplate, persistableDev);
}

---

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

@Test
public void testOverrideReserve() {
    long expiryTime = System.currentTimeMillis() + TestConstants.ONE_DAY_IN_MILLIS;

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

    DawgPoundMongoService dawgPoundService = new DawgPoundMongoService();
    ReflectionTestUtils.setField(dawgPoundService, "mongoTemplate", mockMongoTemplate);

    EasyMock.expect(mockMongoTemplate.findById(TestConstants.DEVICE_ID, PersistableDevice.class,
            TestConstants.COLLECTION_NAME)).andReturn(persistableDev);
    EasyMock.expect(persistableDev.unreserve()).andReturn("previous_token");
    EasyMock.expect(persistableDev.reserve(TestConstants.TOKEN)).andReturn(TestConstants.TOKEN);
    persistableDev.setExpiration(expiryTime);
    mockMongoTemplate.save(persistableDev, TestConstants.COLLECTION_NAME);
    EasyMock.replay(mockMongoTemplate, persistableDev);

    Assert.assertTrue(//from   w  w  w.ja  v a  2 s.  c o m
            dawgPoundService.overrideReserve(TestConstants.DEVICE_ID, TestConstants.TOKEN, expiryTime));
    EasyMock.verify(mockMongoTemplate, persistableDev);
}

---

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

@Test
public void testIsReservedWithDeviceReserved() {
    long expiryTime = System.currentTimeMillis() + TestConstants.ONE_DAY_IN_MILLIS;
    MongoTemplate mockMongoTemplate = EasyMock.createMock(MongoTemplate.class);
    PersistableDevice persistableDev = EasyMock.createMock(PersistableDevice.class);

    DawgPoundMongoService dawgPoundService = new DawgPoundMongoService();
    ReflectionTestUtils.setField(dawgPoundService, "mongoTemplate", mockMongoTemplate);

    EasyMock.expect(mockMongoTemplate.findById(TestConstants.DEVICE_ID, PersistableDevice.class,
            TestConstants.COLLECTION_NAME)).andReturn(persistableDev);
    EasyMock.expect(persistableDev.isReserved()).andReturn(true);
    EasyMock.expect(persistableDev.getReserver()).andReturn(TestConstants.TOKEN);
    EasyMock.expect(persistableDev.getExpiration()).andReturn(expiryTime);

    EasyMock.replay(mockMongoTemplate, persistableDev);
    Map<String, String> map = dawgPoundService.isReserved(TestConstants.DEVICE_ID);
    Assert.assertNotNull(map);// w w w  .  j a  v  a2  s  . c o  m
    Assert.assertEquals(map.size(), 2);
    Assert.assertEquals(map.get("reserver"), TestConstants.TOKEN);
    Assert.assertEquals(map.get("expiration"), String.valueOf(expiryTime));
    EasyMock.verify(mockMongoTemplate, persistableDev);
}

---

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

@Test
public void testValidReservation() {
    String newToken = "8e6488ed-60f9-487f-b4bf-83c74fa0b06a";
    long expiryTime = System.currentTimeMillis() + TestConstants.ONE_DAY_IN_MILLIS;

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

    DawgPoundMongoService dawgPoundService = new DawgPoundMongoService();
    ReflectionTestUtils.setField(dawgPoundService, "mongoTemplate", mockMongoTemplate);

    EasyMock.expect(mockMongoTemplate.findById(TestConstants.DEVICE_ID, PersistableDevice.class,
            TestConstants.COLLECTION_NAME)).andReturn(persistableDev);
    EasyMock.expect(persistableDev.isReserved()).andReturn(false);
    EasyMock.expect(persistableDev.unreserve()).andReturn("previous_token");
    EasyMock.expect(persistableDev.reserve(newToken)).andReturn(newToken);
    persistableDev.setExpiration(expiryTime);
    EasyMock.expect(persistableDev.getKeys()).andReturn(new String[] {});
    persistableDev.setKeys(EasyMock.aryEq(new String[] { newToken }));
    mockMongoTemplate.save(persistableDev, TestConstants.COLLECTION_NAME);

    EasyMock.replay(mockMongoTemplate, persistableDev);

    Assert.assertEquals(dawgPoundService.reserve(TestConstants.DEVICE_ID, newToken, expiryTime), newToken);
    EasyMock.verify(mockMongoTemplate, persistableDev);
}

---