3
# Copyright 2006, Google Inc.
6
# Redistribution and use in source and binary forms, with or without
7
# modification, are permitted provided that the following conditions are
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
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.
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.
32
"""gen_gtest_pred_impl.py v0.1
34
Generates the implementation of Google Test predicate assertions and
39
gen_gtest_pred_impl.py MAX_ARITY
41
where MAX_ARITY is a positive integer.
43
The command generates the implementation of up-to MAX_ARITY-ary
44
predicate assertions, and writes it to file gtest_pred_impl.h in the
45
directory where the script is. It also generates the accompanying
46
unit test in file gtest_pred_impl_unittest.cc.
49
__author__ = 'wan@google.com (Zhanyong Wan)'
55
# Where this script is.
56
SCRIPT_DIR = os.path.dirname(sys.argv[0])
58
# Where to store the generated header.
59
HEADER = os.path.join(SCRIPT_DIR, '../include/gtest/gtest_pred_impl.h')
61
# Where to store the generated unit test.
62
UNIT_TEST = os.path.join(SCRIPT_DIR, '../test/gtest_pred_impl_unittest.cc')
65
def HeaderPreamble(n):
66
"""Returns the preamble for the header file.
69
n: the maximum arity of the predicate macros to be generated.
72
# A map that defines the values used in the preamble template.
74
'today' : time.strftime('%m/%d/%Y'),
75
'year' : time.strftime('%Y'),
76
'command' : '%s %s' % (os.path.basename(sys.argv[0]), n),
81
"""// Copyright 2006, Google Inc.
82
// All rights reserved.
84
// Redistribution and use in source and binary forms, with or without
85
// modification, are permitted provided that the following conditions are
88
// * Redistributions of source code must retain the above copyright
89
// notice, this list of conditions and the following disclaimer.
90
// * Redistributions in binary form must reproduce the above
91
// copyright notice, this list of conditions and the following disclaimer
92
// in the documentation and/or other materials provided with the
94
// * Neither the name of Google Inc. nor the names of its
95
// contributors may be used to endorse or promote products derived from
96
// this software without specific prior written permission.
98
// THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
99
// "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
100
// LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
101
// A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
102
// OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
103
// SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
104
// LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
105
// DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
106
// THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
107
// (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
108
// OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
110
// This file is AUTOMATICALLY GENERATED on %(today)s by command
111
// '%(command)s'. DO NOT EDIT BY HAND!
113
// Implements a family of generic predicate assertion macros.
115
#ifndef GTEST_INCLUDE_GTEST_GTEST_PRED_IMPL_H_
116
#define GTEST_INCLUDE_GTEST_GTEST_PRED_IMPL_H_
118
// Makes sure this header is not included before gtest.h.
119
#ifndef GTEST_INCLUDE_GTEST_GTEST_H_
120
#error Do not include gtest_pred_impl.h directly. Include gtest.h instead.
121
#endif // GTEST_INCLUDE_GTEST_GTEST_H_
123
// This header implements a family of generic predicate assertion
126
// ASSERT_PRED_FORMAT1(pred_format, v1)
127
// ASSERT_PRED_FORMAT2(pred_format, v1, v2)
130
// where pred_format is a function or functor that takes n (in the
131
// case of ASSERT_PRED_FORMATn) values and their source expression
132
// text, and returns a testing::AssertionResult. See the definition
133
// of ASSERT_EQ in gtest.h for an example.
135
// If you don't care about formatting, you can use the more
136
// restrictive version:
138
// ASSERT_PRED1(pred, v1)
139
// ASSERT_PRED2(pred, v1, v2)
142
// where pred is an n-ary function or functor that returns bool,
143
// and the values v1, v2, ..., must support the << operator for
144
// streaming to std::ostream.
146
// We also define the EXPECT_* variations.
148
// For now we only support predicates whose arity is at most %(n)s.
149
// Please email googletestframework@googlegroups.com if you need
150
// support for higher arities.
152
// GTEST_ASSERT is the basic statement to which all of the assertions
153
// in this file reduce. Don't use this in your code.
155
#define GTEST_ASSERT(expression, on_failure) \\
156
GTEST_AMBIGUOUS_ELSE_BLOCKER \\
157
if (const ::testing::AssertionResult gtest_ar = (expression)) \\
160
on_failure(gtest_ar.failure_message())
165
"""Returns the English name of the given arity."""
170
return ['nullary', 'unary', 'binary', 'ternary'][n]
176
"""Returns the given word in title case. The difference between
177
this and string's title() method is that Title('4-ary') is '4-ary'
178
while '4-ary'.title() is '4-Ary'."""
180
return word[0].upper() + word[1:]
184
"""Returns the list [1, 2, 3, ..., n]."""
186
return range(1, n + 1)
189
def Iter(n, format, sep=''):
190
"""Given a positive integer n, a format string that contains 0 or
191
more '%s' format specs, and optionally a separator string, returns
192
the join of n strings, each formatted with the format string on an
193
iterator ranged from 1 to n.
197
Iter(3, 'v%s', sep=', ') returns 'v1, v2, v3'.
200
# How many '%s' specs are in format?
201
spec_count = len(format.split('%s')) - 1
202
return sep.join([format % (spec_count * (i,)) for i in OneTo(n)])
205
def ImplementationForArity(n):
206
"""Returns the implementation of n-ary predicate assertions."""
208
# A map the defines the values used in the implementation template.
211
'vs' : Iter(n, 'v%s', sep=', '),
212
'vts' : Iter(n, '#v%s', sep=', '),
214
'Arity' : Title(Arity(n))
219
// Helper function for implementing {EXPECT|ASSERT}_PRED%(n)s. Don't use
220
// this in your code.
221
template <typename Pred""" % DEFS
227
AssertionResult AssertPred%(n)sHelper(const char* pred_text""" % DEFS
239
if (pred(%(vs)s)) return AssertionSuccess();
244
impl += ' msg << pred_text << "("'
247
<< e%s""", sep=' << ", "')
249
impl += ' << ") evaluates to false, where"'
252
<< "\\n" << e%s << " evaluates to " << v%s""")
255
return AssertionFailure(msg);
258
// Internal macro for implementing {EXPECT|ASSERT}_PRED_FORMAT%(n)s.
259
// Don't use this in your code.
260
#define GTEST_PRED_FORMAT%(n)s(pred_format, %(vs)s, on_failure)\\
261
GTEST_ASSERT(pred_format(%(vts)s, %(vs)s),\\
264
// Internal macro for implementing {EXPECT|ASSERT}_PRED%(n)s. Don't use
265
// this in your code.
266
#define GTEST_PRED%(n)s(pred, %(vs)s, on_failure)\\
267
GTEST_ASSERT(::testing::AssertPred%(n)sHelper(#pred""" % DEFS
269
impl += Iter(n, """, \\
275
impl += Iter(n, """, \\
278
impl += """), on_failure)
280
// %(Arity)s predicate assertion macros.
281
#define EXPECT_PRED_FORMAT%(n)s(pred_format, %(vs)s) \\
282
GTEST_PRED_FORMAT%(n)s(pred_format, %(vs)s, GTEST_NONFATAL_FAILURE)
283
#define EXPECT_PRED%(n)s(pred, %(vs)s) \\
284
GTEST_PRED%(n)s(pred, %(vs)s, GTEST_NONFATAL_FAILURE)
285
#define ASSERT_PRED_FORMAT%(n)s(pred_format, %(vs)s) \\
286
GTEST_PRED_FORMAT%(n)s(pred_format, %(vs)s, GTEST_FATAL_FAILURE)
287
#define ASSERT_PRED%(n)s(pred, %(vs)s) \\
288
GTEST_PRED%(n)s(pred, %(vs)s, GTEST_FATAL_FAILURE)
295
def HeaderPostamble():
296
"""Returns the postamble for the header file."""
300
#endif // GTEST_INCLUDE_GTEST_GTEST_PRED_IMPL_H_
304
def GenerateFile(path, content):
305
"""Given a file path and a content string, overwrites it with the
308
print 'Updating file %s . . .' % path
314
print 'File %s has been updated.' % path
317
def GenerateHeader(n):
318
"""Given the maximum arity n, updates the header file that implements
319
the predicate assertions."""
323
+ ''.join([ImplementationForArity(i) for i in OneTo(n)])
327
def UnitTestPreamble():
328
"""Returns the preamble for the unit test file."""
330
# A map that defines the values used in the preamble template.
332
'today' : time.strftime('%m/%d/%Y'),
333
'year' : time.strftime('%Y'),
334
'command' : '%s %s' % (os.path.basename(sys.argv[0]), sys.argv[1]),
338
"""// Copyright 2006, Google Inc.
339
// All rights reserved.
341
// Redistribution and use in source and binary forms, with or without
342
// modification, are permitted provided that the following conditions are
345
// * Redistributions of source code must retain the above copyright
346
// notice, this list of conditions and the following disclaimer.
347
// * Redistributions in binary form must reproduce the above
348
// copyright notice, this list of conditions and the following disclaimer
349
// in the documentation and/or other materials provided with the
351
// * Neither the name of Google Inc. nor the names of its
352
// contributors may be used to endorse or promote products derived from
353
// this software without specific prior written permission.
355
// THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
356
// "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
357
// LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
358
// A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
359
// OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
360
// SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
361
// LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
362
// DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
363
// THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
364
// (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
365
// OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
367
// This file is AUTOMATICALLY GENERATED on %(today)s by command
368
// '%(command)s'. DO NOT EDIT BY HAND!
370
// Regression test for gtest_pred_impl.h
372
// This file is generated by a script and quite long. If you intend to
373
// learn how Google Test works by reading its unit tests, read
374
// gtest_unittest.cc instead.
376
// This is intended as a regression test for the Google Test predicate
377
// assertions. We compile it as part of the gtest_unittest target
378
// only to keep the implementation tidy and compact, as it is quite
379
// involved to set up the stage for testing Google Test using Google
382
// Currently, gtest_unittest takes ~11 seconds to run in the testing
383
// daemon. In the future, if it grows too large and needs much more
384
// time to finish, we should consider separating this file into a
385
// stand-alone regression test.
389
#include <gtest/gtest.h>
390
#include <gtest/gtest-spi.h>
392
// A user-defined data type.
394
explicit Bool(int val) : value(val != 0) {}
396
bool operator>(int n) const { return value > Bool(n).value; }
398
Bool operator+(const Bool& rhs) const { return Bool(value + rhs.value); }
400
bool operator==(const Bool& rhs) const { return value == rhs.value; }
405
// Enables Bool to be used in assertions.
406
std::ostream& operator<<(std::ostream& os, const Bool& x) {
407
return os << (x.value ? "true" : "false");
413
def TestsForArity(n):
414
"""Returns the tests for n-ary predicate assertions."""
416
# A map that defines the values used in the template for the tests.
419
'es' : Iter(n, 'e%s', sep=', '),
420
'vs' : Iter(n, 'v%s', sep=', '),
421
'vts' : Iter(n, '#v%s', sep=', '),
422
'tvs' : Iter(n, 'T%s v%s', sep=', '),
423
'int_vs' : Iter(n, 'int v%s', sep=', '),
424
'Bool_vs' : Iter(n, 'Bool v%s', sep=', '),
425
'types' : Iter(n, 'typename T%s', sep=', '),
426
'v_sum' : Iter(n, 'v%s', sep=' + '),
428
'Arity' : Title(Arity(n)),
432
"""// Sample functions/functors for testing %(arity)s predicate assertions.
434
// A %(arity)s predicate function.
436
bool PredFunction%(n)s(%(tvs)s) {
437
return %(v_sum)s > 0;
440
// The following two functions are needed to circumvent a bug in
441
// gcc 2.95.3, which sometimes has problem with the above template
443
bool PredFunction%(n)sInt(%(int_vs)s) {
444
return %(v_sum)s > 0;
446
bool PredFunction%(n)sBool(%(Bool_vs)s) {
447
return %(v_sum)s > 0;
452
// A %(arity)s predicate functor.
453
struct PredFunctor%(n)s {
455
bool operator()(""" % DEFS
457
tests += Iter(n, 'const T%s& v%s', sep=""",
461
return %(v_sum)s > 0;
467
// A %(arity)s predicate-formatter function.
469
testing::AssertionResult PredFormatFunction%(n)s(""" % DEFS
471
tests += Iter(n, 'const char* e%s', sep=""",
474
tests += Iter(n, """,
478
if (PredFunction%(n)s(%(vs)s))
479
return testing::AssertionSuccess();
481
testing::Message msg;
484
tests += Iter(n, 'e%s', sep=' << " + " << ')
487
<< " is expected to be positive, but evaluates to "
489
return testing::AssertionFailure(msg);
494
// A %(arity)s predicate-formatter functor.
495
struct PredFormatFunctor%(n)s {
497
testing::AssertionResult operator()(""" % DEFS
499
tests += Iter(n, 'const char* e%s', sep=""",
502
tests += Iter(n, """,
505
tests += """) const {
506
return PredFormatFunction%(n)s(%(es)s, %(vs)s);
512
// Tests for {EXPECT|ASSERT}_PRED_FORMAT%(n)s.
514
class Predicate%(n)sTest : public testing::Test {
516
virtual void SetUp() {
517
expected_to_finish_ = true;
518
finished_ = false;""" % DEFS
521
""" + Iter(n, 'n%s_ = ') + """0;
526
virtual void TearDown() {
527
// Verifies that each of the predicate's arguments was evaluated
530
tests += ''.join(["""
531
EXPECT_EQ(1, n%s_) <<
532
"The predicate assertion didn't evaluate argument %s "
533
"exactly once.";""" % (i, i + 1) for i in OneTo(n)])
537
// Verifies that the control flow in the test function is expected.
538
if (expected_to_finish_ && !finished_) {
539
FAIL() << "The predicate assertion unexpactedly aborted the test.";
540
} else if (!expected_to_finish_ && finished_) {
541
FAIL() << "The failed predicate assertion didn't abort the test "
546
// true iff the test function is expected to run to finish.
547
static bool expected_to_finish_;
549
// true iff the test function did run to finish.
550
static bool finished_;
559
bool Predicate%(n)sTest::expected_to_finish_;
560
bool Predicate%(n)sTest::finished_;
563
tests += Iter(n, """int Predicate%%(n)sTest::n%s_;
567
typedef Predicate%(n)sTest EXPECT_PRED_FORMAT%(n)sTest;
568
typedef Predicate%(n)sTest ASSERT_PRED_FORMAT%(n)sTest;
569
typedef Predicate%(n)sTest EXPECT_PRED%(n)sTest;
570
typedef Predicate%(n)sTest ASSERT_PRED%(n)sTest;
573
def GenTest(use_format, use_assert, expect_failure,
574
use_functor, use_user_type):
575
"""Returns the test for a predicate assertion macro.
578
use_format: true iff the assertion is a *_PRED_FORMAT*.
579
use_assert: true iff the assertion is a ASSERT_*.
580
expect_failure: true iff the assertion is expected to fail.
581
use_functor: true iff the first argument of the assertion is
582
a functor (as opposed to a function)
583
use_user_type: true iff the predicate functor/function takes
584
argument(s) of a user-defined type.
588
GenTest(1, 0, 0, 1, 0) returns a test that tests the behavior
589
of a successful EXPECT_PRED_FORMATn() that takes a functor
590
whose arguments have built-in types."""
593
assrt = 'ASSERT' # 'assert' is reserved, so we cannot use
594
# that identifier here.
598
assertion = assrt + '_PRED'
601
pred_format = 'PredFormat'
602
assertion += '_FORMAT'
606
assertion += '%(n)s' % DEFS
609
pred_format_type = 'functor'
610
pred_format += 'Functor%(n)s()'
612
pred_format_type = 'function'
613
pred_format += 'Function%(n)s'
616
pred_format += 'Bool'
620
test_name = pred_format_type.title()
623
arg_type = 'user-defined type (Bool)'
624
test_name += 'OnUserType'
630
arg_type = 'built-in type (int)'
631
test_name += 'OnBuiltInType'
638
successful_or_failed = 'failed'
639
expected_or_not = 'expected.'
640
test_name += 'Failure'
642
successful_or_failed = 'successful'
643
expected_or_not = 'UNEXPECTED!'
644
test_name += 'Success'
646
# A map that defines the values used in the test template.
650
'assertion' : assertion,
651
'test_name' : test_name,
652
'pf_type' : pred_format_type,
654
'arg_type' : arg_type,
656
'successful' : successful_or_failed,
657
'expected' : expected_or_not,
661
// Tests a %(successful)s %(assertion)s where the
662
// predicate-formatter is a %(pf_type)s on a %(arg_type)s.
663
TEST_F(%(assertion)sTest, %(test_name)s) {""" % defs
665
indent = (len(assertion) + 3)*' '
672
expected_to_finish_ = false;
673
EXPECT_FATAL_FAILURE({ // NOLINT"""
676
EXPECT_NONFATAL_FAILURE({ // NOLINT"""
678
test += '\n' + extra_indent + """ %(assertion)s(%(pf)s""" % defs
681
test += Iter(n, ',\n' + indent + extra_indent + '%(arg)s' % defs)
682
test += ');\n' + extra_indent + ' finished_ = true;\n'
690
# Generates tests for all 2**6 = 64 combinations.
691
tests += ''.join([GenTest(use_format, use_assert, expect_failure,
692
use_functor, use_user_type)
693
for use_format in [0, 1]
694
for use_assert in [0, 1]
695
for expect_failure in [0, 1]
696
for use_functor in [0, 1]
697
for use_user_type in [0, 1]
703
def UnitTestPostamble():
704
"""Returns the postamble for the tests."""
709
def GenerateUnitTest(n):
710
"""Returns the tests for up-to n-ary predicate assertions."""
712
GenerateFile(UNIT_TEST,
714
+ ''.join([TestsForArity(i) for i in OneTo(n)])
715
+ UnitTestPostamble())
719
"""The entry point of the script. Generates the header file and its
722
if len(sys.argv) != 2:
724
print 'Author: ' + __author__
732
if __name__ == '__main__':