Тестирование баз данных с помощью JUnit и Hibernate. Часть 3. Очистка и дальнейшие идеи

    public class SuperHeroMother {
     private final Session session;
     
     private String secretIdentity = "Mr. Jones";
     private String name = "Name";
     private String weakness = "None";
     private SuperPower power = null;
     
     public SuperHeroMother(Session s) {
      session = s;
      power = new SuperPowerMother(session).instance();
     }
     
     public SuperHero instance() {
      SuperHero hero = loadInstance();
      if (hero == null) {
       hero = createInstance();
      }
      hero.power = power;
      hero.weakness = weakness;
      hero.secretIdentity = secretIdentity;
      session.save(hero);
      return hero;
     }
     
     private SuperHero loadInstance() {
      return (SuperHero) session.createCriteria(SuperHero.class)
        .add(Restrictions.eq("name", name)).uniqueResult();
     
     }
     
     private SuperHero createInstance() {
      SuperHero hero = new SuperHero();
      hero.name = name;
      return hero;
     }
     
     public SuperHeroMother name(String aName) {
      name = aName;
      return this;
     }
     
     public SuperHeroMother secretIdentity(String aSecretIdentity) {
      secretIdentity = aSecretIdentity;
      return this;
     }
     
     public SuperHeroMother power(SuperPower aPower) {
      power = aPower;
      return this;
     }
     
     public SuperHeroMother weaknes(String aWeakness) {
      weakness = aWeakness;
      return this;
     }
    }

    /** create instances of type <tt>T</tt> */
    public abstract class ObjectMother<T> {
     private final Session session;
     
     public ObjectMother(Session s) {
      session = s;
     }
     
     /** returns an instance based on the configuration of this object mother */
     public T instance() {
      T t = loadInstance(session);
      if (t == null)
       t = createInstance();
      configureInstance(t);
      session.save(t);
      return t;
     }
     
     /**
      * configure the instance <tt>t</tt> according to the configuration of this
      * ObjectMother
      */
     abstract protected void configureInstance(T t);
     
     /**
      * try to load an instance based on the alternate key. Returns null if no
      * such instance exists
      */
     abstract protected T loadInstance(Session session);
     
     /**
      * create a fresh instance with the alternate key set according to the
      * configuration of this ObjectMother
      */
     abstract protected T createInstance();
    }

    public abstract class SingleAlternateKeyObjectMother<T, A, S extends SingleAlternateKeyObjectMother<T, A, S>>
      extends ObjectMother<T> {
     
     private final Class<T> objectType;
     private final String alternateKeyName;
     private A alternateKey;
     
     public SingleAlternateKeyObjectMother(Session s, Class<T> theObjectType,
       A defaultAlternateKey, String theAlternateKeyName) {
      super(s);
      objectType = theObjectType;
      alternateKeyName = theAlternateKeyName;
      alternateKey(defaultAlternateKey);
     }
     
     @SuppressWarnings("unchecked")
     @Override
     final protected T loadInstance(Session session) {
      return (T) session.createCriteria(objectType)
        .add(Restrictions.eq(alternateKeyName, getAlternateKey()))
        .uniqueResult();
     
     }
     
     @SuppressWarnings("unchecked")
     public S alternateKey(A theAlternateKey) {
      alternateKey = theAlternateKey;
      return (S) this;
     }
     
     public A getAlternateKey() {
      return alternateKey;
     }
    }

    public class SuperHeroMother extends
      SingleAlternateKeyObjectMother<SuperHero, String, SuperHeroMother> {
     
     private String secretIdentity = "Mr. Jones";
     private String weakness = "None";
     private SuperPower power;
     
     public SuperHeroMother(Session s) {
      super(s, SuperHero.class, "Name", "name");
      power = new SuperPowerMother(s).instance();
     }
     
     @Override
     protected void configureInstance(SuperHero hero) {
      hero.power = power;
      hero.weakness = weakness;
      hero.secretIdentity = secretIdentity;
     }
     
     @Override
     protected SuperHero createInstance() {
      SuperHero hero = new SuperHero();
      hero.name = getAlternateKey();
      return hero;
     }
     
     public SuperHeroMother secretIdentity(String aSecretIdentity) {
      secretIdentity = aSecretIdentity;
      return this;
     }
     
     public SuperHeroMother power(SuperPower aPower) {
      power = aPower;
      return this;
     }
     
     public SuperHeroMother weaknes(String aWeakness) {
      weakness = aWeakness;
      return this;
     }
    }