build-system/gradle/src/main/groovy/com/android/build/gradle/TestPlugin.groovy

/*
 * Copyright (C) 2015 The Android Open Source Project
 *
 * 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.android.build.gradle

import com.android.build.gradle.internal.DependencyManager
import com.android.build.gradle.internal.SdkHandler
import com.android.build.gradle.internal.TaskManager
import com.android.build.gradle.internal.TestApplicationTaskManager
import com.android.build.gradle.internal.variant.TestVariantFactory
import com.android.build.gradle.internal.variant.VariantFactory
import com.android.builder.core.AndroidBuilder
import org.gradle.api.Plugin
import org.gradle.api.Project
import org.gradle.internal.reflect.Instantiator
import org.gradle.tooling.provider.model.ToolingModelBuilderRegistry

import javax.inject.Inject

/**
 * Gradle plugin class for 'application' projects.
 */
class TestPlugin extends BasePlugin implements Plugin<Project> {
    @Inject
    public TestPlugin(Instantiator instantiator, ToolingModelBuilderRegistry registry) {
        super(instantiator, registry)
    }

    @Override
    protected Class<? extends BaseExtension> getExtensionClass() {
        return TestExtension.class
    }

    @Override
    protected TaskManager createTaskManager(
            Project project,
            AndroidBuilder androidBuilder,
            AndroidConfig extension,
            SdkHandler sdkHandler,
            DependencyManager dependencyManager,
            ToolingModelBuilderRegistry toolingRegistry) {
        return new TestApplicationTaskManager (
                project,
                androidBuilder,
                extension,
                sdkHandler,
                dependencyManager,
                toolingRegistry)
    }

    @Override
    void apply(Project project) {
        super.apply(project)
    }

    @Override
    protected VariantFactory createVariantFactory() {
        return new TestVariantFactory(instantiator, androidBuilder, extension)
    }
}