List of usage examples for org.springframework.data.mongodb.core MongoTemplate save
@Override @SuppressWarnings("unchecked") public <T> T save(T objectToSave, String collectionName)
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 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. jav a2 s. co 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 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); }
From source file:example.springdata.mongodb.fluent.ApplicationConfiguration.java
@Bean CommandLineRunner init(MongoTemplate template) { return (args) -> { if (template.collectionExists(COLLECTION)) { template.dropCollection(COLLECTION); }/*from ww w . jav a 2 s. c o m*/ GeospatialIndex index = new GeospatialIndex("homePlanet.coordinates") // .typed(GeoSpatialIndexType.GEO_2DSPHERE) // .named("planet-coordinate-idx"); template.createCollection(COLLECTION); template.indexOps(SWCharacter.class).ensureIndex(index); Planet alderaan = new Planet("alderaan", new Point(-73.9667, 40.78)); Planet stewjon = new Planet("stewjon", new Point(-73.9836, 40.7538)); Planet tatooine = new Planet("tatooine", new Point(-73.9928, 40.7193)); Jedi anakin = new Jedi("anakin", "skywalker"); anakin.setHomePlanet(tatooine); Jedi luke = new Jedi("luke", "skywalker"); luke.setHomePlanet(tatooine); Jedi leia = new Jedi("leia", "organa"); leia.setHomePlanet(alderaan); Jedi obiWan = new Jedi("obi-wan", "kenobi"); obiWan.setHomePlanet(stewjon); Human han = new Human("han", "solo"); template.save(anakin, COLLECTION); template.save(luke, COLLECTION); template.save(leia, COLLECTION); template.save(obiWan, COLLECTION); template.save(han, COLLECTION); }; }