blob: 6e8965860bb59190e1edcdc902498c2404aac3b7 [file] [log] [blame]
#!/usr/bin/env python
# Copyright 2016 The Chromium Authors
# Use of this source code is governed by a BSD-style license that can be
# found in the LICENSE file.
import collections
import compact_blocklist as cb
import unittest
class CompactBlocklistTest(unittest.TestCase):
def TestValidUUID(self):
self.assertTrue( cb.ValidUUID('00000000-0000-0000-0000-000000000000'))
self.assertTrue( cb.ValidUUID('01234567-89ab-cdef-0123-456789abcdef'))
self.assertTrue( cb.ValidUUID('00001812-0000-1000-8000-00805f9b34fb'))
self.assertFalse(cb.ValidUUID('g1234567-89ab-cdef-0123-456789abcdef'))
self.assertFalse(cb.ValidUUID('01234567-89ab-cdef-0123-456789abcdef0'))
self.assertFalse(cb.ValidUUID('0123456789abcdef0123456789abcdef'))
self.assertFalse(cb.ValidUUID('01234567089ab0cdef001230456789abcdef'))
def TestShortenUUID(self):
self.assertEqual(cb.ShortenUUID(
'00000000-0000-0000-0000-000000000000'),
'00000000-0000-0000-0000-000000000000')
self.assertEqual(cb.ShortenUUID(
'01234567-89ab-cdef-0123-456789abcdef'),
'01234567-89ab-cdef-0123-456789abcdef')
self.assertEqual(cb.ShortenUUID(
'00001812-0000-1000-8000-00805f9b34fb'),
'1812')
self.assertEqual(cb.ShortenUUID(
'g1234567-89ab-cdef-0123-456789abcdef'),
'g1234567-89ab-cdef-0123-456789abcdef')
self.assertEqual(cb.ShortenUUID(
'01234567-89ab-cdef-0123-456789abcdef0'),
'01234567-89ab-cdef-0123-456789abcdef0')
self.assertEqual(cb.ShortenUUID(
'0123456789abcdef0123456789abcdef'),
'0123456789abcdef0123456789abcdef')
self.assertEqual(cb.ShortenUUID(
'01234567089ab0cdef001230456789abcdef'),
'01234567089ab0cdef001230456789abcdef')
def TestProcess(self):
blocklist = collections.OrderedDict()
try:
cb.Process('# comment', blocklist)
cb.Process('', blocklist)
except Exception:
self.fail('Failed test for comment or blank line.')
self.assertRaises(cb.BadLineException, cb.Process,
'00001812-0000-1000-8000-00805f9b34fb exclude-write exclude', blocklist)
self.assertRaises(cb.InvalidExclusionException, cb.Process,
'00001812-0000-1000-8000-00805f9b34fb exclude-write', blocklist)
self.assertRaises(cb.InvalidExclusionException, cb.Process,
'00001812-0000-1000-8000-00805f9b34fb exclude', blocklist)
try:
cb.Process('00001812-0000-1000-8000-00805f9b34fa exclude-writes',
blocklist)
cb.Process('00001812-0000-1000-8000-00805f9b34fb exclude-reads',
blocklist)
cb.Process('00001812-0000-1000-8000-00805f9b34fc', blocklist)
except Exception:
self.fail('Failed test for valid lines.')
self.assertRaises(cb.DuplicateUUIDException, cb.Process,
'00001812-0000-1000-8000-00805f9b34fa exclude-writes', blocklist)
if __name__ == '__main__':
unittest.main()