source: gtest-1.7.0/scripts/fuse_gtest_files.py

Last change on this file was 12746, checked in by hock@…, 11 years ago

integrated the Google Testing Framework (gtest)

and wrote an Hello World test, to ensure the framework is working..

  • Property svn:executable set to *
File size: 8.6 KB
Line 
1#!/usr/bin/env python
2#
3# Copyright 2009, Google Inc.
4# All rights reserved.
5#
6# Redistribution and use in source and binary forms, with or without
7# modification, are permitted provided that the following conditions are
8# met:
9#
10# * Redistributions of source code must retain the above copyright
11# notice, this list of conditions and the following disclaimer.
12# * Redistributions in binary form must reproduce the above
13# copyright notice, this list of conditions and the following disclaimer
14# in the documentation and/or other materials provided with the
15# distribution.
16# * Neither the name of Google Inc. nor the names of its
17# contributors may be used to endorse or promote products derived from
18# this software without specific prior written permission.
19#
20# THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
21# "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
22# LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
23# A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
24# OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
25# SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
26# LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
27# DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
28# THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
29# (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
30# OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
31
32"""fuse_gtest_files.py v0.2.0
33Fuses Google Test source code into a .h file and a .cc file.
34
35SYNOPSIS
36 fuse_gtest_files.py [GTEST_ROOT_DIR] OUTPUT_DIR
37
38 Scans GTEST_ROOT_DIR for Google Test source code, and generates
39 two files: OUTPUT_DIR/gtest/gtest.h and OUTPUT_DIR/gtest/gtest-all.cc.
40 Then you can build your tests by adding OUTPUT_DIR to the include
41 search path and linking with OUTPUT_DIR/gtest/gtest-all.cc. These
42 two files contain everything you need to use Google Test. Hence
43 you can "install" Google Test by copying them to wherever you want.
44
45 GTEST_ROOT_DIR can be omitted and defaults to the parent
46 directory of the directory holding this script.
47
48EXAMPLES
49 ./fuse_gtest_files.py fused_gtest
50 ./fuse_gtest_files.py path/to/unpacked/gtest fused_gtest
51
52This tool is experimental. In particular, it assumes that there is no
53conditional inclusion of Google Test headers. Please report any
54problems to googletestframework@googlegroups.com. You can read
55http://code.google.com/p/googletest/wiki/GoogleTestAdvancedGuide for
56more information.
57"""
58
59__author__ = 'wan@google.com (Zhanyong Wan)'
60
61import os
62import re
63import sets
64import sys
65
66# We assume that this file is in the scripts/ directory in the Google
67# Test root directory.
68DEFAULT_GTEST_ROOT_DIR = os.path.join(os.path.dirname(__file__), '..')
69
70# Regex for matching '#include "gtest/..."'.
71INCLUDE_GTEST_FILE_REGEX = re.compile(r'^\s*#\s*include\s*"(gtest/.+)"')
72
73# Regex for matching '#include "src/..."'.
74INCLUDE_SRC_FILE_REGEX = re.compile(r'^\s*#\s*include\s*"(src/.+)"')
75
76# Where to find the source seed files.
77GTEST_H_SEED = 'include/gtest/gtest.h'
78GTEST_SPI_H_SEED = 'include/gtest/gtest-spi.h'
79GTEST_ALL_CC_SEED = 'src/gtest-all.cc'
80
81# Where to put the generated files.
82GTEST_H_OUTPUT = 'gtest/gtest.h'
83GTEST_ALL_CC_OUTPUT = 'gtest/gtest-all.cc'
84
85
86def VerifyFileExists(directory, relative_path):
87 """Verifies that the given file exists; aborts on failure.
88
89 relative_path is the file path relative to the given directory.
90 """
91
92 if not os.path.isfile(os.path.join(directory, relative_path)):
93 print 'ERROR: Cannot find %s in directory %s.' % (relative_path,
94 directory)
95 print ('Please either specify a valid project root directory '
96 'or omit it on the command line.')
97 sys.exit(1)
98
99
100def ValidateGTestRootDir(gtest_root):
101 """Makes sure gtest_root points to a valid gtest root directory.
102
103 The function aborts the program on failure.
104 """
105
106 VerifyFileExists(gtest_root, GTEST_H_SEED)
107 VerifyFileExists(gtest_root, GTEST_ALL_CC_SEED)
108
109
110def VerifyOutputFile(output_dir, relative_path):
111 """Verifies that the given output file path is valid.
112
113 relative_path is relative to the output_dir directory.
114 """
115
116 # Makes sure the output file either doesn't exist or can be overwritten.
117 output_file = os.path.join(output_dir, relative_path)
118 if os.path.exists(output_file):
119 # TODO(wan@google.com): The following user-interaction doesn't
120 # work with automated processes. We should provide a way for the
121 # Makefile to force overwriting the files.
122 print ('%s already exists in directory %s - overwrite it? (y/N) ' %
123 (relative_path, output_dir))
124 answer = sys.stdin.readline().strip()
125 if answer not in ['y', 'Y']:
126 print 'ABORTED.'
127 sys.exit(1)
128
129 # Makes sure the directory holding the output file exists; creates
130 # it and all its ancestors if necessary.
131 parent_directory = os.path.dirname(output_file)
132 if not os.path.isdir(parent_directory):
133 os.makedirs(parent_directory)
134
135
136def ValidateOutputDir(output_dir):
137 """Makes sure output_dir points to a valid output directory.
138
139 The function aborts the program on failure.
140 """
141
142 VerifyOutputFile(output_dir, GTEST_H_OUTPUT)
143 VerifyOutputFile(output_dir, GTEST_ALL_CC_OUTPUT)
144
145
146def FuseGTestH(gtest_root, output_dir):
147 """Scans folder gtest_root to generate gtest/gtest.h in output_dir."""
148
149 output_file = file(os.path.join(output_dir, GTEST_H_OUTPUT), 'w')
150 processed_files = sets.Set() # Holds all gtest headers we've processed.
151
152 def ProcessFile(gtest_header_path):
153 """Processes the given gtest header file."""
154
155 # We don't process the same header twice.
156 if gtest_header_path in processed_files:
157 return
158
159 processed_files.add(gtest_header_path)
160
161 # Reads each line in the given gtest header.
162 for line in file(os.path.join(gtest_root, gtest_header_path), 'r'):
163 m = INCLUDE_GTEST_FILE_REGEX.match(line)
164 if m:
165 # It's '#include "gtest/..."' - let's process it recursively.
166 ProcessFile('include/' + m.group(1))
167 else:
168 # Otherwise we copy the line unchanged to the output file.
169 output_file.write(line)
170
171 ProcessFile(GTEST_H_SEED)
172 output_file.close()
173
174
175def FuseGTestAllCcToFile(gtest_root, output_file):
176 """Scans folder gtest_root to generate gtest/gtest-all.cc in output_file."""
177
178 processed_files = sets.Set()
179
180 def ProcessFile(gtest_source_file):
181 """Processes the given gtest source file."""
182
183 # We don't process the same #included file twice.
184 if gtest_source_file in processed_files:
185 return
186
187 processed_files.add(gtest_source_file)
188
189 # Reads each line in the given gtest source file.
190 for line in file(os.path.join(gtest_root, gtest_source_file), 'r'):
191 m = INCLUDE_GTEST_FILE_REGEX.match(line)
192 if m:
193 if 'include/' + m.group(1) == GTEST_SPI_H_SEED:
194 # It's '#include "gtest/gtest-spi.h"'. This file is not
195 # #included by "gtest/gtest.h", so we need to process it.
196 ProcessFile(GTEST_SPI_H_SEED)
197 else:
198 # It's '#include "gtest/foo.h"' where foo is not gtest-spi.
199 # We treat it as '#include "gtest/gtest.h"', as all other
200 # gtest headers are being fused into gtest.h and cannot be
201 # #included directly.
202
203 # There is no need to #include "gtest/gtest.h" more than once.
204 if not GTEST_H_SEED in processed_files:
205 processed_files.add(GTEST_H_SEED)
206 output_file.write('#include "%s"\n' % (GTEST_H_OUTPUT,))
207 else:
208 m = INCLUDE_SRC_FILE_REGEX.match(line)
209 if m:
210 # It's '#include "src/foo"' - let's process it recursively.
211 ProcessFile(m.group(1))
212 else:
213 output_file.write(line)
214
215 ProcessFile(GTEST_ALL_CC_SEED)
216
217
218def FuseGTestAllCc(gtest_root, output_dir):
219 """Scans folder gtest_root to generate gtest/gtest-all.cc in output_dir."""
220
221 output_file = file(os.path.join(output_dir, GTEST_ALL_CC_OUTPUT), 'w')
222 FuseGTestAllCcToFile(gtest_root, output_file)
223 output_file.close()
224
225
226def FuseGTest(gtest_root, output_dir):
227 """Fuses gtest.h and gtest-all.cc."""
228
229 ValidateGTestRootDir(gtest_root)
230 ValidateOutputDir(output_dir)
231
232 FuseGTestH(gtest_root, output_dir)
233 FuseGTestAllCc(gtest_root, output_dir)
234
235
236def main():
237 argc = len(sys.argv)
238 if argc == 2:
239 # fuse_gtest_files.py OUTPUT_DIR
240 FuseGTest(DEFAULT_GTEST_ROOT_DIR, sys.argv[1])
241 elif argc == 3:
242 # fuse_gtest_files.py GTEST_ROOT_DIR OUTPUT_DIR
243 FuseGTest(sys.argv[1], sys.argv[2])
244 else:
245 print __doc__
246 sys.exit(1)
247
248
249if __name__ == '__main__':
250 main()
Note: See TracBrowser for help on using the repository browser.