com.speedment.internal.util.testing.MockManagerImpl Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of speedment Show documentation
There is a newer version: 3.0.0-EA
/**
 *
 * Copyright (c) 2006-2016, Speedment, Inc. All Rights Reserved.
 *
 * Licensed under the Apache License, Version 2.0 (the "License"); You may not
 * use this file except in compliance with the License. You may obtain a copy of
 * the License at:
 *
 * http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
 * WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
 * License for the specific language governing permissions and limitations under
 * the License.
 */
package com.speedment.internal.util.testing;

import com.speedment.Speedment;
import com.speedment.config.db.Table;
import com.speedment.db.MetaResult;
import com.speedment.exception.SpeedmentException;
import com.speedment.field.FieldIdentifier;
import com.speedment.field.trait.ComparableFieldTrait;
import com.speedment.field.trait.FieldTrait;
import com.speedment.field.trait.ReferenceFieldTrait;
import com.speedment.manager.Manager;
import com.speedment.stream.StreamDecorator;
import com.speedment.util.tuple.Tuple;
import java.sql.Array;
import java.sql.Blob;
import java.sql.Clob;
import java.sql.NClob;
import java.sql.SQLException;
import java.sql.SQLXML;
import java.sql.Struct;
import java.util.Optional;
import java.util.function.BiFunction;
import java.util.function.Consumer;
import java.util.function.Function;
import java.util.function.Supplier;
import java.util.stream.Stream;

/**
 *
 * @author pemi
 * @param  type
 */
public class MockManagerImpl implements MockManager {

    private final Manager inner;
    private Supplier instanceSupplier;
    private Function> nativeStreamer;
    private Function> streamer;
    private Function persister;
    private Function updater;
    private Function remover;
    private BiFunction, Optional> finder;

    public MockManagerImpl(Manager inner) {
        this.inner = inner;
        this.instanceSupplier = inner::newEmptyEntity;
        this.nativeStreamer = inner::nativeStream;
        this.streamer = inner::stream;
        this.persister = inner::persist;
        this.updater = inner::update;
        this.remover = inner::remove;
    }

    // MockManager
    @Override
    public MockManager setInstanceFactory(Supplier factory) {
        instanceSupplier = factory;
        return this;
    }

    @Override
    public MockManager setNativeStreamer(Function> nativeStreamer) {
        this.nativeStreamer = nativeStreamer;
        return this;
    }

    @Override
    public MockManager setStreamer(Function> streamer) {
        this.streamer = streamer;
        return this;
    }

    @Override
    public MockManager setPersister(Function persister) {
        this.persister = persister;
        return this;
    }

    @Override
    public MockManager setUpdater(Function updater) {
        this.updater = updater;
        return this;
    }

    @Override
    public MockManager setRemover(Function remover) {
        this.remover = remover;
        return this;
    }

    @Override
    public , F extends FieldTrait & ReferenceFieldTrait & ComparableFieldTrait>
        MockManager setFinder(BiFunction> finder) {
        @SuppressWarnings("unchecked")
        final BiFunction, Optional> castedFinder
            = (BiFunction, Optional>) (BiFunction) finder;
        this.finder = castedFinder;

        return this;
    }

    // Manager
    @Override
    public Object primaryKeyFor(ENTITY entity) {
        return inner.primaryKeyFor(entity);
    }

    @Override
    public Object get(ENTITY entity, FieldIdentifier identifier) {
        return inner.get(entity, identifier);
    }

    @Override
    public void set(ENTITY entity, FieldIdentifier identifier, Object value) {
        inner.set(entity, identifier, value);
    }

    @Override
    public Table getTable() {
        return inner.getTable();
    }

    @Override
    public ENTITY newEmptyEntity() {
        return instanceSupplier.get();
    }

    @Override
    public Class getEntityClass() {
        return inner.getEntityClass();
    }

    @Override
    public Class> getManagerClass() {
        return inner.getManagerClass();
    }

    @Override
    public Tuple getPrimaryKeyClasses() {
        return inner.getPrimaryKeyClasses();
    }

    @Override
    public String toJson(ENTITY entity) {
        return inner.toJson(entity);
    }

    @Override
    public Stream stream(StreamDecorator decorator) {
        return streamer.apply(decorator);
    }

    @Override
    public Stream nativeStream(StreamDecorator decorator) {
        return nativeStreamer.apply(decorator);
    }

    @Override
    public ENTITY persist(ENTITY entity) throws SpeedmentException {
        return persister.apply(entity);
    }

    @Override
    public ENTITY update(ENTITY entity) throws SpeedmentException {
        return updater.apply(entity);
    }

    @Override
    public ENTITY remove(ENTITY entity) throws SpeedmentException {
        return remover.apply(entity);
    }

    @Override
    public , F extends FieldTrait & ReferenceFieldTrait & ComparableFieldTrait>
        Optional findAny(F field, V value) {
        return finder.apply(field, value);
    }

    @Override
    public ENTITY persist(ENTITY entity, Consumer> consumer) throws SpeedmentException {
        return inner.persist(entity, consumer);
    }

    @Override
    public ENTITY update(ENTITY entity, Consumer> consumer) throws SpeedmentException {
        return inner.update(entity, consumer);
    }

    @Override
    public ENTITY remove(ENTITY entity, Consumer> consumer) throws SpeedmentException {
        return inner.remove(entity, consumer);
    }

    @Override
    public Manager initialize() {
        return inner.initialize();
    }

    @Override
    public Manager load() {
        return inner.load();
    }

    @Override
    public Manager resolve() {
        return inner.resolve();
    }

    @Override
    public Manager start() {
        return inner.start();
    }

    @Override
    public Manager stop() {
        return inner.stop();
    }

    @Override
    public State getState() {
        return inner.getState();
    }

    @Override
    public void setState(State state) {
        inner.setState(state);
    }

    @Override
    public boolean isInitialized() {
        return inner.isInitialized();
    }

    @Override
    public boolean isLoaded() {
        return inner.isLoaded();
    }

    @Override
    public boolean isResolved() {
        return inner.isResolved();
    }

    @Override
    public boolean isStarted() {
        return inner.isStarted();
    }

    @Override
    public boolean isStopped() {
        return inner.isStopped();
    }

    @Override
    public Speedment speedment() {
        return inner.speedment();
    }

    @Override
    public Stream fields() {
        return inner.fields();
    }

    @Override
    public Stream primaryKeyFields() {
        return inner.primaryKeyFields();
    }

    @Override
    public Clob createClob() throws SQLException {
        return inner.createClob();
    }

    @Override
    public Blob createBlob() throws SQLException {
        return inner.createBlob();
    }

    @Override
    public NClob createNClob() throws SQLException {
        return createNClob();
    }

    @Override
    public SQLXML createSQLXML() throws SQLException {
        return createSQLXML();
    }

    @Override
    public Array createArrayOf(String typeName, Object[] elements) throws SQLException {
        return inner.createArrayOf(typeName, elements);
    }

    @Override
    public Struct createStruct(String typeName, Object[] attributes) throws SQLException {
        return inner.createStruct(typeName, attributes);
    }

    @Override
    public ENTITY newCopyOf(ENTITY source) {
        return inner.newCopyOf(source);
    }

}