Examples with SessionAwareMethodInterceptor org.springframework.data.mongodb.SessionAwareMethodInterceptor used on opensource projects

Example 1 with SessionAwareMethodInterceptor

use of org.springframework.data.mongodb.SessionAwareMethodInterceptor in project spring-data-mongodb by spring-projects.

the class SessionBoundMongoTemplateTests method setUp.

@Before
public void setUp() {
    MongoClient client = new MongoClient();
    MongoDbFactory factory = new SimpleMongoDbFactory(client, "session-bound-mongo-template-tests") {

        @Override
        public MongoDatabase getDb() throws DataAccessException {
            MongoDatabase spiedDatabse = Mockito.spy(super.getDb());
            spiedDatabases.add(spiedDatabse);
            return spiedDatabse;
        }
    };
    session = client.startSession(ClientSessionOptions.builder().build());
    this.template = new MongoTemplate(factory);
    this.sessionBoundTemplate = new SessionBoundMongoTemplate(session, new MongoTemplate(factory, getDefaultMongoConverter(factory))) {

        @Override
        protected MongoCollection<Document> prepareCollection(MongoCollection<Document> collection) {
            injectCollectionSpy(collection);
            return super.prepareCollection(collection);
        }

        @SuppressWarnings({ "ConstantConditions", "unchecked" })
        private void injectCollectionSpy(MongoCollection<Document> collection) {
            InvocationHandler handler = Proxy.getInvocationHandler(collection);
            Advised advised = (Advised) ReflectionTestUtils.getField(handler, "advised");
            for (Advisor advisor : advised.getAdvisors()) {
                Advice advice = advisor.getAdvice();
                if (advice instanceof SessionAwareMethodInterceptor) {
                    MongoCollection<Document> spiedCollection = Mockito.spy((MongoCollection<Document>) ReflectionTestUtils.getField(advice, "target"));
                    spiedCollections.add(spiedCollection);
                    ReflectionTestUtils.setField(advice, "target", spiedCollection);
                }
            }
        }
    };
}