# 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.
"""Tests for disable.py"""

import unittest
import disable


class DisableTest(unittest.TestCase):
  def test_parse_bug(self):
    self.assertEqual(disable.parse_bug('1234'), (1234, 'chromium'))
    self.assertEqual(disable.parse_bug('crbug/9871'), (9871, 'chromium'))
    self.assertEqual(
      disable.parse_bug('https://crbug/9871'), (9871, 'chromium')
    )
    self.assertEqual(disable.parse_bug('crbug/v8/111111'), (111111, 'v8'))
    self.assertEqual(
      disable.parse_bug('https://crbug/v8/111111'), (111111, 'v8')
    )
    self.assertEqual(disable.parse_bug('crbug.com/8'), (8, 'chromium'))
    self.assertEqual(disable.parse_bug('https://crbug.com/8'), (8, 'chromium'))
    self.assertEqual(
      disable.parse_bug('crbug.com/monorail/19782757'), (19782757, 'monorail')
    )
    self.assertEqual(
      disable.parse_bug('https://crbug.com/monorail/19782757'),
      (19782757, 'monorail'),
    )
    self.assertEqual(
      disable.parse_bug('bugs.chromium.org/p/foo/issues/detail?id=6060842'),
      (6060842, 'foo'),
    )
    self.assertEqual(
      disable.parse_bug(
        'http://bugs.chromium.org/p/foo/issues/detail?id=6060842'
      ),
      (6060842, 'foo'),
    )
    self.assertEqual(
      disable.parse_bug(
        'https://bugs.chromium.org/p/foo/issues/detail?id=6060842'
      ),
      (6060842, 'foo'),
    )
    self.assertEqual(
      disable.parse_bug(
        'https://bugs.chromium.org/p/foo/issues/detail'
        + '?id=191972&q=owner%3Ame%20link&can=2'
      ),
      (191972, 'foo'),
    )


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