TestMapping.java

package org.mongodb.morphia.issue241;


import org.bson.types.ObjectId;
import org.junit.Assert;
import org.junit.Test;
import org.mongodb.morphia.TestBase;
import org.mongodb.morphia.annotations.Entity;
import org.mongodb.morphia.annotations.Id;
import org.mongodb.morphia.dao.BasicDAO;


/**
 * Unit test for testing morphia mappings with generics.
 */
public class TestMapping extends TestBase {

    @Test
    public void testMapping() {
        final BasicDAO<Message, ObjectId> messageDAO = new BasicDAO<Message, ObjectId>(Message.class, getDs());
        Assert.assertNotNull(messageDAO);
    }

    @Entity
    private static class Message<U extends User> {

        @Id
        private ObjectId id;
        private U user;

        public U getUser() {
            return user;
        }

        public void setUser(final U user) {
            this.user = user;
        }
    }

    @Entity
    private static class User {
        @Id
        private ObjectId id;

        @Override
        public int hashCode() {
            int hash = 3;
            hash = 97 * hash + (id != null ? id.hashCode() : 0);
            return hash;
        }

        @Override
        public boolean equals(final Object obj) {
            if (obj == null) {
                return false;
            }
            if (getClass() != obj.getClass()) {
                return false;
            }
            final User other = (User) obj;
            return !(id != other.id && (id == null || !id.equals(other.id)));
        }
    }
}