/*
 * Copyright 2017, TeamDev Ltd. All rights reserved.
 *
 * Redistribution and use in source and/or binary forms, with or without
 * modification, must retain the above copyright notice and the following
 * disclaimer.
 *
 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
 * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
 * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
 * A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
 * OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
 * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
 * LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
 * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
 * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
 * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
 */

package org.spine3.server.aggregate;

import com.google.protobuf.Message;
import org.spine3.server.entity.AbstractEntity;
import org.spine3.server.entity.EntityBuilder;

import java.lang.reflect.Constructor;

/**
 * Utility class for building {@code AggregatePart}s for tests.
 *
 * @author Alexander Yevsyukov
 */
public class AggregatePartBuilder<A extends AggregatePart<I, S, ?>, I, S extends Message>
        extends AggregateBuilder<A, I, S> {

    private AggregateRoot<I> aggregateRoot;

    /**
     * {@inheritDoc}
     */
    public AggregatePartBuilder() {
        super();
        // Have the constructor for easier location of usages.
    }

    @Override
    public AggregatePartBuilder<A, I, S> setResultClass(Class<A> entityClass) {
        super.setResultClass(entityClass);
        return this;
    }

    @Override
    protected A createEntity(I id) {
        final Constructor<A> constructor = getConstructor();
        final AggregateRoot<I> root = getAggregateRoot();
        final A result = AbstractEntity.createEntity(constructor, id, root);
        return result;
    }

    @Override
    protected Constructor<A> getConstructor() {
        final Constructor<A> constructor = AbstractEntity.getAggregatePartConstructor(
                getResultClass(), getIdClass());
        constructor.setAccessible(true);
        return constructor;
    }

    public EntityBuilder<A, I, S> withRoot(AggregateRoot<I> root) {
        this.aggregateRoot = root;
        return this;
    }

    public AggregateRoot<I> getAggregateRoot() {
        return aggregateRoot;
    }
}

@@ -1657,32 +1657,181 @@
  usages.%0A    %7D%0A%0A
+    public EntityBuilder%3CA, I, S%3E withRoot(AggregateRoot%3CI%3E aggregateRoot) %7B%0A        this.aggregateRoot = aggregateRoot;%0A        return this;%0A    %7D%0A%0A
     @Override%0A  
@@ -1824,24 +1824,24 @@
   @Override%0A
-
     public A
@@ -2129,20 +2129,17 @@
  root = 
-getA
+a
 ggregate
@@ -2142,18 +2142,16 @@
 gateRoot
-()
 ;%0A      
@@ -2490,16 +2490,16 @@
 (true);%0A
+
         
@@ -2527,224 +2527,7 @@
    %7D
-%0A%0A    public EntityBuilder%3CA, I, S%3E withRoot(AggregateRoot%3CI%3E root) %7B%0A        this.aggregateRoot = root;%0A        return this;%0A    %7D%0A%0A    public AggregateRoot%3CI%3E getAggregateRoot() %7B%0A        return aggregateRoot;%0A    %7D
 %0A%7D%0A

Remove useless getter.