#
# Copyright 2017 The Cobalt Authors. 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.
"""Class used to filter out unit tests."""

# Constant to use for filtering all unit tests from a specific test executable.
# Provide it as the "test_name" argument to TestFilter's constructor.
FILTER_ALL = 'FILTER_ALL'
DISABLE_TESTING = 'DISABLE_TESTING'

EVERGREEN_ONLY_TESTS = {
    'elf_loader_test': {FILTER_ALL},
    'installation_manager_test': {FILTER_ALL},
    'reset_evergreen_update_test': {FILTER_ALL},
    'slot_management_test': {FILTER_ALL},
}


class TestFilter(object):
  """Container for data used to filter out a unit test.

  Initialization is done with three arguments:

  target_name:  The name of the unit test binary from which to remove tests.
  test_name:  The name of a specific test from the provided test target, or a
    constant from this module defining a group of tests to filter.
  config: An optional argument specifying for which build configuration
    the test should be excluded.  If this name is not provided, the test
    will be excluded from all test runs.
  """

  def __init__(self, target_name, test_name, config=None):
    self.target_name = target_name
    self.test_name = test_name
    self.config = config