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

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

Introduction

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

Prototype

@Override
    @SuppressWarnings("unchecked")
    public <T> T save(T objectToSave, String collectionName) 

Source Link

Usage

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);
    };
}