test/groovy/org/codehaus/groovy/grails/orm/hibernate/UserTypeMappingTests.groovy

   1 package org.codehaus.groovy.grails.orm.hibernate
   2
   3 import javax.sql.DataSource
   4
   5 /**
   6 * @author Graeme Rocher
   7 */
   8 class UserTypeMappingTests extends AbstractGrailsHibernateTests{
   9
  10     protected void onSetUp() {
  11         gcl.parseClass '''
  12 import org.hibernate.type.*
  13 class UserTypeMappingTest
  14 {
  15   Long id
  16   Long version
  17
  18   Boolean active
  19
  20   static mapping = {
  21     table 'type_test'
  22     columns {
  23       active (column: 'active', type: YesNoType)
  24     }
  25   }
  26 }
  27
  28
  29 '''
  30        gcl.parseClass '''
  31 import org.hibernate.usertype.UserType
  32
  33 import org.hibernate.type.Type
  34 import java.sql.PreparedStatement
  35 import java.sql.ResultSet
  36 import java.sql.SQLException
  37 import org.hibernate.Hibernate
  38 import org.hibernate.HibernateException
  39 import java.sql.Types
  40
  41 class WeightUserType implements UserType {
  42
  43     private static final int[] SQL_TYPES = [ Types.INTEGER ]
  44     public int[] sqlTypes() {
  45         return SQL_TYPES
  46     }
  47
  48     public Class returnedClass() {
  49         return Weight.class
  50     }
  51
  52     public boolean equals(Object x, Object y) throws HibernateException {
  53         if (x == y) {
  54           return true;
  55         } else if (x == null || y == null) {
  56           return false;
  57         } else {
  58           return x.equals(y);
  59         }
  60     }
  61
  62     public int hashCode(Object x) throws HibernateException {
  63         return x.hashCode()
  64     }
  65
  66   public Object nullSafeGet(ResultSet resultSet,  String[] names, Object owner) throws HibernateException, SQLException {
  67     Weight result = null;
  68     int pounds = resultSet.getInt(names[0])
  69     if (!resultSet.wasNull()) {
  70       result = new Weight(pounds)
  71     }
  72     return result;
  73   }
  74
  75     public void nullSafeSet(PreparedStatement statement,  Object value, int index)   throws HibernateException, SQLException {
  76         if (value == null) {
  77             statement.setNull(index);
  78         } else {
  79             Integer pounds = value.pounds
  80             statement.setInt(index, pounds);
  81         }
  82     }
  83
  84   public Object deepCopy(Object value) throws HibernateException {
  85     return value;
  86   }
  87
  88   public boolean isMutable() {
  89     return false;
  90   }
  91
  92     public Serializable disassemble(Object value) throws HibernateException {
  93         return (Serializable) value;
  94     }
  95
  96     public Object assemble(Serializable state, Object owner) throws HibernateException {
  97         return state;
  98     }
  99
 100     public Object replace(Object original, Object target, Object owner) throws HibernateException {
 101         return original;
 102     }
 103 }
 104
 105 class Weight {
 106     Integer pounds
 107     Weight(Integer pounds)
 108     {
 109         this.pounds= pounds
 110     }
 111 }
 112 '''
 113
 114        gcl.parseClass '''
 115 class UserTypeMappingTestsPerson {
 116     Long id
 117     Long version
 118     String name
 119     Weight weight
 120
 121     static constraints = {
 122         name(unique: true)
 123         weight(nullable: true)
 124     }
 125
 126     static mapping = {
 127         columns {
 128             weight( type:WeightUserType)
 129         }
 130     }
 131
 132 }
 133
 134 '''
 135     }
 136
 137
 138     void testCustomUserType() {
 139         def personClass = ga.getDomainClass("UserTypeMappingTestsPerson").clazz
 140         def weightClass = ga.classLoader.loadClass("Weight")
 141
 142         def person = personClass.newInstance(name:"Fred", weight:weightClass.newInstance(200))
 143
 144         person.save(flush:true)
 145         session.clear()
 146
 147         person = personClass.get(1)
 148
 149         assert person
 150         assert person.weight
 151         assertEquals 200, person.weight.pounds
 152     }
 153
 154     void testUserTypeMapping() {
 155
 156         def clz = ga.getDomainClass("UserTypeMappingTest").clazz
 157
 158
 159         assert clz.newInstance(active:true).save(flush:true)
 160
 161         DataSource ds = (DataSource)applicationContext.getBean('dataSource')
 162
 163          def con
 164          try {
 165              con = ds.getConnection()
 166              def statement = con.prepareStatement("select * from type_test")
 167              def result = statement.executeQuery()
 168              assert result.next()
 169              def value = result.getString('active')
 170
 171              assertEquals "Y", value
 172
 173          } finally {
 174              con.close()
 175          }
 176     }
 177
 178
 179 }