#!/usr/bin/env python3
# Copyright 2022 The Chromium Authors. All rights reserved.
# 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


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'])


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