1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
|
#!/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"
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()
|
