lint/libs/lint-tests/src/main/java/com/android/tools/lint/checks/infrastructure/TestConfiguration.kt

/*
 * Copyright (C) 2016 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.tools.lint.checks.infrastructure

import com.android.tools.lint.client.api.Configuration
import com.android.tools.lint.client.api.DefaultConfiguration
import com.android.tools.lint.client.api.IssueRegistry
import com.android.tools.lint.client.api.LintClient
import com.android.tools.lint.detector.api.Context
import com.android.tools.lint.detector.api.Issue
import com.android.tools.lint.detector.api.Location
import com.android.tools.lint.detector.api.Project
import com.android.tools.lint.detector.api.Severity
import org.junit.Assert.fail

class TestConfiguration(
    private val task: TestLintTask,
    client: LintClient,
    project: Project,
    parent: Configuration?
) : DefaultConfiguration(client, project, parent) {

    override fun getDefaultSeverity(issue: Issue): Severity {
        // In unit tests, include issues that are ignored by default
        val severity = super.getDefaultSeverity(issue)
        if (severity == Severity.IGNORE) {
            if (issue.defaultSeverity != Severity.IGNORE) {
                return issue.defaultSeverity
            }
            return Severity.WARNING
        }
        return severity
    }

    override fun isEnabled(issue: Issue): Boolean {
        when {
            issue == IssueRegistry.LINT_ERROR -> return task.allowSystemErrors ||
                    !task.allowCompilationErrors
            issue == IssueRegistry.PARSER_ERROR -> return !task.allowSystemErrors
            issue == IssueRegistry.OBSOLETE_LINT_CHECK -> return !task.allowObsoleteLintChecks
            task.issueIds != null -> for (id in task.issueIds!!) {
                if (issue.id == id) {
                    return true
                }
            }
        }

        return task.checkedIssues.contains(issue)
    }

    override fun ignore(
        context: Context,
        issue: Issue,
        location: Location?,
        message: String
    ) = fail("Not supported in tests.")

    override fun setSeverity(issue: Issue, severity: Severity?) = fail("Not supported in tests.")
}