93 строки
2.2 KiB
Python
93 строки
2.2 KiB
Python
|
#!/usr/bin/python
|
||
|
|
# Copyright (c) 2011 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 os
|
||
|
|
import scan_sources
|
||
|
|
import unittest
|
||
|
|
|
||
|
|
|
||
|
|
class MockSimplePath(object):
|
||
|
|
def exists(self, pathname):
|
||
|
|
return True
|
||
|
|
|
||
|
|
def getcwd(self):
|
||
|
|
return '.'
|
||
|
|
|
||
|
|
def isdir(self, pathname):
|
||
|
|
return True
|
||
|
|
|
||
|
|
def realpath(self, pathname):
|
||
|
|
return pathname
|
||
|
|
|
||
|
|
|
||
|
|
class MockScanner(object):
|
||
|
|
def __init__(self, filelists):
|
||
|
|
self.filelists = filelists
|
||
|
|
|
||
|
|
def ScanFile(self, filename):
|
||
|
|
if not self.filelists:
|
||
|
|
return []
|
||
|
|
return self.filelists.pop()
|
||
|
|
|
||
|
|
|
||
|
|
class MockResolver(object):
|
||
|
|
def FindFile(self, filename):
|
||
|
|
return filename
|
||
|
|
|
||
|
|
|
||
|
|
class ScannerUnitTest(unittest.TestCase):
|
||
|
|
|
||
|
|
def testScanData(self):
|
||
|
|
scanner = scan_sources.Scanner()
|
||
|
|
test = """
|
||
|
|
#This is not an include
|
||
|
|
#include is <bogus>
|
||
|
|
#include <x1>
|
||
|
|
#include "x2"
|
||
|
|
#include <x3>
|
||
|
|
#include "x4"
|
||
|
|
# include <x5>
|
||
|
|
# include "x6"
|
||
|
|
# include "x7"
|
||
|
|
Not
|
||
|
|
"""
|
||
|
|
results = scanner.ScanData(test)
|
||
|
|
self.assertEqual(results, ['x1', 'x2', 'x3', 'x4', 'x5', 'x6', 'x7'])
|
||
|
|
|
||
|
|
def testResolverAddDirectories(self):
|
||
|
|
resolver = scan_sources.Resolver(MockSimplePath())
|
||
|
|
resolver.AddDirectories(['Dir1 Dir2', 'Dir3', 'Dir4', 'Dir5'])
|
||
|
|
results = resolver.GetDirectories()
|
||
|
|
self.assertEqual(results, ['Dir1','Dir2','Dir3','Dir4','Dir5'])
|
||
|
|
|
||
|
|
def testResolverRelative(self):
|
||
|
|
resolver = scan_sources.Resolver()
|
||
|
|
tests = [
|
||
|
|
('/foo/bar','/foo/bar/file1','file1'),
|
||
|
|
('/foo/bar/extra', '/foo/bar/file2', '../file2'),
|
||
|
|
('/foo/bar', '/foo/bar/extra/file3', 'extra/file3'),
|
||
|
|
]
|
||
|
|
for (base, full, rel) in tests:
|
||
|
|
self.assertEqual(rel, resolver.RealToRelative(full, base))
|
||
|
|
|
||
|
|
def testWorkQ(self):
|
||
|
|
filelists = [['file1', 'file4', 'file2'], ['file3'], ['file5', 'file2']]
|
||
|
|
resolver = MockResolver()
|
||
|
|
scanner = MockScanner(filelists)
|
||
|
|
workq = scan_sources.WorkQueue(resolver, scanner)
|
||
|
|
workq.PushIfNew('file3')
|
||
|
|
result = workq.Run()
|
||
|
|
|
||
|
|
flat = set([item for item in [sublist for sublist in filelists]])
|
||
|
|
flat = [item for item in flat]
|
||
|
|
|
||
|
|
flat = sorted(flat)
|
||
|
|
result = sorted(flat)
|
||
|
|
self.assertEqual(flat, result)
|
||
|
|
|
||
|
|
if __name__ == '__main__':
|
||
|
|
unittest.main()
|
||
|
|
|