src/main/java/org/mockito/internal/session/DefaultMockitoSessionBuilder.java

/*
 * Copyright (c) 2017 Mockito contributors
 * This program is made available under the terms of the MIT License.
 */
package org.mockito.internal.session;

import org.mockito.MockitoSession;
import org.mockito.internal.framework.DefaultMockitoSession;
import org.mockito.internal.util.ConsoleMockitoLogger;
import org.mockito.quality.Strictness;
import org.mockito.session.MockitoSessionBuilder;

public class DefaultMockitoSessionBuilder implements MockitoSessionBuilder {

    private Object testClassInstance;
    private Strictness strictness;

    @Override
    public MockitoSessionBuilder initMocks(Object testClassInstance) {
        this.testClassInstance = testClassInstance;
        return this;
    }

    @Override
    public MockitoSessionBuilder strictness(Strictness strictness) {
        this.strictness = strictness;
        return this;
    }

    @Override
    public MockitoSession startMocking() {
        //Configure default values
        Object effectiveTest = this.testClassInstance == null ? new Object() : this.testClassInstance;
        Strictness effectiveStrictness = this.strictness == null ? Strictness.STRICT_STUBS : this.strictness;
        return new DefaultMockitoSession(effectiveTest, effectiveStrictness, new ConsoleMockitoLogger());
    }
}