#!/usr/bin/env python3
# Copyright 2022 The Chromium Authors
# Use of this source code is governed by a BSD-style license that can be
# found in the LICENSE file.

import unittest

from make_gtest_filter import PascalCaseSplit, CompressWithWildcards, GetFiltersForTests


class Foo(unittest.TestCase):
  def testSplit(self):
    self.assertEqual(list(PascalCaseSplit('testTerm')), ['test', 'Term'])
    self.assertEqual(list(PascalCaseSplit('TestTerm')), ['Test', 'Term'])
    self.assertEqual(list(PascalCaseSplit('TestTerm80')),
                     ['Test', 'Term', '80'])
    self.assertEqual(list(PascalCaseSplit('TestTerm80Foo')),
                     ['Test', 'Term', '80', 'Foo'])
    self.assertEqual(list(PascalCaseSplit('TestTerm80/Foo')),
                     ['Test', 'Term', '80', '/', 'Foo'])

  def testDepth(self):
    melons = ['A.DoomMelonFooBar', 'A.DoomMelonFooBaz']
    self.assertEqual(list(CompressWithWildcards(melons, 1, 0)), ['A.Doom*'])
    self.assertEqual(list(CompressWithWildcards(melons, 2, 0)),
                     ['A.DoomMelon*'])
    self.assertEqual(list(CompressWithWildcards(melons, 3, 0)),
                     ['A.DoomMelonFoo*'])
    self.assertEqual(list(CompressWithWildcards(melons, 4, 0)),
                     ['A.DoomMelonFooBar', 'A.DoomMelonFooBaz'])

  def testDontWildcardAcrossSuites(self):
    self.assertEqual(list(CompressWithWildcards(['A.X', 'B.X'], 1, 0)),
                     ['A.X', 'B.X'])

  def testCaseNumBoundaryBeforeWildcard(self):
    fruit = ['A.DoomMelonFooBar', 'A.DoomMelonFooBaz', 'A.DoomBanana']
    self.assertEqual(list(CompressWithWildcards(fruit, 2, 0)),
                     ['A.DoomBanana', 'A.DoomMelon*'])
    self.assertEqual(list(CompressWithWildcards(fruit, 2, 1)),
                     ['A.DoomBanana', 'A.DoomMelon*'])
    self.assertEqual(list(CompressWithWildcards(fruit, 2, 2)),
                     ['A.DoomBanana', 'A.DoomMelonFooBar', 'A.DoomMelonFooBaz'])

  def testGetFiltersForTests(self):
    tests = ['TestSuite.TestName']
    self.assertEqual(list(GetFiltersForTests(tests, class_only=True)), [
        'TestSuite.*', '*/TestSuite.*/*', '*/TestSuite/*.*', 'TestSuite.*/*',
        'TestSuite/*.*'
    ])
    self.assertEqual(list(GetFiltersForTests(tests, class_only=False)), [
        'TestSuite.TestName', '*/TestSuite.TestName/*', 'TestSuite.TestName/*',
        'TestSuite/*.TestName'
    ])


if __name__ == '__main__':
  unittest.main()