From 4c871b51bfd1cefa812b1b2b54b22c9482361aa3 Mon Sep 17 00:00:00 2001
From: Michael Hunger <github@jexp.de>
Date: Tue, 24 Sep 2013 14:19:41 +0200
Subject: [PATCH] DATAGRAPH-393 - Use supplied RestGraphDatabase as
 GraphDatabase in MappingInfrastructure.

---
 .../data/neo4j/support/MappingInfrastructureFactoryBean.java   | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/spring-data-neo4j/src/main/java/org/springframework/data/neo4j/support/MappingInfrastructureFactoryBean.java b/spring-data-neo4j/src/main/java/org/springframework/data/neo4j/support/MappingInfrastructureFactoryBean.java
index 4a82fb9f3..8184f49f5 100644
--- a/spring-data-neo4j/src/main/java/org/springframework/data/neo4j/support/MappingInfrastructureFactoryBean.java
+++ b/spring-data-neo4j/src/main/java/org/springframework/data/neo4j/support/MappingInfrastructureFactoryBean.java
@@ -107,6 +107,9 @@ public class MappingInfrastructureFactoryBean implements FactoryBean<Infrastruct
         if (this.graphDatabaseService == null && graphDatabase instanceof DelegatingGraphDatabase) {
             this.graphDatabaseService = ((DelegatingGraphDatabase) graphDatabase).getGraphDatabaseService();
         }
+        if (this.graphDatabase == null && graphDatabaseService instanceof GraphDatabase) {
+            this.graphDatabase = (GraphDatabase)graphDatabaseService;
+        }
         if (this.graphDatabase == null) {
             this.graphDatabase = new DelegatingGraphDatabase(graphDatabaseService);
         }