blob: 0d98030a9af0cf7d396248d6a3b7709f334a1f99 [file] [log] [blame]
#!/usr/bin/env 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"
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/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)
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__':