test: TestDataBuilders in den master gemergt

Thomas Ziemer
2 parents 40a078fc 16a60304
Showing 5 changed files with 184 additions and 1 deletions
src/main/java/net/ziemers/swxercise/lg/model/user/Profile.java
src/main/java/net/ziemers/swxercise/lg/model/user/User.java
src/test/java/net/ziemers/swxercise/lg/testdata/testdatabuilder/AbstractTestDataBuilder.java
src/test/java/net/ziemers/swxercise/lg/testdata/testdatabuilder/user/ProfileTestDataBuilder.java
src/test/java/net/ziemers/swxercise/lg/testdata/testdatabuilder/user/UserTestDataBuilder.java
@@ -133,7 +133,7 @@ public class Profile extends BaseEntity {
         return username;
     }
-    public void setUsername(String username) {
+    private void setUsername(String username) {
         this.username = username;
     }
@@ -183,4 +183,9 @@ public class Profile extends BaseEntity {
         this.mailaddress = mailaddress;
     }
+    public Profile withMailaddress(final String mailaddress) {
+        setMailaddress(mailaddress);
+        return this;
+    }
+
 }
@@ -71,4 +71,9 @@ public class User extends BaseEntity {
         this.address = address;
     }
+    public User withAddress(final Address address) {
+        setAddress(address);
+        return this;
+    }
+
 }
+package net.ziemers.swxercise.lg.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.testdata.testdatabuilder.user;
+
+import net.ziemers.swxercise.lg.model.user.Profile;
+import net.ziemers.swxercise.lg.testdata.testdatabuilder.AbstractTestDataBuilder;
+
+import javax.persistence.EntityManager;
+
+public class ProfileTestDataBuilder extends AbstractTestDataBuilder {
+
+    private String username = "tziemer";
+
+    private String password = "secret";
+
+    private String mailaddress = "tziemer@ziemers.net";
+
+    public ProfileTestDataBuilder(final EntityManager em) {
+        super(em);
+    }
+
+    @Override
+    public Profile build() {
+        return new Profile(username, password)
+                .withMailaddress(mailaddress);
+    }
+
+    public ProfileTestDataBuilder withUsername(final String username) {
+        this.username = username;
+        return this;
+    }
+
+    public ProfileTestDataBuilder withPassword(final String password) {
+        this.password = password;
+        return this;
+    }
+
+    public ProfileTestDataBuilder withMailaddress(final String mailaddress) {
+        this.mailaddress = mailaddress;
+        return this;
+    }
+
+}
+package net.ziemers.swxercise.lg.testdata.testdatabuilder.user;
+
+import net.ziemers.swxercise.lg.model.user.Address;
+import net.ziemers.swxercise.lg.model.user.Profile;
+import net.ziemers.swxercise.lg.model.user.User;
+import net.ziemers.swxercise.lg.testdata.testdatabuilder.AbstractTestDataBuilder;
+
+import javax.persistence.EntityManager;
+
+public class UserTestDataBuilder extends AbstractTestDataBuilder<User> {
+
+    private String firstname = "Hein";
+
+    private String lastname = "Bloed";
+
+    private Profile profile;
+
+    private Address address = null;
+
+    public UserTestDataBuilder(final EntityManager 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 UserTestDataBuilder withProfile(final Profile profile) {
+        this.profile = profile;
+        return this;
+    }
+
+    public UserTestDataBuilder withAddress(final Address address) {
+        this.address = address;
+        return this;
+    }
+
+}
+