test: einige TestDataBuilders (WIP)

Thomas Ziemer
1 parent 619ec32c
Showing 3 changed files with 134 additions and 0 deletions
src/main/java/net/ziemers/swxercise/lg/model/testdata/testdatabuilder/AbstractTestDataBuilder.java
src/main/java/net/ziemers/swxercise/lg/model/testdata/testdatabuilder/UserTestDataBuilder.java
src/main/java/net/ziemers/swxercise/lg/model/user/User.java
+package net.ziemers.swxercise.lg.model.testdata.testdatabuilder;
+
+import javax.persistence.EntityManager;
+
+public abstract class AbstractTestDataBuilder<T> {
+
+    private static int count = 0;
+
+    private EntityManager entityManager;
+
+    /**
+     * Creates an new TestdataBuilder with persistence.
+     *
+     * @param entityManager
+     */
+    public AbstractTestDataBuilder(final EntityManager entityManager) {
+        this.entityManager = entityManager;
+    }
+
+    /**
+     * Creates an new TestdataBuilder without persistence.
+     */
+    public AbstractTestDataBuilder() {
+    }
+
+    public abstract T build();
+
+    /**
+     * Returns the EntityManager or null.
+     *
+     * @return {@link EntityManager} or null
+     */
+    protected final EntityManager getEntityManager() {
+        return entityManager;
+    }
+
+    /**
+     * Ensure the TestdataBuilder is constructed with a
+     * {@link EntityManager}
+     *
+     * @throws IllegalStateException
+     *             if the TestdataBuilder is constructed without a
+     *             {@link EntityManager}
+     */
+    protected final void ensureEntityManager() {
+        if (entityManager == null) {
+            throw new IllegalStateException("Cannot persist w/o entity manager");
+        }
+    }
+
+    /**
+     * {@inheritDoc} Executed within a new transaction.
+     *
+     * @throws IllegalStateException
+     *             if the TestdataBuilder is constructed without a
+     *             {@link EntityManager}
+     */
+    public final T buildAndSave() {
+        ensureEntityManager();
+        try {
+            final T obj = build();
+            entityManager.persist(obj);
+            return obj;
+        } catch (Exception e) {
+            throw new RuntimeException(e);
+        }
+    }
+
+    /**
+     * Returns an integer value from an static counter.
+     *
+     * @return value of the static counter.
+     */
+    protected final int getId() {
+        return count++;
+    }
+
+}
+
+package net.ziemers.swxercise.lg.model.testdata.testdatabuilder;
+
+import net.ziemers.swxercise.lg.model.user.Address;
+import net.ziemers.swxercise.lg.model.user.Profile;
+import net.ziemers.swxercise.lg.model.user.User;
+
+public class UserTestDataBuilder extends AbstractTestDataBuilder<User> {
+
+    private String firstname = "Hein";
+
+    private String lastname = "Bloed";
+
+    private Profile profile;
+
+    private Address address = null;
+
+    public UserTestDataBuilder(final EntityManger em) {
+        super(em);
+        profile = new ProfileTestDataBuilder(em).build();
+    }
+
+    @Override
+    public User build() {
+        return new User(firstname, lastname)
+                .withProfile(profile)
+                .withAddress(address);
+    }
+
+    public UserTestDataBuilder withFirstname(final String firstname) {
+        this.firstname = firstname;
+        return this;
+    }
+
+    public UserTestDataBuilder withLastname(final String lastname) {
+        this.lastname = lastname;
+        return this;
+    }
+
+    public UserTestDataBuidler withProfile(final Profile profile) {
+        this.profile = profile;
+        return this;
+    }
+
+    public UserTestDataBuilder withAddress(final Address address) {
+        this.address = address;
+        return this;
+    }
+
+}
+
@@ -71,4 +71,9 @@ public class User extends BaseEntity {
         this.address = address;
     }
  
+    public User withAddress(final Address address) {
+        setAddress(address);
+        return this;
+    }
+
 }