convert key values properly with annotated with idOnly = true

fixes #794 (cherry picked from commit d67d0ae)
MorphiaOrg · Jun 11, 2015 · 1e9b185 · 1e9b185
1 parent 833fd62
commit 1e9b185
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 1 deletion.
diff --git a/morphia/src/main/java/org/mongodb/morphia/mapping/Mapper.java b/morphia/src/main/java/org/mongodb/morphia/mapping/Mapper.java
@@ -416,7 +416,10 @@ public Object toMongoObject(final MappedField mf, final MappedClass mc, final Ob
                     if (key == null) {
                         mappedValue = toMongoObject(value, false);
                     } else {
-                        mappedValue = keyToRef(key);
+                        final Reference refAnn = mf.getAnnotation(Reference.class);
+                        mappedValue = refAnn != null && refAnn.idOnly()
+                                      ? keyToManualRef(key)
+                                      : keyToRef(key);
                         if (mappedValue == value) {
                             throw new ValidationException("cannot map to @Reference/Key<T>/DBRef field: " + value);
                         }

diff --git a/morphia/src/test/java/org/mongodb/morphia/mapping/ReferencesNotUsingDBRefTest.java b/morphia/src/test/java/org/mongodb/morphia/mapping/ReferencesNotUsingDBRefTest.java
@@ -216,6 +216,18 @@ public void testNullReferences() {
         getDs().save(container);
         allNull(container);
     }
+
+    @Test
+    public void testFindByEntityReference() {
+        final Ref ref = new Ref("refId");
+        getDs().save(ref);
+
+        final Container container = new Container();
+        container.singleRef = ref;
+        getDs().save(container);
+
+        Assert.assertNotNull(getDs().find(Container.class, "singleRef", ref).get());
+    }
 
     private void allNull(final Container container) {
         Assert.assertNull(container.lazyMapRef);