You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
730 lines
22 KiB
730 lines
22 KiB
#!/usr/bin/env python |
|
# |
|
# Copyright 2006, Google Inc. |
|
# All rights reserved. |
|
# |
|
# Redistribution and use in source and binary forms, with or without |
|
# modification, are permitted provided that the following conditions are |
|
# met: |
|
# |
|
# * Redistributions of source code must retain the above copyright |
|
# notice, this list of conditions and the following disclaimer. |
|
# * Redistributions in binary form must reproduce the above |
|
# copyright notice, this list of conditions and the following disclaimer |
|
# in the documentation and/or other materials provided with the |
|
# distribution. |
|
# * Neither the name of Google Inc. nor the names of its |
|
# contributors may be used to endorse or promote products derived from |
|
# this software without specific prior written permission. |
|
# |
|
# THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS |
|
# "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT |
|
# LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR |
|
# A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT |
|
# OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, |
|
# SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT |
|
# LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, |
|
# DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY |
|
# THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT |
|
# (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE |
|
# OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. |
|
|
|
"""gen_gtest_pred_impl.py v0.1 |
|
|
|
Generates the implementation of Google Test predicate assertions and |
|
accompanying tests. |
|
|
|
Usage: |
|
|
|
gen_gtest_pred_impl.py MAX_ARITY |
|
|
|
where MAX_ARITY is a positive integer. |
|
|
|
The command generates the implementation of up-to MAX_ARITY-ary |
|
predicate assertions, and writes it to file gtest_pred_impl.h in the |
|
directory where the script is. It also generates the accompanying |
|
unit test in file gtest_pred_impl_unittest.cc. |
|
""" |
|
|
|
__author__ = 'wan@google.com (Zhanyong Wan)' |
|
|
|
import os |
|
import sys |
|
import time |
|
|
|
# Where this script is. |
|
SCRIPT_DIR = os.path.dirname(sys.argv[0]) |
|
|
|
# Where to store the generated header. |
|
HEADER = os.path.join(SCRIPT_DIR, '../include/gtest/gtest_pred_impl.h') |
|
|
|
# Where to store the generated unit test. |
|
UNIT_TEST = os.path.join(SCRIPT_DIR, '../test/gtest_pred_impl_unittest.cc') |
|
|
|
|
|
def HeaderPreamble(n): |
|
"""Returns the preamble for the header file. |
|
|
|
Args: |
|
n: the maximum arity of the predicate macros to be generated. |
|
""" |
|
|
|
# A map that defines the values used in the preamble template. |
|
DEFS = { |
|
'today' : time.strftime('%m/%d/%Y'), |
|
'year' : time.strftime('%Y'), |
|
'command' : '%s %s' % (os.path.basename(sys.argv[0]), n), |
|
'n' : n |
|
} |
|
|
|
return ( |
|
"""// Copyright 2006, Google Inc. |
|
// All rights reserved. |
|
// |
|
// Redistribution and use in source and binary forms, with or without |
|
// modification, are permitted provided that the following conditions are |
|
// met: |
|
// |
|
// * Redistributions of source code must retain the above copyright |
|
// notice, this list of conditions and the following disclaimer. |
|
// * Redistributions in binary form must reproduce the above |
|
// copyright notice, this list of conditions and the following disclaimer |
|
// in the documentation and/or other materials provided with the |
|
// distribution. |
|
// * Neither the name of Google Inc. nor the names of its |
|
// contributors may be used to endorse or promote products derived from |
|
// this software without specific prior written permission. |
|
// |
|
// THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS |
|
// "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT |
|
// LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR |
|
// A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT |
|
// OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, |
|
// SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT |
|
// LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, |
|
// DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY |
|
// THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT |
|
// (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE |
|
// OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. |
|
|
|
// This file is AUTOMATICALLY GENERATED on %(today)s by command |
|
// '%(command)s'. DO NOT EDIT BY HAND! |
|
// |
|
// Implements a family of generic predicate assertion macros. |
|
|
|
#ifndef GTEST_INCLUDE_GTEST_GTEST_PRED_IMPL_H_ |
|
#define GTEST_INCLUDE_GTEST_GTEST_PRED_IMPL_H_ |
|
|
|
// Makes sure this header is not included before gtest.h. |
|
#ifndef GTEST_INCLUDE_GTEST_GTEST_H_ |
|
# error Do not include gtest_pred_impl.h directly. Include gtest.h instead. |
|
#endif // GTEST_INCLUDE_GTEST_GTEST_H_ |
|
|
|
// This header implements a family of generic predicate assertion |
|
// macros: |
|
// |
|
// ASSERT_PRED_FORMAT1(pred_format, v1) |
|
// ASSERT_PRED_FORMAT2(pred_format, v1, v2) |
|
// ... |
|
// |
|
// where pred_format is a function or functor that takes n (in the |
|
// case of ASSERT_PRED_FORMATn) values and their source expression |
|
// text, and returns a testing::AssertionResult. See the definition |
|
// of ASSERT_EQ in gtest.h for an example. |
|
// |
|
// If you don't care about formatting, you can use the more |
|
// restrictive version: |
|
// |
|
// ASSERT_PRED1(pred, v1) |
|
// ASSERT_PRED2(pred, v1, v2) |
|
// ... |
|
// |
|
// where pred is an n-ary function or functor that returns bool, |
|
// and the values v1, v2, ..., must support the << operator for |
|
// streaming to std::ostream. |
|
// |
|
// We also define the EXPECT_* variations. |
|
// |
|
// For now we only support predicates whose arity is at most %(n)s. |
|
// Please email googletestframework@googlegroups.com if you need |
|
// support for higher arities. |
|
|
|
// GTEST_ASSERT_ is the basic statement to which all of the assertions |
|
// in this file reduce. Don't use this in your code. |
|
|
|
#define GTEST_ASSERT_(expression, on_failure) \\ |
|
GTEST_AMBIGUOUS_ELSE_BLOCKER_ \\ |
|
if (const ::testing::AssertionResult gtest_ar = (expression)) \\ |
|
; \\ |
|
else \\ |
|
on_failure(gtest_ar.failure_message()) |
|
""" % DEFS) |
|
|
|
|
|
def Arity(n): |
|
"""Returns the English name of the given arity.""" |
|
|
|
if n < 0: |
|
return None |
|
elif n <= 3: |
|
return ['nullary', 'unary', 'binary', 'ternary'][n] |
|
else: |
|
return '%s-ary' % n |
|
|
|
|
|
def Title(word): |
|
"""Returns the given word in title case. The difference between |
|
this and string's title() method is that Title('4-ary') is '4-ary' |
|
while '4-ary'.title() is '4-Ary'.""" |
|
|
|
return word[0].upper() + word[1:] |
|
|
|
|
|
def OneTo(n): |
|
"""Returns the list [1, 2, 3, ..., n].""" |
|
|
|
return range(1, n + 1) |
|
|
|
|
|
def Iter(n, format, sep=''): |
|
"""Given a positive integer n, a format string that contains 0 or |
|
more '%s' format specs, and optionally a separator string, returns |
|
the join of n strings, each formatted with the format string on an |
|
iterator ranged from 1 to n. |
|
|
|
Example: |
|
|
|
Iter(3, 'v%s', sep=', ') returns 'v1, v2, v3'. |
|
""" |
|
|
|
# How many '%s' specs are in format? |
|
spec_count = len(format.split('%s')) - 1 |
|
return sep.join([format % (spec_count * (i,)) for i in OneTo(n)]) |
|
|
|
|
|
def ImplementationForArity(n): |
|
"""Returns the implementation of n-ary predicate assertions.""" |
|
|
|
# A map the defines the values used in the implementation template. |
|
DEFS = { |
|
'n' : str(n), |
|
'vs' : Iter(n, 'v%s', sep=', '), |
|
'vts' : Iter(n, '#v%s', sep=', '), |
|
'arity' : Arity(n), |
|
'Arity' : Title(Arity(n)) |
|
} |
|
|
|
impl = """ |
|
|
|
// Helper function for implementing {EXPECT|ASSERT}_PRED%(n)s. Don't use |
|
// this in your code. |
|
template <typename Pred""" % DEFS |
|
|
|
impl += Iter(n, """, |
|
typename T%s""") |
|
|
|
impl += """> |
|
AssertionResult AssertPred%(n)sHelper(const char* pred_text""" % DEFS |
|
|
|
impl += Iter(n, """, |
|
const char* e%s""") |
|
|
|
impl += """, |
|
Pred pred""" |
|
|
|
impl += Iter(n, """, |
|
const T%s& v%s""") |
|
|
|
impl += """) { |
|
if (pred(%(vs)s)) return AssertionSuccess(); |
|
|
|
""" % DEFS |
|
|
|
impl += ' return AssertionFailure() << pred_text << "("' |
|
|
|
impl += Iter(n, """ |
|
<< e%s""", sep=' << ", "') |
|
|
|
impl += ' << ") evaluates to false, where"' |
|
|
|
impl += Iter(n, """ |
|
<< "\\n" << e%s << " evaluates to " << v%s""") |
|
|
|
impl += """; |
|
} |
|
|
|
// Internal macro for implementing {EXPECT|ASSERT}_PRED_FORMAT%(n)s. |
|
// Don't use this in your code. |
|
#define GTEST_PRED_FORMAT%(n)s_(pred_format, %(vs)s, on_failure)\\ |
|
GTEST_ASSERT_(pred_format(%(vts)s, %(vs)s), \\ |
|
on_failure) |
|
|
|
// Internal macro for implementing {EXPECT|ASSERT}_PRED%(n)s. Don't use |
|
// this in your code. |
|
#define GTEST_PRED%(n)s_(pred, %(vs)s, on_failure)\\ |
|
GTEST_ASSERT_(::testing::AssertPred%(n)sHelper(#pred""" % DEFS |
|
|
|
impl += Iter(n, """, \\ |
|
#v%s""") |
|
|
|
impl += """, \\ |
|
pred""" |
|
|
|
impl += Iter(n, """, \\ |
|
v%s""") |
|
|
|
impl += """), on_failure) |
|
|
|
// %(Arity)s predicate assertion macros. |
|
#define EXPECT_PRED_FORMAT%(n)s(pred_format, %(vs)s) \\ |
|
GTEST_PRED_FORMAT%(n)s_(pred_format, %(vs)s, GTEST_NONFATAL_FAILURE_) |
|
#define EXPECT_PRED%(n)s(pred, %(vs)s) \\ |
|
GTEST_PRED%(n)s_(pred, %(vs)s, GTEST_NONFATAL_FAILURE_) |
|
#define ASSERT_PRED_FORMAT%(n)s(pred_format, %(vs)s) \\ |
|
GTEST_PRED_FORMAT%(n)s_(pred_format, %(vs)s, GTEST_FATAL_FAILURE_) |
|
#define ASSERT_PRED%(n)s(pred, %(vs)s) \\ |
|
GTEST_PRED%(n)s_(pred, %(vs)s, GTEST_FATAL_FAILURE_) |
|
|
|
""" % DEFS |
|
|
|
return impl |
|
|
|
|
|
def HeaderPostamble(): |
|
"""Returns the postamble for the header file.""" |
|
|
|
return """ |
|
|
|
#endif // GTEST_INCLUDE_GTEST_GTEST_PRED_IMPL_H_ |
|
""" |
|
|
|
|
|
def GenerateFile(path, content): |
|
"""Given a file path and a content string, overwrites it with the |
|
given content.""" |
|
|
|
print 'Updating file %s . . .' % path |
|
|
|
f = file(path, 'w+') |
|
print >>f, content, |
|
f.close() |
|
|
|
print 'File %s has been updated.' % path |
|
|
|
|
|
def GenerateHeader(n): |
|
"""Given the maximum arity n, updates the header file that implements |
|
the predicate assertions.""" |
|
|
|
GenerateFile(HEADER, |
|
HeaderPreamble(n) |
|
+ ''.join([ImplementationForArity(i) for i in OneTo(n)]) |
|
+ HeaderPostamble()) |
|
|
|
|
|
def UnitTestPreamble(): |
|
"""Returns the preamble for the unit test file.""" |
|
|
|
# A map that defines the values used in the preamble template. |
|
DEFS = { |
|
'today' : time.strftime('%m/%d/%Y'), |
|
'year' : time.strftime('%Y'), |
|
'command' : '%s %s' % (os.path.basename(sys.argv[0]), sys.argv[1]), |
|
} |
|
|
|
return ( |
|
"""// Copyright 2006, Google Inc. |
|
// All rights reserved. |
|
// |
|
// Redistribution and use in source and binary forms, with or without |
|
// modification, are permitted provided that the following conditions are |
|
// met: |
|
// |
|
// * Redistributions of source code must retain the above copyright |
|
// notice, this list of conditions and the following disclaimer. |
|
// * Redistributions in binary form must reproduce the above |
|
// copyright notice, this list of conditions and the following disclaimer |
|
// in the documentation and/or other materials provided with the |
|
// distribution. |
|
// * Neither the name of Google Inc. nor the names of its |
|
// contributors may be used to endorse or promote products derived from |
|
// this software without specific prior written permission. |
|
// |
|
// THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS |
|
// "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT |
|
// LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR |
|
// A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT |
|
// OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, |
|
// SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT |
|
// LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, |
|
// DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY |
|
// THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT |
|
// (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE |
|
// OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. |
|
|
|
// This file is AUTOMATICALLY GENERATED on %(today)s by command |
|
// '%(command)s'. DO NOT EDIT BY HAND! |
|
|
|
// Regression test for gtest_pred_impl.h |
|
// |
|
// This file is generated by a script and quite long. If you intend to |
|
// learn how Google Test works by reading its unit tests, read |
|
// gtest_unittest.cc instead. |
|
// |
|
// This is intended as a regression test for the Google Test predicate |
|
// assertions. We compile it as part of the gtest_unittest target |
|
// only to keep the implementation tidy and compact, as it is quite |
|
// involved to set up the stage for testing Google Test using Google |
|
// Test itself. |
|
// |
|
// Currently, gtest_unittest takes ~11 seconds to run in the testing |
|
// daemon. In the future, if it grows too large and needs much more |
|
// time to finish, we should consider separating this file into a |
|
// stand-alone regression test. |
|
|
|
#include <iostream> |
|
|
|
#include "gtest/gtest.h" |
|
#include "gtest/gtest-spi.h" |
|
|
|
// A user-defined data type. |
|
struct Bool { |
|
explicit Bool(int val) : value(val != 0) {} |
|
|
|
bool operator>(int n) const { return value > Bool(n).value; } |
|
|
|
Bool operator+(const Bool& rhs) const { return Bool(value + rhs.value); } |
|
|
|
bool operator==(const Bool& rhs) const { return value == rhs.value; } |
|
|
|
bool value; |
|
}; |
|
|
|
// Enables Bool to be used in assertions. |
|
std::ostream& operator<<(std::ostream& os, const Bool& x) { |
|
return os << (x.value ? "true" : "false"); |
|
} |
|
|
|
""" % DEFS) |
|
|
|
|
|
def TestsForArity(n): |
|
"""Returns the tests for n-ary predicate assertions.""" |
|
|
|
# A map that defines the values used in the template for the tests. |
|
DEFS = { |
|
'n' : n, |
|
'es' : Iter(n, 'e%s', sep=', '), |
|
'vs' : Iter(n, 'v%s', sep=', '), |
|
'vts' : Iter(n, '#v%s', sep=', '), |
|
'tvs' : Iter(n, 'T%s v%s', sep=', '), |
|
'int_vs' : Iter(n, 'int v%s', sep=', '), |
|
'Bool_vs' : Iter(n, 'Bool v%s', sep=', '), |
|
'types' : Iter(n, 'typename T%s', sep=', '), |
|
'v_sum' : Iter(n, 'v%s', sep=' + '), |
|
'arity' : Arity(n), |
|
'Arity' : Title(Arity(n)), |
|
} |
|
|
|
tests = ( |
|
"""// Sample functions/functors for testing %(arity)s predicate assertions. |
|
|
|
// A %(arity)s predicate function. |
|
template <%(types)s> |
|
bool PredFunction%(n)s(%(tvs)s) { |
|
return %(v_sum)s > 0; |
|
} |
|
|
|
// The following two functions are needed to circumvent a bug in |
|
// gcc 2.95.3, which sometimes has problem with the above template |
|
// function. |
|
bool PredFunction%(n)sInt(%(int_vs)s) { |
|
return %(v_sum)s > 0; |
|
} |
|
bool PredFunction%(n)sBool(%(Bool_vs)s) { |
|
return %(v_sum)s > 0; |
|
} |
|
""" % DEFS) |
|
|
|
tests += """ |
|
// A %(arity)s predicate functor. |
|
struct PredFunctor%(n)s { |
|
template <%(types)s> |
|
bool operator()(""" % DEFS |
|
|
|
tests += Iter(n, 'const T%s& v%s', sep=""", |
|
""") |
|
|
|
tests += """) { |
|
return %(v_sum)s > 0; |
|
} |
|
}; |
|
""" % DEFS |
|
|
|
tests += """ |
|
// A %(arity)s predicate-formatter function. |
|
template <%(types)s> |
|
testing::AssertionResult PredFormatFunction%(n)s(""" % DEFS |
|
|
|
tests += Iter(n, 'const char* e%s', sep=""", |
|
""") |
|
|
|
tests += Iter(n, """, |
|
const T%s& v%s""") |
|
|
|
tests += """) { |
|
if (PredFunction%(n)s(%(vs)s)) |
|
return testing::AssertionSuccess(); |
|
|
|
return testing::AssertionFailure() |
|
<< """ % DEFS |
|
|
|
tests += Iter(n, 'e%s', sep=' << " + " << ') |
|
|
|
tests += """ |
|
<< " is expected to be positive, but evaluates to " |
|
<< %(v_sum)s << "."; |
|
} |
|
""" % DEFS |
|
|
|
tests += """ |
|
// A %(arity)s predicate-formatter functor. |
|
struct PredFormatFunctor%(n)s { |
|
template <%(types)s> |
|
testing::AssertionResult operator()(""" % DEFS |
|
|
|
tests += Iter(n, 'const char* e%s', sep=""", |
|
""") |
|
|
|
tests += Iter(n, """, |
|
const T%s& v%s""") |
|
|
|
tests += """) const { |
|
return PredFormatFunction%(n)s(%(es)s, %(vs)s); |
|
} |
|
}; |
|
""" % DEFS |
|
|
|
tests += """ |
|
// Tests for {EXPECT|ASSERT}_PRED_FORMAT%(n)s. |
|
|
|
class Predicate%(n)sTest : public testing::Test { |
|
protected: |
|
virtual void SetUp() { |
|
expected_to_finish_ = true; |
|
finished_ = false;""" % DEFS |
|
|
|
tests += """ |
|
""" + Iter(n, 'n%s_ = ') + """0; |
|
} |
|
""" |
|
|
|
tests += """ |
|
virtual void TearDown() { |
|
// Verifies that each of the predicate's arguments was evaluated |
|
// exactly once.""" |
|
|
|
tests += ''.join([""" |
|
EXPECT_EQ(1, n%s_) << |
|
"The predicate assertion didn't evaluate argument %s " |
|
"exactly once.";""" % (i, i + 1) for i in OneTo(n)]) |
|
|
|
tests += """ |
|
|
|
// Verifies that the control flow in the test function is expected. |
|
if (expected_to_finish_ && !finished_) { |
|
FAIL() << "The predicate assertion unexpactedly aborted the test."; |
|
} else if (!expected_to_finish_ && finished_) { |
|
FAIL() << "The failed predicate assertion didn't abort the test " |
|
"as expected."; |
|
} |
|
} |
|
|
|
// true iff the test function is expected to run to finish. |
|
static bool expected_to_finish_; |
|
|
|
// true iff the test function did run to finish. |
|
static bool finished_; |
|
""" % DEFS |
|
|
|
tests += Iter(n, """ |
|
static int n%s_;""") |
|
|
|
tests += """ |
|
}; |
|
|
|
bool Predicate%(n)sTest::expected_to_finish_; |
|
bool Predicate%(n)sTest::finished_; |
|
""" % DEFS |
|
|
|
tests += Iter(n, """int Predicate%%(n)sTest::n%s_; |
|
""") % DEFS |
|
|
|
tests += """ |
|
typedef Predicate%(n)sTest EXPECT_PRED_FORMAT%(n)sTest; |
|
typedef Predicate%(n)sTest ASSERT_PRED_FORMAT%(n)sTest; |
|
typedef Predicate%(n)sTest EXPECT_PRED%(n)sTest; |
|
typedef Predicate%(n)sTest ASSERT_PRED%(n)sTest; |
|
""" % DEFS |
|
|
|
def GenTest(use_format, use_assert, expect_failure, |
|
use_functor, use_user_type): |
|
"""Returns the test for a predicate assertion macro. |
|
|
|
Args: |
|
use_format: true iff the assertion is a *_PRED_FORMAT*. |
|
use_assert: true iff the assertion is a ASSERT_*. |
|
expect_failure: true iff the assertion is expected to fail. |
|
use_functor: true iff the first argument of the assertion is |
|
a functor (as opposed to a function) |
|
use_user_type: true iff the predicate functor/function takes |
|
argument(s) of a user-defined type. |
|
|
|
Example: |
|
|
|
GenTest(1, 0, 0, 1, 0) returns a test that tests the behavior |
|
of a successful EXPECT_PRED_FORMATn() that takes a functor |
|
whose arguments have built-in types.""" |
|
|
|
if use_assert: |
|
assrt = 'ASSERT' # 'assert' is reserved, so we cannot use |
|
# that identifier here. |
|
else: |
|
assrt = 'EXPECT' |
|
|
|
assertion = assrt + '_PRED' |
|
|
|
if use_format: |
|
pred_format = 'PredFormat' |
|
assertion += '_FORMAT' |
|
else: |
|
pred_format = 'Pred' |
|
|
|
assertion += '%(n)s' % DEFS |
|
|
|
if use_functor: |
|
pred_format_type = 'functor' |
|
pred_format += 'Functor%(n)s()' |
|
else: |
|
pred_format_type = 'function' |
|
pred_format += 'Function%(n)s' |
|
if not use_format: |
|
if use_user_type: |
|
pred_format += 'Bool' |
|
else: |
|
pred_format += 'Int' |
|
|
|
test_name = pred_format_type.title() |
|
|
|
if use_user_type: |
|
arg_type = 'user-defined type (Bool)' |
|
test_name += 'OnUserType' |
|
if expect_failure: |
|
arg = 'Bool(n%s_++)' |
|
else: |
|
arg = 'Bool(++n%s_)' |
|
else: |
|
arg_type = 'built-in type (int)' |
|
test_name += 'OnBuiltInType' |
|
if expect_failure: |
|
arg = 'n%s_++' |
|
else: |
|
arg = '++n%s_' |
|
|
|
if expect_failure: |
|
successful_or_failed = 'failed' |
|
expected_or_not = 'expected.' |
|
test_name += 'Failure' |
|
else: |
|
successful_or_failed = 'successful' |
|
expected_or_not = 'UNEXPECTED!' |
|
test_name += 'Success' |
|
|
|
# A map that defines the values used in the test template. |
|
defs = DEFS.copy() |
|
defs.update({ |
|
'assert' : assrt, |
|
'assertion' : assertion, |
|
'test_name' : test_name, |
|
'pf_type' : pred_format_type, |
|
'pf' : pred_format, |
|
'arg_type' : arg_type, |
|
'arg' : arg, |
|
'successful' : successful_or_failed, |
|
'expected' : expected_or_not, |
|
}) |
|
|
|
test = """ |
|
// Tests a %(successful)s %(assertion)s where the |
|
// predicate-formatter is a %(pf_type)s on a %(arg_type)s. |
|
TEST_F(%(assertion)sTest, %(test_name)s) {""" % defs |
|
|
|
indent = (len(assertion) + 3)*' ' |
|
extra_indent = '' |
|
|
|
if expect_failure: |
|
extra_indent = ' ' |
|
if use_assert: |
|
test += """ |
|
expected_to_finish_ = false; |
|
EXPECT_FATAL_FAILURE({ // NOLINT""" |
|
else: |
|
test += """ |
|
EXPECT_NONFATAL_FAILURE({ // NOLINT""" |
|
|
|
test += '\n' + extra_indent + """ %(assertion)s(%(pf)s""" % defs |
|
|
|
test = test % defs |
|
test += Iter(n, ',\n' + indent + extra_indent + '%(arg)s' % defs) |
|
test += ');\n' + extra_indent + ' finished_ = true;\n' |
|
|
|
if expect_failure: |
|
test += ' }, "");\n' |
|
|
|
test += '}\n' |
|
return test |
|
|
|
# Generates tests for all 2**6 = 64 combinations. |
|
tests += ''.join([GenTest(use_format, use_assert, expect_failure, |
|
use_functor, use_user_type) |
|
for use_format in [0, 1] |
|
for use_assert in [0, 1] |
|
for expect_failure in [0, 1] |
|
for use_functor in [0, 1] |
|
for use_user_type in [0, 1] |
|
]) |
|
|
|
return tests |
|
|
|
|
|
def UnitTestPostamble(): |
|
"""Returns the postamble for the tests.""" |
|
|
|
return '' |
|
|
|
|
|
def GenerateUnitTest(n): |
|
"""Returns the tests for up-to n-ary predicate assertions.""" |
|
|
|
GenerateFile(UNIT_TEST, |
|
UnitTestPreamble() |
|
+ ''.join([TestsForArity(i) for i in OneTo(n)]) |
|
+ UnitTestPostamble()) |
|
|
|
|
|
def _Main(): |
|
"""The entry point of the script. Generates the header file and its |
|
unit test.""" |
|
|
|
if len(sys.argv) != 2: |
|
print __doc__ |
|
print 'Author: ' + __author__ |
|
sys.exit(1) |
|
|
|
n = int(sys.argv[1]) |
|
GenerateHeader(n) |
|
GenerateUnitTest(n) |
|
|
|
|
|
if __name__ == '__main__': |
|
_Main()
|
|
|